Breast cancer detection using deep learning techniques: An investigation using the CBIS-DDSM dataset and customized neural network model, ResNetV2 and YOLO

Document Type : Research Paper

Authors

1 Media Technology and Communications Eng. Department, College of Engineering, University of Information Technology and Communications, Baghdad, Iraq.

2 Department of Basic Sciences, College of Nursing, University of Baghdad, Baghdad, Iraq.

3 School of Engineering, Jawaharlal Nehru University, New Delhi 110067, India.

4 Department of CSE, College of Engineering and Technology, Acharya Nagarjuna University, Nagarjuna Nagar, Guntur, 522510, India.

10.30772/qjes.2025.155626.1429
Abstract
The current work investigates the application of deep learning methodologies for detecting and classifying tumors in mammograms. A comprehensive analysis was conducted by evaluating various pre-trained neural network architectures to identify the model that delivers optimal performance. Among the tested architectures, Residual Neural Networks and Densely Connected Convolutional Networks were explored extensively. Data preprocessing, including data augmentation, was a critical step due to the limited availability of public medical imaging datasets. This process ensured diversity in the data and improved model robustness.  This paper evaluates deep-learning models for mammogram tumor detection. Experiments without the AdamW optimizer and horizontal flip showed overfitting and low precision (below 40\%). Densely Connected Convolutional Networks achieved high precision but exhibited overfitting with noisy validation loss curves. Adding AdamW and horizontal flip reduced overfitting but lowered overall performance. In classification, the model detected tumors in 9 out of 16 images, showing potential but requiring improvement. The model's ability to detect calcification tumors enhances robustness. YOLO network metrics were modest, reflecting the task's complexity, but results were acceptable for tumor classification challenges.

Keywords

Crossmark

  1. M. Arnolda, E. Morgana, H. Rumgay, and et al., “Current and future burden of breast cancer: Global statistics for 2020 and 2040,” The Breast, vol. 66, pp.
    15–23, 2022. [Online]. Available: https://doi.org/10.1016/j.breast.2022.08.010
  2. H. M. Ahmed and K. Khoshnood, “Breast cancer stigma in iraq,” Current Breast Cancer Reports, vol. 15, no. 4, p. 345–349, 2023. [Online]. Available:
    https://doi.org/10.1007/s12609-023-00513-2
  3. N. N. R. Alrawi, “A review on breast cancer in iraq and future therapies insights,” Chemical Engineering Journal, vol. 3, no. 01, p. 4–16, 2022. [Online].
    Available: https://doi.org/10.47419/bjbabs.v3i01.64
  4. P. Rawla, T. Sunkara, and A. Barsouk, “Epidemiology of colorectal cancer: incidence, mortality, survival, and risk factors,” Prz Gastroentero, vol. 14, no. 2,
    pp. 89–103, 2019. [Online]. Available: :https://doi.org/10.5114/pg.2018.81072
  5. H. Swaminathan, K. Saravanamurali, and A. Y. Sangilimuthu, “Extensive review on breast cancer its etiology, progression, prognostic markers, and treatment,”
    Medical Oncology, vol. 40, no. 8, p. 238, 2023. [Online]. Available: https://doi.org/10.1007/s12032-023-02111-9
  6.  M. Cardoso, P. Poortmans, E. Senkus, D. Gentilini, and N. Houssami, “Breast cancer highlights from 2023: Knowledge to guide practice and future research,”
    Breast, vol. 74, p. 103674, 2024. [Online]. Available: https://doi.org/10.1016/j.breast.2024.103674
  7.  X. Mao and et al., “Breast cancer incidence after a false-positive mammography result,” JAMA Oncology, vol. 10, no. 1, pp. 63–70, 2024. [Online]. Available:
    https://doi.org/10.1001/jamaoncol.2023.4519
  8.  N. Moshina, A. Gr ¨awingholt, K. L ˚ang, and et al., “Digital breast tomosynthesis in mammographic screening: false negative cancer cases in the to-be 1 trial,”
    Insights Imaging, vol. 15, p. 38, 2024. [Online]. Available: https://doi.org/10.1186/s13244-023-01604-5
  9.  Z. Tasnim, S. Chakraborty, F. M. J. M. Shamrat, A. Newaz, and et al, “Deep learning predictive model for colon cancer patient using cnn-based classification,”
    International Journal of Advanced Computer Science and Applications, vol. 12, no. 8, 2021. [Online]. Available: https://doi.org/10.14569/ijacsa.2021.0120880
  10. C. Kavyashree, H. Vimala, and J. Shreyas, “Improving oral cancer detection using pretrained model,” 2022 IEEE 6th Conference on Information and
    Communication Technology (CICT), pp. 1–5, 2022. [Online]. Available: https://doi.org/10.1109/cict56698.2022.9997897
  11. A. Sinha, M. Pandey, M. Nazma, J. Naskar, and S. S. Rautaray, “Segmentation-based classification deep learning model for breast cancer detection using
    mammogram images,” 2023 IEEE 3rd Mysore Sub Section International Conference (MysuruCon), HASSAN, India, pp. 1–8, 2023. [Online]. Available:
    https://doi.org/10.1109/mysurucon59703.2023.10397015
  12.  U. Sirisha, S. Praveen, P. N. Srinivasu, P. Barsocchi, and A. K. Bhoi, “Statistical analysis of design aspects of various yolo-based deep learning
    models for object detection,” International Journal of Computational Intelligence Systems, vol. 16, no. 1, p. 126, 2023. [Online]. Available:
    https://doi.org/10.1007/s44196-023-00302-w
  13.  M. Panda, A. Abraham, B. Gopi, and R. Ajith, “Computational intelligence for oncology and neurological disorders,” Informa,CRC Press, vol. 35, no. 1, p.
    292, 2024. [Online]. Available: https://doi.org/10.1201/9781003450153
  14. A. Carriero, L. Groenhoff, E. Vologina, P. Basile, and M. Albera, “Deep learning in breast cancer imaging: State of the art and recent advancements in early
    2024,” Diagnostics, vol. 14, no. 8, p. 848, 2024. [Online]. Available: https://doi.org/10.3390/diagnostics14080848
  15. C. Rajeshkumar, K. R. Soundar, M. Sneha, S. S. Maheswari, M. S. Lakshmi, and R. Priyanka, “Convolutional neural networks (cnn) based brain tumor
    detection in mri images,” 2023 5th International Conference on Smart Systems and Inventive Technology (ICSSIT), pp. 976–979, 2023. [Online]. Available:
    https://doi.org/10.1109/icssit55814.2023.10060968
  16. Y. Kadhim, U. Khan, and A. Mishra, “Deep learning-based computer-aided diagnosis (cad): Applications for medical image datasets,” Sensors, vol. 22,
    no. 22, p. 8999, 2022. [Online]. Available: https://doi.org/10.3390/s22228999
  17. R. Yamashita, M. Nishio, K. Do, and K. Togashi, “Convolutional neural networks: an overview and application in radiology,” Insights Imaging, vol. 9, no. 4,
    p. 611–629, 2018. [Online]. Available: https://doi.org/10.1007/s13244-018-0639-9
  18. L. Alzubaidi and et al., “Review of deep learning: concepts, cnn architectures, challenges, applications, future directions,” Journal of Big Data, vol. 8, no. 1,
    p. 53, 2021. [Online]. Available: https://doi.org/10.1186/s40537-021-00444-8
  19. O. Olayemi, O. Salako, A. Jinadu, A. M. Obalalu, , and B. E. Anyaegbuna, “Aerodynamic lift coefficient prediction of supercritical airfoils at transonic flow
    regime using convolutional neural networks (cnns) and multi-layer perceptions (MLPs),” Al-Qadisiyah Journal for Engineering Sciences, vol. 16, no. 2, p.
    108–115, 2023. [Online]. Available: https://doi.org/10.30772/qjes.v16i2.955
  20. T. Zhou, X. Ye, H. Lu, X. Zheng, S. Qiu, and Y. Liu, “Dense convolutional network and its application in medical image analysis,” BioMed Research
    International, vol. 2022, no. 1, p. 2384830, 202. [Online]. Available: https://doi.org/10.1155/2022/2384830
  21. M. Shafiq and Z. Gu, “Deep residual learning for image recognition: A survey,” Applied Sciences, vol. 12, no. 18, p. 8972, 2022. [Online]. Available:
    https://doi.org/10.3390/app12188972
  22. L. He, P. Tiwari, C. Lv, W. Wu, and L. Guo, “Reducing noisy annotations for depression estimation from facial images,” Neural Networks, vol. 153, p.
    120–129, 2022. [Online]. Available: https://doi.org/10.1016/j.neunet.2022.05.025
  23. M. Anas, I. U. Haq, G. Husnain, and A. Faraz, “Advancing breast cancer detection: Enhancing yolov5 network for accurate classification in mammogram
    images,” IEEE Access, vol. 12, pp. 16 474–16 488, 2024. [Online]. Available: https://doi.org/10.1109/access.2024.3358686
  24. S. A. Hasanah, A. A. Pravitasari, A. S. Abdullah, I. N. Yulita, and M. H. Asnawi, “A deep learning review of resnet architecture for lung disease identification
    in cxr image,” Applied sciences, vol. 13, no. 24, p. 13111, 2023. [Online]. Available: https://doi.org/10.3390/app132413111
  25. T. Ochiai and A. Fujishima, “Comparison of yolo versions for object detection from aerial images,” International Journal of Environment and Geoinformatics,
    vol. 9, no. 2, p. 87–93, 2022. [Online]. Available: https://doi.org/10.30897/ijegeo.1010741
  26. R. Lee, F. Gimenez, A. Hoogi, K. Miyake, M. Gorovoy, and D. Rubin, “A curated mammography data set for use in computer-aided detection and diagnosis
    research,” Scientific Data, vol. 4, no. 1, p. 170177, 2017. [Online]. Available: https://doi.org/10.1038/sdata.2017.177
  27. M. Brahim, K. Westerkamp, L. Hempel, R. Lehmann, D. Hempel, and P. Philipp, “Automated assessment of breast positioning quality in screening
    mammography,” Cancers, vol. 14, no. 19, p. 4704, 2022. [Online]. Available: https://doi.org/10.3390/cancers14194704
  28. J. Suckling, J. Parker, D. Dance, S. Astley, I. Hutt, C. Boggis, I. Ricketts, E. Stamatakis, N. Cerneaz, S. Kok, P. Taylor, D. Betal, and
    J. Savage, “Mammographic image analysis society (MIAS) database v1.21,” Apollo - University of Cambridge Repository, 2015. [Online]. Available:
    https://doi.org/10.17863/CAM.105113
  29. P. Park, H. Oh, and H. Kim, “Dual-color space network with global priors for photo retouching,” Scientific Reports, vol. 13, no. 1, p. 19798, 2023. [Online].
    Available: https://doi.org/10.1038/s41598-023-47186-6
  30. K. Saa, “Method for detection and diagnosis of the area of skin disease based on color by wavelet transform and artificial neural network,” Al-Qadisiyah
    Journal for Engineering Sciences, vol. 2, no. 4, p. 799–829, 2022.
  31. A. Bugeau, R. Giraud, and L. Raad, “Influence of color spaces for deep learning image colorization,” Springer eBooks, p. 847–878, 2023. [Online]. Available:
    https://doi.org/10.1007/978-3-030-98661-2 125
  32. A. Mumuni and F. Mumuni, “Data augmentation: A comprehensive survey of modern approaches,” Array, vol. 16, p. 100258, 2022. [Online]. Available:
    https://doi.org/10.1016/j.array.2022.100258
  33. S. Ni, F. Shao, X. Chai, H. Chen, and Y.-S. Ho, “Composition-guided neural network for image cropping aesthetic assessment,” IEEE Transactions on
    Multimedia, vol. 25, no. 3, p. 6836–6851, 2022. [Online]. Available: https://doi.org/10.1109/tmm.2022.3215003
  34. A. H. Abdulaal, M. Valizadeh, C. Amirani, and M. S. Shah, “A self-learning deep neural network for classification of breast histopathological images,”
    Biomedical Signal Processing and Control, vol. 87, no. part B, p. 105418, 2024. [Online]. Available: https://doi.org/10.1016/j.bspc.2023.105418
  35. M. A. Wakili, H. A. Shehu, M. H. Sharif, and et al., “Classification of breast cancer histopathological images using densenet and transfer learning,”
    Computational Intelligence and Neuroscience, vol. 2022, no. 1, p. 8904768, 2022. [Online]. Available: https://doi.org/10.1155/2022/8904768
  36. X. Jiang, Z. Hu, S. Wang, and Y. Zhang, “Deep learning for medical image-based cancer diagnosis,” Cancers, vol. 15, no. 14, p. 3608, 2023. [Online].
    Available: https://doi.org/10.3390/cancers15143608
  37. M. Nasser and U. K. Yusof, “Deep learning based methods for breast cancer diagnosis: A systematic review and future direction,” Diagnostics, vol. 13, no. 1,
    p. 161, 2023. [Online]. Available: https://doi.org/10.3390/diagnostics13010161
  38. K. Uddin, N. Biswas, T. Rikta, and K. Dey, “Machine learning-based diagnosis of breast cancer utilizing feature optimization technique,” Computer Methods
    and Programs in Biomedicine Update, vol. 3, p. 100098, 2023. [Online]. Available: https://doi.org/10.1016/j.cmpbup.2023.100098
  39. J. Guo and et al., “A novel breast cancer image classification model based on multiscale texture feature analysis and dynamic learning,” Scientific Reports,
    vol. 14, no. 1, p. 7216, 2024. [Online]. Available: https://doi.org/10.1038/s41598-024-57891-5
  40. M. D. Ali, A. Saleem, H. Elahi, M. A. Khan, and et al., “Breast cancer classification through meta-learning ensemble technique using convolution neural
    networks,” Diagnostics, vol. 13, no. 13, p. 2242, 2023. [Online]. Available: https://doi.org/10.3390/diagnostics13132242âĂŔ
  41. R. G. Martinez and M. van Dongen, “Deep learning algorithms for the early detection of breast cancer: A comparative study with traditional machine
    learning,” Informatics in Medicine Unlocked, vol. 41, p. 101317, 2023. [Online]. Available: https://doi.org/10.1016/j.imu.2023.101317
  42. A. Raza, N. Ullah, J. A. Khan, M. Assam, A. Guzzo, and H. Aljuaid, “Deepbreastcancernet: A novel deep learning model for breast cancer detection using
    ultrasound images,” Applied sciences, vol. 13, no. 4, p. 2082, 2023. [Online]. Available: https://doi.org/10.3390/app13042082
  43. A. M. Alhassan, “An improved breast cancer classification with hybrid chaotic sand cat and remora optimization feature selection algorithm,” PLoS ONE,
    vol. 19, no. 4, 2024. [Online]. Available: https://doi.org/10.1371/journal.pone.0300622
  44. A. Boulenger, Y. Luo, C. Zhang, and et al, “Deep learning-based system for automatic prediction oftriple-negative breast cancer from ultrasound images,”
    Medical Biological Engineering Computing, vol. 61, p. 567–578, 2023. [Online]. Available: https://doi.org/10.1007/s11517-022-02728-4
  45. F. Prinzi, M. Insalaco, A. Orlando, S. Gaglio, and S. Vitabile, “A yolo-based model for breast cancer detection in mammograms,” Cognitive Computation,
    vol. 16, no. 1, p. 107–120, 2023. [Online]. Available: https://doi.org/10.1007/s12559-023-10189-6
  46. N. Hassan, S. Hamad, and K. Mahar, “Yolo-based cad framework with vit transformer for breast mass detection and classification in cesm and ffdm images,”
    Neural computing applications, vol. 36, p. 6467–6496, 2024. [Online]. Available: https://doi.org/10.1007/s00521-023-09364-5
  47. A. Mohammed and D. Ekmekci, “Breast cancer diagnosis using yolo-based multiscale parallel cnn and flattened threshold swish,” Applied Sciences, vol. 14,
    no. 7, p. 2680, 2024. [Online]. Available: https://doi.org/10.3390/app14072680
  48. S. R. Kebede and et al., “Dual view deep learning for enhanced breast cancer screening using mammography,” Scientific Reports, vol. 14, no. 1, p. 3839, 2024.
    [Online]. Available: https://doi.org/10.1038/s41598-023-50797-8
  49. G. Aly, M. Marey, S. El-Sayed, and F. Tolba, “Yolo based breast masses detection and classification in full-field digital mammograms,” Computer Methods
    and Programs in Biomedicine, vol. 200, p. 105823, 2021. [Online]. Available: https://doi.org/10.1016/j.cmpb.2020.105823
  50.  J. Terven, M. C ´ordova-Esparza, and A. Romero-Gonz ´alez, “A comprehensive review of yolo architectures in computer vision: From yolov1 to yolov8 and
    yolo-nas,” Machine Learning and Knowledge Extraction, vol. 5, no. 4, p. 1680–1716, 2023. [Online]. Available: https://doi.org/10.3390/make5040083
  51. X. Wang, H. Gao, Z. Jia, and Z. Li, “Bl-yolov8: An improved road defect detection model based on yolov8,” Sensors, vol. 23, no. 20, p. 8361, 2023. [Online].
    Available: https://doi.org/10.3390/s23208361
  52. G. Yang, J. Wang, Z. Nie, H. Yang, and S. Yu, “A lightweight yolov8 tomato detection algorithm combining feature enhancement and attention,” Agronomy,
    vol. 13, no. 7, p. 1824, 2023. [Online]. Available: https://doi.org/10.3390/agronomy13071824
  53. Q. Hu, M. Whitney, and L. Giger, “A deep learning methodology for improved breast cancer diagnosis using multiparametric mri,” Scientific Reports, vol. 10,
    no. 1, p. 10536, 2020. [Online]. Available: https://doi.org/10.1038/s41598-020-67441-4
  54. R. Adam, K. Dell’Aquila, T. M. L. Hodges, and Q. Duong, “Deep learning applications to breast cancer detection by magnetic resonance imaging: a literature
    review,” Breast Cancer Research, vol. 25, no. 1, p. 87, 2023. [Online]. Available: https://doi.org/10.1186/s13058-023-01687-4
  55. L. Shen, R. Margolies, H. Rothstein, E. Fluder, R. McBride, and W. Sieh, “Deep learning to improve breast cancer detection on screening mammography,”
    Scientific Reports, vol. 9, no. 1, p. 12495, 2019. [Online]. Available: https://doi.org/10.1038/s41598-019-48995-4
Volume 19, Issue 1
Winter 2026
Pages 59-70

  • Receive Date 26 December 2024
  • Revise Date 09 April 2025
  • Accept Date 08 October 2025