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 条
  • [1] Weighted ensemble deep learning approach for classification of gastrointestinal diseases in colonoscopy images aided by explainable AI
    Enes Oğuz, Faruk
    Alkan, Ahmet
    [J]. Displays, 2024, 85
  • [2] Automated Gastrointestinal Tract Classification Via Deep Learning and The Ensemble Method
    Almanifi, Omair Rashed Abdulwareth
    Razman, Mohd Azraai Mohd
    Khairuddin, Ismail Mohd
    Abdullah, Muhammad Amirul
    Majeedi, Anwar P. P. Abdul
    [J]. 2021 21ST INTERNATIONAL CONFERENCE ON CONTROL, AUTOMATION AND SYSTEMS (ICCAS 2021), 2021, : 602 - 606
  • [3] Plant Leaf Disease Detection Using Ensemble Learning and Explainable AI
    Ammar, O.A.D.
    Abbas, Syed Shoaib
    Zafar, Amna
    Akram, Beenish Ayesha
    Dong, Feng
    Talpur, Mir Sajjad Hussain
    Uddin, Mueen
    [J]. IEEE Access, 2024, 12 : 156038 - 156049
  • [4] Explainable ensemble learning method for OCT detection with transfer learning
    Yang, Jiasheng
    Wang, Guanfang
    Xiao, Xu
    Bao, Meihua
    Tian, Geng
    [J]. PLOS ONE, 2024, 19 (03):
  • [5] Deep convolutional neural networks with ensemble learning and transfer learning for automated detection of gastrointestinal diseases
    Su, Qiaosen
    Wang, Fengsheng
    Chen, Dong
    Chen, Gang
    Li, Chao
    Wei, Leyi
    [J]. COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 150
  • [6] Improving Hate Speech Classification Through Ensemble Learning and Explainable AI Techniques
    Garg, Priya
    Sharma, M. K.
    Kumar, Parteek
    [J]. ARABIAN JOURNAL FOR SCIENCE AND ENGINEERING, 2024,
  • [7] Advanced Fraud Detection in Blockchain Transactions: An Ensemble Learning and Explainable AI Approach
    Taher, Shimal Sh.
    Ameen, Siddeeq Y.
    Ahmed, Jihan A.
    [J]. ENGINEERING TECHNOLOGY & APPLIED SCIENCE RESEARCH, 2024, 14 (01) : 12822 - 12830
  • [8] Enhancing intrusion detection performance using explainable ensemble deep learning
    Ncir, Chiheb Eddine Ben
    Hajkacem, Mohamed Aymen Ben
    Alattas, Mohammed
    [J]. PEERJ COMPUTER SCIENCE, 2024, 10
  • [9] Enhancing intrusion detection performance using explainable ensemble deep learning
    Ben Ncir, Chiheb Eddine
    Ben HajKacem, Mohamed Aymen
    Alattas, Mohammed
    [J]. PeerJ Computer Science, 2024, 10
  • [10] Explainable AI for deep learning based potato leaf disease detection
    Bengamra, Siwar
    Zagrouba, Ezzeddine
    Bigand, Andre
    [J]. 2023 IEEE INTERNATIONAL CONFERENCE ON FUZZY SYSTEMS, FUZZ, 2023,