Enhancing breast cancer segmentation and classification: An Ensemble Deep Convolutional Neural Network and U-net approach on ultrasound images

被引:9
|
作者
Islam, Md Rakibul [2 ,3 ]
Rahman, Md Mahbubur [2 ]
Ali, Md Shahin [1 ,3 ]
Nafi, Abdullah Al Nomaan [2 ]
Alam, Md Shahariar [2 ]
Godder, Tapan Kumar [2 ]
Miah, Md Sipon [2 ]
Islam, Md Khairul [1 ,3 ]
机构
[1] Islamic Univ, Dept Biomed Engn, Kushtia 7003, Bangladesh
[2] Islamic Univ, Dept Informat & Commun Technol, Kushtia 7003, Bangladesh
[3] Islamic Univ, Dept Biomed Engn, Bioimaging Res Lab, Kushtia 7003, Bangladesh
来源
MACHINE LEARNING WITH APPLICATIONS | 2024年 / 16卷
关键词
Breast cancer; Ultrasound images; Preprocessing; Segmentation; U -net model; Ensemble model;
D O I
10.1016/j.mlwa.2024.100555
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Breast cancer is a condition where the irregular growth of breast cells occurs uncontrollably, leading to the formation of tumors. It poses a significant threat to women's lives globally, emphasizing the need for enhanced methods of detecting and categorizing the disease. In this work, we propose an Ensemble Deep Convolutional Neural Network (EDCNN) model that exhibits superior accuracy compared to several transfer learning models and the Vision Transformer model. Our EDCNN model integrates the strengths of the MobileNet and Xception models to improve its performance in breast cancer detection and classification. We employ various preprocessing techniques, including image resizing, data normalization, and data augmentation, to prepare the data for analysis. By following these measures, the formatting is optimized, and the model's capacity to make generalizations is improved. We trained and evaluated our proposed EDCNN model using ultrasound images, a widely available modality for breast cancer imaging. The outcomes of our experiments illustrate that the EDCNN model attains an exceptional accuracy of 87.82% on Dataset 1 and 85.69% on Dataset 2, surpassing the performance of several well-known transfer learning models and the Vision Transformer model. Furthermore, an AUC value of 0.91 on Dataset 1 highlights the robustness and effectiveness of our proposed model. Moreover, we highlight the incorporation of the Grad-CAM Explainable Artificial Intelligence (XAI) technique to improve the interpretability and transparency of our proposed model. Additionally, we performed image segmentation using the U-Net segmentation technique on the input ultrasound images. This segmentation process allowed for the identification and isolation of specific regions of interest, facilitating a more comprehensive analysis of breast cancer characteristics. In conclusion, the study presents a creative approach to detecting and categorizing breast cancer, demonstrating the superior performance of the EDCNN model compared to well-established transfer learning models. Through advanced deep learning techniques and image segmentation, this study contributes to improving diagnosis and treatment outcomes in breast cancer.
引用
收藏
页数:12
相关论文
共 50 条
  • [31] Road Detection and Centerline Extraction Via Deep Recurrent Convolutional Neural Network U-Net
    Yang, Xiaofei
    Li, Xutao
    Ye, Yunming
    Lau, Raymond Y. K.
    Zhang, Xiaofeng
    Huang, Xiaohui
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (09): : 7209 - 7220
  • [32] Segmentation for mammography classification utilizing deep convolutional neural network
    Kumar Saha, Dip
    Hossain, Tuhin
    Safran, Mejdl
    Alfarhood, Sultan
    Mridha, M. F.
    Che, Dunren
    BMC MEDICAL IMAGING, 2024, 24 (01):
  • [33] Segmentation and Classification for Breast Cancer Ultrasound Images Using Deep Learning Techniques: A Review
    Jahwar, Alan Fuad
    Abdulazeez, Adnan Mohsin
    2022 IEEE 18TH INTERNATIONAL COLLOQUIUM ON SIGNAL PROCESSING & APPLICATIONS (CSPA 2022), 2022, : 225 - 230
  • [34] EfficientU-Net: A Novel Deep Learning Method for Breast Tumor Segmentation and Classification in Ultrasound Images
    Dar, Mohsin Furkh
    Ganivada, Avatharam
    NEURAL PROCESSING LETTERS, 2023, 55 (08) : 10439 - 10462
  • [35] EfficientU-Net: A Novel Deep Learning Method for Breast Tumor Segmentation and Classification in Ultrasound Images
    Mohsin Furkh Dar
    Avatharam Ganivada
    Neural Processing Letters, 2023, 55 : 10439 - 10462
  • [36] Automatic Tumor Segmentation by Means of Deep Convolutional U-Net With Pre-Trained Encoder in PET Images
    Lu, Yongzhou
    Lin, Jinqiu
    Chen, Sheng
    He, Hui
    Cai, Yuantao
    IEEE ACCESS, 2020, 8 : 113636 - 113648
  • [37] Convolutional Neural Network for Classification of Histopathology Images for Breast Cancer Detection
    Narayanan, Barath Narayanan
    Krishnaraja, Vignesh
    Ali, Redha
    PROCEEDINGS OF THE 2019 IEEE NATIONAL AEROSPACE AND ELECTRONICS CONFERENCE (NAECON), 2019, : 291 - 295
  • [38] A Lightweight Deep Convolutional Neural Network Implemented on FPGA and Android Devices for Detection of Breast Cancer Using Ultrasound Images
    Vinod, Aditya
    Guddati, Prabhav
    Panda, Amit Kumar
    Tripathy, Rajesh Kumar
    IEEE ACCESS, 2024, 12 : 179190 - 179203
  • [39] Accurate segmentation for different types of lung nodules on CT images using improved U-Net convolutional network
    Zhang, Xiaofang
    Liu, Xiaomin
    Zhang, Bin
    Dong, Jie
    Zhang, Bin
    Zhao, Shujun
    Li, Suxiao
    MEDICINE, 2021, 100 (40) : E27491
  • [40] Acral melanocytic lesion segmentation with a convolution neural network (U-Net)
    Jaworek-Korjakowska, Joanna
    MEDICAL IMAGING 2019: COMPUTER-AIDED DIAGNOSIS, 2019, 10950