Source-linked AI summary

Simple and Effective Semi-Supervised Question Answering

Bhuwan Dhingra, Danish Pruthi, Dheeraj Rajagopal

arXiv:1804.00720v1cs.CLcs.LG

TL;DR

Domain-specific QA can require large annotated corpora that are limited and expensive to collect. The paper generates cloze questions from base documents, pretrains QA models on them, and fine-tunes on labeled examples; it reports low-resource gains across three datasets, including over 50% F1 on SQuAD and TriviaQA using 1% of training data.

  • Problem

    Large annotated corpora for domain-specific extractive QA are limited and expensive to construct, constraining practitioners who lack resources to collect many questions.

    Source: §1 · p. 1
  • Method

    The system constructs cloze questions from base documents, pretrains an extractive QA model on them, and fine-tunes it on a small labeled set.

    Source: §1 · p. 1
  • Results

    Cloze pretraining improves low-resource performance across SQuAD, TriviaQA, and BioASQ; on SQuAD and TriviaQA, it exceeds 50% F1 with 1% of training data.

    Source: §1 · p. 1, §4.2 · p. 4
  • Takeaways & Limitations

    Pretraining most helps questions asking for factual information located in specific context passages, while fine-tuning remains necessary.

    Source: §5 · p. 5

    Cloze generation assumes an introduction-body structure and uses the first 20% of a document when no clear introduction boundary exists, despite noise.

    Source: §3 · p. 2

Abstract

from arXiv · show

Recent success of deep learning models for the task of extractive Question Answering (QA) is hinged on the availability of large annotated corpora. However, large domain specific annotated corpora are limited and expensive to construct. In this work, we envision a system where the end user specifies a set of base documents and only a few labelled examples. Our system exploits the document structure to create cloze-style questions from these base documents; pre-trains a powerful neural network on the cloze style questions; and further fine-tunes the model on the labeled examples. We evaluate our proposed system across three diverse datasets from different domains, and find it to be highly effective with very little labeled data. We attain more than 50% F1 score on SQuAD and TriviaQA with less than a thousand labelled examples. We are also releasing a set of 3.2M cloze-style questions for practitioners to use while building QA systems.

1 Introduction

The system uses document structure to generate cloze questions for QA pretraining, then fine-tunes on a small labeled set. Across three datasets, it improves low-resource performance, with especially strong reported results on SQuAD and TriviaQA.

1 Introduction

  • The approach targets the scarcity and expense of large annotated corpora, especially for domain-specific QA and low-resource settings.It hypothesizes particular benefit for factual questions lexically similar to the base documents.Source: §1 · p. 1
  • The system generates cloze questions from base documents, pretrains an extractive QA model on them, and fine-tunes it using a small set of labeled question-answer pairs.The user specifies the base documents and labeled examples.Source: §1 · p. 1
  • More than 50% F1 on SQuAD and TriviaQA is achieved with 1% of training data, and the system improves low-resource performance across all three datasets.The datasets are SQuAD, TriviaQA-Web, and BioASQ.Source: §1 · p. 1

2 Related Work

Prior semi-supervised QA and domain adaptation methods use generated questions or manually curated source datasets. This work instead constructs a source dataset from unlabeled documents in the target domain.

2 Related Work

  • GDAN uses reinforcement learning to generate questions from passages, whereas this work uses a simpler heuristic and reports superior performance to GDAN.Other natural-question generation approaches have not shown significant improvement from generated questions in semi-supervised QA.Source: §2 · p. 2
  • Earlier QA transfer learning uses manually curated source datasets, while this work automatically constructs a source dataset from the target domain and reports better performance.Prior examples transfer from SQuAD to BioASQ or from Book Test to SQuAD.Source: §2 · p. 2
  • Cloze datasets have been constructed with heuristics, and this work examines their use in a semi-supervised QA setting.The cited cloze datasets include those of Hermann et al. (2015), Onishi et al. (2016), and Hill et al. (2016).Source: §2 · p. 2

3 Methodology

The method creates cloze QA triples by matching answer-bearing introduction text to body passages, then uses the resulting corpora for pretraining and fine-tunes on labeled examples. It retains noisy clozes and finds the full corpus preferable to selected subsets.

3 Methodology

  • Cloze generation assumes documents have introductions that summarize their bodies, using the first 20% when no clear boundary exists; matched answer candidates are drawn from body passages.The heuristic is noisy, but the authors found it beneficial; exact matching spans must be recognized phrases or named entities, and triples with fewer than two non-stopword overlaps are pruned.Source: §3 · p. 2
  • The generated Wikipedia and PubMed corpora contain 2.2M and 1M clozes, respectively; manual analysis found their passages answerable in 76% and 80% of cases.The clozes often paraphrase passage information.Source: §3 · p. 3
  • Using the entire cloze corpus performed better than selecting subsets by sentence-passage Jaccard similarity, answer-candidate tf-idf, or answer length.Source: §3 · p. 3
  • The cloze corpus pretrains GA Reader and BiDAF+SA; BiDAF+SA reaches 0.58 F1 and 0.55 EM on Wikipedia cloze development questions.The authors interpret this as evidence that the cloze corpus is neither too easy nor too difficult to answer.Source: §3 · p. 3
  • The pretrained model is fine-tuned on a small set of labeled question-answer pairs, a step the authors say is crucial for achieving substantial performance with few examples.The paper contrasts this with the tens of thousands of questions typically used in training.Source: §3 · p. 3

4 Experiments & Results

Cloze pretraining improves low-resource QA results across SQuAD, TriviaQA, and BioASQ. Its benefits vary with question wording, answer span length, and question type.

4.1 Datasets

  • Evaluation covers SQuAD and TriviaQA span QA plus BioASQ factoid and list questions, using training-data subsets and five-fold cross-validation for BioASQ.The BioASQ setup uses all 899 available training questions.Source: §4.1 · p. 3

4.2 Main Results

  • With 1% of training data (454 SQuAD questions and 746 TriviaQA questions), BiDAF+SA exceeds 50% F1; at 10%, it approaches 90% of best performance.The comparison covers SQuAD and TriviaQA across training-data fractions.Source: Table 2 · p. 4, §4.2 · p. 4
  • Cloze pretraining outperforms GDAN on SQuAD and same-data language-model pretraining; its gains over supervised learning shrink as labeled data grows.The authors attribute the language-model baseline's weaker performance partly to only a subset of its parameters being pretrained.Source: Table 2 · p. 4, §4.2 · p. 4
  • On BioASQ, cloze pretraining significantly improves over supervised learning, with a larger gain than prior SQuAD pretraining on that target.The authors believe the improvement is smaller than on SQuAD and TriviaQA because BioASQ questions are generally more difficult.Source: Table 3 · p. 4, §4.2 · p. 4

4.3 Analysis

  • In SQuAD regression analysis, answer-sentence lexical overlap predicts larger pretraining gains, while whole-passage overlap predicts the opposite; short answers and rare tokens also benefit.The authors connect these patterns to cloze construction, which tends to produce short answer spans and covers a broader domain than SQuAD training data.Source: §4.3 · pp. 4–5, Figure 1 · p. 5
  • Across question types, pretraining improves broadly, especially for abbreviation questions, while why questions improve least, consistent with their reasoning or world-knowledge demands.The analysis uses SQuAD's 10% training split and two question classifications.Source: Figure 2 · p. 5, §4.3 · p. 5

5 Conclusion

Pre-training QA models with automatically constructed cloze questions significantly improves performance, especially with few labeled examples. The paper also finds that fine-tuning is needed and that pre-training benefits factual questions whose answers occur in a specific context location.

  • Automatically constructed cloze-question pre-training significantly improves QA performance, especially when labeled examples are scarce.Source: §5 · p. 5
  • Models trained only on cloze questions perform poorly, supporting the need to fine-tune them on labeled examples.Source: §5 · p. 5
  • Regression analysis finds pre-training helps with factual questions whose answers are located in a specific part of the context.Source: §5 · p. 5
  • Future work will explore active learning for choosing passages or question types to annotate, and adapting cloze pre-training to other NLP tasks.Source: §5 · p. 5
Loading 1804.00720v1…