解耦虚假相关:IRM 如何处理文本分类中的多样捷径
Disentangling Spurious Correlations: How IRM Handles Diverse Shortcuts in Text Classification
Paul Grunenwald, Benoît Crabbé, Guillaume Wisniewski, Pirmin Lemberger
AI 速读 基于摘要
该研究用因果驱动框架评估不变风险最小化(IRM)在多种捷径生成机制下的表现,并进一步考察其在两个更真实文本分类场景中的效果。
cs.CL · stat.ML
查看原文摘要
In NLP, models trained via Empirical Risk Minimization (ERM) often exploit spurious correlations, such as dialectal markers, at the expense of robust linguistic features, thereby degrading their out-of-distribution (OOD) generalization. Invariant Risk Minimization (IRM) aims to suppress such shortcuts by identifying predictive relationships that remain stable across environments, but its practical utility for NLP remains debated. We use a causality-driven framework to evaluate IRM under three shortcut-generating mechanisms: label-generated features, confounding, and selection bias. Our results show that IRM successfully rejects shortcuts across the evaluated causal and anti-causal structures, including under limited environmental variation and increasing label noise. We then extend our analysis beyond controlled synthetic artifacts to two less constrained settings: gender-profession associations in Bias in Bios and semantically loaded actor-name shortcuts in ActorCorr. These experiments show that invariant learning can mitigate more realistic textual shortcuts, while semantically meaningful features require careful calibration of the invariance penalty.