Deep learning based capsule networks for breast cancer classification using ultrasound images
Main Article Content
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.
Article Details
![Creative Commons License](/wp-content/uploads/88x31.png)
This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
References
- 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
- WHO. WHO launches new roadmap on breast cancer, 2023. https://www.who.int
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- QAMEBI. Breast Ultrasound Images Database - QAMEBI, 2023. https://qamebi.com/breast-ultrasound-images-database
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- Sabour S, Nov CV, Hinton GE. Dynamic Routing Between Capsules. no. Nips, 2017.
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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
- 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