提出MiMu方法,解决Transformer模型多重捷径依赖问题。
MiMu: Mitigating Multiple Shortcut Learning Behavior of Transformers
- 通过自校准与自提升策略,抑制模型对多种捷径的依赖。
- 在多个数据集上显著提升模型在分布外任务中的泛化性能。
- 适用于需要强鲁棒性的NLP与计算机视觉场景。
经验风险最小化(ERM)模型在学习过程中常依赖特征与标签间的虚假关联,导致捷径学习行为,损害泛化能力。现有研究多聚焦单一捷径的识别与缓解,但在真实场景中,数据中的线索多样且未知。实证研究表明,模型对不同捷径的依赖程度各异,尤其更依赖强捷径,从而削弱泛化表现。为此,我们提出MiMu,一种集成于Transformer-based ERM的新方法,用于缓解多重捷径学习行为。该方法包含自校准策略(在源模型中防止对捷径过度依赖及过度自信预测)和自提升策略(在目标模型中减少对多种捷径的依赖)。后者采用随机掩码机制,随机遮蔽部分注意力位置以分散关注区域;同时引入自适应注意力对齐模块,使目标模型注意力权重与经校准的源模型对齐,无需后处理注意力图或监督信号。大量实验在自然语言处理(NLP)与计算机视觉(CV)任务上验证了MiMu的有效性,显著提升了模型的鲁棒泛化能力。
原文摘要 · Abstract (English)
Empirical Risk Minimization (ERM) models often rely on spurious correlations between features and labels during the learning process, leading to shortcut learning behavior that undermines robustness generalization performance. Current research mainly targets identifying or mitigating a single shortcut; however, in real-world scenarios, cues within the data are diverse and unknown. In empirical studies, we reveal that the models rely to varying extents on different shortcuts. Compared to weak shortcuts, models depend more heavily on strong shortcuts, resulting in their poor generalization ability. To address these challenges, we propose MiMu, a novel method integrated with Transformer-based ERMs designed to Mitigate Multiple shortcut learning behavior, which incorporates self-calibration strategy and self-improvement strategy. In the source model, we preliminarily propose the self-calibration strategy to prevent the model from relying on shortcuts and make overconfident predictions. Then, we further design self-improvement strategy in target model to reduce the reliance on multiple shortcuts. The random mask strategy involves randomly masking partial attention positions to diversify the focus of target model other than concentrating on a fixed region. Meanwhile, the adaptive attention alignment module facilitates the alignment of attention weights to the calibrated source model, without the need for post-hoc attention maps or supervision. Finally, extensive experiments conducted on Natural Language Processing (NLP) and Computer Vision (CV) demonstrate the effectiveness of MiMu in improving robustness generalization abilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。