Dicer: Dialogue-Centric Representation for Knowledge-Grounded Dialogue through Contrastive Learning

Knowledge-grounded dialogue is a task that utilizes external knowledge to generate appropriate and fluent responses to statements. Owing to the relevance of generating responses based on relevant knowledge in diverse fields, the knowledge selection task has been spotlighted. In this study, we propose a novel selection model that applies contrastive-learning with negative sampling loss to create dialoguecentric representation of knowledge. A two-part loss is considered knowledge selection loss and topic prediction loss. The former increases the similarity between content representations of related knowledge and dialogue history, while the latter increases the similarity between their topic representations. The proposed model was evaluated on two well-known datasets,Wizard of Wikipedia and Holl-E, in terms of the knowledge-grounded dialogue task exhibiting remarkable improvement over previously proposed methods on both knowledge selection and response generation tasks.& COPY; 2023 Elsevier B.V. All rights reserved.

Automated summary

Knowledge-grounded dialogue is a task that utilizes external knowledge to generate appropriate responses. In this study, a novel selection model using contrastive learning and negative sampling is proposed to create dialogue-centric representations of knowledge. The model incorporates knowledge selection loss and topic prediction loss to improve similarity between knowledge and dialogue history. The proposed model shows remarkable improvement on knowledge selection and response generation tasks.