Automatic Classification of Cervical Cells Using Deep Learning Method

Abstract

Cervical cancer is the fourth most prevalent disease among women. Prompt diagnosis and its management can significantly improve patients’ survival rates. Therefore, routine screening for cervical cancer is of paramount importance. Herein, we explore the potential of a deep learning model to automatically distinguish abnormal cells from normal cells. The ThinPrep cytologic test dataset was collected from the fourth central hospital of Baoding city, China. Based on the dataset, four classification models were developed. The first model was a 10-layer convolutional neural network (CNN). The second model was an advancement of the first model equipped with a spatial pyramid pooling (SPP) layer (CNN + SPP) to treat cell images based on their sizes. Based on the first model, the third model replaced the CNN layers with the inception module (CNN + Inception). However, the fourth model incorporated both the SPP layer and the inception module into the first model (CNN + inception + SPP). The performances of the four models are estimated and compared by using the same testing data and evaluation index. The testing results demonstrated that the fourth model yields the best performance. Moreover, the area under the curve (AUC) for module four was 0.997.