arXiv:2604.17360cs.AI2026-04被引 1

用教师指导的双路原型检索提升医学图像细粒度分类准确率

T-DuMpRa: Teacher-guided Dual-path Multi-prototype Retrieval Augmented framework for fine-grained medical image classification

论文配图:T-DuMpRa: Teacher-guided Dual-path Multi-prototype Retrieval Augmented framework for fine-grained medical image classification
图 1 · 摘自论文原文
  • 双路径融合分类与原型检索,联合优化嵌入几何
  • 在五个骨干网络上提升0.21%-2.69%准确率,尤其改善模糊病例
  • 仅需构建小型原型库,可无缝集成现有模型

细粒度医学图像分类面临类别间差异微小、视觉模糊等挑战,传统判别模型虽整体准确率高,却常在相似类别间误判,预测置信度失准。本文提出T-DuMpRa框架,通过教师引导的双路径多原型检索增强机制,在训练中联合优化交叉熵与监督对比损失,学习兼容余弦距离的嵌入结构;利用指数移动平均(EMA)教师生成平滑表示,并在教师嵌入空间聚类构建多原型记忆库。推理时,仅当分类器预测不确定且检索证据明确冲突时,才启用检索结果进行保守融合,否则保留原有预测。在HAM10000和ISIC2019数据集上,该方法在5种骨干网络上分别实现0.68%-0.21%和0.44%-2.69%的性能提升,可视化验证其显著增强了对视觉模糊病例的处理能力。

原文摘要 · Abstract (English)

Fine-grained medical image classification is challenged by subtle inter-class variations and visually ambiguous cases, where confidence estimates often exhibit uncertainty rather than being overconfident. In such scenarios, purely discriminative classifiers may achieve high overall accuracy yet still fail to distinguish between highly similar categories, leading to miscalibrated predictions. We propose T-DuMpRa, a teacher-guided dual-path multi-prototype retrieval-augmented framework, where discriminative classification and multi-prototype retrieval jointly drive both training and prediction. During training, we jointly optimize cross-entropy and supervised contrastive objectives to learn a cosine-compatible embedding geometry for reliable prototype matching. We further employ an exponential moving average (EMA) teacher to obtain smoother representations and build a multi-prototype memory bank by clustering teacher embeddings in the teacher embedding space. Our framework is plug-and-play: it can be easily integrated into existing classification models by constructing a compact prototype bank, thereby improving performance on visually ambiguous cases. At inference, we combine the classifier's predicted distribution with a similarity-based distribution computed via cosine matching to prototypes, and apply a conservative confidence-gated fusion that activates retrieval only when the classifier's prediction is uncertain and the retrieval evidence is decisive and conflicting, otherwise keeping confident predictions unchanged. On HAM10000 and ISIC2019, our method yields 0.68%-0.21% and 0.44%-2.69% improvements on 5 different backbones. And visualization analysis proves our model can enhance the model's ability to handle visually ambiguous cases.

医学图像细粒度分类原型检索置信度校准

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