References & Citations
Computer Science > Computation and Language
Title: Better Robustness by More Coverage: Adversarial Training with Mixup Augmentation for Robust Fine-tuning
(Submitted on 31 Dec 2020 (v1), last revised 6 Jun 2021 (this version, v3))
Abstract: Pretrained language models (PLMs) perform poorly under adversarial attacks. To improve the adversarial robustness, adversarial data augmentation (ADA) has been widely adopted to cover more search space of adversarial attacks by adding textual adversarial examples during training. However, the number of adversarial examples for text augmentation is still extremely insufficient due to the exponentially large attack search space. In this work, we propose a simple and effective method to cover a much larger proportion of the attack search space, called Adversarial and Mixup Data Augmentation (AMDA). Specifically, AMDA linearly interpolates the representations of pairs of training samples to form new virtual samples, which are more abundant and diverse than the discrete text adversarial examples in conventional ADA. Moreover, to fairly evaluate the robustness of different models, we adopt a challenging evaluation setup, which generates a new set of adversarial examples targeting each model. In text classification experiments of BERT and RoBERTa, AMDA achieves significant robustness gains under two strong adversarial attacks and alleviates the performance degradation of ADA on the clean data. Our code is available at: this https URL .
Submission history
From: Chenglei Si [view email][v1] Thu, 31 Dec 2020 16:28:07 GMT (448kb,D)
[v2] Sat, 22 May 2021 09:19:34 GMT (108kb,D)
[v3] Sun, 6 Jun 2021 02:56:56 GMT (108kb,D)
Link back to: arXiv, form interface, contact.