Classification of Mammography based on Semi-supervised Learning

Lilei Sun,Jie Wen,Junqian Wang,Yong Zhao,Yong Xu
DOI: https://doi.org/10.1109/PIC50277.2020.9350835
2020-01-01
Abstract:In recent years, deep learning, especially the convo-lutional neural network (CNN), has been widely used in the field of molybdenum target image classification. Generally speaking, the CNN based methods train the network with a huge number of samples in a supervised way. However, it is too expensive to collect such a large-scale labeled mammogram dataset for training. In this paper, we propose a semi-supervised mammography classification approach to address this issue. Compared with the existing method, our method can simultaneously use the labeled and unlabeled samples to train the mammogram classification network. Specifically, for the breast mass images represented by Mediolateral Oblique (MLO) and Craniocaudal (CC) views, a new classification network is designed to extract features from both views simultaneously. The compact bilinear pooling (CBP) is introduced to extract the fine-grained features of breast mass images. Moreover, considering that the samples near the decision boundary are easy to be misclassified and those misclassified samples are harmful to the model training, we develop a K-Means based strategy, which first clusters the unlabeled samples and then removes those samples far away from the clustering center during model training. The method proposed in this paper is compared with the mainstream deep learning classification methods and semi-supervised methods on the Digital Database for Screening Mammography (DDSM) public mammogram database. The experimental results show that the method proposed is superior to most of the compared methods in terms of processing time and classification performance.
What problem does this paper attempt to address?