提出新方法提升医疗AI对未知病例的识别能力,保障临床安全。
MARVEL: Margin-Aware Robust von Mises-Fischer Expert Learning for Long-Tailed Out-of-Distribution Detection

- 用非线性冯·米塞斯-费舍尔分类器学习复杂决策边界。
- 多专家机制结合边界感知,解决数据不平衡问题,提升检测精度。
- 专设异常点专家,显著降低误判率,适合真实医疗场景应用。
为实现临床部署,自动化诊断系统需在面对未见过的病例时仍保持可靠性,但深度模型常对分布外(OOD)输入给出高置信度错误分类,凸显对更鲁棒的OOD检测方法的需求。尽管已有大量研究提升模型鲁棒性,多数工作假设数据均衡,评估基于粗粒度或非临床的OOD来源,且缺乏在多样化OOD场景下的全面评估。为此,我们提出一种在多样且不均衡医学数据集上训练、并在临床相关OOD谱系中评估的新方法。框架包含三个关键组件:(1) 非线性冯·米塞斯-费舍尔(NvMF)分类器,可学习非线性决策边界,并有理论证明其渐近等价于余弦分类器;(2) 多专家架构,其中边界感知的NvMF分类器在标签分布不同区域进行专业化,以更好处理不平衡问题;(3) 显式训练的异常点专家,专门区分内点与异常点数据,强化OOD检测。在RFMiD、ISIC2019和NCTCRC数据集上的评估显示,相较现有最优方法,平均FPR95分别降低8.45%、13.02%和36.90%。全面消融实验验证了各组件的有效性。该方法能可靠识别陌生病例并交由医生处理,支持真实世界中更安全的AI辅助诊断。代码已公开于https://github.com/redboxup/MARVEL。
原文摘要 · Abstract (English)
For clinical deployment, it is essential that automated diagnostic systems remain reliable when confronted with previously unseen cases, yet deep models routinely misclassify out-of-distribution (OOD) inputs with high confidence, underscoring the need for more robust OOD detection methods. Although substantial effort has been devoted to improving model robustness, most of the existing literature assumes balanced datasets, evaluates OOD detection on coarse or non-clinical OOD sources, or lacks comprehensive assessment across diverse OOD scenarios. To address the gaps, we propose a novel methodology trained on diverse and imbalanced medical datasets and evaluated across a clinically reflective OOD spectrum. Our framework comprises three key components: (1) a Nonlinear von Mises-Fisher (NvMF) classifier capable of learning non-linear decision boundaries, with theoretical proof of its asymptotic connection to cosine classifiers; (2) a multi-expert framework in which margin-aware NvMF classifiers specialise in different regions of label distribution to better handle imbalance; and (3) an outlier expert trained explicitly to distinguish inlier from outlier data, thereby strengthening OOD detection. Evaluation on RFMiD, ISIC2019, and NCTCRC datasets demonstrates consistent improvements over state-of-the-art methods, achieving mean FPR95 reductions of 8.45%, 13.02%, and 36.90% respectively. These gains are further supported by comprehensive ablations that validated the contributions of each component. This enables reliable identification of unfamiliar cases for deferral to clinicians, supporting safer AI-assisted diagnosis in real-world workflows. Our code is available at https://github.com/redboxup/MARVEL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。