Ensembles of deep learning framework for stomach abnormalities classification

Saeed, Talha and Loo, Chu Kiong and Kassim, Muhammad Shahreeza Safiruz (2022) Ensembles of deep learning framework for stomach abnormalities classification. CMC-Computers Materials & Continua, 70 (3). pp. 4357-4372. ISSN 1546-2218, DOI https://doi.org/10.32604/cmc.2022.019076.

Full text not available from this repository.


Abnormalities of the gastrointestinal tract are widespread worldwide today. Generally, an effective way to diagnose these life-threatening diseases is based on endoscopy, which comprises a vast number of images. However, the main challenge in this area is that the process is time-consuming and fatiguing for a gastroenterologist to examine every image in the set. Thus, this led to the rise of studies on designing AI-based systems to assist physicians in the diagnosis. In several medical imaging tasks, deep learning methods, especially convolutional neural networks (CNNs), have contributed to the stateof-the-art outcomes, where the complicated nonlinear relation between target classes and data can be learned and not limit to hand-crafted features. On the other hand, hyperparameters are commonly set manually, which may take a long time and leave the risk of non-optimal hyperparameters for classification. An effective tool for tuning optimal hyperparameters of deep CNN is Bayesian optimization. However, due to the complexity of the CNN, the network can be regarded as a black-box model where the information stored within it is hard to interpret. Hence, Explainable Artificial Intelligence (XAI) techniques are applied to overcome this issue by interpreting the decisions of the CNNs in such wise the physicians can trust. To play an essential role in real-time medical diagnosis, CNN-based models need to be accurate and interpretable, while the uncertainty must be handled. Therefore, a novel method comprising of three phases is proposed to classify these life-threatening diseases. At first, hyperparameter tuning is performed using Bayesian optimization for two state-of-the-art deep CNNs, and then Darknet53 and InceptionV3 features are extracted from these fine-tunned models. Secondly, XAI techniques are used to interpret which part of the images CNN takes for feature extraction. At last, the features are fused, and uncertainties are handled by selecting entropy based features. The experimental results show that the proposed method outperforms existing methods by achieving an accuracy of 97% based on a Bayesian optimized Support Vector Machine classifier.

Item Type: Article
Funders: Universiti Malaya Impact-oriented Interdisciplinary Research Grant Programme (IIRG) [IIRG002C-19HWB], Universiti Malaya Covid-19 Related Special Research Grant (UMCSRG) [CSRG008-2020ST], Universiti Malaya [RK0122019]
Uncontrolled Keywords: Gastrointestinal tract; Deep learning; Bayesian optimization hyperparameters; Explainable AI; Uncertainty handling; Feature fusion
Subjects: Q Science > QA Mathematics
Q Science > QA Mathematics > QA75 Electronic computers. Computer science
T Technology > T Technology (General)
Divisions: Faculty of Computer Science & Information Technology > Department of Artificial Intelligence
Depositing User: Ms. Juhaida Abd Rahim
Date Deposited: 29 Jul 2022 01:13
Last Modified: 29 Jul 2022 01:13
URI: http://eprints.um.edu.my/id/eprint/33591

Actions (login required)

View Item View Item