科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明提供一种基于多任务学习的回复多样性多轮对话生成方法、系统、存储介质和电子设备,涉及多轮对话生成技术领域。本发明首先构建多任务学习模型,所述多任务学习模型包括预先训练的多轮对话模型和VAE模型,所述多轮对话模型包括话语级编码器、话语间编码器和第一解码器；将预处理后的多轮对话历史信息中的每个句子序列向量依次输入所述话语级编码器和话语间编码器,获取包含整个所述多轮对话历史信息的隐藏向量；将所述隐藏向量输入第一解码器,所述第一解码器的参数加载自所述VAE模型中的第二解码器参数,获取回复句子序列。将VAE模型解码器参数与多轮对话生成模型共享,增强了多轮对话生成模型里解码器的解码能力,提高了生成回复的文本质量。
- 单位
