English

Robustifying Language Models with Test-Time Adaptation

Computation and Language 2023-10-31 v1 Machine Learning

Abstract

Large-scale language models achieved state-of-the-art performance over a number of language tasks. However, they fail on adversarial language examples, which are sentences optimized to fool the language models but with similar semantic meanings for humans. While prior work focuses on making the language model robust at training time, retraining for robustness is often unrealistic for large-scale foundation models. Instead, we propose to make the language models robust at test time. By dynamically adapting the input sentence with predictions from masked words, we show that we can reverse many language adversarial attacks. Since our approach does not require any training, it works for novel tasks at test time and can adapt to novel adversarial corruptions. Visualizations and empirical results on two popular sentence classification datasets demonstrate that our method can repair adversarial language attacks over 65% o

Keywords

Cite

@article{arxiv.2310.19177,
  title  = {Robustifying Language Models with Test-Time Adaptation},
  author = {Noah Thomas McDermott and Junfeng Yang and Chengzhi Mao},
  journal= {arXiv preprint arXiv:2310.19177},
  year   = {2023}
}

Comments

8 Pages 2 Figures Submitted to ICLR Workshop

R2 v1 2026-06-28T13:05:20.794Z