A memory network based end-to-end personalized task-oriented dialogue generation

摘要

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.