AAAI Publications, Thirty-Second AAAI Conference on Artificial Intelligence

Font Size: 
Personalizing a Dialogue System With Transfer Reinforcement Learning
Kaixiang Mo, Yu Zhang, Shuangyin Li, Jiajun Li, Qiang Yang

Last modified: 2018-04-27


It is difficult to train a personalized task-oriented dialogue system because the data collected from each individual is often insufficient. Personalized dialogue systems trained on a small dataset is likely to overfit and make it difficult to adapt to different user needs. One way to solve this problem is to consider a collection of multiple users as a source domain and an individual user as a target domain, and to perform transfer learning from the source domain to the target domain. By following this idea, we propose a PErsonalized Task-oriented diALogue (PETAL) system, a transfer reinforcement learning framework based on POMDP, to construct a personalized dialogue system. The PETAL system first learns common dialogue knowledge from the source domain and then adapts this knowledge to the target domain. The proposed PETAL system can avoid the negative transfer problem by considering differences between the source and target users in a personalized Q-function. Experimental results on a real-world coffee-shopping data and simulation data show that the proposed PETAL system can learn optimal policies for different users, and thus effectively improve the dialogue quality under the personalized setting.


transfer learning; reinforcement learning; task-oriented dialogue system; personalized dialogue system; transfer reinforcement learning; multi-turn dialogue system;

Full Text: PDF