Abnormal breast identification by nine-layer convolutional neural network with parametric rectified linear unit and rank-based stochastic pooling

Yu-Dong Zhang,Chichun Pan,Xianqing Chen,Fubin Wang
DOI: https://doi.org/10.1016/j.jocs.2018.05.005
IF: 3.817
2018-07-01
Journal of Computational Science
Abstract:Aim Abnormal breast appears similar as dense breast in mammography, which makes it a challenge for radiologists to identify. Scholars have proposed numerous computer-vision and machine-learning based approaches. Nevertheless, the features were manually designed. Method In this study, the breast dataset was chosen as the open-access mini MIAS dataset. Cost-sensitive learning was used to balance the dataset. Data augmentation was used to increase the size of training set. We proposed an improved nine-layer convolutional neural network (CNN). In addition, we compared three activation functions: rectified linear unit (ReLU), leaky ReLU, and parametric ReLU. Besides, six pooling techniques were compared: average pooling, max pooling, stochastic pooling, rank-based average pooling, rank-based weighted pooling, and rank-based stochastic pooling. Results The results over 100 test set showed the combination of parametric ReLU and rank-based stochastic pooling performed the best, with sensitivity of 93.4%, specificity of 94.6%, precision of 94.5%, and accuracy of 94.0%. This result is better than six state-of-the-art breast cancer detection approaches. Conclusion Deep learning can provide better detection results than traditional artificial intelligence methods. We validate why we set the number of convolution layers as 2. We shall try to further improve the performance of this proposed method.
computer science, theory & methods, interdisciplinary applications
What problem does this paper attempt to address?