Span Selection Pre-training for Question Answering

By Michael R. Glass, Alfio Massimiliano Gliozzo, Rishav Chakravarti, Anthony Ferritto, Lin Pan, G P Shrivatsa Bhargav, Dinesh Garg, Avirup Sil. 2019

76 citations recorded by the source corpus.

BERT (Bidirectional Encoder Representations from Transformers) and related pre-trained Transformers have provided large gains across many language understanding tasks, achieving a new state-of-the-art (SOTA). BERT is pretrained on two auxiliary tasks: Masked Language Model and Next Sentence Prediction. In this paper we introduce a new pre-training task inspired by reading comprehension to better align the pre-training from memorization to understanding. Span Selection PreTraining (SSPT) poses cloze-like training instances, but rather than draw the answer from the model’s parameters, it is selected from a relevant passage. We find significant and consistent improvements over both BERT-BASE and BERT-LARGE on multiple Machine Reading Comprehension (MRC) datasets. Specifically, our proposed model has strong empirical evidence as it obtains SOTA results on Natural Questions, a new benchmark MRC dataset, outperforming BERT-LARGE by 3 F1 points on short answer prediction. We also show significant impact in HotpotQA, improving answer prediction F1 by 4 points and supporting fact prediction F1 by 1 point and outperforming the previous best system. Moreover, we show that our pre-training approach is particularly effective when training data is limited, improving the learning curve by a large amount.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Natural Language Processing, Question Answering & Reading Comprehension

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. Mohit Bansal
  2. Jianfeng Gao
  3. Nan Duan
  4. Jonathan Berant
  5. Percy Liang
  6. Luke S. Zettlemoyer
  7. Graham Neubig
  8. Yejin Choi

In Computer Vision and Pattern Recognition

  1. Nan Duan
  2. J. Scott McCarley
  3. Jonathan Berant
  4. Omer Levy
  5. Ori Ram
  6. Huan Sun
  7. Yu Su
  8. Amir Globerson

In Natural Language Processing

  1. Mohit Bansal
  2. Nan Duan
  3. Jonathan Berant
  4. Graham Neubig
  5. Yejin Choi
  6. Omer Levy
  7. Percy Liang
  8. Jianfeng Gao

In Question Answering & Reading Comprehension

  1. J. Scott McCarley
  2. Huan Sun
  3. Xiang Deng
  4. Radu Florian
  5. Jonathan Berant
  6. Omer Levy
  7. Nan Duan
  8. Daxin Jiang

Citations in the corpus