Models Know Their Shortcuts: Deployment-Time Shortcut Mitigation

Jiayi Li, Shijie Tang, Gün Kaynar, Shiyi Du, Carl Kingsford


Abstract
Pretrained text encoders are prone to shortcut learning, relying on token-label correlations that fail once the distribution shifts in deployment. Existing shortcut mitigation methods mainly operate at training time and assume access to training data, training dynamics, or shortcut annotations, which are hardly available during deployment, where only the converged model remains. We show that this model alone suffices to mitigate shortcuts during deployment: a biased model internalizes a signal of its learned shortcuts that can be captured via unsupervised gradient-based attribution. We further prove that deployment-time mitigation is information-theoretically upper-bounded by training-time mitigation. Nevertheless, exploiting this gradient signal, our proposed unsupervised deployment-time shortcut mitigation framework for pretrained text encoders, Shortcut Guardrail, recovers substantial performance under shortcut distribution shift, matching or outperforming training-time baselines across sentiment classification, toxicity detection, and natural language inference.
Anthology ID:
2026.findings-aacl.171
Volume:
Findings of the Association for Computational Linguistics: IJCNLP-AACL 2026
Month:
November
Year:
2026
Address:
Hengqin, China
Editors:
Yue Zhang, Ruifeng Xu, Ayu Purwarianti, Tong Rong, Jun Suzuki
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
3408–3430
Language:
URL:
https://aclanthology.org/2026.findings-aacl.171/
DOI:
Bibkey:
Cite (ACL):
Jiayi Li, Shijie Tang, Gün Kaynar, Shiyi Du, and Carl Kingsford. 2026. Models Know Their Shortcuts: Deployment-Time Shortcut Mitigation. In Findings of the Association for Computational Linguistics: IJCNLP-AACL 2026, pages 3408–3430, Hengqin, China. Association for Computational Linguistics.
Cite (Informal):
Models Know Their Shortcuts: Deployment-Time Shortcut Mitigation (Li et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-aacl.171.pdf