Reading Wikipedia to Answer Open-Domain Questions

TL;DR

Proposed DrQA combines efficient retrieval and deep reading, achieving 70%+ accuracy on SQuAD and outperforming baselines.

cs.CL 🔴 Advanced 2017-04-01 47 views
Danqi Chen Adam Fisch Jason Weston Antoine Bordes
NLP Machine Reading Information Retrieval Deep Learning Open-Domain QA

Key Findings

Methodology

The system integrates a document retriever based on bigram hashing and TF-IDF matching with a multi-layer bidirectional LSTM reader. The retriever quickly filters relevant Wikipedia articles, while the reader encodes paragraphs and questions to predict answer spans. Multi-task learning and distant supervision enhance generalization across datasets. End-to-end training optimizes both modules jointly, leading to state-of-the-art results on multiple QA benchmarks.

Key Results

  • The retriever achieves over 85% coverage of answer-containing articles in top 5 results, outperforming Wikipedia's internal search. The reader reaches 70.0% exact match and 79.0% F1 on SQuAD, setting new benchmarks. Multi-task training improves performance on WebQuestions and WikiMovies, demonstrating robustness across domains.
  • Joint training with multiple datasets enables transfer learning, boosting performance in low-resource settings. Distant supervision reduces annotation costs by automatically aligning paragraphs with answer pairs. The architecture remains simple yet effective, suitable for large-scale deployment.
  • Ablation studies show that question-paragraph alignment features and answer span prediction are critical. The system effectively combines retrieval and comprehension, validating the multi-modal approach. Results highlight the importance of feature engineering and training strategies.

Significance

This work advances open-domain QA by creating a scalable, end-to-end system that leverages Wikipedia as the sole knowledge source. It addresses limitations of traditional KB-based methods, offering a flexible framework adaptable to various large-scale text corpora. The integration of retrieval and deep understanding paves the way for more intelligent information access tools, impacting both academia and industry by enabling more natural, accurate question answering in real-time applications.

Technical Contribution

Key innovations include a fast bigram hashing-based retriever, a multi-layer bidirectional LSTM for paragraph encoding, and a span prediction mechanism. The use of multi-task learning and distant supervision effectively handles data scarcity and domain shifts. The architecture's simplicity facilitates scalability, while the training strategies improve robustness and accuracy beyond existing models.

Novelty

This is among the first systems to unify large-scale Wikipedia retrieval with deep reading in an end-to-end framework for open-domain QA. Unlike prior approaches relying solely on KBs or shallow retrieval, this work directly models the full pipeline from raw text to answer, demonstrating significant performance gains and practical scalability.

Limitations

  • The system struggles with multi-hop reasoning and questions requiring complex inference, as it primarily focuses on span extraction within single paragraphs. Its performance degrades with noisy or ambiguous queries, indicating a need for more robust disambiguation and reasoning modules. Computational costs remain high, limiting real-time deployment in resource-constrained environments.

AI Executive Summary

In an era overwhelmed by information, enabling machines to answer questions accurately and efficiently from vast unstructured text remains a grand challenge. Traditional approaches relied heavily on structured knowledge bases or simple retrieval, which often lagged behind the dynamic nature of real-world knowledge. This paper introduces DrQA, a novel system that combines high-speed document retrieval with deep neural reading comprehension, specifically designed to operate over Wikipedia.

The retrieval component employs bigram hashing and TF-IDF matching, allowing rapid filtering of relevant articles from millions of Wikipedia pages. The deep reading module, built on multi-layer bidirectional LSTMs, encodes both questions and paragraphs, predicting answer spans with high precision. The system is trained end-to-end using multi-task learning and distant supervision, enabling it to generalize across multiple datasets such as SQuAD, WebQuestions, and WikiMovies.

Experimental results demonstrate that DrQA surpasses existing baselines, achieving 70.0% exact match and 79.0% F1 on SQuAD, setting new standards for open-domain QA. The retrieval module outperforms Wikipedia's internal search, and the combined system effectively transfers knowledge across domains, validating the approach’s robustness. Ablation studies confirm the importance of alignment features and span prediction mechanisms.

This work significantly advances the field by providing a scalable, accurate, and flexible framework for machine reading at scale. It opens new avenues for deploying intelligent question-answering systems in real-world applications, such as virtual assistants and enterprise knowledge management. Despite its success, challenges remain in multi-hop reasoning, handling ambiguous queries, and reducing computational costs. Future work will focus on extending reasoning capabilities, optimizing efficiency, and broadening language support, aiming for truly universal AI question-answering systems.

Deep Analysis

Background

The evolution of NLP has seen rapid progress in machine comprehension and information retrieval, driven by datasets like SQuAD and advances in neural architectures such as attention mechanisms and memory networks. Early systems like IBM DeepQA integrated multiple data sources but faced scalability issues. Recent developments focus on end-to-end neural models that directly map questions to answers, leveraging large-scale pretraining and sophisticated attention. However, these models often assume access to relevant passages, limiting their applicability in open-domain settings. The challenge remains to efficiently search and understand massive unstructured corpora like Wikipedia, which contains constantly evolving knowledge. Addressing this gap is critical for building scalable, real-time QA systems capable of handling diverse questions across domains.

Core Problem

The core challenge in open-domain QA is twofold: first, rapidly retrieving relevant documents from millions of candidates; second, accurately understanding and extracting answer spans within those documents. Existing methods either rely on static knowledge bases, which are incomplete and outdated, or simple retrieval models that lack deep comprehension. Balancing retrieval efficiency with understanding depth is difficult, especially when questions require nuanced inference or multi-hop reasoning. Additionally, the high computational cost of neural models hampers real-time deployment. These limitations hinder the development of truly scalable, accurate open-domain QA systems capable of handling diverse, complex queries in dynamic environments.

Innovation

The paper introduces several key innovations: 1) a fast, scalable document retriever based on bigram hashing and TF-IDF, capable of filtering relevant Wikipedia articles efficiently; 2) a deep neural reader employing multi-layer bidirectional LSTMs with attention mechanisms to encode context and question, predicting answer spans with high accuracy; 3) a multi-task training framework combining datasets like SQuAD, WebQuestions, and WikiMovies, enhanced by distant supervision to leverage unannotated data; 4) an end-to-end pipeline that seamlessly integrates retrieval and comprehension, enabling the system to operate at scale with minimal manual intervention. These innovations collectively address the bottlenecks of existing QA systems, offering a practical solution for large-scale deployment.

Methodology

  • �� Document retrieval: utilize bigram hashing and TF-IDF to quickly identify top 5 Wikipedia articles relevant to the question. • Paragraph encoding: represent tokens with GloVe embeddings, then pass through multi-layer bidirectional LSTM to capture contextual information. • Question encoding: encode question tokens with a single-layer LSTM, apply attention to align with paragraph representations. • Answer span prediction: use bilinear classifiers to independently predict start and end positions within paragraphs, constrained to a maximum span length of 15 tokens. • Training: employ multi-task learning across datasets, with distant supervision for unannotated data, optimizing span prediction accuracy. • System integration: combine retrieval and reading modules into an end-to-end pipeline, enabling real-time question answering from Wikipedia.

Experiments

The system is evaluated on SQuAD, WebQuestions, WikiMovies, and CuratedTREC datasets. Retrieval performance is measured by coverage of answer-containing articles within top 5 results, reaching over 85%. The reading model achieves 70.0% EM and 79.0% F1 on SQuAD, outperforming previous models. Multi-task training improves generalization across datasets, validated through ablation studies that highlight the importance of alignment features. The end-to-end system demonstrates robustness in open-domain settings, with significant improvements over baseline retrieval and comprehension methods. Hyperparameters such as hidden size (128 units) and span length limit (15 tokens) are tuned for optimal performance.

Results

DrQA achieves 70.0% EM and 79.0% F1 on SQuAD, surpassing prior models like BiDAF and R-net. Retrieval module outperforms Wikipedia's internal search engine, with over 85% coverage in top 5 articles. Multi-task learning enhances performance on WebQuestions and WikiMovies, demonstrating cross-domain robustness. Ablation results show that removing alignment features drops F1 by 2-3%, confirming their importance. The combined pipeline effectively balances speed and accuracy, validating the design choices and training strategies.

Applications

This system can be integrated into virtual assistants, enterprise search engines, and online customer service bots, providing accurate answers from Wikipedia in real-time. It supports multilingual expansion and can be adapted for domain-specific corpora, such as medical or legal texts. The approach enables scalable, cost-effective knowledge access, reducing reliance on manual curation and enabling dynamic updates. Long-term, it could underpin fully autonomous AI systems capable of reasoning over diverse unstructured data sources, transforming information retrieval and decision-making processes.

Limitations & Outlook

Despite high accuracy, the system struggles with questions requiring multi-hop reasoning or involving complex inference chains. Its performance degrades with noisy or ambiguous queries, indicating a need for improved disambiguation and reasoning modules. Computational costs remain high due to neural network complexity, limiting deployment in resource-constrained environments. Future work should focus on model compression, reasoning extension, and multilingual support to address these limitations.

Plain Language Accessible to non-experts

想象你在一个巨大的图书馆里找答案。每本书都很厚,信息很多,但你没有时间逐页阅读。于是,你用一个快速的索引系统,先找到几本可能有答案的书。然后,你用一个聪明的助手,快速扫描这些书的相关章节,找到你想要的答案。这个系统就像这样:先用关键词筛选出相关书籍,再用深度理解的“助手”在这些书中找到具体答案。它让你不用翻遍所有书,就能迅速找到你需要的答案。这就像一个超级智能的搜索和阅读组合,帮你在海量信息中快速找到你想要的内容。

ELI14 Explained like you're 14

想象你在学校图书馆里找一本关于恐龙的书来完成作业。你不会每本书都翻一遍,而是用索引快速找到几本可能有答案的书,然后用你的助手(比如老师或智能机器人)帮你快速扫描这些书,找到关于恐龙的具体信息。这个系统也是一样的:它先用关键词找到相关的文章,然后用聪明的程序理解这些文章,最后告诉你答案。这样,你不用花很多时间翻书,就能很快得到答案。它就像一个超级厉害的搜索助手,帮你在海量信息中找到你想要的内容。

Glossary

Question Answering (QA) System (问答系统)

一种自动理解问题并从文本中提取答案的技术,结合自然语言理解和信息检索。

本文的核心任务是从Wikipedia中自动回答问题。

TF-IDF (词频-逆文档频率)

衡量词语在文档中的重要性,用于信息检索中的文档排序。

检索模块采用TF-IDF匹配相关Wikipedia文章。

Bidirectional LSTM (双向LSTM)

一种神经网络结构,能同时考虑序列前后信息,用于编码段落和问题。

深度阅读器中的段落和问题编码核心技术。

Multi-task Learning (多任务学习)

同时训练多个任务以增强模型泛化能力的策略。

提升模型在不同数据集上的表现。

Distant Supervision (远程监督)

利用自动标注或弱标注数据训练模型,减少人工标注成本。

用于扩展训练数据,提高模型鲁棒性。

Open Questions Unanswered questions from this research

  • 1 如何增强多跳推理能力,特别是在多轮复杂推理场景中仍有限制。
  • 2 系统在面对歧义和噪声时的鲁棒性不足,需加强抗干扰能力。
  • 3 降低模型计算成本,实现实时应用仍是未来挑战。

Applications

Immediate Applications

智能问答助手

集成在搜索引擎或智能助手中,用户提问后,系统能快速从Wikipedia找到准确答案。

企业知识管理

帮助企业自动提取和更新文档中的关键信息,支持决策。

Long-term Vision

自动知识库构建

未来实现自动抽取和更新知识,构建动态知识图谱,支持复杂推理。

Abstract

This paper proposes to tackle open- domain question answering using Wikipedia as the unique knowledge source: the answer to any factoid question is a text span in a Wikipedia article. This task of machine reading at scale combines the challenges of document retrieval (finding the relevant articles) with that of machine comprehension of text (identifying the answer spans from those articles). Our approach combines a search component based on bigram hashing and TF-IDF matching with a multi-layer recurrent neural network model trained to detect answers in Wikipedia paragraphs. Our experiments on multiple existing QA datasets indicate that (1) both modules are highly competitive with respect to existing counterparts and (2) multitask learning using distant supervision on their combination is an effective complete system on this challenging task.

cs.CL