Skip to main navigation Skip to search Skip to main content

Infusing internalized knowledge of language models into hybrid prompts for knowledgeable dialogue generation

  • Jiaqi Bai
  • , Zhao Yan
  • , Shun Zhang
  • , Jian Yang
  • , Hongcheng Guo
  • , Zhoujun Li*
  • *Corresponding author for this work
  • Beihang University
  • Tencent

Research output: Contribution to journal › Article › peer-review

Abstract

Existing knowledge-grounded dialogue (KGD) systems access the knowledge from an external knowledge base, then generate the context-coherent response accordingly. However, the knowledge access capability is constrained to the scale of a knowledge base. On the one hand, a small-scale knowledge base makes a model hard to generalize on unseen topics, while the improper shift of topics may induce an unsmooth conversation flow. On the other hand, a large-scale knowledge base requires a strong retrieval component to accurately index the context-relevant knowledge from many plausible candidates, costing significant amounts of time and resources. To address this, we regard the language model as a virtual knowledge base and propose homogenizing internalized knowledge of different language models into hybrid prompts. The hybrid prompts are a set of continuous vectors learned to represent knowledge inherently encoded in different language models. Furthermore, we devise a two-stage knowledge-grounding manner, in which both the knowledge internalized in language models and the knowledge provided by evidence can be jointly optimized to generate a knowledgeable response. We compare our proposed method with two groups of methods, including methods with explicit knowledge retrieval and those with implicit knowledge access. Experimental results on three knowledge-grounded dialogue corpora demonstrate advantages over these competitive methods.

Original languageEnglish
Article number111874
JournalKnowledge-Based Systems
Volume296
DOIs
StatePublished - 19 Jul 2024

Keywords

  • Dialogue response generation
  • Knowledge-grounded dialogue
  • Parameter-efficient fine-tuning
  • Pre-trained language models

Fingerprint

Dive into the research topics of 'Infusing internalized knowledge of language models into hybrid prompts for knowledgeable dialogue generation'. Together they form a unique fingerprint.

Cite this