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

摘要

AimAbnormal 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.MethodIn 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.ResultsThe 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.ConclusionDeep 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.