[发明专利]一种对话预测的方法及装置

说明书

本申请提供一种对话预测的方法及装置，属于计算机技术领域，涉及人工智能和计算机视觉技术，用于提高对话预测的准确性。方法包括：获取聊天过程中的当前输入文本的当前文本特征；确定当前输入文本对应的辅助聊天信息；通过编码器提取辅助聊天信息的编码特征；以编码特征和当前文本特征作为解码器的输入，确定编码特征和当前文本特征分别对应的权重，并根据权重将编码特征和当前文本特征进行融合，得到与当前输入文本对应的对话输出特征；基于对话输出特征输出与天过程匹配的对话信息。

技术领域

本申请涉及计算机技术领域，尤其涉及一种对话预测的方法及装置。

背景技术

目前聊天机器人广泛运用于即时通讯平台，一般以娱乐、零售行销、以及客服为目的。聊天机器人是经由对话或文字进行交谈的计算机程序，能够模拟人类对话。程序将通过算法把预先设定好的回答进行回复。研发者把人类自然对话内容放到数据库中供程序学习，当聊天机器人接收到对话时，识别其中的词组和句子，并通过对话预测，从数据库中找到最贴切的答案进行回复。

利用编码器-解码器结构进行对话预测时，将聊天历史文本和当前文本作为输入源，分别先使用编码器生成对应的隐变量特征，这些特征会在解码器端使用自注意力机制分别生成新的隐变量，而不同的输入源对应的变量之间通过平均的方法生成单一变量，作为最终的解码特征进行对话预测。这种方式生成的输出结果与聊天历史的一致性较低，准确性较低，从而对话的流畅度和自然度较低。

发明内容

为解决相关技术中存在的技术问题，本申请实施例提供一种对话预测的方法及装置，可以提高对话预测的准确性。

为达到上述目的，本申请实施例的技术方案是这样实现的：

一方面，本申请实施例提供一种对话预测方法，所述方法包括：

获取聊天过程中的当前输入文本的当前文本特征；

确定所述当前输入文本对应的辅助聊天信息；

通过编码器提取所述辅助聊天信息的编码特征；

以所述编码特征和所述当前文本特征作为解码器的输入，确定所述编码特征和所述当前文本特征分别对应的权重，并根据权重将所述编码特征和所述当前文本特征进行融合，得到与所述当前输入文本对应的对话输出特征；

基于所述对话输出特征输出与所述聊天过程匹配的对话信息。

另一方面，本申请实施例还提供一种对话预测装置，所述装置包括：

特征单元，用于获取聊天过程中的当前输入文本的当前文本特征；

参考单元，用于确定所述当前输入文本对应的辅助聊天信息；

所述特征单元，还用于通过编码器提取所述辅助聊天信息的编码特征；

确定单元，用于以所述编码特征和所述当前文本特征作为所述解码器的输入，确定所述编码特征和所述当前文本特征分别对应的权重，并根据权重将所述编码特征和所述当前文本特征进行融合，得到与所述当前输入文本对应的对话输出特征；

输出单元，用于基于所述对话输出特征输出与所述聊天过程匹配的对话信息。

在一种可选的实施例中，辅助聊天信息包括人物个性信息和历史聊天信息；

特征单元，具体用于：

通过所述编码器，确定所述历史聊天信息对应的历史编码特征，以及所述人物个性信息对应的辅助编码特征。

在一种可选的实施例中，确定单元，具体用于：

将所述编码特征以及所述当前文本特征输入所述编码器对应的解码器中，利用所述解码器中的注意力模型确定所述当前文本特征、所述编码特征分别对应的权重；