DeepGI: Explainable Deep Learning for Gastrointestinal Image Classification

TL;DR

This paper introduces a deep learning approach for gastrointestinal disease classification using endoscopic images, achieving high accuracy and incorporating explainability through Grad-CAM, thus enhancing clinical interpretability and establishing new benchmarks.

Contribution

It provides a comprehensive comparison of deep learning models on a novel dataset, introduces explainability methods, and sets new performance benchmarks for gastrointestinal image classification.

Findings

VGG16 and MobileNetV2 achieved 96.5% accuracy

Models demonstrated robustness against imaging artifacts

Explainability via Grad-CAM improved clinical interpretability

Abstract

This paper presents a comprehensive comparative model analysis on a novel gastrointestinal medical imaging dataset, comprised of 4,000 endoscopic images spanning four critical disease classes: Diverticulosis, Neoplasm, Peritonitis, and Ureters. Leveraging state-of-the-art deep learning techniques, the study confronts common endoscopic challenges such as variable lighting, fluctuating camera angles, and frequent imaging artifacts. The best performing models, VGG16 and MobileNetV2, each achieved a test accuracy of 96.5%, while Xception reached 94.24%, establishing robust benchmarks and baselines for automated disease classification. In addition to strong classification performance, the approach includes explainable AI via Grad-CAM visualization, enabling identification of image regions most influential to model predictions and enhancing clinical interpretability. Experimental results demonstrate the potential for robust, accurate, and interpretable medical image analysis even in complex real-world conditions. This work contributes original benchmarks, comparative insights, and visual explanations, advancing the landscape of gastrointestinal computer-aided diagnosis and underscoring the importance of diverse, clinically relevant datasets and model explainability in medical AI research.