KG-FiD: Infusing Knowledge Graph in Fusion-in-Decoder for Open-Domain Question Answering
Donghan Yu Thanks: ˜˜Work done during internship at Microsoft. Chenguang Zhu Affiliation: Carnegie Mellon University Microsoft Cognitive Services Research Group Affiliation: Yuwei Fang Affiliation: Carnegie Mellon University Microsoft Cognitive Services Research Group Affiliation: Wenhao Yu Shuohang Wang Affiliation: Carnegie Mellon University Microsoft Cognitive Services Research Group Affiliation: Yichong Xu Affiliation: Carnegie Mellon University Microsoft Cognitive Services Research Group Affiliation: Xiang Ren Affiliation: University of Notre Dame University of Southern California Yiming Yang Michael Zeng Affiliation: Carnegie Mellon University Microsoft Cognitive Services Research Group Affiliation:
Abstract
Current Open-Domain Question Answering (ODQA) models typically include a retrieving module and a reading module, where the retriever selects potentially relevant passages from open-source documents for a given question, and the reader produces an answer based on the retrieved passages. The recently proposed Fusion-in-Decoder (FiD) framework is a representative example, which is built on top of a dense passage retriever and a generative reader, achieving the state-of-the-art performance. In this paper we further improve the FiD approach by introducing a knowledge-enhanced version, namely KG-FiD. Our new model uses a knowledge graph to establish the structural relationship among the retrieved passages, and a graph neural network (GNN) to re-rank the passages and select only a top few for further processing. Our experiments on common ODQA benchmark datasets (Natural Questions and TriviaQA) demonstrate that KG-FiD can achieve comparable or better performance in answer prediction than FiD, with less than 40% of the computation cost.
原文 arXiv:2110.04330;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2110.04330v2