Breast cancer mitotic cell detection using cascade convolutional neural network with U-Net

Abstract

The number of mitotic tumor cells detected in each slide is one of the key indicators of breast cancer prognosis. However, accurate mitotic cell counts are still a difficult problem for pathologists and related experts. Traditional methods use manual design algorithms to extract features of mitotic cells, and most methods rely on sliding windows to achieve pixel-level classification through deep learning. However, the complex background and high resolution of pathological images make the above methods time-consuming and ineffective. In order to solve the above problems, we propose a new cascaded convolutional neural network UBCNN (cascaded CNN based on UNet), which consists of three parts: semantic segmentation and classification to detect mitosis. First, we use an improved UNet ++ segmentation network to locate the candidate set of mitotic targets. Secondly, an adequately labeled cell nucleus data set is sent to an improved two-dimensional VNet network, and the cell nucleus is located by means of semantic segmentation to obtain accurate image blocks of mitotic and non-mitotic cells. Finally, the obtained cell image block is used to train a convolutional neural network to achieve binary classification, and the candidate set area is screened to retain the final result of mitosis cells. This paper verifies the detection effect of the above-mentioned cascade detection algorithm on the ICPR 2012 and 2014 mitosis automatic detection competition data sets. The evaluation indicators include accuracy, recall and F-score. Our cascade detection algorithm based on segmentation and classification reached 0.831 on the ICPR 2012 data set and 0.576 on the ICPR 2014 data set. Compared with other existing algorithms, the detection effect was improved, which was very competitive.

Keywords

• mitosis automatic detection
• breast cancer
• deep learning
• cascade detection
• semantic segmentation
• binary classification