Pretrained Transformers Improve Out-of-Distribution Robustness
arXiv:2004.06100
Abstract
Although pretrained Transformers such as BERT achieve high accuracy on in-distribution examples, do they generalize to new distributions? We systematically measure out-of-distribution (OOD) generalization for seven NLP datasets by constructing a new robustness benchmark with realistic distribution shifts. We measure the generalization of previous models including bag-of-words models, ConvNets, and LSTMs, and we show that pretrained Transformers' performance declines are substantially smaller. Pretrained transformers are also more effective at detecting anomalous or OOD examples, while many previous models are frequently worse than chance. We examine which factors affect robustness, finding that larger models are not necessarily more robust, distillation can be harmful, and more diverse pretraining data can enhance robustness. Finally, we show where future work can improve OOD robustness.
ACL 2020
References in corpus (2)
Cited by in corpus (5)
- Uncertainty-Aware Reliable Text Classification
- Estimating the Brittleness of AI: Safety Integrity Levels and the Need for Testing Out-Of-Distribution Performance
- A Concept Knowledge-Driven Keywords Retrieval Framework for Sponsored Search
- On a Benefit of Mask Language Modeling: Robustness to Simplicity Bias
- X-GGM: Graph Generative Modeling for Out-of-Distribution Generalization in Visual Question Answering