arXiv:2609.07922cs.LGcs.CV2026-09

通过校准缓解模型依赖表面线索,提升公平性与准确性。

Prevalence calibration as shortcut mitigation

论文配图:Prevalence calibration as shortcut mitigation
图 1 · 摘自论文原文
  • 将捷径学习视为校准问题,通过均衡不同组别流行率来纠正偏差。
  • 后处理校准使错误组别AUROC从0.23提升至0.73,显著改善性能。
  • 适用于冻结主干网络的模型,对医疗影像分类尤其有效。

捷径学习指分类器利用虚假相关而非诊断特征的现象。现有方法多聚焦于学习对捷径不变的表征,但效果有限且不适用于使用冻结基础模型编码器的分类器。本文将捷径学习重新定义为校准问题:无约束学习会隐式将每个捷径组校准至训练集疾病流行率,导致某组过度自信、另一组则低估。基于此,我们提出两种无需依赖编码器的方法——一种过程内正则化与一种后处理流行率均衡重校准。在CheXpert与SIIM-ACR的胸腔引流-气胸基准测试中,涵盖微调CNN与冻结基础模型骨干,两种方法均显著优于所有基线。对标准ERM训练的DenseNet进行后处理校准,使误配组的AUROC从0.23提升至0.73,表明捷径依赖损害的是分类头而非底层表征。除两种新状态最优的捷径缓解方法外,研究还首次将捷径学习与校准理论及算法公平性深度关联。

原文摘要 · Abstract (English)

Shortcut learning denotes the widespread situation in which a classifier exploits spurious correlations rather than diagnostic features. Existing mitigation strategies mostly aim to learn shortcut-invariant representations; their empirical success is limited and they cannot be applied to classifiers using frozen foundation model encoders. We propose to reframe shortcut learning as fundamentally a calibration problem: unconstrained learning implicitly calibrates each shortcut group to its training set disease prevalence, rendering the resulting classifier necessarily over-confident in one group and under-confident in the other. Building on this insight, we prevalence-equalize calibration between shortcut groups through two encoder-agnostic methods, an in-processing regularizer and a post-hoc prevalence-equalized recalibration step. Across chest-drain-pneumothorax benchmarks on CheXpert and SIIM-ACR, spanning fine-tuned CNNs and frozen foundation-model backbones, both methods substantially outperform all baselines. Post-hoc recalibration of a standard ERM-trained DenseNet raises misaligned-group AUROC from 0.23 to 0.73, indicating that shortcut reliance degrades the classification head rather than the underlying representation. Besides two new state-of-the-art shortcut mitigation approaches, our findings more fundamentally connect shortcut learning to calibration theory and algorithmic fairness.

捷径学习校准公平性医疗影像

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