Magnetic Resonance Imaging for Breast Cancer Classification Using Convolutional Neural Networks
DOI:
https://doi.org/10.30871/jaic.v9i2.9101Keywords:
Breast Cancer, Computer Vision, Convolutional Neural Networks, Confusion Matrix, Image ClassificationAbstract
Breast cancer remains a leading cause of mortality among women worldwide, emphasizing the urgent need for accurate diagnostic methods. This research addresses the challenges of early detection by leveraging Convolutional Neural Networks (CNNs) for the classification of Magnetic Resonance Imaging (MRI) data. Using a publicly available Kaggle dataset consisting of 54,676 MRI images categorized into "Normal" and "Cancer" classes, the dataset was split into 80% for training and 20% for validation. A modified CNN architecture was developed, incorporating optimized layers and hyperparameters, such as the ADAM optimizer, a learning rate of 0.0001, and a mini-batch size of 128. The proposed model achieved exceptional performance, with an accuracy of 99.72%, precision and recall of 99.98% and 99.97%, respectively, and an F1-score of 99.98%, as evaluated through a confusion matrix. These results demonstrate the model’s robustness in distinguishing between healthy and cancerous tissues, providing a reliable and efficient diagnostic tool. This study highlights the potential of CNNs to improve diagnostic precision in medical imaging, aiding clinicians and advancing AI applications in healthcare.
Downloads
References
[1] T. Trongtirakul, S. Agaian, and A. Oulefki, “Automated tumor segmentation in thermographic breast images,” Mathematical Biosciences and Engineering, vol. 20, no. 9, pp. 16786–16806, 2023, doi: 10.3934/mbe.2023748.
[2] A. A. Amadea, C. A. Sari, E. H. Rachmawanto, W. S. Sari, M. S. Noori, and M. A. Alkhafaji, “Mammogram Breast Cancer Classification Based on Deep-Convolutional Neural Network,” in 2023 International Seminar on Application for Technology of Information and Communication (iSemantic), IEEE, Sep. 2023, pp. 394–399. doi: 10.1109/iSemantic59612.2023.10295318.
[3] A. Kumar, “Study and analysis of different segmentation methods for brain tumor MRI application,” Multimed Tools Appl, vol. 82, no. 5, pp. 7117–7139, Feb. 2023, doi: 10.1007/s11042-022-13636-y.
[4] N. R. D. Cahyo, C. A. Sari, E. H. Rachmawanto, C. Jatmoko, R. R. A. Al-Jawry, and M. A. Alkhafaji, “A Comparison of Multi Class Support Vector Machine vs Deep Convolutional Neural Network for Brain Tumor Classification,” in 2023 International Seminar on Application for Technology of Information and Communication (iSemantic), IEEE, Sep. 2023, pp. 358–363. doi: 10.1109/iSemantic59612.2023.10295336.
[5] C. A. Sari, E. H. Rachmawanto, E. Daniati, F. A. Setiawan, A. S. Hyperastuty, and E. Mintorini, “Breast tumor classification using adam and optuna model optimization based on CNN architecture,” Journal of Soft Computing Exploration, vol. 5, no. 2, pp. 153–160, Jun. 2024, doi: 10.52465/joscex.v5i2.373.
[6] V. K. Reshma et al., “Detection of Breast Cancer Using Histopathological Image Classification Dataset with Deep Learning Techniques,” Biomed Res Int, vol. 2022, 2022, doi: 10.1155/2022/8363850.
[7] Y. Zheng et al., “Application of transfer learning and ensemble learning in image-level classification for breast histopathology,” Intelligent Medicine, vol. 3, no. 2, pp. 115–128, May 2023, doi: 10.1016/j.imed.2022.05.004.
[8] E. Manouchehri, A. Taghipour, A. Ebadi, F. Homaei Shandiz, and R. Latifnejad Roudsari, “Understanding breast cancer risk factors: is there any mismatch between laywomen perceptions and expert opinions,” BMC Cancer, vol. 22, no. 1, Dec. 2022, doi: 10.1186/s12885-022-09372-z.
[9] A. A. Khan, A. A. Laghari, and S. A. Awan, “Machine Learning in Computer Vision: A Review,” EAI Endorsed Transactions on Scalable Information Systems, vol. 8, no. 32, pp. 1–11, 2021, doi: 10.4108/eai.21-4-2021.169418.
[10] D. Bhatt et al., “Cnn variants for computer vision: History, architecture, application, challenges and future scope,” Oct. 01, 2021, MDPI. doi: 10.3390/electronics10202470.
[11] A. F. A. Fernandes, J. R. R. Dórea, and G. J. de M. Rosa, “Image Analysis and Computer Vision Applications in Animal Sciences: An Overview,” Oct. 21, 2020, Frontiers Media S.A. doi: 10.3389/fvets.2020.551269.
[12] S. Deepak and P. M. Ameer, “Automated Categorization of Brain Tumor from MRI Using CNN features and SVM,” J Ambient Intell Humaniz Comput, vol. 12, no. 8, pp. 8357–8369, Aug. 2021, doi: 10.1007/s12652-020-02568-w.
[13] W. Sterkens et al., “Computer vision and optical character recognition for the classification of batteries from WEEE,” in Procedia CIRP, Elsevier B.V., 2022, pp. 110–115. doi: 10.1016/j.procir.2022.02.019.
[14] E. H. Rachmawanto, C. A. Sari, and F. O. Isinkaye, “A good result of brain tumor classification based on simple convolutional neural network architecture,” Telkomnika (Telecommunication Computing Electronics and Control), vol. 22, no. 3, pp. 711–719, Jun. 2024, doi: 10.12928/TELKOMNIKA.v22i3.25863.
[15] I. P. Kamila, C. A. Sari, E. H. Rachmawanto, and N. R. D. Cahyo, “A Good Evaluation Based on Confusion Matrix for Lung Diseases Classification using Convolutional Neural Networks,” Advance Sustainable Science, Engineering and Technology, vol. 6, no. 1, p. 0240102, Dec. 2023, doi: 10.26877/asset.v6i1.17330.
[16] A. Susanto, C. A. Sari, E. H. Rachmawanto, I. U. W. Mulyono, and N. Mohd Yaacob, “A Comparative Study of Javanese Script Classification with GoogleNet, DenseNet, ResNet, VGG16 and VGG19,” Scientific Journal of Informatics, vol. 11, no. 1, pp. 31–40, Jan. 2024, doi: 10.15294/sji.v11i1.47305.
[17] R. K. Pathan et al., “Breast Cancer Classification by Using Multi-Headed Convolutional Neural Network Modeling,” Healthcare (Switzerland), vol. 10, no. 12, Dec. 2022, doi: 10.3390/healthcare10122367.
[18] I. U. W. Mulyono, E. H. Rachmawanto, C. A. Sari, and M. K. Sarker, “A high accuracy of deep learning based CNN architecture: classic, VGGNet, and RestNet50 for Covid-19 image classification,” TELKOMNIKA (Telecommunication Computing Electronics and Control), vol. 22, no. 5, p. 1187, Jul. 2024, doi: 10.12928/telkomnika.v22i5.26017.
[19] N. R. D. Cahyo and M. M. I. Al-Ghiffary, “An Image Processing Study: Image Enhancement, Image Segmentation, and Image Classification using Milkfish Freshness Images,” IJECAR) International Journal of Engineering Computing Advanced Research, vol. 1, no. 1, pp. 11–22, 2024.
[20] F. Farhan, C. A. Sari, E. H. Rachmawanto, and N. R. D. Cahyo, “Mangrove Tree Species Classification Based on Leaf, Stem, and Seed Characteristics Using Convolutional Neural Networks with K-Folds Cross Validation Optimalization,” Advance Sustainable Science Engineering and Technology, vol. 5, no. 3, p. 02303011, Oct. 2023, doi: 10.26877/asset.v5i3.17188.
[21] M. M. I. Al-Ghiffary, N. R. D. Cahyo, E. H. Rachmawanto, C. Irawan, and N. Hendriyanto, “Adaptive deep learning based on FaceNet convolutional neural network for facial expression recognition,” Journal of Soft Computing, vol. 05, no. 03, pp. 271–280, 2024, doi: https://doi.org/10.52465/joscex.v5i3.450.
[22] Y. Alqahtani, U. Mandawkar, A. Sharma, M. N. S. Hasan, M. H. Kulkarni, and R. Sugumar, “Breast Cancer Pathological Image Classification Based on the Multiscale CNN Squeeze Model,” Comput Intell Neurosci, vol. 2022, 2022, doi: 10.1155/2022/7075408.
[23] W. Rahman, M. G. G. Faruque, K. Roksana, A. H. M. S. Sadi, M. M. Rahman, and M. M. Azad, “Multiclass blood cancer classification using deep CNN with optimized features,” Array, vol. 18, Jul. 2023, doi: 10.1016/j.array.2023.100292.
[24] M. Seetha, N. Kalyani, and Y. Sravani Devi, “An Ensemble CNN Model for Identification of Diabetic Retinopathy Eye Disease,” in Smart Intelligent Computing and Applications, Volume 2, S. C. Satapathy, V. Bhateja, M. N. Favorskaya, and T. Adilakshmi, Eds., Singapore: Springer Nature Singapore, 2022, pp. 191–198. doi: 10.1007/978-981-16-9705-0_19.
[25] B. Hemalatha, B. Karthik, C. V. Krishna Reddy, and A. Latha, “Deep learning approach for segmentation and classification of blood cells using enhanced CNN,” Measurement: Sensors, vol. 24, Dec. 2022, doi: 10.1016/j.measen.2022.100582.
[26] D. Kilichev and W. Kim, “Hyperparameter Optimization for 1D-CNN-Based Network Intrusion Detection Using GA and PSO,” Mathematics, vol. 11, no. 17, Sep. 2023, doi: 10.3390/math11173724.
[27] I. Markoulidakis and G. Markoulidakis, “Probabilistic Confusion Matrix: A Novel Method for Machine Learning Algorithm Generalized Performance Analysis,” Technologies (Basel), vol. 12, no. 7, p. 113, Jul. 2024, doi: 10.3390/technologies12070113.
[28] A. Theissler, M. Thomas, M. Burch, and F. Gerschner, “ConfusionVis: Comparative evaluation and selection of multi-class classifiers based on confusion matrices,” Knowl Based Syst, vol. 247, Jul. 2022, doi: 10.1016/j.knosys.2022.108651.
[29] Y. A. Nisa, C. A. Sari, E. H. Rachmawanto, and N. Mohd Yaacob, “Ambon Banana Maturity Classification Based On Convolutional Neural Network (CNN),” sinkron, vol. 8, no. 4, pp. 2568–2578, Oct. 2023, doi: 10.33395/sinkron.v8i4.12961.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Adiyah Mahiruna, Rachmat Destriana, Rahmat Riansyah
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).
