A Memory Network Based End-to-end Personalized Task-Oriented Dialogue Generation

Bowen Zhang,Xiaofei Xu,Xutao Li,Yunming Ye,Xiaojun Chen,Zhongjie Wang
DOI: https://doi.org/10.1016/j.knosys.2020.106398
IF: 8.139
2020-01-01
Knowledge-Based Systems
Abstract:Building a personalized task-oriented dialogue system is an important but challenging task. Significant success has been achieved in the template selection responses. However, preparing a massive response template is time-consuming and human-labor intensive. In this paper, we propose an end-to-end framework based on memory networks for response generation in a personalized task-oriented dialogue system. Our model consists of three parts: a retrieval module, a memory encoder network and a memory decoder network. Retrieval module employs the user utterances and user attributes to collect relevant responses from other users. Memory encoder is trained with textual features to obtain dialogue representation. Memory decoder is composed of an RNN and a rule-memory network for response generation. Experiments on the benchmark dataset show that our model achieves better performance than strong baselines in personalized task-oriented dialogue generation.
What problem does this paper attempt to address?