Emotion Controllable Dialogue Generation Method Combining Fine-Tuning and Reranking

DU Baoxiang,MA Zhiqiang,WANG Chunyu,JIA Wenchao,WANG Hongbin

DOI: https://doi.org/10.3778/j.issn.1673-9418.2106060

2023-01-01

Abstract:The existing emotional dialogue generation methods are usually based on the sequence-to-sequence（Seq2Seq） dialogue generation models, and improve the encoding or decoding in terms of emotion. This method can achieve a certain extent of emotional response generation abilities but is prone to low-quality and common response problems. To solve these problems and achieve emotion controllable and high-quality response generation, this paper proposes an emotion controllable dialogue generation method combining fine-tuning and reranking, called EmoGPT（emotional generative pre-trained transformer）. In the model training phase, a method for fine-tuning the GPT-2model using dialogue corpus with emotion category labels is proposed to learn the semantic and emotional dependence of the sentences; in the response generation phase, an emotion reranking strategy is proposed to score and sort the generated multiple responses to improve the controllability of the response emotions. Experimental results of emotional response generation using a dialogue dataset with emotion labels show that EmoGPT with an emotion reranking strategy achieves better performance than comparison models in terms of content relevance and emotion consistency of the generated responses, thus verifying the ability of the proposed method to generate emotion controllable and high-quality responses.