Detection of various gastrointestinal tract diseases through a deep learning method with ensemble ELM and explainable AI

被引:0
|
作者
Ahamed, Md. Faysal [1 ]
Nahiduzzaman, Md. [1 ]
Islam, Md. Rabiul [2 ]
Naznine, Mansura [2 ]
Ayari, Mohamed Arselene [3 ]
Khandakar, Amith [3 ]
Haider, Julfikar [4 ]
机构
[1] Rajshahi Univ Engn & Technol, Dept Elect & Comp Engn, Rajshahi 6204, Bangladesh
[2] Rajshahi Univ Engn & Technol, Dept Comp Sci & Engn, Rajshahi 6204, Bangladesh
[3] Qatar Univ, Dept Elect Engn, Doha 2713, Qatar
[4] Manchester Metropolitan Univ, Dept Engn, Chester St, Manchester M1 5GD, England
关键词
Gastrointestinal tract; Depthwise separable convolutional neural; network; Ensemble Extreme Learning Machine (EELM); Grad-CAM; Shapley Additive Explanations (SHAP); Explainable AI (XAI); WIRELESS CAPSULE ENDOSCOPY; AMERICAN SOCIETY; ADVERSE EVENTS; CLASSIFICATION; DIAGNOSIS; RULE;
D O I
10.1016/j.eswa.2024.124908
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The rising prevalence of gastrointestinal (GI) tract disorders worldwide highlights the urgent need for precise diagnosis, as these diseases greatly affect human life and contribute to high mortality rates. Fast identification, accurate classification, and efficient treatment approaches are essential for addressing this critical health issue. Common side effects include abdominal pain, bloating, and discomfort, which can be chronic and debilitating. Nausea and vomiting are also frequent, leading to difficulties in maintaining adequate nutrition and hydration. The current study intends to develop a deep learning (DL)-based approach that automatically classifies GI tract diseases. For the first time, a GastroVision dataset with 8000 images of 27 different GI diseases was utilized in this work to design a computer-aided diagnosis (CAD) system. This study presents a novel lightweight feature extractor with a compact size and minimum number of layers named Parallel Depthwise Separable Convolutional Neural Network (PD-CNN) and a Pearson Correlation Coefficient (PCC) as the feature selector. Furthermore, a robust classifier named the Ensemble Extreme Learning Machine (EELM), combined with pseudo inverse ELM (ELM) and L1 Regularized ELM (RELM), has been proposed to identify diseases more precisely. A hybrid preprocessing technique, including scaling, normalization, and image enhancement techniques such as erosion, CLAHE, sharpening, and Gaussian filtering, are employed to enhance image representation and improve classification performance. The proposed approach consists of twenty-four layers and only 0.815 million parameters with a 9.79 MB model size. The proposed PD-CNN-PCC-EELM extracts essential features, reduces computational overhead, and achieves excellent classification performance on multiclass GI images. The PD-CNN-PCC-EELM achieved the highest precision, recall, f1, accuracy, ROC-AUC, and AUC-PR values of 88.12 +/- 0.332 %, 87.75 +/- 0.348 %, 87.12 +/- 0.324 %, 87.75 %, 98.89 %, and 92 %, respectively, while maintaining a minimum testing time of 0.000001 s. A comparative study utilizes 10-fold cross-validation, ablation study and various state-of-theart (SOTA) transfer learning (TL) models as feature extractors. Then, the PCC and EELM are integrated with TL to generate predictions, notably in terms of performance and real-time processing capability; the proposed model significantly outperforms the other models. Moreover, various explainable AI (XAI) methods, such as SHAP (Shapley Additive Explanations), heatmap, guided heatmap, Grad-Cam (Gradient-weighted Class Activation Mapping), guided Grad-CAM, and guided Saliency mapping, have been employed to explore the interpretability and decision-making capability of the proposed model. Therefore, the model provides practical intelligence for increasing confidence in diagnosing GI diseases in real-world scenarios.
引用
收藏
页数:31
相关论文
共 50 条
  • [41] A novel deep learning method for detection and classification of plant diseases
    Albattah, Waleed
    Nawaz, Marriam
    Javed, Ali
    Masood, Momina
    Albahli, Saleh
    [J]. COMPLEX & INTELLIGENT SYSTEMS, 2022, 8 (01) : 507 - 524
  • [42] A study and comparison of deep learning based potato leaf disease detection and classification techniques using explainable AI
    Paul, Hrithik
    Ghatak, Sayani
    Chakraborty, Sanjay
    Pandey, Saroj Kumar
    Dey, Lopamudra
    Show, Debashis
    Maity, Saikat
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (14) : 42485 - 42518
  • [43] A study and comparison of deep learning based potato leaf disease detection and classification techniques using explainable AI
    Hrithik Paul
    Sayani Ghatak
    Sanjay Chakraborty
    Saroj Kumar Pandey
    Lopamudra Dey
    Debashis Show
    Saikat Maity
    [J]. Multimedia Tools and Applications, 2024, 83 : 42485 - 42518
  • [44] A deep learning-based ensemble method for helmet-wearing detection
    Fan, Zheming
    Peng, Chengbin
    Dai, Licun
    Cao, Feng
    Qi, Jianyu
    Hua, Wenyi
    [J]. PEERJ COMPUTER SCIENCE, 2020, : 1 - 21
  • [45] Explainable Deep Learning-Based Feature Selection and Intrusion Detection Method on the Internet of Things
    Chen, Xuejiao
    Liu, Minyao
    Wang, Zixuan
    Wang, Yun
    [J]. SENSORS, 2024, 24 (16)
  • [46] FireXnet: an explainable AI-based tailored deep learning model for wildfire detection on resource-constrained devices
    Ahmad, Khubab
    Khan, Muhammad Shahbaz
    Ahmed, Fawad
    Driss, Maha
    Boulila, Wadii
    Alazeb, Abdulwahab
    Alsulami, Mohammad
    Alshehri, Mohammed S.
    Ghadi, Yazeed Yasin
    Ahmad, Jawad
    [J]. FIRE ECOLOGY, 2023, 19 (01)
  • [47] GI-Net: Anomalies Classification in Gastrointestinal Tract through Endoscopic Imagery with Deep Learning
    Gamage, Chathurika
    Wijesinghe, Isuru
    Chitraranjan, Charith
    Perera, Indika
    [J]. 2019 MORATUWA ENGINEERING RESEARCH CONFERENCE (MERCON) / 5TH INTERNATIONAL MULTIDISCIPLINARY ENGINEERING RESEARCH CONFERENCE, 2019, : 66 - 71
  • [48] GastroNet: Abnormalities Recognition in Gastrointestinal Tract through Endoscopic Imagery using Deep Learning Techniques
    Lafraxo, Samira
    El Ansari, Mohamed
    [J]. 2020 8TH INTERNATIONAL CONFERENCE ON WIRELESS NETWORKS AND MOBILE COMMUNICATIONS (WINCOM 2020), 2020, : 130 - 134
  • [49] Augmented histopathology: Enhancing colon cancer detection through deep learning and ensemble techniques
    Gowthamy, J.
    Ramesh, S. S. Subashka
    [J]. MICROSCOPY RESEARCH AND TECHNIQUE, 2024,
  • [50] Detection of gastrointestinal tract disorders using deep learning methods from colonoscopy images and videos
    Aliyi, Salih
    Dese, Kokeb
    Raj, Hakkins
    [J]. SCIENTIFIC AFRICAN, 2023, 20