Open-Domain Dialogue Generation Based on Pre-trained Language Models

Abstract

Pre-trained language models have been successfully used in response generation for open-domain dialogue. Four main frameworks have been proposed: (1) Transformer-ED using Transformer encoder and decoder separately for source and target sentences; (2) Transformer-Dec using Transformer decoder for both source and target sentences; (3) Transformer-MLM using Transformer decoder that applies bi-directional attention on the source side and left-to-right attention on the target side with masked language model objective; and (4) Transformer-AR that uses auto-regressive objective instead. In this study, we compare these frameworks on 3 datasets, and our comparison reveals that the best framework uses bidirectional attention on the source side and does not separate encoder and decoder. We also examine model discrepancy, and our experiments confirm that the performance of a model is directly impacted by the underlying discrepancies. We then propose two correction methods to reduce the discrepancies, and both improve the model performance. These results show that discrepancies is an important factor to consider when we use a pre-trained model, and a reduction in discrepancies can lead to improved performance.

arXiv comment[v0], 10 pages, 4 figures

References in corpus (11)

• A Neural Conversational Model
• TransferTransfo: A Transfer Learning Approach for Neural Network Based Conversational Agents
• Towards a Human-like Open-Domain Chatbot
• UniLMv2: Pseudo-Masked Language Models for Unified Language Model Pre-Training
• DialoGPT: Large-Scale Generative Pre-training for Conversational Response Generation
• Mutual Information and Diverse Decoding Improve Neural Machine Translation
• Generating Wikipedia by Summarizing Long Sequences
• PLATO-2: Towards Building an Open-Domain Chatbot via Curriculum Learning
• CAiRE: An Empathetic Neural Chatbot
• A Pre-training Based Personalized Dialogue Generation Model with Persona-sparse Data
• Multi-Domain Dialogue State Tracking based on State Graph

Cited by in corpus (1)

• A Simple and Efficient Multi-Task Learning Approach for Conditioned Dialogue Generation