Improved Breast Cancer Classification Through Combining Graph Convolutional Network and Convolutional Neural Network

Yu-Dong Zhang,Suresh Chandra Satapathy,David S. Guttery,Juan Manuel Górriz,Shui-Hua Wang
DOI: https://doi.org/10.1016/j.ipm.2020.102439
2021-03-01
Abstract:<h3 class="u-h4 u-margin-m-top u-margin-xs-bottom">Aim</h3><p>In a pilot study to improve detection of malignant lesions in breast mammograms, we aimed to develop a new method called BDR-CNN-GCN, combining two advanced neural networks: (i) graph convolutional network (GCN); and (ii) convolutional neural network (CNN).</p><h3 class="u-h4 u-margin-m-top u-margin-xs-bottom">Method</h3><p>We utilised a standard 8-layer CNN, then integrated two improvement techniques: (i) batch normalization (BN) and (ii) dropout (DO). Finally, we utilized rank-based stochastic pooling (RSP) to substitute the traditional max pooling. This resulted in BDR-CNN, which is a combination of CNN, BN, DO, and RSP. This BDR-CNN was hybridized with a two-layer GCN, and yielded our BDR-CNN-GCN model which was then utilized for analysis of breast mammograms as a 14-way data augmentation method.</p><h3 class="u-h4 u-margin-m-top u-margin-xs-bottom">Results</h3><p>As proof of concept, we ran our BDR-CNN-GCN algorithm 10 times on the breast mini-MIAS dataset (containing 322 mammographic images), achieving a sensitivity of 96.20±2.90%, a specificity of 96.00±2.31% and an accuracy of 96.10±1.60%.</p><h3 class="u-h4 u-margin-m-top u-margin-xs-bottom">Conclusion</h3><p>Our BDR-CNN-GCN showed improved performance compared to five proposed neural network models and 15 state-of-the-art breast cancer detection approaches, proving to be an effective method for data augmentation and improved detection of malignant breast masses.</p>
computer science, information systems,information science & library science
What problem does this paper attempt to address?