BreastCDNet: Breast Cancer Detection Neural Network, Classification and Localization
Author:
Mahichi Hassan1, Ghods Vahid1, Sohrabi Mohammad Karim1, Sabbaghi Arash1
Affiliation:
1. Islamic Azad University
Abstract
Abstract
Breast cancer remains a significant health concern globally, necessitating advanced detection methods for improved patient outcomes. Convolutional neural networks, CNN, have revolutionized object detection by automating the feature extraction process and allowing for the efficient localization and classification of objects within images. BreastCDNet introduces a novel approach to breast cancer detection using CNN. This pioneering CNN method is tailored specifically for the task of ultrasound breast cancer detection and localization, offering a streamlined and effective approach that significantly enhances diagnostic accuracy. Multi-task learning is leveraged by the proposed model, with simultaneous consideration of breast ultrasound image classification and bounding box regression for lesion localization. Intricate features from medical images are extracted by BreastCDNet architecture, facilitating both classification (benign or malignant) and object localization. Separate data generators are employed for classification and bounding box labels to optimize model training. The model's effectiveness is demonstrated by experimental results on the BUSI dataset, where exceptional classification accuracy and precise bounding box predictions are achieved. Key components, including convolutional layers for feature extraction, max-pooling layers for spatial reduction, fully connected layers for predictions, and specific output layers for classification and bounding box regression, are encompassed by the model architecture. The BreastCDNet model is a high-performance breast cancer detection model that achieved 99.14% training accuracy, 97.70% validation accuracy, 0.99 ROC AUC, and 0.97 F1-score on the training set. It is also capable of accurately localizing breast cancer lesions with an IOU score of 0.95.
Publisher
Research Square Platform LLC
Reference45 articles.
1. Bray, F., Ferlay, J., Soerjomataram, I., Siegel, R.: “Global Cancer Statistics : GLOBOCAN Estimates of Incidence and Mortality Worldwide for 36 Cancers in 185 Countries. CA: A Cancer Journal for Clinicians. CA,” CA Cancer J. Clin., vol. 68, 2018. (2018) 2. Cancer statistics, 2023;Siegel RL;CA Cancer J Clin 3. Bray, F., Ferlay, J., Soerjomataram, I., Siegel, R.L., Torre, L.A., Jemal, A.: “Global cancer statistics 2018: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries,” CA Cancer J Clin, vol. 68, no. 6, pp. 394–424, Nov. (2018). 10.3322/caac.21492 4. Rony, J., Belharbi, S., Dolz, J., Ben Ayed, I., McCaffrey, L., Granger, E.: “Deep Weakly-Supervised Learning Methods for Classification and Localization in Histology Images: A Survey,” Machine Learning for Biomedical Imaging, vol. 2, no. March 2023, (2023). 10.59275/j.melba.2023-5g54 5. American Cancer Society nutrition and physical activity guideline for cancer survivors;Rock CL;CA Cancer J Clin,2022
|
|