Timezone: »

Empowering Language Models with Knowledge Graph Reasoning for Question Answering
Ziniu Hu · Yichong Xu · Wenhao Yu · Shuohang Wang · Ziyi Yang · Chenguang Zhu · Kai-Wei Chang · Yizhou Sun
Event URL: https://openreview.net/forum?id=ADs62ZDRf84 »

Answering open-domain questions requires world knowledge about in-context entities. As pre-trained Language Models (LMs) lack the power to store required knowledge, external knowledge sources, such as knowledge graphs, are often used to augment LMs. In this work, we propose knOwledge REasOning empowered Language Model (OREO-LM), which consists of a novel Knowledge Interaction Layer that can be flexibly plugged into existing Transformer-based LMs to interact with a differentiable Knowledge Graph Reasoning module collaboratively. In this way, LM guides KG to walk towards the desired answer, while the retrieved knowledge improves LM.By adopting OREO-LM to RoBERTa and T5, we show significant performance gain, achieving state-of-art results in the Closed-Book setting. The performance enhancement is mainly from the KG reasoning's capacity to infer missing relational facts. In addition, OREO-LM provides reasoning paths as rationales to interpret the model's decision.

Author Information

Ziniu Hu (UCLA)
Yichong Xu (Microsoft)
Wenhao Yu (University of Notre Dame)
Shuohang Wang (Microsoft)
Ziyi Yang (Stanford University)
Chenguang Zhu (Stanford University)
Kai-Wei Chang (UCLA)
Yizhou Sun (UCLA)

More from the Same Authors