arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1911.04700cs.CLcs.AIcs.LG

A Pre-training Based Personalized Dialogue Generation Model with Persona-sparse Data

Tsinghua University(清华大学);Samsung Research China - Beijing (SRC-B)(三星中国研究院-北京(SRC-B));NetEase Inc.(网易公司)

Yinhe Zheng, Rongsheng Zhang, Xiaoxi Mao, Minlie Huang

英文摘要:

Endowing dialogue systems with personas is essential to deliver more human-like conversations. However, this problem is still far from well explored due to the difficulties of both embodying personalities in natural languages and the persona sparsity issue observed in most dialogue corpora. This paper proposes a pre-training based personalized dialogue model that can generate coherent responses using persona-sparse dialogue data. In this method, a pre-trained language model is used to initialize an encoder and decoder, and personal attribute embeddings are devised to model richer dialogue contexts by encoding speakers' personas together with dialogue histories. Further, to incorporate the target persona in the decoding process and to balance its contribution, an attention routing structure is devised in the decoder to merge features extracted from the target persona and dialogue contexts using dynamically predicted weights. Our model can utilize persona-sparse dialogues in a unified manner during the training process, and can also control the amount of persona-related features to exhibit during the inference process. Both automatic and manual evaluation demonstrates that the proposed model outperforms state-of-the-art methods for generating more coherent and persona consistent responses with persona-sparse data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑