arXiv:2608.07953cs.LG2026-08

让模型不仅能发现异常数据,还能说出失败原因。

From Uncertainty to Failure Attribution: Self-Diagnosing Models for Failure Attribution under Distribution Shift

  • 设计自诊断模型,同时预测输出、不确定性和失败类型
  • 识别四种失败类型:协方差偏移、语义偏移、噪声、对抗攻击
  • 适合需要可解释性与鲁棒性的实际部署场景

分布偏移严重威胁机器学习模型的鲁棒性,现有方法仅能检测分布外样本并估计不确定性。本文提出分布偏移下的失败归因新问题,使模型不仅能检测分布外样本,还能定位失败原因。所提自诊断模型联合学习预测输出、预测不确定性及失败归因信号。通过神经网络生成失败归因向量,结构化区分四类失败:协方差偏移、语义偏移、噪声污染、对抗扰动。即从标量不确定性迈向失败识别。训练中引入一致性正则化,促进不确定性与失败归因预测的一致性。为评估模型归因能力,构建多个带有预定义分布偏移机制的基准测试集。

原文摘要 · Abstract (English)

Distribution shift poses a significant challenge to the robustness of machine learning models, but the current solutions only aim to detect out-of-distribution (OOD) samples and predict uncertainty levels. We introduce a problem setting for failure attribution under distribution shift, which enables the models not only to detect OOD samples, but also to find out the reason for their failure. The solution we propose is called self-diagnosing models, which are capable of jointly learning predictive output, predictive uncertainty, and a failure attribution signal. In particular, we use the failure attribution vector, produced by a neural network, which provides a structured representation of predictive unreliability by distinguishing four different types of failures: covariance shift, semantic shift, noise corruption, and adversarial perturbation. In other words, we move from scalar uncertainty towards failure identification. For training the model, we introduce a consistency regularizer that encourages consistency between uncertainty and failure attribution predictions. Moreover, to be able to evaluate the model on its ability to find the reasons for failure, we construct several distribution shift benchmarks with predefined mechanisms for generating distribution shifts.

失败归因分布偏移可解释性鲁棒性

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