[发明专利]神经网络的训练、对话生成方法及装置、存储介质在审
| 申请号: | 202010620299.6 | 申请日: | 2020-06-30 | 
| 公开(公告)号: | CN111783955A | 公开(公告)日: | 2020-10-16 | 
| 发明(设计)人: | 章学妙;曹阳 | 申请(专利权)人: | 北京市商汤科技开发有限公司 | 
| 主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08;G06F16/332;G06F16/35 | 
| 代理公司: | 北京博思佳知识产权代理有限公司 11415 | 代理人: | 靳玫 | 
| 地址: | 100080 北京市海淀区北*** | 国省代码: | 北京;11 | 
| 权利要求书: | 查看更多 | 说明书: | 查看更多 | 
| 摘要: | |||
| 搜索关键词: | 神经网络 训练 对话 生成 方法 装置 存储 介质 | ||
1.一种神经网络的训练方法,其特征在于,所述神经网络包括生成器模块以及多个判别器模块,其中,多个判别器模块对应不同优化目标,所述方法包括:
将样本查询语句输入所述生成器模块,得到第一回复语句;
在所述多个判别器模块中,确定第一轮对抗训练采用的第一判别器模块;
获取与所述样本查询语句对应的样本回复语句,并根据所述样本回复语句和所述第一回复语句,确定所述第一判别器模块在对应的第一优化目标上产生的第一奖励信号;
在所述第一轮对抗训练中,根据所述样本回复语句、所述第一回复语句、所述第一奖励信号,对所述神经网络进行训练。
2.根据权利要求1所述的方法,其特征在于,所述在所述第一轮对抗训练中,根据所述样本回复语句、所述第一回复语句、所述第一奖励信号,对所述神经网络进行训练,包括如下至少一项:
在所述第一轮对抗训练中,根据所述第一奖励信号,对所述生成器模块进行训练;
在所述第一轮对抗训练中,根据所述样本回复语句和所述第一回复语句,对所述第一判别器模块进行训练。
3.根据权利要求1或2所述的方法,其特征在于,所述将样本查询语句输入所述生成器模块,得到第一回复语句,包括:
通过叠加的多层长短期记忆网络对输入的所述样本查询语句进行编码,得到与所述样本查询语句对应的至少一个隐状态向量;
再通过叠加的多层长短期记忆网络对与所述样本查询语句对应的至少一个隐状态向量进行解码后,得到所述第一回复语句。
4.根据权利要求1-3任一项所述的方法,其特征在于,所述多个判别器模块对应的采样概率值满足采样分布函数;
所述在所述多个判别器模块中,确定第一轮对抗训练采用的第一判别器模块,包括:
根据所述采样分布函数,将所述多个判别器模块中所述采样概率值最大的判别器模块作为所述第一判别器模块。
5.根据权利要求1-4任一项所述的方法,其特征在于,所述根据所述样本回复语句和所述第一回复语句,确定所述第一判别器模块在对应的第一优化目标上产生的第一奖励信号,包括:
通过第一判别器模块从所述样本回复语句和所述第一回复语句中提取与所述第一优化目标对应的特征信息;
根据与所述第一优化目标对应的所述特征信息,确定所述第一回复语句在所述第一优化目标上取得的第一分值;
在所述第一轮对抗训练中,将所述第一分值的变化量作为所述第一奖励信号对应的数值。
6.根据权利要求5所述的方法,其特征在于,所述第一优化目标包括语言优化目标;
所述通过第一判别器模块从所述样本回复语句和所述第一回复语句中提取与所述第一优化目标对应的特征信息,包括:
通过叠加的多层长短期记忆网络从所述样本回复语句和所述第一回复语句中提取与所述语言优化目标对应的语言特征信息;
所述根据与所述第一优化目标对应的所述特征信息,确定所述第一回复语句在所述第一优化目标上取得的第一分值,包括:
通过二分类判别器对从所述样本回复语句和所述第一回复语句中提取的所述语言特征信息进行分类,得到分类结果;
将所述分类结果为真值的所述语言特征信息的概率值作为所述第一回复语句在所述语言优化目标上取得的所述第一分值。
7.根据权利要求5所述的方法,其特征在于,所述第一优化目标包括语法优化目标;
所述通过第一判别器模块从所述样本回复语句和所述第一回复语句中提取与所述第一优化目标对应的特征信息,包括:
通过预先训练好的语法图卷积神经网络,从所述样本回复语句和所述第一回复语句中提取与所述语法优化目标对应的语法特征信息;
所述根据与所述第一优化目标对应的所述特征信息,确定所述第一回复语句在所述第一优化目标上取得的第一分值,包括:
对从所述样本回复语句和所述第一回复语句中提取的所述语法特征信息依次进行批正则化处理和线性回归处理,确定所述第一回复语句在所述语法优化目标上取得的所述第一分值。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京市商汤科技开发有限公司,未经北京市商汤科技开发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010620299.6/1.html,转载请声明来源钻瓜专利网。





