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 条
  • [21] Automatic breast mass segmentation in ultrasound images with U-Net and resolution enhancement blocks
    Rahmani, Ali Ahmad
    Shirazi, Ali Asghar Beheshti
    Behnam, Hamid
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 94
  • [22] Comparison of deep convolutional neural networks for classification of breast ultrasound images
    Park J.
    Kim Y.
    Ryu C.-W.
    Kim H.
    Transactions of the Korean Institute of Electrical Engineers, 2021, 70 (01) : 176 - 183
  • [23] An Ensemble of U-Net Models for Kidney Tumor Segmentation With CT Images
    Causey, Jason
    Stubblefield, Jonathan
    Qualls, Jake
    Fowler, Jennifer
    Cai, Lingrui
    Walker, Karl
    Guan, Yuanfang
    Huang, Xiuzhen
    IEEE-ACM TRANSACTIONS ON COMPUTATIONAL BIOLOGY AND BIOINFORMATICS, 2022, 19 (03) : 1387 - 1392
  • [24] Residual U-Net approach for thyroid nodule detection and classification from thyroid ultrasound images
    Agustin, Sheeja
    Sruthy, S.
    James, Ajay
    Simon, Philomina
    AUTOMATIKA, 2024, 65 (03) : 726 - 737
  • [25] Segmentation of Lymph Nodes in Ultrasound Images Using U-Net Convolutional Neural Networks and Gabor-Based Anisotropic Diffusion
    Chen, Haobo
    Wang, Yuqun
    Shi, Jie
    Xiong, Jingyu
    Jiang, Jianwei
    Chang, Wanying
    Chen, Man
    Zhang, Qi
    JOURNAL OF MEDICAL AND BIOLOGICAL ENGINEERING, 2021, 41 (06) : 942 - 952
  • [26] Segmentation of Lymph Nodes in Ultrasound Images Using U-Net Convolutional Neural Networks and Gabor-Based Anisotropic Diffusion
    Haobo Chen
    Yuqun Wang
    Jie Shi
    Jingyu Xiong
    Jianwei Jiang
    Wanying Chang
    Man Chen
    Qi Zhang
    Journal of Medical and Biological Engineering, 2021, 41 : 942 - 952
  • [27] U-Net Convolutional Neural Networks for breast IR imaging segmentation on frontal and lateral view
    Carlos de Carvalho, Elisson
    Martins Coelho, Alessandra
    Conci, Aura
    de Freitas Oliveira Baffa, Matheus
    COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2023, 11 (03) : 311 - 316
  • [28] Segmentation of Breast Ultrasound Images using Densely Connected Deep Convolutional Neural Network and Attention Gates
    Thirusangu, Niranjan
    Almekkawy, Mohamed
    2021 IEEE UFFC LATIN AMERICA ULTRASONICS SYMPOSIUM (LAUS), 2021,
  • [29] SEGMENTATION OF PORES IN CARBON FIBER REINFORCED POLYMERS USING THE U-NET CONVOLUTIONAL NEURAL NETWORK
    Yosifov, Miroslav
    Weinberger, Patrick
    Plank, Bernhard
    Froehler, Bernhard
    Hoeglinger, Markus
    Kastner, Johann
    Heinzl, Christoph
    18TH YOUTH SYMPOSIUM ON EXPERIMENTAL SOLID MECHANICS, YSESM 2023, 2023, 42 : 87 - 93
  • [30] Breast Ultrasound Images Augmentation and Segmentation Using GAN with Identity Block and Modified U-Net 3+
    Alruily, Meshrif
    Said, Wael
    Mostafa, Ayman Mohamed
    Ezz, Mohamed
    Elmezain, Mahmoud
    SENSORS, 2023, 23 (20)