Multimodal breast cancer hybrid explainable computer-aided diagnosis using medical mammograms and ultrasound Images

被引:7
作者
Al-Tam, Riyadh M. [1 ,2 ]
Al-Hejri, Aymen M. [1 ,2 ]
Alshamrani, Sultan S. [3 ]
Al-antari, Mugahed A. [4 ]
Narangale, Sachin M. [5 ]
机构
[1] Swami Ramanand Teerth Marathwada Univ, Sch Computat Sci, Nanded 431606, Maharashtra, India
[2] Univ Albaydha, Fac Adm & Comp Sci, Albaydha, Yemen
[3] Taif Univ, Coll Comp & Informat Technol, Dept Informat Technol, POB 11099, Taif 21944, Saudi Arabia
[4] Sejong Univ, Coll AI Convergence, Daeyang AI Ctr, Dept Artificial Intelligence & Data Sci, Seoul 05006, South Korea
[5] Swami Ramanand Teerth Marathwada Univ, Sch Media Studies, Nanded 431606, Maharashtra, India
基金
新加坡国家研究基金会;
关键词
Breast cancer detection and classification; BI-RAD scores; Ensemble transfer learning; Hybrid CAD system; Explainable heatmap; DIGITAL MAMMOGRAMS; AUTOMATIC DETECTION; SEGMENTATION; FULL; CLASSIFICATION;
D O I
10.1016/j.bbe.2024.08.007
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Breast cancer is a prevalent global disease where early detection is crucial for effective treatment and reducing mortality rates. To address this challenge, a novel Computer-Aided Diagnosis (CAD) framework leveraging Artificial Intelligence (AI) techniques has been developed. This framework integrates capabilities for the simultaneous detection and classification of breast lesions. The AI-based CAD framework is meticulously structured into two pipelines (Stage 1 and Stage 2). The first pipeline (Stage 1) focuses on detectable cases where lesions are identified during the detection task. The second pipeline (Stage 2) is dedicated to cases where lesions are not initially detected. Various experimental scenarios, including binary (benign vs. malignant) and multi- class classifications based on BI-RADS scores, were conducted for training and evaluation. Additionally, a verification and validation (V&V) &V) scenario was implemented to assess the reliability of the framework using unseen multimodal datasets for both binary and multi-class tasks. For the detection tasks, the recent AI detectors like YOLO (You Only Look Once) variants were fine-tuned and optimized to localize breast lesions. For classification tasks, hybrid AI models incorporating ensemble convolutional neural networks (CNNs) and the attention mechanism of Vision Transformers were proposed to enhance prediction performance. The proposed AI-based CAD framework was trained and evaluated using various multimodal ultrasound datasets (BUSI and US2) and mammogram datasets (MIAS, INbreast, real private mammograms, KAU-BCMD, and CBIS-DDSM), either individually or in merged forms. Visual t-SNE techniques were applied to visually harmonize data distributions across ultrasound and mammogram datasets for effective various datasets merging. To generate visually explainable heatmaps in both pipelines (stages 1 and 2), Grad-CAM was utilized. These heatmaps assisted in finalizing detected boxes, especially in stage 2 when the AI detector failed to automatically detect breast lesions. The highest evaluation metrics achieved for merged dataset (BUSI, INbreast, and MIAS) were 97.73% accuracy and 97.27% mAP50 in the first pipeline. In the second pipeline, the proposed CAD achieved 91.66% accuracy with 95.65% mAP50 on MIAS and 95.65% accuracy with 96.10% mAP50 on the merged dataset (INbreast and MIAS). Meanwhile, exceptional performance was demonstrated using BI-RADS scores, achieving 87.29% accuracy, 91.68% AUC, 86.72% mAP50, and 64.75% mAP50-95 on a combined dataset of INbreast and CBIS-DDSM. These results underscore the practical significance of the proposed CAD framework in automatically annotating suspected lesions for radiologists.
引用
收藏
页码:731 / 758
页数:28
相关论文
共 106 条
[1]   A hybrid lightweight breast cancer classification framework using the histopathological images [J].
Addo, Daniel ;
Zhou, Shijie ;
Sarpong, Kwabena ;
Nartey, Obed T. ;
Abdullah, Muhammed A. ;
Ukwuoma, Chiagoziem C. ;
Al-antari, Mugahed A. .
BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2024, 44 (01) :31-54
[2]   Novel prediction model on OSCC histopathological images via deep transfer learning combined with Grad-CAM interpretation [J].
Afify, Heba M. ;
Mohammed, Kamel K. ;
Hassanien, Aboul Ella .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 83
[3]   A combined feature-vector based multiple instance learning convolutional neural network in breast cancer classification from histopathological images [J].
Ahmed, Mohiuddin ;
Islam, Md Rabiul .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 84
[4]   Deep YOLO-Based Detection of Breast Cancer Mitotic-Cells in Histopathological Images [J].
Al Zorgani, Maisun Mohamed ;
Mehmood, Irfan ;
Ugail, Hassan .
PROCEEDINGS OF 2021 INTERNATIONAL CONFERENCE ON MEDICAL IMAGING AND COMPUTER-AIDED DIAGNOSIS, MICAD 2021, 2022, 784 :335-342
[5]   Evaluation of deep learning detection and classification towards computer-aided diagnosis of breast lesions in digital X-ray mammograms [J].
Al-antari, Mugahed A. ;
Han, Seung-Moo ;
Kim, Tae-Seong .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 196
[6]   An Automatic Computer-Aided Diagnosis System for Breast Cancer in Digital Mammograms via Deep Belief Network [J].
Al-antari, Mugahed A. ;
Al-masni, Mohammed A. ;
Park, Sung-Un ;
Park, JunHyeok ;
Metwally, Mohamed K. ;
Kadah, Yasser M. ;
Han, Seung-Moo ;
Kim, Tae-Seong .
JOURNAL OF MEDICAL AND BIOLOGICAL ENGINEERING, 2018, 38 (03) :443-456
[7]   Dataset of breast ultrasound images [J].
Al-Dhabyani, Walid ;
Gomaa, Mohammed ;
Khaled, Hussien ;
Fahmy, Aly .
DATA IN BRIEF, 2020, 28
[8]   ETECADx: Ensemble Self-Attention Transformer Encoder for Breast Cancer Diagnosis Using Full-Field Digital X-ray Breast Images [J].
Al-Hejri, Aymen M. ;
Al-Tam, Riyadh M. ;
Fazea, Muneer ;
Sable, Archana Harsing ;
Lee, Soojeong ;
Al-antari, Mugahed A. .
DIAGNOSTICS, 2023, 13 (01)
[9]   Analyzing Histological Images Using Hybrid Techniques for Early Detection of Multi-Class Breast Cancer Based on Fusion Features of CNN and Handcrafted [J].
Al-Jabbar, Mohammed ;
Alshahrani, Mohammed ;
Senan, Ebrahim Mohammed ;
Ahmed, Ibrahim Abdulrab .
DIAGNOSTICS, 2023, 13 (10)
[10]   Multi-Method Diagnosis of Histopathological Images for Early Detection of Breast Cancer Based on Hybrid and Deep Learning [J].
Al-Jabbar, Mohammed ;
Alshahrani, Mohammed ;
Senan, Ebrahim Mohammed ;
Ahmed, Ibrahim Abdulrab .
MATHEMATICS, 2023, 11 (06)