Open Access Peer-reviewed Research Article

Deep learning based capsule networks for breast cancer classification using ultrasound images

Article Sidebar

Grad-CAM of the models
Download PDF
Submitted   Mar 28, 2024
Published   Aug 27, 2024
Issue    Vol 6 (2024)
DOI   10.25082/CCR.2024.01.002

Views

447

Downloads

60

Citations

PlumX Metrics

Main Article Content

Stephen Afrifa corresponding author
[1] Department of Information and Communication Engineering, Tianjin University, Tianjin 300072, China [2] Department of Information Technology and Decision Sciences, University of Energy and Natural Resources, Sunyani 00233, Ghana
Vijayakumar Varadarajan
[1] Research Division, Swiss School of Business and Management, Geneva 1213, Switzerland [2] International Divisions, Ajeenkya D. Y. Patil University, Pune 412105, India
Tao Zhang
Department of Information and Communication Engineering, Tianjin University, Tianjin 300072, China
Peter Appiahene
Department of Information Technology and Decision Sciences, University of Energy and Natural Resources, Sunyani 00233, Ghana
Daniel Gyamfi
Department of Mathematics and Statistics, Saint Louis University, MO 63103, Missouri, USA
Rose-Mary Owusuaa Mensah Gyening
Department of Computer Science, Kwame Nkrumah University of Science and Technology, Kumasi 00233, Ghana
Jacob Mensah
Department of Information Technology and Decision Sciences, University of Energy and Natural Resources, Sunyani 00233, Ghana
Samuel Opoku Berchie
Department of Information Technology and Decision Sciences, University of Energy and Natural Resources, Sunyani 00233, Ghana

Abstract

Purposes: Breast cancer (BC) is a disease in which the breast cells multiply uncontrolled. Breast cancer is one of the most often diagnosed malignancies in women worldwide. Early identification of breast cancer is critical for limiting the impact on affected people's health conditions. The influence of technology and artificial intelligence approaches (AI) in the health industry is tremendous as technology advances. Deep learning (DL) techniques are used in this study to classify breast lumps.
Materials and Methods: The study makes use of two distinct breast ultrasound images (BUSI) with binary and multiclass classification. To assist the models in understanding the data, the datasets are exposed to numerous preprocessing and hyperparameter approaches. With data imbalance being a key difficulty in health analysis, due to the likelihood of not having a condition exceeding that of having the disease, this study applies a cutoff stage to impact the decision threshold in the datasets data augmentation procedures. The capsule neural network (CapsNet), Gabor capsule network (GCN), and convolutional neural network (CNN) are the DL models used to train the various datasets.
Results: The findings showed that the CapsNet earned the maximum accuracy value of 93.62% while training the multiclass data, while the GCN achieved the highest model accuracy of 97.08% when training the binary data. The models were also evaluated using a variety of performance assessment parameters, which yielded consistent results across all datasets.
Conclusion: The study provides a non-invasive approach to detect breast cancer; and enables stakeholders, medical practitioners, and health research enthusiasts a fresh view into the analysis of breast cancer detection with DL techniques to make educated judgements.

Keywords
breast cancer, capsule network, deep learning, Gabor capsule, medical imaging

Article Details

Supporting Agencies
The authors are grateful to Adwoa Afriyie, popularly called Amanpene, and Malcolm Afrifa for their encouragement and advise throughout the studies.
How to Cite
Afrifa, S., Varadarajan, V., Zhang, T., Appiahene, P., Gyamfi, D., Gyening, R.-M. O. M., Mensah, J., & Opoku Berchie, S. (2024). Deep learning based capsule networks for breast cancer classification using ultrasound images. Current Cancer Reports, 6(1), 205-224. https://doi.org/10.25082/CCR.2024.01.002
Creative Commons License

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.

References

  1. Bokhare A, Jha P. Machine learning models applied in analyzing breast cancer classification accuracy. IAES International Journal of Artificial Intelligence (IJ-AI). 2023, 12(3): 1370. https://doi.org/10.11591/ijai.v12.i3.pp1370-1377
  2. WHO. WHO launches new roadmap on breast cancer, 2023. https://www.who.int
  3. Jabeen K, Khan MA, Alhaisoni M, et al. Breast Cancer Classification from Ultrasound Images Using Probability-Based Optimal Deep Learning Feature Fusion. Sensors. 2022, 22(3): 807. https://doi.org/10.3390/s22030807
  4. de Caldas Filho FL, Soares SCM, Oroski E, et al. Botnet Detection and Mitigation Model for IoT Networks Using Federated Learning. Sensors. 2023, 23(14): 6305. https://doi.org/10.3390/s23146305
  5. Hamzeh O, Alkhateeb A, Zheng J, et al. Prediction of tumor location in prostate cancer tissue using a machine learning system on gene expression data. BMC Bioinformatics. 2020, 21(S2). https://doi.org/10.1186/s12859-020-3345-9
  6. Ding IJ, Zheng NW. CNN Deep Learning with Wavelet Image Fusion of CCD RGB-IR and Depth-Grayscale Sensor Data for Hand Gesture Intention Recognition. Sensors. 2022, 22(3): 803. https://doi.org/10.3390/s22030803
  7. Al-Dhabyani W, Gomaa M, Khaled H, et al. Deep Learning Approaches for Data Augmentation and Classification of Breast Masses using Ultrasound Images. International Journal of Advanced Computer Science and Applications. 2019, 10(5). https://doi.org/10.14569/ijacsa.2019.0100579
  8. Zhao X, Jiang C. The prediction of distant metastasis risk for male breast cancer patients based on an interpretable machine learning model. BMC Medical Informatics and Decision Making. 2023, 23(1). https://doi.org/10.1186/s12911-023-02166-8
  9. Prodan M, Paraschiv E, Stanciu A. Applying Deep Learning Methods for Mammography Analysis and Breast Cancer Detection. Applied Sciences. 2023, 13(7): 4272. https://doi.org/10.3390/app13074272
  10. Ardakani AA, Mohammadi A, Faeghi F, et al. Performance evaluation of 67 denoising filters in ultrasound images: A systematic comparison analysis. International Journal of Imaging Systems and Technology. 2023, 33(2): 445-464. https://doi.org/10.1002/ima.22843
  11. Boumaraf S, Liu X, Wan Y, et al. Conventional Machine Learning versus Deep Learning for Magnification Dependent Histopathological Breast Cancer Image Classification: A Comparative Study with Visual Explanation. Diagnostics. 2021, 11(3): 528. https://doi.org/10.3390/diagnostics11030528
  12. Yang X, Fan X, Lin S, et al. Assessment of Lymphovascular Invasion in Breast Cancer Using a Combined MRI Morphological Features, Radiomics, and Deep Learning Approach Based on Dynamic Contrast‐Enhanced MRI. Journal of Magnetic Resonance Imaging. 2023, 59(6): 2238-2249. https://doi.org/10.1002/jmri.29060
  13. Sahu A, Das PK, Meher S. High accuracy hybrid CNN classifiers for breast cancer detection using mammogram and ultrasound datasets. Biomedical Signal Processing and Control. 2023, 80: 104292. https://doi.org/10.1016/j.bspc.2022.104292
  14. Sirjani N, Ghelich Oghli M, Kazem Tarzamni M, et al. A novel deep learning model for breast lesion classification using ultrasound Images: A multicenter data evaluation. Physica Medica. 2023, 107: 102560. https://doi.org/10.1016/j.ejmp.2023.102560
  15. Yu S, Jin M, Wen T, et al. Accurate breast cancer diagnosis using a stable feature ranking algorithm. BMC Medical Informatics and Decision Making. 2023, 23(1). https://doi.org/10.1186/s12911-023-02142-2
  16. Sulu SMM, Mukuku O, et al. Women’s breast cancer risk factors in Kinshasa, Democratic Republic of the Congo. Current Cancer Reports. 2022, 4(1): 139-143. https://doi.org/10.25082/ccr.2022.01.003
  17. Lee S, Jung H, Park J, et al. Accurate Prediction of Cancer Prognosis by Exploiting Patient-Specific Cancer Driver Genes. International Journal of Molecular Sciences. 2023, 24(7): 6445. https://doi.org/10.3390/ijms24076445
  18. QAMEBI. Breast Ultrasound Images Database - QAMEBI, 2023. https://qamebi.com/breast-ultrasound-images-database
  19. Al-Dhabyani W, Gomaa M, Khaled H, et al. Dataset of breast ultrasound images. Data in Brief. 2020, 28: 104863. https://doi.org/10.1016/j.dib.2019.104863
  20. Afrifa S, Zhang T, Zhao X, et al. Climate change impact assessment on groundwater level changes: A study of hybrid model techniques. IET Signal Processing. 2023, 17(6). https://doi.org/10.1049/sil2.12227
  21. Appiahene P, Varadarajan V, Zhang T, et al. Experiences of sexual minorities on social media: A study of sentiment analysis and machine learning approaches. Journal of Autonomous Intelligence. 2023, 6(2): 623. https://doi.org/10.32629/jai.v6i2.623
  22. Ma R, Ma Y, Li C, et al. Potential mechanism exploration of San Wei Tan Xiang capsule for depression treatment by network pharmacology and molecular docking. Medicine in Novel Technology and Devices. 2022, 16: 100160. https://doi.org/10.1016/j.medntd.2022.100160
  23. Afrifa S, Zhang T, Appiahene P, et al. Mathematical and Machine Learning Models for Groundwater Level Changes: A Systematic Review and Bibliographic Analysis. Future Internet. 2022, 14(9): 259. https://doi.org/10.3390/fi14090259
  24. MISHRA A. Contrast Limited Adaptive Histogram Equalization (CLAHE) Approach for Enhancement of the Microstructures of Friction Stir Welded Joints. Published online June 10, 2021. https://doi.org/10.21203/rs.3.rs-607179/v1
  25. Afrifa S, Varadarajan V. Cyberbullying Detection on Twitter Using Natural Language Processing and Machine Learning Techniques. International Journal of Innovative Technology and Interdisciplinary Sciences. 2022, 5(4): 1069-1080. https://doi.org/10.15157/IJITIS.2022.5.4.1069-1080
  26. Sabour S, Nov CV, Hinton GE. Dynamic Routing Between Capsules. no. Nips, 2017.
  27. Hasani M, Saravi AN, Khotanlou H. An Efficient Approach for Using Expectation Maximization Algorithm in Capsule Networks. 2020 International Conference on Machine Vision and Image Processing (MVIP). Published online February 2020. https://doi.org/10.1109/mvip49855.2020.9116870
  28. Toraman S, Alakus TB, Turkoglu I. Convolutional capsnet: A novel artificial neural network approach to detect COVID-19 disease from X-ray images using capsule networks. Chaos, Solitons & Fractals. 2020, 140: 110122. https://doi.org/10.1016/j.chaos.2020.110122
  29. Alekseev A, Bobe A. GaborNet: Gabor filters with learnable parameters in deep convolutional neural network. 2019 International Conference on Engineering and Telecommunication (EnT). Published online November 2019. https://doi.org/10.1109/ent47717.2019.9030571
  30. He Y, Song F, Wu W, et al. MultiTrans: Multi-scale feature fusion transformer with transfer learning strategy for multiple organs segmentation of head and neck CT images. Medicine in Novel Technology and Devices. 2023, 18: 100235. https://doi.org/10.1016/j.medntd.2023.100235
  31. Appiahene P, Arthur EJ, Korankye S, et al. Detection of anemia using conjunctiva images: A smartphone application approach. Medicine in Novel Technology and Devices. 2023, 18: 100237. https://doi.org/10.1016/j.medntd.2023.100237
  32. Appiahene P, Asare JW, Donkoh ET, et al. Detection of iron deficiency anemia by medical images: a comparative study of machine learning algorithms. BioData Mining. 2023, 16(1). https://doi.org/10.1186/s13040-023-00319-z
  33. Podda AS, Balia R, Barra S, et al. Fully-automated deep learning pipeline for segmentation and classification of breast ultrasound images. Journal of Computational Science. 2022, 63: 101816. https://doi.org/10.1016/j.jocs.2022.101816
  34. Kaur P, Singh A, Chana I. BSense: A parallel Bayesian hyperparameter optimized Stacked ensemble model for breast cancer survival prediction. Journal of Computational Science. 2022, 60: 101570. https://doi.org/10.1016/j.jocs.2022.101570
  35. Afrifa S, Varadarajan V, Appiahene P, et al. Ensemble Machine Learning Techniques for Accurate and Efficient Detection of Botnet Attacks in Connected Computers. Eng. 2023, 4(1): 650-664. https://doi.org/10.3390/eng4010039
  36. Adu WK, Appiahene P, Afrifa S. VAR, ARIMAX and ARIMA models for nowcasting unemployment rate in Ghana using Google trends. Journal of Electrical Systems and Information Technology. 2023, 10(1). https://doi.org/10.1186/s43067-023-00078-1
  37. Zhang YD, Pan C, Chen X, et al. Abnormal breast identification by nine-layer convolutional neural network with parametric rectified linear unit and rank-based stochastic pooling. Journal of Computational Science. 2018, 27: 57-68. https://doi.org/10.1016/j.jocs.2018.05.005
  38. Yansari RT, Mirzarezaee M, Sadeghi M, et al. A new survival analysis model in adjuvant Tamoxifen-treated breast cancer patients using manifold-based semi-supervised learning. Journal of Computational Science. 2022, 61: 101645. https://doi.org/10.1016/j.jocs.2022.101645
  39. Phillips CM, Lima EABF, Wu C, et al. Assessing the identifiability of model selection frameworks for the prediction of patient outcomes in the clinical breast cancer setting. Journal of Computational Science. 2023, 69: 102006. https://doi.org/10.1016/j.jocs.2023.102006