一种利用微调和重排序策略的情感可控回复生成方法

摘要

一种利用微调和重排序策略的情感可控回复生成方法，利用微调算法对大规模开放域语言预训练模型GPT‑2进行训练，即，使用带有情感类别标签的开放领域多轮对话语料训练GPT‑2，通过将对话语句与其对应情感类别标签混合训练的方式，使语言模型学习语义和情感的依赖关系，得到EmoGPT模型，利用EmoGPT模型，根据指定的情感类别标签，将生成的K个回复进行情感重排序，得到最终的生成回复，本发明解决了多轮对话情况下由于对话生成模型中缺乏情感控制方法导致的生成回复的情感与指定情感不一致的问题，实现了多轮对话的情感可控回复生成，提高了对话生成回复的情感一致性。