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 条
  • [31] A Novel Breast Tumor Classification in Ultrasound Images, Using Deep Convolutional Neural Network
    Zeimarani, Bashir
    Costa, M. G. F.
    Nurani, Nilufar Z.
    Costa Filho, Cicero F. F.
    XXVI BRAZILIAN CONGRESS ON BIOMEDICAL ENGINEERING, CBEB 2018, VOL. 2, 2019, 70 (02): : 89 - 94
  • [32] DBU-Net: Dual branch U-Net for tumor segmentation in breast ultrasound images
    Pramanik, Payel
    Pramanik, Rishav
    Schwenker, Friedhelm
    Sarkar, Ram
    PLOS ONE, 2023, 18 (11):
  • [33] AAU-Net: An Adaptive Attention U-Net for Breast Lesions Segmentation in Ultrasound Images
    Chen, Gongping
    Li, Lei
    Dai, Yu
    Zhang, Jianxun
    Yap, Moi Hoon
    IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (05) : 1289 - 1300
  • [34] Utilization of the U-Net Convolutional Neural Network and Its Modifications for Segmentation of Tundra Lakes in Satellite Optical Images
    Abramova, I. A.
    Demchev, D. M.
    Kharyutkina, E. V.
    Savenkova, E. N.
    Sudakow, I. A.
    ATMOSPHERIC AND OCEANIC OPTICS, 2024, 37 (02) : 205 - 210
  • [35] Classification of Deep Convolutional Neural Network in Thyroid Ultrasound Images
    Hui, Ran
    Chen, Jiaxing
    Liu, Yu
    Shi, Lin
    Fu, Chao
    Ishsay, Ostfeld
    JOURNAL OF MEDICAL IMAGING AND HEALTH INFORMATICS, 2020, 10 (08) : 1943 - 1948
  • [36] Segmentation of the Retinal Reflex in Bruckner Test Images Using U-Net Convolutional Network
    Santos da Silva, Italo Francyles
    Sousa de Almeida, Joao Dallyson
    Meireles Teixeira, Jorge Antonio
    Braz Junior, Geraldo
    de Paiva, Anselmo Cardoso
    IMAGE ANALYSIS AND RECOGNITION (ICIAR 2018), 2018, 10882 : 679 - 686
  • [37] Semantic Segmentation of Lung Tissues in HRCT Images by Means of a U-Net Convolutional Network
    Hernandez-Juarez, Sarahi
    Mejia-Rodriguez, Aldo R.
    Arce-Santana, Edgar R.
    Charleston-Villalobos, S.
    Aljama-Corrales, A. T.
    Gonzalez-Camarena, R.
    Mejia-Avila, M.
    VIII LATIN AMERICAN CONFERENCE ON BIOMEDICAL ENGINEERING AND XLII NATIONAL CONFERENCE ON BIOMEDICAL ENGINEERING, 2020, 75 : 426 - 434
  • [38] Classification of Breast Cancer Lesions in Ultrasound Images by Using Attention Layer and Loss Ensemble in Deep Convolutional Neural Networks
    Kalafi, Elham Yousef
    Jodeiri, Ata
    Setarehdan, Seyed Kamaledin
    Lin, Ng Wei
    Rahmat, Kartini
    Taib, Nur Aishah
    Ganggayah, Mogana Darshini
    Dhillon, Sarinder Kaur
    DIAGNOSTICS, 2021, 11 (10)
  • [39] Deep Learning Based Automatic Liver Volume Estimation and Segmentation via U-net Convolutional Neural Network
    Marlatt, B.
    Pettit, R.
    Havelka, J.
    Corr, S. J.
    Rana, A.
    AMERICAN JOURNAL OF TRANSPLANTATION, 2021, 21 : 797 - 797
  • [40] HIC-net: A deep convolutional neural network model for classification of histopathological breast images
    Ozturk, Saban
    Akdemir, Bayram
    COMPUTERS & ELECTRICAL ENGINEERING, 2019, 76 : 299 - 310