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 条
  • [21] Modified U-Net and Convolutional Networks for Breast Cancer Segmentation and Classification with New Texture Descriptors
    Porwal, Priya
    Singh, Ajay Shanker
    Kannapiran, Thirunavukkarasu
    INTERNATIONAL JOURNAL OF IMAGE AND GRAPHICS, 2024,
  • [22] Retinal Vessel Segmentation Algorithm Based on U-NET Convolutional Neural Network
    Zhang, Yun-Hao
    Wang, Jie-Sheng
    Zhang, Zhi-Hao
    ENGINEERING LETTERS, 2023, 31 (04) : 1837 - 1846
  • [23] UCapsNet: A Two-Stage Deep Learning Model Using U-Net and Capsule Network for Breast Cancer Segmentation and Classification in Ultrasound Imaging
    Madhu, Golla
    Bonasi, Avinash Meher
    Kautish, Sandeep
    Almazyad, Abdulaziz S.
    Mohamed, Ali Wagdy
    Werner, Frank
    Hosseinzadeh, Mehdi
    Shokouhifar, Mohammad
    CANCERS, 2024, 16 (22)
  • [24] Microaneurysms segmentation with a U-Net based on recurrent residual convolutional neural network
    Kou, Caixia
    Li, Wei
    Liang, Wei
    Yu, Zekuan
    Hao, Jianchen
    JOURNAL OF MEDICAL IMAGING, 2019, 6 (02)
  • [25] Improved U-net MALF model for lesion segmentation in breast ultrasound images
    Tong, Ying
    Liu, Yangyang
    Zhao, Manxue
    Meng, Lin
    Zhang, Jiachao
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2021, 68
  • [26] Nipple Segmentation and Localization Using Modified U-Net on Breast Ultrasound Images
    Zhuang, Zhemin
    Raj, Alex Noel Joseph
    Jain, Atyant
    Ruban, Nersisson
    Chaurasia, Saksham
    Li, Nan
    Lakshmanan, M.
    Murugappan, M.
    JOURNAL OF MEDICAL IMAGING AND HEALTH INFORMATICS, 2019, 9 (09) : 1827 - 1837
  • [27] LIGHTWEIGHT U-NET FOR LESION SEGMENTATION IN ULTRASOUND IMAGES
    Li, Yingping
    Chouzenoux, Emilie
    Charmettant, Benoit
    Benatsou, Baya
    Lamarque, Jean-Philippe
    Lassau, Nathalie
    2021 IEEE 18TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (ISBI), 2021, : 611 - 615
  • [28] Residual u-net with Self-Attention based deep convolutional adaptive capsule network for liver cancer segmentation and classification
    Archana, R.
    Anand, L.
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 105
  • [29] Automatic Breast and Fibroglandular Tissue Segmentation in Breast MRI Using Deep Learning by a Fully-Convolutional Residual Neural Network U-Net
    Zhang, Yang
    Chen, Jeon-Hor
    Chang, Kai-Ting
    Park, Vivian Youngjean
    Kim, Min Jung
    Chan, Siwa
    Chang, Peter
    Chow, Daniel
    Luk, Alex
    Kwong, Tiffany
    Su, Min-Ying
    ACADEMIC RADIOLOGY, 2019, 26 (11) : 1526 - 1535
  • [30] Breast cancer mitotic cell detection using cascade convolutional neural network with U-Net
    Lu, Xi
    You, Zejun
    Sun, Miaomiao
    Wu, Jing
    Zhang, Zhihong
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2020, 18 (01) : 673 - 695