QANet: Combining Local Convolution with Global Self-Attention for Reading Comprehension

By Adams Wei Yu, David Dohan, Minh-Thang Luong, Rui Zhao, Kai Chen, Mohammad Norouzi, Quoc V. Le. 2018

1,133 citations recorded by the source corpus.

Current end-to-end machine reading and question answering (Q\&A) models are primarily based on recurrent neural networks (RNNs) with attention. Despite their success, these models are often slow for both training and inference due to the sequential nature of RNNs. We propose a new Q\&A architecture called QANet, which does not require recurrent networks: Its encoder consists exclusively of convolution and self-attention, where convolution models local interactions and self-attention models global interactions. On the SQuAD dataset, our model is 3x to 13x faster in training and 4x to 9x faster in inference, while achieving equivalent accuracy to recurrent models. The speed-up gain allows us to train the model with much more data. We hence combine our model with data generated by backtranslation from a neural machine translation model. On the SQuAD dataset, our single model, trained with augmented data, achieves 84.6 F1 score on the test set, which is significantly better than the best published F1 score of 81.8.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Natural Language Processing, Model Architectures

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. Luke S. Zettlemoyer
  4. Percy Liang
  5. Hannaneh Hajishirzi
  6. Yejin Choi
  7. Quoc V. Le
  8. Caiming Xiong

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Jianfeng Gao
  4. Trevor Darrell
  5. Mohit Bansal
  6. Xiaogang Wang
  7. Ming-Hsuan Yang
  8. Cordelia Schmid

In Natural Language Processing

  1. Mohit Bansal
  2. Jianfeng Gao
  3. Luke S. Zettlemoyer
  4. Yejin Choi
  5. Hannaneh Hajishirzi
  6. Percy Liang
  7. Graham Neubig
  8. Caiming Xiong

In Model Architectures

  1. Quoc V. Le
  2. Tie-Yan Liu
  3. Tao Qin
  4. Mohit Bansal
  5. Yoshua Bengio
  6. Richard Socher
  7. Ruslan Salakhutdinov
  8. Jianfeng Gao

Citations in the corpus