arXiv:2607.12663cs.CV2026-07中稿 · MICCAI 2026

通过结构化特征学习提升胃部肿瘤分类准确率

MAGE: Color-Invariant and Spatial Knowledge Distillation for Gastric Neoplasm Classification

论文配图:MAGE: Color-Invariant and Spatial Knowledge Distillation for Gastric Neoplasm Classification
图 1 · 摘自论文原文
  • 用去色图像训练辅助分支,强制模型关注形态结构而非颜色
  • 双目标蒸馏同时传递分类结果和注意力图,提升模型可靠性
  • 无需标注掩码,可实时部署,适合临床实际场景

内镜下准确区分胃腺瘤与癌变对临床决策至关重要,但二者类间相似度高、边界模糊,现有基于区域的方法易受检测/分割误差影响且丢失全局上下文,全图分类则缺乏空间聚焦。我们观察到深度网络会依赖出血、光照等域特异性纹理偏差,导致基于虚假相关性的错误预测。为此提出新框架MAGE:训练时引入一个在掩码去色图像上训练的局部专家分支,抑制背景与颜色干扰,强制学习纯结构特征;采用双目标蒸馏策略,将分类输出与空间注意力图同时传给主分支(接收全色WLI输入),实现隐式空间监督,使模型基于形态学特征而非捷径进行判断,同时保留临床相关的颜色线索。推理时无需标注掩码,具备实时部署能力。在临床胃镜数据集上的实验表明,该方法显著优于基于检测(如YOLO)和基于分类(如Swin-Transformer)的方法,不仅分类性能更优,且生成可解释的注意力图,增强临床可信度。

原文摘要 · Abstract (English)

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.

胃癌分类知识蒸馏可解释性内镜影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。