您所在的位置: 成果库 一种基于常识扩展的角色一致性对话生成方法

一种基于常识扩展的角色一致性对话生成方法

发布时间: 2023-06-19

来源: 科技服务团

基本信息

合作方式: 作价入股
成果类型: 发明专利,新技术
行业领域:
电子信息技术
成果介绍
本发明属于自然语言处理领域,具体涉及一种基于常识扩展的角色一致性对话生成方法;
成果亮点
包括:获取常识扩展数据集并进行人物角色语句提取和对话历史语句的有效语句提取,得到第一对话数据集和第二对话数据集;组合第一对话数据集和第二对话数据集,对组合后的数据集和常识扩展数据集进行训练,直到两数据集相互收敛,对收敛后的组合数据集中的人物角色语句和对话历史语句进行嵌入编码;将编码后的结果输入到GPT‑2模型中进行训练,得到初始对话回复;将初始对话回复和编码后的人物角色语句输入到一致性理解解码器中进行训练,得到最终的一致性对话回复;本发明有效提高了对话生成模型的角色一致性和回复多样性。
团队介绍
唐宏,朱传润,杨浩澜,王宁喆,刘钟
成果资料