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

被引:8
|
作者
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
来源
关键词
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 条
  • [41] Vessel lumen segmentation in carotid artery ultrasounds with the U-Net convolutional neural network
    Xie, Meiyan
    Li, Yunzhi
    Xue, Yunzhe
    Huntress, Lauren
    Beckerman, William
    Rahimi, Saum
    Ady, Justin
    Roshan, Usman
    2020 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE, 2020, : 2680 - 2684
  • [42] A U-Net Ensemble for breast lesion segmentation in DCE MRI
    Khaled, Roa'a
    Vidal, Joel
    Vilanova, Joan C.
    Marti, Robert
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 140
  • [43] Lung-Nodule Segmentation Using a Convolutional Neural Network with the U-Net Architecture
    Hernandez-Solis, Vicente
    Tellez-Velazquez, Arturo
    Orantes-Molina, Antonio
    Cruz-Barbosa, Raul
    PATTERN RECOGNITION (MCPR 2021), 2021, 12725 : 335 - 344
  • [44] A Convolutional Neural Network for Skin Lesion Segmentation Using Double U-Net Architecture
    Abid, Iqra
    Almakdi, Sultan
    Rahman, Hameedur
    Almulihi, Ahmed
    Alqahtani, Ali
    Rajab, Khairan
    Alqhatani, Abdulmajeed
    Shaikh, Asadullah
    INTELLIGENT AUTOMATION AND SOFT COMPUTING, 2022, 33 (03): : 1407 - 1421
  • [45] Study on the Optic Cup Segmentation Method With an Improved u-net Convolutional Neural Network
    Wei, Haicheng
    Hu, Wenrui
    Wang, Shengying
    Xiao, Mingxia
    Zhao, Jing
    BASIC & CLINICAL PHARMACOLOGY & TOXICOLOGY, 2020, 126 : 42 - 42
  • [46] Atrous residual convolutional neural network based on U-Net for retinal vessel segmentation
    Wu, Jin
    Liu, Yong
    Zhu, Yuanpei
    Li, Zun
    PLOS ONE, 2022, 17 (08):
  • [47] Automatic lung field segmentation based on the U-net deep neural network
    Zhang Kunpeng
    Sun Xin
    PROCEEDINGS OF 2019 14TH IEEE INTERNATIONAL CONFERENCE ON ELECTRONIC MEASUREMENT & INSTRUMENTS (ICEMI), 2019, : 1670 - 1676
  • [48] REGION SEGMENTATION IN HISTOPATHOLOGICAL BREAST CANCER IMAGES USING DEEP CONVOLUTIONAL NEURAL NETWORK
    Su, Hai
    Liu, Fujun
    Xie, Yuanpu
    Xing, Fuyong
    Meyyappan, Sreenivasan
    Yang, Lin
    2015 IEEE 12TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2015, : 55 - 58
  • [49] 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
  • [50] Nerve Segmentation of Ultrasound Images Bayesian U-Net Models
    Michael, Taryn
    Obagbuwa, Ibidun Christiana
    INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2024, 2024 (01)