Robust Machine Comprehension Models via Adversarial Training

By Yicheng Wang, Mohit Bansal. 2018

125 citations recorded by the source corpus.

It is shown that many published models for the Stanford Question Answering Dataset (Rajpurkar et al., 2016) lack robustness, suffering an over 50% decrease in F1 score during adversarial evaluation based on the AddSent (Jia and Liang, 2017) algorithm. It has also been shown that retraining models on data generated by AddSent has limited effect on their robustness. We propose a novel alternative adversary-generation algorithm, AddSentDiverse, that significantly increases the variance within the adversarial training data by providing effective examples that punish the model for making certain superficial assumptions. Further, in order to improve robustness to AddSent’s semantic perturbations (e.g., antonyms), we jointly improve the model’s semantic-relationship learning capabilities in addition to our AddSentDiverse-based adversarial training data augmentation. With these additions, we show that we can make a state-of-the-art model significantly more robust, achieving a 36.5% increase in F1 score under many different types of adversarial evaluation while maintaining performance on the regular SQuAD task.

Subfields: Artificial Intelligence, Natural Language Processing, Safety, Robustness & Interpretability

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Mohit Bansal
  3. Zhiyuan Liu
  4. Maosong Sun
  5. Yejin Choi
  6. Sebastian Riedel
  7. Luke S. Zettlemoyer
  8. Hannaneh Hajishirzi

In Natural Language Processing

  1. Mohit Bansal
  2. Jianfeng Gao
  3. Zhiyuan Liu
  4. Sebastian Riedel
  5. Maosong Sun
  6. Yejin Choi
  7. Xiaodong Liu
  8. Hannaneh Hajishirzi

In Safety, Robustness & Interpretability

  1. Riza Theresa Batista-Navarro
  2. Goran Nenadic
  3. Viktor Schlegel
  4. Yichen Jiang
  5. Mohit Bansal
  6. Xiaodong Liu
  7. Jianfeng Gao
  8. W. Zhang

Citations in the corpus