Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces the Masked Achromatic Guidance Expert (MAGE) framework to improve the classification of gastric adenoma and carcinoma during endoscopy by addressing the challenges of inter-class similarity and spurious texture biases in deep learning models. MAGE employs a dual-objective distillation strategy that trains a local expert on masked achromatic views to focus on structural features while transferring classification logits and spatial attention maps to the main branch. Experimental results demonstrate that MAGE significantly outperforms existing methods, achieving superior classification accuracy and providing interpretable attention maps for enhanced clinical reliability.
MAGE redefines gastric neoplasm classification by forcing models to prioritize intrinsic morphological features over misleading texture cues, leading to unprecedented accuracy in clinical settings.
Accurate differentiation between gastric adenoma and carcinoma during endoscopy is critical for clinical decision-making. Yet, this task is highly challenging due to high inter-class similarity and ambiguous boundaries between the two classes. Existing ROI-based classification methods often suffer from detection/segmentation error propagation and loss of surrounding global context. In contrast, full-image classification lacks the necessary spatial focus. Furthermore, we observe that deep neural networks gravitate towards domain-specific texture biases(e.g. bleeding, lighting artifacts), often causing models to predict based on spurious correlations instead of intrinsic morphological features. To address these limitations, we propose a novel framework, Masked Achromatic Guidance Expert (MAGE). During training, we introduce an auxiliary local expert branch trained on masked achromatic views of the neoplasm. By suppressing background context and color, this branch is forced to learn highly discriminative, purely structural features. We then employ a dual-objective distillation strategy, transferring both classification logits and spatial attention maps to provide implicit spatial supervision to the main branch that receives full WLI as input. This dual-objective distillation forces the model to ground its predictions in morphology rather than relying on shortcuts, while still retaining clinically relevant color cues. At inference time, our deployable model operates on images without annotated masks, ensuring real-time deployability . Extensive experiments on a clinical gastric endoscopy dataset show that our method significantly outperforms existing detection-based methodologies (e.g. YOLO) and classification-based methodologies (e.g. Swin-Transformer), providing not only superior classification performance but also interpretable attention maps for clinical reliability.