Sign in
BioengineeringbioRxiv
Preprint

CBAM-enhanced MaxViT-Tiny for gastrointestinal endoscopy image classification: Robustness, explainability, and external validation

Sadek Al PrinceiD, Israt Fatema Shorna, Md Mehedi Hasan, Mohammed Ahanaf Akif, Aziz Misher Mishu, Gazi Mehraj Sakib

Abstract

Automated classification of the huge number of images generated during gastrointestinal (GI) endoscopy is required for reliable and interpretable diagnosis, but there is concern about the vulnerability of deep models to adversarial perturbations. In the current study, a MaxViT-Tiny backbone with a Convolutional Block Attention Module (CBAM) head is used for 8-class classification of the Kvasir endoscopic images. After deleting duplicate images, 3,996 images remained, which were split into a training set (70%), a validation set (15%) and a test set (15%); the model was compared with five backbones using the same CBAM head under three random seeds. The proposed model obtained the highest test accuracy of 94.33%, a macro F1 score of 0.9431, and a macro area under the receiver operating characteristic curve (AUC) of 0.9975, with a mean accuracy of 93.72 {+/-} 1.06% across all seeds. White-box FGSM, PGD, CW-L2, and MI-FGSM attacks greatly reduced the accuracy: after adversarial fine-tuning, the accuracy under FGSM was 74.17%, while the accuracy under PGD and MI-FGSM was below 4%, showing poor cross-attack generalization. A preliminary attack detector obtained an AUC of 0.857. Grad-CAM++, Score-CAM, and GradientSHAP sometimes identified clinically relevant regions, but also sometimes focused on non-anatomical artefacts. High accuracy on clean images, even with attention augmentation, therefore does not imply high accuracy on adversarial images or reliable explanations, and the model needs further verification on independent datasets and multi-centre validation before clinical use.

PDFDOI

02 /Specification & Classification

Click any tag to browse matching papers

Reproducibility ledger

Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.

Sign in to file a reproducibility report.