The CNN Model with YOLO Architecture for Ultrasonography Images in Early Breast Cancer Detection

Astika Ayuningtyas, Hero Wintolo, Arwin Datumaya Wahyudi Sumari, Emy Setyaningsih, Asih Pujiastuti, Anton Setiawan Honggowibowo, Edi Triono Nuryatno, Anggraini Kusumaningrum
Pages 1116-1128
Abstract Views 763
PDF Downloads 838
Crossref Citations -
DOI https://doi.org/10.47738/jads.v6i2.587

Abstract


The rapid development of object detection technology has opened new opportunities in the healthcare sector, especially in early cancer detection. This paper presents a deep learning-based breast cancer detection system using ultrasound images. The primary goal of this study is to create a model that can effectively differentiate between malignant and benign breast tumors, assisting in early diagnosis. The proposed system employs the Convolutional Neural Network (CNN) algorithm with You Only Look Once version 5 (YOLOv5) architecture, which is renowned for its high speed and accuracy in object detection tasks. A dataset comprising 10,954 ultrasound images was used to train the model, with 70% allocated for training, 20% for validation, and 10% for testing. The study reveals that the model achieves a high accuracy rate of 92.8% for malignant tumor detection and 99.1% for benign tumors, with precision rates of 99.6% for malignant tumors and 97.5% for benign tumors. These results demonstrate the feasibility of the proposed model as a reliable tool for early breast cancer detection. The findings highlight the potential of deep learning in medical image processing, suggesting that this technology could be further developed into an accessible, efficient early detection system for breast cancer in clinical settings. Future research could explore the integration of additional imaging modalities and the application of this model in real-world healthcare environments

Keywords


CNN; YOLO; Deep Learning; Object Detection; Breast Cancer

Full Text:

PDF

References


K. Alemerien, S. Alsarayreh, and E. Altarawneh, “Diagnosing Cardiovascular Diseases using Optimized Machine Learning Algorithms with GridSearchCV,” Journal of Applied Data Sciences, vol. 5, no. 4, pp. 1539–1552, 2024, doi: 10.47738/jads.v5i4.280.

A. Prasetya Wibawa, T. Widiyaningtyas, R. Ridwan Raja, H. Darwis, and A. Nafalski, “Congestion Predictive Modelling on Network Dataset Using Ensemble Deep Learning,” Journal of Applied Data Sciences, vol. 5, no. 4, pp. 1597–1613, 2024, doi: 10.47738/jads.

A. A. Abu-Shareha, H. Qutaishat, and A. Al-Khayat, “A Framework for Diabetes Detection Using Machine Learning and Data Preprocessing,” Journal of Applied Data Sciences, vol. 5, no. 4, pp. 1654–1667, 2024, doi: 10.47738/jads.v5i4.363.

R. Rizal Isnanto and B. Warsito, “Applied Data Science and Artificial Intelligence for Tourism and Hospitality Industry in Society 5.0: A Review,” Journal of Applied Data Sciences, vol. 5, no. 4, pp. 1566–1578, 2024, doi: 10.47738/jads.v5i4.300.

D. Dimitrov et al., “Nationwide analysis of the breast cancer guidelines adherence in Bulgaria,” Clinical Surgical Oncology, vol. 3, no. 2, p. 100044, 2024, doi: https://doi.org/10.1016/j.cson.2024.100044.

S.-A. Brown et al., “Advancing the care of individuals with cancer through innovation & technology: Proceedings from the cardiology oncology innovation summit 2020 and 2021,” American Heart Journal Plus: Cardiology Research and Practice, vol. 38, p. 10035

M. Minnoor and V. Baths, “Diagnosis of Breast Cancer Using Random Forests,” Procedia Comput Sci, vol. 218, pp. 429–437, 2023, doi: https://doi.org/10.1016/j.procs.2023.01.025.

A. Peryanto, A. Yudhana, and R. Umar, “Convolutional neural network and support vector machine in classification of flower images,” Khazanah Informatika: Jurnal Ilmu Komputer dan Informatika, vol. 8, no. 1, pp. 1–7, 2022.

M. Fahmi, A. Yudhana, and S. Sunardi, “Pemilahan Sampah Menggunakan Model Klasifikasi Support Vector Machine Gabungan dengan Convolutional Neural Network,” JURIKOM (Jurnal Riset Komputer), vol. 10, no. 1, pp. 76–81, 2023.

M. Sathianarayanan, P.-H. Hsu, and C.-C. Chang, “Extracting disaster location identification from social media images using deep learning,” International Journal of Disaster Risk Reduction, vol. 104, p. 104352, 2024.

F. Alhomayani and M. H. Mahoor, “Deep learning methods for fingerprint-based indoor positioning: A review,” Journal of Location Based Services, vol. 14, no. 3, pp. 129–200, 2020.

L. Alzubaidi et al., “Review of deep learning: concepts, CNN architectures, challenges, applications, future directions,” J Big Data, vol. 8, pp. 1–74, 2021.

V. Sinitsin, O. Ibryaeva, V. Sakovskaya, and V. Eremeeva, “Intelligent bearing fault diagnosis method combining mixed input and hybrid CNN-MLP model,” Mech Syst Signal Process, vol. 180, p. 109454, 2022.

F.-J. Du and S.-J. Jiao, “Improvement of lightweight convolutional neural network model based on YOLO algorithm and its research in pavement defect detection,” Sensors, vol. 22, no. 9, p. 3537, 2022.

Y. Yin, H. Li, and W. Fu, “Faster-YOLO: An accurate and faster object detection method,” Digit Signal Process, vol. 102, p. 102756, 2020.

F. P. May et al., “Community Collaboration to Advance Racial/Ethnic Equity in Colorectal Cancer Screening: Protocol for a Multilevel Intervention to Improve Screening and Follow-up in Community Health Centers,” Contemp Clin Trials, vol. 145, p. 107639, 20

I. A. Kandhro et al., “Performance evaluation of E-VGG19 model: Enhancing real-time skin cancer detection and classification,” Heliyon, vol. 10, no. 10, p. e31488, 2024, doi: https://doi.org/10.1016/j.heliyon.2024.e31488.

H. El Massari, N. Gherabi, S. Mhammedi, Z. Sabouri, H. Ghandi, and F. Qanouni, “Effectiveness of applying Machine Learning techniques and Ontologies in Breast Cancer detection,” Procedia Comput Sci, vol. 218, pp. 2392–2400, 2023, doi: https://doi.org/10.1.

D. Truhn et al., “Encrypted federated learning for secure decentralized collaboration in cancer image analysis,” Med Image Anal, vol. 92, p. 103059, 2024, doi: https://doi.org/10.1016/j.media.2023.103059.

R. Padilla, S. L. Netto, and E. A. B. Da Silva, “A survey on performance metrics for object-detection algorithms,” in 2020 international conference on systems, signals and image processing (IWSSIP), IEEE, 2020, pp. 237–242.

Z.-Q. Zhao, P. Zheng, S. Xu, and X. Wu, “Object detection with deep learning: A review,” IEEE Trans Neural Netw Learn Syst, vol. 30, no. 11, pp. 3212–3232, 2019.

E. H. Lysø, M. B. Hesjedal, J.-A. Skolbekken, and M. Solbjør, “Men’s sociotechnical imaginaries of artificial intelligence for prostate cancer diagnostics – A focus group study,” Soc Sci Med, vol. 347, p. 116771, 2024, doi: https://doi.org/10.1016/j.socsc

S. Mishra, H. K. Thakkar, P. K. Mallick, P. Tiwari, and A. Alamri, “A sustainable IoHT based computationally intelligent healthcare monitoring system for lung cancer risk detection,” Sustain Cities Soc, vol. 72, p. 103079, 2021, doi: https://doi.org/10.10

D. Sharma, R. Kumar, and A. Jain, “An adaptive framework for predicting breast cancer at an early stage,” Measurement: Sensors, vol. 30, p. 100901, 2023, doi: https://doi.org/10.1016/j.measen.2023.100901.

X. Zhang and V. C. S. Lee, “Deep Learning Empowered Decision Support Systems for Thyroid Cancer Detection and Management,” Procedia Comput Sci, vol. 237, pp. 945–954, 2024, doi: https://doi.org/10.1016/j.procs.2024.05.183.

E. Farea, R. A. A. Saleh, H. AbuAlkebash, A. A. R. Farea, and M. A. Al-Ansari, "A hybrid deep learning skin cancer prediction framework," Engineering Science and Technology, an International Journal, vol. 57, p. 101818, 2024, doi: https://doi.org/10.1016/

S. Chaudhury and K. Sau, “A blockchain-enabled internet of medical things system for breast cancer detection in healthcare,” Healthcare Analytics, vol. 4, p. 100221, 2023, doi: https://doi.org/10.1016/j.health.2023.100221.

J. Zhang et al., “A robust and efficient AI assistant for breast tumor segmentation from DCE-MRI via a spatial-temporal framework,” Patterns, vol. 4, no. 9, p. 100826, 2023, doi: https://doi.org/10.1016/j.patter.2023.100826.

A. Singhal, K. K. Agrawal, A. Quezada, A. R. Aguiñaga, S. Jiménez, and S. P. Yadav, “Explainable Artificial Intelligence (XAI) Model for Cancer Image Classification,” CMES - Computer Modeling in Engineering and Sciences, vol. 141, no. 1, pp. 401–441, 2024

M. Idrees and A. Sohail, “Explainable machine learning of the breast cancer staging for designing smart biomarker sensors,” Sensors International, vol. 3, p. 100202, 2022, doi: https://doi.org/10.1016/j.sintl.2022.100202.

N. Papachristou et al., “A Smart Digital Health Platform to Enable Monitoring of Quality of Life and Frailty in Older Patients with Cancer: A Mixed-Methods, Feasibility Study Protocol,” Semin Oncol Nurs, vol. 39, no. 3, p. 151437, 2023, doi: https://doi.o

M. Suleman et al., “Smart MobiNet: A Deep Learning Approach for Accurate Skin Cancer Diagnosis,” Computers, Materials and Continua, vol. 77, no. 3, pp. 3533–3549, 2023, doi: https://doi.org/10.32604/cmc.2023.042365.

H. E. Saroğlu et al., “Machine learning, IoT and 5G technologies for breast cancer studies: A review,” Alexandria Engineering Journal, vol. 89, pp. 210–223, 2024, doi: https://doi.org/10.1016/j.aej.2024.01.043.

K. Bin Abu Bakar, F. T. Zuhra, B. Isyaku, and F. A. Ghaleb, “Optimized Identification with Severity Factors of Gastric Cancer for Internet of Medical Things,” Computers, Materials and Continua, vol. 75, no. 1, pp. 785–798, 2023, doi: https://doi.org/10.32

V. A. Sairam, "Ultrasound Breast Images for Breast Cancer." Accessed: October 26, 2024. [Online]. Available: https://www.kaggle.com/datasets/lytranhoanghieu/ultrasound-images-with-masks-breast-cancer

A. Iqbal and M. Sharif, “PDF-UNet: A semi-supervised method for segmentation of breast tumor images using a U-shaped pyramid-dilated network,” Expert Syst Appl, vol. 221, p. 119718, Jul. 2023, doi: 10.1016/j.eswa.2023.119718.

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.

M. Maaz, "Breast Ultrasound classification." Accessed: October 26, 2024. [Online]. Available: https://www.kaggle.com/datasets/mrmaazoo/breast-ultrasound-classification

The Cancer Imaging Archive, "Breast-Lesions-USG | A Curated Benchmark Dataset for Ultrasound Based Breast Lesion Analysis." Accessed: October 26, 2024. [Online]. Available: https://www.cancerimagingarchive.net/collection/breast-lesions-usg/

M. Hussain, “YOLO-v1 to YOLO-v8, the rise of YOLO and its complementary nature toward digital manufacturing and industrial defect detection,” Machines, vol. 11, no. 7, p. 677, 2023.

D. Wan, R. Lu, B. Hu, J. Yin, S. Shen, and X. Lang, "YOLO-MIF: Improved YOLOv8 with Multi-Information fusion for object detection in Grayscale images," Advanced Engineering Informatics, vol. 62, p. 102709, 2024.

S. Sapakova, A. Sapakov, and Y. Yilibule, “A YOLOv5-Based Model for Real-Time Mask Detection in Challenging Environments,” Procedia Comput Sci, vol. 231, pp. 267–274, 2024.

L. Du and Y. Wang, “Bi-YOLO: A novel object detection network and dataset for components of China heritage buildings,” Journal of Building Engineering, p. 110817, 2024.

G. Yasmine, G. Maha, and M. Hicham, “Anti-drone systems: An attention based improved YOLOv7 model for a real-time detection and identification of multi-airborne target,” Intelligent Systems with Applications, vol. 20, Nov. 2023, doi: 10.1016/j.iswa.2023.2

Q. Aini, N. Lutfiani, H. Kusumah, and M. S. Zahran, “Deteksi dan Pengenalan Objek Dengan Model Machine Learning: Model Yolo,” CESS (Journal of Computer Engineering, System and Science), vol. 6, no. 2, p. 192, 2021.

D. J. N. Patel and D. S. B. Patel, “Real-Time Object Detection: A Benchmarking Study of Yolo Versions.,” Journal of Scholastic Engineering Science and Management, vol. 3, no. 8, pp. 1–6, 2024.

J. S. Murthy, G. M. Siddesh, W.-C. Lai, B. D. Parameshachari, S. N. Patil, and K. L. Hemalatha, “ObjectDetect: A Real‐Time Object Detection Framework for Advanced Driver Assistant Systems Using YOLOv5,” Wirel Commun Mob Comput, vol. 2022, no. 1, p. 944436

C. Dewi, R.-C. Chen, Y.-C. Zhuang, X. Jiang, and H. Yu, “Recognizing road surface traffic signs based on YOLO models considering image flips,” Big data and cognitive computing, vol. 7, no. 1, p. 54, 2023.

A. N. Kadhum and A. N. Kadhum, “Literature Survey on YOLO Models for Face Recognition in Covid-19 Pandemic,” Journal of Image Processing and Intelligent Remote Sensing, no. 34, pp. 27–35, Jul. 2023, doi: 10.55529/jipirs.34.27.35.




DOI: https://doi.org/10.47738/jads.v6i2.587

Refbacks

  • There are currently no refbacks.


Barcode

Journal of Applied Data Sciences

ISSN:2723-6471 (Online)
Publisher:Bright Publisher
Website:http://bright-journal.org/JADS
Email:taqwa@amikompurwokerto.ac.id (principal contact)
  support@bright-journal.org (technical issues)

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0