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 条
  • [1] Breast mass segmentation in ultrasound with selective kernel U-Net convolutional neural network
    Byra, Michal
    Jarosik, Piotr
    Szubert, Aleksandra
    Galperin, Michael
    Ojeda-Fournier, Haydee
    Olson, Linda
    O'Boyle, Mary
    Comstock, Christopher
    Andre, Michael
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 61
  • [2] Enhanced U-Net segmentation with ensemble convolutional neural network for automated skin disease classification
    Dasari Anantha Reddy
    Swarup Roy
    Sanjay Kumar
    Rakesh Tripathi
    Knowledge and Information Systems, 2023, 65 : 4111 - 4156
  • [3] Enhanced U-Net segmentation with ensemble convolutional neural network for automated skin disease classification
    Reddy, Dasari Anantha
    Roy, Swarup
    Kumar, Sanjay
    Tripathi, Rakesh
    KNOWLEDGE AND INFORMATION SYSTEMS, 2023, 65 (10) : 4111 - 4156
  • [4] Breast tumor segmentation in ultrasound images: comparing U-net and U-net + +
    de Oliveira, Carlos Eduardo Gonçalves
    Vieira, Sílvio Leão
    Paranaiba, Caio Felipe Brito
    Itikawa, Emerson Nobuyuki
    Research on Biomedical Engineering, 2025, 41 (01)
  • [5] Breast Segmentation in MRI via U-Net Deep Convolutional Neural Networks
    Piantadosi, Gabriele
    Sansone, Mario
    Sansone, Carlo
    2018 24TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2018, : 3917 - 3922
  • [6] A deep Residual U-Net convolutional neural network for automated lung segmentation in computed tomography images
    Khanna, Anita
    Londhe, Narendra D.
    Gupta, S.
    Semwal, Ashish
    BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2020, 40 (03) : 1314 - 1327
  • [7] Investigation of the applicability of the convolutional neural network U-Net to a problem of segmentation of aircraft images
    Gavrilov, D. A.
    COMPUTER OPTICS, 2021, 45 (04) : 575 - +
  • [8] Segmentation of Breast Cancer on Ultrasound Images using Attention U-Net Model
    Laghmati, Sara
    Hicham, Khadija
    Cherradi, Bouchaib
    Hamida, Soufiane
    Tmiri, Amal
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (08) : 770 - 778
  • [9] Nanoparticle Segmentation Based on U-Net Convolutional Neural Network
    Zhang Fang
    Wu Yue
    Xiao Zhitao
    Geng Lei
    Wu Jun
    Liu Yanbei
    Wang Wen
    LASER & OPTOELECTRONICS PROGRESS, 2019, 56 (06)
  • [10] BREAST LESION SEGMENTATION AND CLASSIFICATION USING U-NET SALIENCY ESTIMATION AND EXPLAINABLE RESIDUAL CONVOLUTIONAL NEURAL NETWORK
    Fatima, Mamuna
    Khan, Muhammad attique
    Shaheen, Saima
    Albarakati, Hussain mobarak
    Wang, Shuihua
    Jilani, Syeda fizzah
    Shabaz, Mohammad
    FRACTALS-COMPLEX GEOMETRY PATTERNS AND SCALING IN NATURE AND SOCIETY, 2024,