Group-wise Contrastive Learning for Neural Dialogue Generation

Abstract

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.

References in corpus (8)

• A Simple Framework for Contrastive Learning of Visual Representations
• Deep Reinforcement Learning for Dialogue Generation
• Topic Aware Neural Response Generation
• A Fast and Simple Algorithm for Training Neural Probabilistic Language Models
• Personalizing Dialogue Agents: I have a dog, do you have pets too?
• Generating Informative and Diverse Conversational Responses via Adversarial Information Maximization
• Hierarchical Recurrent Attention Network for Response Generation
• Contrastive Learning for Image Captioning