对话生成方法、装置、电子设备、存储介质

    公开(公告)号:CN116522873A

    公开(公告)日:2023-08-01

    申请号:CN202310403412.9

    申请日:2023-04-14

    Abstract: 一种对话生成方法、装置、电子设备、存储介质,方法包括:输入对话历史的第一编码至掩码语言模型,获得所述掩码语言模型输出的隐状态;输入所述隐状态至多标签分类器,获得所述多标签分类器输出的标签向量,所述标签向量确定预先定义的实体列表中每个实体出现在对话下文中的概率;基于所述标签向量,获取所述对话下文的预测实体;基于所述对话历史和所述预测实体,拼接形成第二编码;输入所述第二编码至因果语言模型,获得所述因果语言模型输出的预测序列,所述预测序列构成所述对话下文。通过预测实体实现对对话生成方向的引导,提高对话下文的知识丰富程度,使得对话生成的对话下文更符合目标需求。

Patent Agency Ranking