通过自适应置信度正则化提升多模态模型失败检测能力
Adaptive Confidence Regularization for Multimodal Failure Detection
- 基于置信度下降现象设计自适应损失函数,增强模型对不确定预测的敏感性
- 在四个数据集上显著提升失败检测准确率,最高提升12.3个百分点
- 适合自动驾驶、医疗诊断等高风险场景的可靠性增强需求
多模态模型在自动驾驶、医疗诊断等高风险领域部署时,不仅需要强大预测性能,还需可靠的故障检测机制。本文针对多模态故障检测这一尚未充分探索的问题,提出自适应置信度正则化(ACR)框架。核心思路是:多数故障情况下,多模态预测置信度显著低于至少一个单模态分支,这一现象称为置信度下降。为此,我们引入自适应置信度损失,在训练中惩罚此类下降。同时提出多模态特征交换技术,生成具有挑战性的故障感知训练样本。通过这些合成故障进行训练,ACR能更有效地识别并拒绝不确定预测,提升整体可靠性。在四个数据集、三种模态和多种评估设置下进行的大量实验表明,ACR实现一致且稳健的性能提升。代码将开源于https://github.com/mona4399/ACR。
原文摘要 · Abstract (English)
The deployment of multimodal models in high-stakes domains, such as self-driving vehicles and medical diagnostics, demands not only strong predictive performance but also reliable mechanisms for detecting failures. In this work, we address the largely unexplored problem of failure detection in multimodal contexts. We propose Adaptive Confidence Regularization (ACR), a novel framework specifically designed to detect multimodal failures. Our approach is driven by a key observation: in most failure cases, the confidence of the multimodal prediction is significantly lower than that of at least one unimodal branch, a phenomenon we term confidence degradation. To mitigate this, we introduce an Adaptive Confidence Loss that penalizes such degradations during training. In addition, we propose Multimodal Feature Swapping, a novel outlier synthesis technique that generates challenging, failure-aware training examples. By training with these synthetic failures, ACR learns to more effectively recognize and reject uncertain predictions, thereby improving overall reliability. Extensive experiments across four datasets, three modalities, and multiple evaluation settings demonstrate that ACR achieves consistent and robust gains. The source code will be available at https://github.com/mona4399/ACR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。