Multi-Task Learning of Generation and Classification for Emotion-Aware Dialogue Response Generation

Abstract

For a computer to naturally interact with a human, it needs to be human-like. In this paper, we propose a neural response generation model with multi-task learning of generation and classification, focusing on emotion. Our model based on BART (Lewis et al., 2020), a pre-trained transformer encoder-decoder model, is trained to generate responses and recognize emotions simultaneously. Furthermore, we weight the losses for the tasks to control the update of parameters. Automatic evaluations and crowdsourced manual evaluations show that the proposed model makes generated responses more emotionally aware.

arXiv commentNAACL Student Research Workshop (SRW) 2021

References in corpus (4)

• Wizard of Wikipedia: Knowledge-Powered Conversational agents
• How NOT To Evaluate Your Dialogue System: An Empirical Study of Unsupervised Evaluation Metrics for Dialogue Response Generation
• EmpTransfo: A Multi-head Transformer Architecture for Creating Empathetic Dialog Systems
• Target Guided Emotion Aware Chat Machine