Latent Retrieval for Weakly Supervised Open Domain Question Answering

TL;DR

Proposes Latent Retrieval with BERT and ICT pretraining, achieving 19-point EM gains over BM25 in open-domain QA.

cs.CL 🔴 Advanced 2019-06-02 59 views
Kenton Lee Ming-Wei Chang Kristina Toutanova
NLP QA Information Retrieval Deep Learning Pretraining

Key Findings

Methodology

This work introduces an end-to-end Latent Retrieval framework combining BERT-based encoders with an unsupervised Inverse Cloze Task (ICT) pretraining. The model comprises a retriever that encodes questions and evidence blocks into dense vectors, and a reader that predicts answer spans. The retriever is trained via inner product similarity, enabling dynamic retrieval from the entire Wikipedia corpus without reliance on gold evidence. The training optimizes marginal likelihood of correct answers, using a combination of maximum likelihood and early updates. Experiments on five datasets demonstrate significant improvements, especially in real user scenarios, with up to 19 EM points over BM25.

Key Results

  • On Natural Questions and WebQuestions, the model outperforms BM25 by 6-19 EM points, reaching up to 36.4, showing strong ability to handle real information-seeking queries.
  • In datasets with known answers like SQuAD and TriviaQA, performance matches or slightly exceeds traditional IR, validating robustness across scenarios.
  • Pretraining with ICT enables the model to learn effective retrieval strategies from unlabeled data, reducing supervision needs and enhancing end-to-end training.

Significance

This research addresses fundamental limitations of traditional IR-based QA, enabling models to learn retrieval strategies directly from question-answer pairs without gold evidence. It advances the field toward more autonomous, scalable, and semantically aware systems capable of understanding complex queries. The approach bridges the gap between retrieval and comprehension, opening new avenues for real-world applications such as virtual assistants and knowledge bases, with profound implications for both academia and industry.

Technical Contribution

The key technical innovation is the integration of the ICT pretraining with a dense vector retrieval mechanism, allowing the retriever to be learned jointly with the reader. This contrasts with prior work that relied on static, heuristic-based retrieval. The model’s ability to encode all evidence into a high-dimensional vector space and perform end-to-end optimization significantly improves retrieval quality. The framework also introduces a scalable, probabilistic training objective that handles the vast search space efficiently, making large-scale open-domain QA feasible.

Novelty

This is the first work to leverage an unsupervised inverse cloze task for pretraining a dense retriever in an open-domain QA setting, enabling fully end-to-end learning from question-answer pairs alone. Unlike previous systems limited to reranking a fixed evidence set, this approach learns to retrieve any evidence from the entire Wikipedia corpus dynamically, representing a major step forward in autonomous information retrieval and comprehension.

Limitations

  • The model requires substantial computational resources for encoding and indexing the entire Wikipedia corpus, which may limit deployment in resource-constrained environments.
  • Despite improved semantic understanding, the system still struggles with highly ambiguous or complex queries requiring multi-hop reasoning.
  • Handling very long answers or multiple answer spans remains challenging due to the span prediction mechanism's constraints.

Future Work

Future research will explore integrating multi-modal data, such as images or structured knowledge graphs, to enhance retrieval and reasoning. Improving efficiency through model compression and indexing optimizations is also a priority. Additionally, extending the framework to multi-hop and multi-answer scenarios, and incorporating explicit reasoning modules, could further elevate system capabilities. The goal is to develop more robust, scalable, and intelligent open-domain QA systems capable of complex real-world tasks.

AI Executive Summary

In the era of information explosion, enabling machines to efficiently find accurate answers from vast textual data remains a grand challenge. Traditional information retrieval methods like BM25 excel at lexical matching but falter in understanding deep semantics and complex queries. Recent advances with pre-trained models like BERT have revolutionized natural language understanding, yet integrating retrieval with comprehension in a fully end-to-end manner has been elusive. This paper introduces a novel Latent Retrieval framework that leverages BERT encoders and an unsupervised Inverse Cloze Task (ICT) pretraining to address these issues.

The core idea is to treat evidence retrieval as a latent variable, enabling the model to learn from question-answer pairs directly without requiring gold evidence annotations. The retriever encodes all Wikipedia paragraphs into dense vectors, facilitating fast similarity-based retrieval via inner product. During training, the model optimizes the likelihood of correct answers, jointly tuning the retriever and reader components. This approach allows the system to dynamically select relevant evidence, even from an open corpus, and generate accurate answers.

Experimental results across five datasets demonstrate the model's robustness. In real user scenarios like Natural Questions and WebQuestions, it surpasses BM25 by 6-19 EM points, reaching up to 36.4. In datasets where the answer is known, performance is comparable to traditional IR, validating its versatility. The key innovation lies in the unsupervised pretraining with ICT, which equips the retriever with semantic understanding, reducing reliance on manual annotations.

This work significantly advances open-domain question answering by enabling fully end-to-end training over large-scale, unannotated corpora. It opens new pathways for autonomous, scalable, and semantically rich QA systems, with broad implications for industry applications such as virtual assistants, search engines, and knowledge bases. Nonetheless, challenges remain in computational efficiency and handling complex reasoning, guiding future research directions.

Deep Analysis

Background

近年来,深度预训练模型如BERT推动了自然语言理解的快速发展,问答系统从封闭式逐步向开放式演进。传统方法依赖于手工标注的金标准证据,如SQuAD中的段落标注,局限于封闭证据集。随着大规模百科知识库的普及,研究者开始探索从全百科中检索答案的可能性,但受限于检索效率和语义理解能力。早期工作如DrQA采用TF-IDF和BM25作为检索基础,结合阅读理解模型实现端到端训练,但仍依赖预定义证据集。近年来,神经检索模型如BERTserini和基于向量的匹配方法取得一定进展,但多依赖于封闭集重排序。真正实现从未标注数据中学习检索策略,仍是学术界的重大挑战。

Core Problem

核心问题在于如何在没有金标准证据的情况下,学习从海量开放文本中动态检索支持答案的证据。传统IR方法受限于词匹配和召回率,难以理解复杂语义,且无法端到端训练。数据偏差、歧义和大规模索引带来的计算成本,使得直接训练检索器成为难题。此外,如何在弱监督条件下,避免模型陷入虚假相关或无关证据,也是一大难点。这些问题限制了问答系统在真实场景中的表现,亟需创新的学习策略和模型架构。

Innovation

本研究的创新点在于引入逆句任务(ICT)作为预训练策略,赋予检索器良好的初始化能力,结合BERT编码实现高效动态检索。模型采用端到端训练框架,融合检索与阅读两个模块,避免中间人工标注依赖。具体创新包括:1)利用无监督的逆句任务,学习语义理解和检索能力;2)通过内积匹配实现高效检索,支持大规模索引;3)采用最大似然和早期更新策略,优化弱监督学习效果。这些创新使模型能够在没有金标准证据的情况下,从全百科中自主学习检索策略,显著提升问答性能。

Methodology

  • �� 预训练阶段:利用逆句任务(ICT)对检索器进行无监督预训练,学习句子与上下文的语义关系。• 构建索引:将所有百科文本编码成128维向量,建立高效索引以支持快速检索。• 端到端训练:输入问题,检索器通过内积匹配筛选Top-k证据块,阅读器对每个证据块中的候选Span进行评分。• 损失函数:最大似然结合早期更新策略,优化检索和答案匹配。• 训练细节:采用多层感知机(MLP)对Span进行评分,利用弱监督信号学习检索策略。• 推理流程:在推理时,先检索Top-k证据块,再由阅读器生成答案,整个流程支持动态调整。• 关键技术:结合BERT编码、向量检索和Span预测,实现端到端学习。

Experiments

  • �� 数据集:在五个公开问答数据集(Natural Questions、WebQuestions、CuratedTrec、TriviaQA、SQuAD)上评估。• 评估指标:采用Exact Match(EM)和F1分数,重点关注真实用户场景。• 实验设置:比较BM25、神经向量检索和提出模型的性能,分析不同预训练策略和索引规模的影响。• 超参数:采用128维向量、Top-k=5检索、学习率10^-5,训练20轮。• 细节:在大规模百科上预编码,利用GPU/TPU加速索引构建和训练。

Results

  • �� 在用户真实需求场景(Natural Questions、WebQuestions)中,提出模型超越BM25,EM提升6-19点,最高达36.4点,验证其在复杂语义理解中的优势。• 在已知答案场景(SQuAD、TriviaQA)中,性能与传统IR相当,显示模型在不同场景的适应性。• 逆句预训练显著改善无监督检索能力,减少对人工标注的依赖。• 端到端训练实现检索与阅读的联合优化,整体性能优于单独优化方案。

Plain Language Accessible to non-experts

想象你在图书馆找一本书里的答案。传统方法就像用目录逐一翻查,效率低,还可能错过重要内容。而新方法像是你提前用一个智能助手学习如何快速找到相关章节:它通过阅读大量书籍,学会了理解句子背后的意思。你只要告诉它问题,它就能从整个图书馆中快速找到最相关的章节,然后帮你找到答案。这就像是它自己学会了在海量信息中寻找线索,而不是依赖事先准备好的索引。这种学习方式让它变得更聪明,也更自主,能应对各种不同的问题,就像你拥有了一个超级聪明的图书馆助手一样。

ELI14 Explained like you're 14

想象你在学校图书馆里找答案。以前,你得用目录一页页翻,花很多时间。而现在,有个聪明的朋友,他提前看过很多书,学会了怎么快速找到你要的内容。你问他问题,他就能告诉你在哪一章、哪一段,甚至直接告诉你答案。这就像这个研究里的新方法,它让电脑自己学会了怎么在大海一样的知识库里找到答案,不需要提前告诉它每个答案在哪里。它通过学习理解句子背后的意思,变得越来越聪明,能帮你在海量信息中找到你想要的答案。这就像拥有一个超级聪明的助手,帮你节省了很多时间,也让搜索变得更聪明、更快!

Abstract

Recent work on open domain question answering (QA) assumes strong supervision of the supporting evidence and/or assumes a blackbox information retrieval (IR) system to retrieve evidence candidates. We argue that both are suboptimal, since gold evidence is not always available, and QA is fundamentally different from IR. We show for the first time that it is possible to jointly learn the retriever and reader from question-answer string pairs and without any IR system. In this setting, evidence retrieval from all of Wikipedia is treated as a latent variable. Since this is impractical to learn from scratch, we pre-train the retriever with an Inverse Cloze Task. We evaluate on open versions of five QA datasets. On datasets where the questioner already knows the answer, a traditional IR system such as BM25 is sufficient. On datasets where a user is genuinely seeking an answer, we show that learned retrieval is crucial, outperforming BM25 by up to 19 points in exact match.

cs.CL