[Paper Review] Machine Comprehension Using Match-LSTM and Answer Pointer

Shuohang Wang, Jing Jiang|arXiv (Cornell University)|Aug 29, 2016

Topic Modeling18 references414 citations

TL;DR

The paper presents two end-to-end neural architectures that combine match-LSTM and Pointer Network to tackle SQuAD-style machine comprehension, achieving strong exact-match and F1 scores and outperforming feature-engineered baselines. An ensemble of models yields the best results on the SQuAD test set.

ABSTRACT

Machine comprehension of text is an important problem in natural language processing. A recently released dataset, the Stanford Question Answering Dataset (SQuAD), offers a large number of real questions and their answers created by humans through crowdsourcing. SQuAD provides a challenging testbed for evaluating machine comprehension algorithms, partly because compared with previous datasets, in SQuAD the answers do not come from a small set of candidate answers and they have variable lengths. We propose an end-to-end neural architecture for the task. The architecture is based on match-LSTM, a model we proposed previously for textual entailment, and Pointer Net, a sequence-to-sequence model proposed by Vinyals et al.(2015) to constrain the output tokens to be from the input sequences. We propose two ways of using Pointer Net for our task. Our experiments show that both of our two models substantially outperform the best results obtained by Rajpurkar et al.(2016) using logistic regression and manually crafted features.

Motivation & Objective

• Motivation: improve machine comprehension on SQuAD, where answers are subsequences of the input text and vary in length.
• Goal: develop end-to-end models that generate answers from input tokens without heavy feature engineering.
• Aim: compare sequence vs. boundary Pointer Network approaches and explore ensembling to boost performance.
• Context: build on match-LSTM for textual entailment and Pointer Network to select answer spans from the passage.

Proposed method

• Adopts a preprocessing LSTM to encode passage and question.
• Implements a match-LSTM layer with attention to align passage tokens to the question.
• Uses an Answer Pointer layer based on Pointer Networks to extract answer tokens from the passage.
• Two answer generation modes: (i) sequence model producing a variable-length token sequence, (ii) boundary model predicting start and end positions of a span.
• Optional enhancements: span search (limit to spans of up to 15 tokens) and bi-directional processing (Bi-Ans-Ptr).
• Ensemble method: combine probabilities from multiple boundary models to select the best span.

Experimental results

Research questions

• RQ1 Can end-to-end neural models using match-LSTM and Pointer Networks accurately locate and extract answer spans from a passage for SQuAD-style questions?
• RQ2 Is a boundary-based output (start/end span) more effective than a sequence of tokens for this task?
• RQ3 Does adding search, bi-directional processing, or ensemble methods improve performance on SQuAD?

Key findings

• Boundary model with search outperforms the sequence model on exact-match and F1 metrics.
• Ensembling boundary models yields the best performance on the dev and test sets.
• On the test set, the Boundary+Search+en model achieves an exact-match of 67.9% and an F1 of 77.0%.
• Single models: Boundary+Search achieves 63.0% EM and 72.7% F1 on development data, with larger L and bi-directional variants offering marginal gains.
• Compared to a feature-engineered logistic regression baseline, the neural models significantly improve performance (e.g., EM 40.0–67.9% on dev/test, F1 51.0–77.0%).
• The authors provide qualitative analyses showing attention alignment and variations by question type and answer length.