通过生成反事实样本来纠正图像分类器的伪相关问题。
Mitigating Clever Hans Strategies in Image Classifiers through Generating Counterexamples
- 用反事实样本生成技术替代依赖分组标签的重加权方法。
- 在低数据场景下显著提升模型对伪相关性的鲁棒性,尤其在多混淆因子时表现更优。
- 无需混淆因子标签,适合工业级低资源场景的模型可靠性优化。
深度学习模型仍易受伪相关影响,导致所谓的‘聪明汉’预测器,即使在大规模基础模型和自监督模型中也损害其鲁棒性。现有基于组分布鲁棒性的方法(如深度特征重加权,DFR)依赖显式分组标签来提升低频子群体权重,但存在三大局限:(1) 分组标签常不可得,(2) 组内样本量小难以覆盖子群分布,(3) 当多个伪相关将数据分割成更小组别时性能急剧下降。本文提出反事实知识蒸馏(CFKD),通过生成多样反事实样本,使人工标注者能高效探索并修正模型决策边界。与DFR不同,本方法不仅重加权未充分采样的组,还通过新增数据点丰富这些组。该方法无需任何混淆因子标签,可有效扩展至多重混淆因子,实现跨组的均衡泛化。我们在五个数据集上验证了其有效性,涵盖合成任务到工业应用,在低数据且存在明显伪相关场景下表现尤为突出。此外,我们对反事实解释器和教师模型的选择进行了消融研究,揭示其对鲁棒性的重要影响。
原文摘要 · Abstract (English)
Deep learning models remain vulnerable to spurious correlations, leading to so-called Clever Hans predictors that undermine robustness even in large-scale foundation and self-supervised models. Group distributional robustness methods, such as Deep Feature Reweighting (DFR) rely on explicit group labels to upweight underrepresented subgroups, but face key limitations: (1) group labels are often unavailable, (2) low within-group sample sizes hinder coverage of the subgroup distribution, and (3) performance degrades sharply when multiple spurious correlations fragment the data into even smaller groups. We propose Counterfactual Knowledge Distillation (CFKD), a framework that sidesteps these issues by generating diverse counterfactuals, enabling a human annotator to efficiently explore and correct the model's decision boundaries through a knowledge distillation step. Unlike DFR, our method not only reweights the undersampled groups, but it also enriches them with new data points. Our method does not require any confounder labels, achieves effective scaling to multiple confounders, and yields balanced generalization across groups. We demonstrate CFKD's efficacy across five datasets, spanning synthetic tasks to an industrial application, with particularly strong gains in low-data regimes with pronounced spurious correlations. Additionally, we provide an ablation study on the effect of the chosen counterfactual explainer and teacher model, highlighting their impact on robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。