Representation Select Deep Neural Network for Mammographic Image Classification

Liyun Xia,Lei Zhang,Xin Shu
DOI: https://doi.org/10.1109/cisai54367.2021.00115
2021-01-01
Abstract:The early detection and timely treatment of breast cancer can save lives. With the development of computer-aided diagnosis systems, the algorithms based deep neural network have been widely used in breast cancer screen. However, the most of the existing methods use deep convolutional neural network to extract feature maps for image classification, which lack attention to the representations of regions with a high probability of containing malignant lesions. In this paper, a strategy is proposed to train the neural network and select the most important representations to make the final classification decision. Specifically, the extracted feature maps are divided into several mutually exclusive parts, and for each part, a trainable filter is used to compute the importance of each element. Then, the most important k representations where most likely to contain malignant lesions are selected from each part and is concatenated as the final decision selected representations. Finally, a new classifier is trained to make the final decision based on the selected representations. Several experiments are conducted and this strategy’s performance proposed in this paper is compared with other similar algorithms’ performance on a public mammographic image dataset, the experimental results show the proposed structure can achieve the stat-of-the-art AUC score.
What problem does this paper attempt to address?