arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01534cs.CLcs.AI
收藏

Towards Efficiently Diversifying Dialogue Generation via Embedding Augmentation

Tencent AI Lab(腾讯AI实验室)

Yu Cao, Liang Ding, Zhiliang Tian, Meng Fang

英文摘要:

Dialogue generation models face the challenge of producing generic and repetitive responses. Unlike previous augmentation methods that mostly focus on token manipulation and ignore the essential variety within a single sample using hard labels, we propose to promote the generation diversity of the neural dialogue models via soft embedding augmentation along with soft labels in this paper. Particularly, we select some key input tokens and fuse their embeddings together with embeddings from their semantic-neighbor tokens. The new embeddings serve as the input of the model to replace the original one. Besides, soft labels are used in loss calculation, resulting in multi-target supervision for a given input. Our experimental results on two datasets illustrate that our proposed method is capable of generating more diverse responses than raw models while remains a similar n-gram accuracy that ensures the quality of generated responses.

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

↑