Implementation of Convolutional Neural Networks (CNN) for Breast Cancer Detection Using ResNet18 Architecture
DOI:
https://doi.org/10.30871/jaic.v9i4.9746Keywords:
CNN, ResNet18, Transfer Learning, Classification, Ultrasound, Breast CancerAbstract
Early detection of breast cancer is crucial for improving patient survival rates. This study implements a Convolutional Neural Network (CNN) architecture based on ResNet18 using a transfer learning approach to classify breast ultrasound (USG) images into three categories: normal, benign, and malignant. The dataset, comprising 1,578 grayscale images collected from Baheya Hospital in Egypt, underwent preprocessing steps including image conversion, normalization, and augmentation. The ResNet18 model was fine-tuned using selective layer unfreezing to better adapt to the medical imaging domain. Evaluation was conducted using stratified 5-fold cross-validation and assessed with accuracy, precision, recall, F1-score, and AUC metrics. The best results were achieved by fine-tuning layer2, layer3, and the fully connected layer, yielding 95% accuracy, a macro F1-score of 0.93, and an AUC of 0.9906. The findings demonstrate that ResNet18, when properly fine-tuned with transfer learning, delivers high performance in breast cancer detection via ultrasound and holds strong potential as a reliable clinical decision-support tool.
Downloads
References
[1] R. L. Siegel, K. D. Miller, N. S. Wagle, and A. Jemal, “Cancer statistics, 2023,” CA Cancer J Clin, vol. 73, no. 1, pp. 17–48, Jan. 2023, doi: 10.3322/caac.21763.
[2] W. A. Berg et al., “Diagnostic Accuracy of Mammography, Clinical Examination, US, and MR Imaging in Preoperative Assessment of Breast Cancer,” Radiology, vol. 233, no. 3, pp. 830–849, Dec. 2004, doi: 10.1148/radiol.2333031484.
[3] C. Li, J. Li, T. Tan, K. Chen, Y. Xu, and R. Wu, “Application of ultrasonic dual-mode artificially intelligent architecture in assisting radiologists with different diagnostic levels on breast masses classification,” Diagnostic and Interventional Radiology, vol. 27, no. 3, pp. 315–322, May 2021, doi: 10.5152/dir.2021.20018.
[4] P. P. M.Sc and M.Phil, “ANALYSIS OF MEDICAL IMAGE CLASSIFICATION BASED ON CONVOLUTION NEURAL NETWORK,” 2022. [Online]. Available: https://api.semanticscholar.org/CorpusID:248047069
[5] A. Boukaache, B. Nasser Edinne, and D. Boudjehem, “Breast Cancer Image Classification using Convolutional Neural Networks (CNN) Models,” International Journal of Informatics and Applied Mathematics, vol. 6, no. 2, pp. 20–34, Jan. 2024, doi: 10.53508/ijiam.1407152.
[6] H. E. Kim, A. Cosa-Linan, N. Santhanam, M. Jannesari, M. E. Maros, and T. Ganslandt, “Transfer learning for medical image classification: a literature review,” BMC Med Imaging, vol. 22, no. 1, p. 69, Dec. 2022, doi: 10.1186/s12880-022-00793-7.
[7] D. Muduli, R. Dash, and B. Majhi, “Automated diagnosis of breast cancer using multi-modal datasets: A deep convolution neural network based approach,” Biomed Signal Process Control, vol. 71, p. 102825, Jan. 2022, doi: 10.1016/j.bspc.2021.102825.
[8] Y.-D. Zhang, S. C. Satapathy, D. S. Guttery, J. M. Górriz, and S.-H. Wang, “Improved Breast Cancer Classification Through Combining Graph Convolutional Network and Convolutional Neural Network,” Inf Process Manag, vol. 58, no. 2, p. 102439, Mar. 2021, doi: 10.1016/j.ipm.2020.102439.
[9] S. A. E. ALBAKIA and R. A. Saputra, “Identifikasi Jenis Daun Tanaman Obat Menggunakan Metode Convolutional Neural Network (CNN) Dengan Model VGG16,” Jurnal Informatika Polinema, vol. 9, no. 4, pp. 451–460, Aug. 2023, doi: 10.33795/jip.v9i4.1420.
[10] J. Potsangbam and S. Shuleenda Devi, “Classification of Breast Cancer Histopathological Images Using Transfer Learning with DenseNet121,” Procedia Comput Sci, vol. 235, pp. 1990–1997, 2024, doi: 10.1016/j.procs.2024.04.188.
[11] R. T. S and S. Joshi, “Breast Cancer Image Classification Using Convolutional Neural Networks,” 2024 International Conference on Integrated Intelligence and Communication Systems (ICIICS), pp. 1–6, 2024, [Online]. Available: https://api.semanticscholar.org/CorpusID:276118181
[12] W. Al-Dhabyani, M. Gomaa, H. Khaled, and A. Fahmy, “Dataset of breast ultrasound images,” Data Brief, vol. 28, p. 104863, Feb. 2020, doi: 10.1016/j.dib.2019.104863.
[13] H. Fathee and S. Sahmoud, “Iris segmentation in uncooperative and unconstrained environments: State-of-the-art, datasets and future research directions,” Digit Signal Process, vol. 118, p. 103244, Nov. 2021, doi: 10.1016/j.dsp.2021.103244.
[14] K. He, X. Zhang, S. Ren, and J. Sun, “Deep Residual Learning for Image Recognition,” in 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), IEEE, Jun. 2016, pp. 770–778. doi: 10.1109/CVPR.2016.90.
[15] “ImageNet. http://www.image-net.org.”
[16] W. Gómez-Flores and W. Coelho de Albuquerque Pereira, “A comparative study of pre-trained convolutional neural networks for semantic segmentation of breast tumors in ultrasound,” Comput Biol Med, vol. 126, p. 104036, Nov. 2020, doi: 10.1016/j.compbiomed.2020.104036.
[17] A. Davila, J. Colan, and Y. Hasegawa, “Comparison of fine-tuning strategies for transfer learning in medical image classification,” Image Vis Comput, vol. 146, p. 105012, Jun. 2024, doi: 10.1016/j.imavis.2024.105012.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Hagia Sofia Siden, I Gusti Ngurah Lanang Wijayakusuma
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License (Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) ) that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
