arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2009.07543cs.CLcs.AI
收藏

神经对话生成的分组对比学习

Group-wise Contrastive Learning for Neural Dialogue Generation

Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学)

Hengyi Cai, Hongshen Chen, Yonghao Song, Zhuoye Ding, Yongjun Bao, Weipeng Yan, Xiaofang Zhao

摘要:

近年来,神经对话响应生成得到了广泛的应用。最大似然估计(MLE)目标被广泛采用在现有的对话模型学习。然而,使用MLE目标函数训练的模型在开放域会话设置时受到低多样性问题的困扰。受人类不仅从积极信号中学习,而且还从纠正不良行为中受益的观察的启发,在这项工作中,我们将对比学习引入对话生成,其中模型明确感知精心选择的积极和消极话语之间的差异。具体来说,我们采用预训练的基线模型作为参考。在对比学习期间,目标对话模型被训练为与参考模型相比,对于肯定样本给出更高的条件概率,并且对于那些否定样本给出更低的条件概率。为了管理人类对话中普遍存在的多映射关系,我们用分组双采样来增强对比对话学习。大量的实验结果表明,所提出的组明智的对比学习框架是适合于训练范围广泛的神经对话生成模型具有非常有利的性能超过基线训练方法。

英文摘要:

Neural dialogue response generation has gained much popularity in recent years. Maximum Likelihood Estimation (MLE) objective is widely adopted in existing dialogue model learning. However, models trained with MLE objective function are plagued by the low-diversity issue when it comes to the open-domain conversational setting. Inspired by the observation that humans not only learn from the positive signals but also benefit from correcting behaviors of undesirable actions, in this work, we introduce contrastive learning into dialogue generation, where the model explicitly perceives the difference between the well-chosen positive and negative utterances. Specifically, we employ a pretrained baseline model as a reference. During contrastive learning, the target dialogue model is trained to give higher conditional probabilities for the positive samples, and lower conditional probabilities for those negative samples, compared to the reference model. To manage the multi-mapping relations prevailed in human conversation, we augment contrastive dialogue learning with group-wise dual sampling. Extensive experimental results show that the proposed group-wise contrastive learning framework is suited for training a wide range of neural dialogue generation models with very favorable performance over the baseline training approaches.

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

↑