TY - JOUR
T1 - Infusing internalized knowledge of language models into hybrid prompts for knowledgeable dialogue generation
AU - Bai, Jiaqi
AU - Yan, Zhao
AU - Zhang, Shun
AU - Yang, Jian
AU - Guo, Hongcheng
AU - Li, Zhoujun
N1 - Publisher Copyright:
© 2024
PY - 2024/7/19
Y1 - 2024/7/19
N2 - 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.
AB - 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.
KW - Dialogue response generation
KW - Knowledge-grounded dialogue
KW - Parameter-efficient fine-tuning
KW - Pre-trained language models
UR - https://www.scopus.com/pages/publications/85192139531
U2 - 10.1016/j.knosys.2024.111874
DO - 10.1016/j.knosys.2024.111874
M3 - 文章
AN - SCOPUS:85192139531
SN - 0950-7051
VL - 296
JO - Knowledge-Based Systems
JF - Knowledge-Based Systems
M1 - 111874
ER -