MDeiT: A lightweight and explainable hybrid model for cancer classification in histopathology images

Read full text
Authors: Getamesay Haile Dagnaw, Yanming Zhu, Yuhong Wang, Muhammad Hassan Maqsood, Xuefei Yin, Alan Wee‐Chung Liew
Cited by: 1
References: 34

Abstract

Accurate and interpretable cancer classification in histopathological images remains a significant challenge due to the complex structural variations in tissue samples. In this paper, we propose MDeiT, a lightweight and interpretable sequential hybrid model that effectively integrates Convolutional Neural Networks (CNNs) and Vision Transformers (ViTs) to enhance both classification accuracy and efficiency. Unlike traditional ensemble-based hybrid models, our framework adopts a streamlined design, leveraging MobileNetV2 and DeiT Tiny as backbone architectures, with an adaptation layer facilitating the transition from CNN-extracted local features to Transformer processing. To improve interpretability, we incorporate Gradient-weighted Class Activation Mapping (Grad-CAM) for visual explanations of model predictions. Furthermore, we introduce expert-driven qualitative validation, where pathologists annotate ground truth to systematically assess the alignment between model-generated saliency maps and clinically relevant diagnostic regions, establishing a high-quality benchmark for interpretability evaluation. Extensive experiments on skin and lung cancer datasets demonstrate that MDeiT consistently outperforms state-of-the-art models across multiple metrics while maintaining computational efficiency. The results demonstrate its effectiveness in capturing both fine-grained tissue details and broader contextual patterns, making it a robust and scalable solution for real-world histopathological image analysis.

Fields of Study & Subjects

AI in cancer detection Cell Image Analysis Techniques Radiomics and Machine Learning in Medical Imaging

Publication details

Journal / source
Biomedical Signal Processing and Control
Publisher
Elsevier BV
Volume
123
Pages
110620–110620
Publication date
17 May 2026
Language
EN