A triple pronged approach for ulcerative colitis severity classification using multimodal, meta, and transformer based learning.

Journal: Scientific reports
Published Date:

Abstract

Ulcerative colitis (UC) is a chronic inflammatory disorder necessitating precise severity stratification to facilitate optimal therapeutic interventions. This study harnesses a triple-pronged deep learning methodology-including multimodal inference pipelines that eliminate domain-specific training, few-shot meta-learning, and Vision Transformer (ViT)-based ensembling-to classify UC severity within the HyperKvasir dataset. We systematically evaluate multiple vision transformer architectures, discovering that a Swin-Base model achieves an accuracy of 90%, while a soft-voting ensemble of diverse ViT backbones boosts performance to 93%. In parallel, we leverage multimodal pre-trained frameworks (e.g., CLIP, BLIP, FLAVA) integrated with conventional machine learning algorithms, yielding an accuracy of 83%. To address limited annotated data, we deploy few-shot meta-learning approaches (e.g., Matching Networks), attaining 83% accuracy in a 5-shot context. Furthermore, interpretability is enhanced via SHapley Additive exPlanations (SHAP), which interpret both local and global model behaviors, thereby fostering clinical trust in the model's inferences. These findings underscore the potential of contemporary representation learning and ensemble strategies for robust UC severity classification, highlighting the pivotal role of model transparency in facilitating medical image analysis.

Authors

  • Md Nasim Ahmed
    Department of Electrical and Computer Engineering, North South University, Dhaka, 1229, Bangladesh.
  • Dipta Neogi
    Department of Electrical and Computer Engineering, North South University, Dhaka, 1229, Bangladesh.
  • Muhammad Rafsan Kabir
    Department of Electrical and Computer Engineering, North South University, Dhaka, 1229, Bangladesh. muhammad.kabir@northsouth.edu.
  • Shafin Rahman
  • Sifat Momen
    Apurba NSU R&D Lab, Department of Electrical and Computer Engineering North South University, Dhaka, Bangladesh.
  • Nabeel Mohammed
    Apurba NSU R&D Lab, Department of Electrical and Computer Engineering North South University, Dhaka, Bangladesh.

Keywords

No keywords available for this article.