一种基于知识选择的对话模型及其训练方法

    公开(公告)号:CN115392263A

    公开(公告)日:2022-11-25

    申请号:CN202211023548.9

    申请日:2022-08-25

    Abstract: 一种基于知识选择的对话模型,所述模型包括:编码模块,其包括文本编码器和知识编码器,所述文本编码器用于对提问者语句进行编码获得提问者语句编码向量,所述知识编码器用于对知识库中的知识进行编码获得知识编码向量;知识选择模块,用于根据编码模块获得的所述提问者语句编码向量和知识编码向量从知识库中为提问者语句选择知识,所述知识选择模块通过采用强化学习的方式以对话文本及其上下文为输入、为对话文本选择的知识为输出训练获得;解码模块,用于根据所述知识选择模块选择的知识与所述提问者语句编码向量进行解码生成提问者语句的回复语句。

Patent Agency Ranking