KG-BART: Knowledge Graph-Augmented BART for Generative Commonsense Reasoning

Authors

  • Ye Liu University Of Illinois at Chicago
  • Yao Wan Huazhong University of Science and Technology
  • Lifang He Lehigh University
  • Hao Peng Beihang University
  • Philip S. Yu University Of Illinois at Chicago

DOI:

https://doi.org/10.1609/aaai.v35i7.16796

Keywords:

Common-Sense Reasoning, Generation

Abstract

Generative commonsense reasoning which aims to empower machines to generate sentences with the capacity of reasoning over a set of concepts is a critical bottleneck for text generation. Even the state-of-the-art pre-trained language generation models struggle at this task and often produce implausible and anomalous sentences. One reason is that they rarely consider incorporating the knowledge graph which can provide rich relational information among the commonsense concepts. To promote the ability of commonsense reasoning for text generation, we propose a novel knowledge graph augmented pre-trained language generation model KG-BART, which encompasses the complex relations of concepts through the knowledge graph and produces more logical and natural sentences as output. Moreover, KG-BART can leverage the graph attention to aggregate the rich concept semantics that enhances the model generalization on unseen concept sets. Experiments on benchmark CommonGen dataset verify the effectiveness of our proposed approach by comparing with several strong pre-trained language generation models, particularly KG-BART outperforms BART by 5.80, 4.60, in terms of BLEU-3, 4. Moreover, we also show that the generated context by our model can work as background scenarios to benefit downstream commonsense QA tasks.

Downloads

Published

2021-05-18

How to Cite

Liu, Y., Wan, Y., He, L., Peng, H., & Yu, P. . S. (2021). KG-BART: Knowledge Graph-Augmented BART for Generative Commonsense Reasoning. Proceedings of the AAAI Conference on Artificial Intelligence, 35(7), 6418-6425. https://doi.org/10.1609/aaai.v35i7.16796

Issue

Section

AAAI Technical Track on Knowledge Representation and Reasoning