通过神经元级相关性实现可解释的医学图像异常检测
NERO: Explainable Out-of-Distribution Detection with Neuron-level Relevance
- 基于特征层神经元相关性聚类构建类别中心点
- 新样本与中心点的距离越远,越可能是分布外数据
- 既提升检测效果又可解释,适合医疗诊断场景
确保深度学习模型的可靠性在医学影像领域至关重要,因为诊断决策常依赖模型输出。识别分布外(OOD)样本是衡量模型可靠性的重要指标。尽管许多方法依赖特征或对数空间表示,但近期研究指出这些方法难以充分捕捉OOD多样性。为此,我们提出一种新型OOD评分机制NERO,利用特征层的神经元级相关性。具体而言,对每个内分布(ID)类别进行神经元相关性聚类,形成代表性中心点,并引入相关性距离度量来量化新样本与这些中心点的偏离程度,从而增强OOD分离能力。此外,通过在偏置项中引入缩放相关性并结合特征范数,进一步优化性能。该框架还支持可解释的OOD检测。我们在胃肠道影像基准Kvasir和GastroVision上验证了其有效性,对多种深度学习架构均取得优于现有方法的性能。
原文摘要 · Abstract (English)
Ensuring reliability is paramount in deep learning, particularly within the domain of medical imaging, where diagnostic decisions often hinge on model outputs. The capacity to separate out-of-distribution (OOD) samples has proven to be a valuable indicator of a model's reliability in research. In medical imaging, this is especially critical, as identifying OOD inputs can help flag potential anomalies that might otherwise go undetected. While many OOD detection methods rely on feature or logit space representations, recent works suggest these approaches may not fully capture OOD diversity. To address this, we propose a novel OOD scoring mechanism, called NERO, that leverages neuron-level relevance at the feature layer. Specifically, we cluster neuron-level relevance for each in-distribution (ID) class to form representative centroids and introduce a relevance distance metric to quantify a new sample's deviation from these centroids, enhancing OOD separability. Additionally, we refine performance by incorporating scaled relevance in the bias term and combining feature norms. Our framework also enables explainable OOD detection. We validate its effectiveness across multiple deep learning architectures on the gastrointestinal imaging benchmarks Kvasir and GastroVision, achieving improvements over state-of-the-art OOD detection methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。