Abstract
Knowledge selection is the key component in knowledge-ground dialogues, which aims to choice correct knowledge based on external knowledge for dialogue generation. The quality of knowledge selection depend on knowledge representation methods. However, the knowledge representation exploration is still challenging. We propose a knowledge-grounded dialogue model, which incorporates a knowledge-grounded module and a dialogue generation module, aiming to choose the most appropriate knowledge and fuse it into response generation. In addition, supervised contrastive knowledge representation signal is designed to obtain knowledge representation dynamically. Experiments on FoCus dataset show that our model outperforms the baseline models. The ablation study further demonstrates the effectiveness of each sub-modules.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Zhang, S., et al.: Personalizing dialogue agents: i have a dog, do you have pets too?. In: ACL (Volume 1: Long Papers) (2018)
Kim, M., et al.: Dual task framework for improving persona-grounded dialogue dataset. In: AAAI, vol. 36, no. 10 (2022)
Brown, T., et al.: Language models are few-shot learners. In: NeurIPS, vol. 33, pp. 1877–1901 (2020)
Raffel, C., et al.: Exploring the limits of transfer learning with a unified text-to-text transformer. J. Mach. Learn. Res. 21(1), 5485–5551 (2020)
Wang, P., et al.: OFA: unifying architectures, tasks, and modalities through a simple sequence-to-sequence learning framework. In: International Conference on Machine Learning. PMLR (2022)
Dinan, E., et al.: Wizard of Wikipedia: knowledge-powered conversational agents. In: ICLR (2018)
Smith, E.M., et al.: Can you put it all together: evaluating conversational agents’ ability to blend skills. In: ACL (2020)
Jang, Y., et al.: Call for customized conversation: customized conversation grounding persona and knowledge. In: AAAI, vol. 36, no. 10 (2022)
Saha, S., Das, S., Srihari, R.K.: Proto-gen: an end-to-end neural generator for persona and knowledge grounded response generation. In: Proceedings of the 1st Workshop on Customized Chat Grounding Persona and Knowledge (2022)
Lee, Y.-J., et al.: PERSONACHATGEN: generating Personalized Dialogues using GPT-3. In: Proceedings of the 1st Workshop on Customized Chat Grounding Persona and Knowledge (2022)
Bak, J., Oh, A.: Variational hierarchical user-based conversation model. In: EMNLP-IJCNLP (2019)
Nakamura, K., et al.: HybriDialogue: an information-seeking dialogue dataset grounded on tabular and textual data. In: Findings of the Association for Computational Linguistics: ACL (2022)
Xu, F., et al.: Diverse dialogue generation by fusing mutual persona-aware and self-transferrer. In: APIN, pp. 1–14 (2022)
Wang, Y., et al.: Improving persona understanding for persona-based dialogue generation with diverse knowledge selection. In: ICPR. IEEE Computer Society (2022)
Smith, E.M., et al.: Can you put it all together: evaluating conversational agents’ ability to blend skills. In: ACL (2020)
Chen, T., et al.: A simple framework for contrastive learning of visual representations. In: ICML. PMLR (2020)
Khosla, P., et al.: Supervised contrastive learning. In: NeurIPS, pp. 18661–18673 (2020)
Gunel, B., et al.: Supervised contrastive learning for pre-trained language model fine-tuning. In: ICLR (2020)
An, C., et al.: CoNT: contrastive neural text generation. In: NeurIPS (2022)
Salton, G., Buckley, C.: Term-weighting approaches in automatic text retrieval. IPM 24(5), 513–523 (1988)
Bahdanau, D., Cho, K.H., Bengio, Y.: Neural machine translation by jointly learning to align and translate. In: ICLR (2015)
Zhang, Yu., Yang, Q.: A survey on multi-task learning. IEEE Trans. Knowl. Data Eng. 34(12), 5586–5609 (2021)
Wolf, T., et al.: Transformers: State-of-the-art natural language processing. In: EMNLP: System Demonstrations (2020)
Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)
Popović, M.: chrF++: words helping character N-grams. In: Proceedings of the second conference on machine translation (2017)
Lin, C.-Y.: ROUGE: a package for automatic evaluation of summaries. In: Text Summarization Branches Out (2004)
Vaswani, A., et al.: Attention is all you need. In: NeurIPS (2017)
Acknowledgment
This work was supported in part Key R & D project of Shandong Province 2019JZZY010129, and in part by the Shandong Provincial Social Science Planning Project under Award 19BJCJ51, Award 18CXWJ01, and Award 18BJYJ04.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Wang, B., Xu, F., Zhu, Z., Liu, P. (2023). Knowledge-Grounded Dialogue Generation with Contrastive Knowledge Selection. In: Zhang, F., Wang, H., Barhamgi, M., Chen, L., Zhou, R. (eds) Web Information Systems Engineering – WISE 2023. WISE 2023. Lecture Notes in Computer Science, vol 14306. Springer, Singapore. https://doi.org/10.1007/978-981-99-7254-8_55
Download citation
DOI: https://doi.org/10.1007/978-981-99-7254-8_55
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7253-1
Online ISBN: 978-981-99-7254-8
eBook Packages: Computer ScienceComputer Science (R0)