[发明专利]神经网络训练方法及装置和环境处理方法及装置有效
| 申请号: | 201810885459.2 | 申请日: | 2018-08-06 |
| 公开(公告)号: | CN109190760B | 公开(公告)日: | 2021-11-30 |
| 发明(设计)人: | 邓煜彬;余可;吕健勤;林达华;汤晓鸥 | 申请(专利权)人: | 北京市商汤科技开发有限公司 |
| 主分类号: | G06N3/08 | 分类号: | G06N3/08;G06N3/04 |
| 代理公司: | 北京林达刘知识产权代理事务所(普通合伙) 11277 | 代理人: | 刘新宇 |
| 地址: | 100084 北京市海淀区中*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 神经网络 训练 方法 装置 环境 处理 | ||
1.一种神经网络训练方法,其特征在于,所述方法包括:
将当前训练周期的环境状态向量输入神经网络中进行处理,获得当前训练周期的动作输出和当前训练周期的衡量输出,其中,所述环境状态向量用于描述当前训练周期的环境,动作输出用于作用于当前周期的环境中,对环境造成改变,所述环境包括游戏环境,所述动作输出包括游戏中的动作指令;或者,所述环境包括交通环境,所述动作输出包括自动驾驶中的操作指令;所述衡量输出用于在对神经网络进行反向调整时,作为调整的参数;
根据所述当前训练周期的环境状态向量和所述当前训练周期的动作输出,确定当前训练周期的第一奖惩反馈,所述第一奖惩反馈表示当前训练周期的动作输出作用于当前周期的环境后,所述环境与理想状态之间的差距的变化;
根据所述当前训练周期的第一奖惩反馈、历史训练周期的第一奖惩反馈和当前训练周期的衡量输出,确定所述神经网络的模型损失,所述历史训练周期包括在所述当前训练周期之前的一个或多个训练周期;
根据所述模型损失,调整所述神经网络的网络参数值;
在所述神经网络满足训练条件时,获得训练后的神经网络。
2.根据权利要求1所述的方法,其特征在于,所述当前训练周期的衡量输出包括当前训练周期的第一衡量输出和当前训练周期的第二衡量输出。
3.根据权利要求2所述的方法,其特征在于,所述模型损失包括与所述第一衡量输出对应的第一模型损失、与所述第二衡量输出对应的第二模型损失以及与所述动作输出对应的第三模型损失。
4.根据权利要求1-3中任一项所述的方法,其特征在于,根据所述当前训练周期的第一奖惩反馈、历史训练周期的第一奖惩反馈和当前训练周期的衡量输出,确定所述神经网络的模型损失,包括:
根据所述当前训练周期的第一奖惩反馈和当前训练周期的第一衡量输出,确定第一模型损失;
根据所述当前训练周期的第一奖惩反馈、历史训练周期的第一奖惩反馈和当前训练周期的第二衡量输出,确定第二模型损失;
根据所述第一模型损失和所述第二模型损失,确定第三模型损失。
5.根据权利要求4所述的方法,其特征在于,根据所述当前训练周期的第一奖惩反馈和当前训练周期的第一衡量输出,确定第一模型损失,包括:
根据所述当前训练周期的第一奖惩反馈、预设的折现率和第一期望奖惩函数,确定第一累积折现奖惩反馈;
根据所述第一累积折现奖惩反馈和所述第一衡量输出,确定所述第一模型损失。
6.根据权利要求4所述的方法,其特征在于,根据所述当前训练周期的第一奖惩反馈、历史训练周期的第一奖惩反馈和当前训练周期的第二衡量输出,确定第二模型损失,包括:
根据所述当前训练周期的第一奖惩反馈和历史训练周期的第一奖惩反馈,确定当前训练周期的第二奖惩反馈;
根据所述当前训练周期的第二奖惩反馈、预设的折现率以及第二期望奖惩函数,确定第二累积折现奖惩反馈;
根据所述第二累积折现奖惩反馈和所述第二衡量输出,确定所述第二模型损失。
7.根据权利要求6所述的方法,其特征在于,根据所述当前训练周期的第一奖惩反馈和历史训练周期的第一奖惩反馈,确定当前训练周期的第二奖惩反馈,包括:
根据历史训练周期的第一奖惩反馈和当前训练周期的第一奖惩反馈,确定奖惩变化向量;
根据所述奖惩变化向量,获得变化累积向量;
根据所述变化累积向量,获得零波动奖惩向量;
根据所述变化累积向量和所述零波动奖惩向量,确定所述第二奖惩反馈。
8.根据权利要求1所述的方法,其特征在于,所述方法还包括:
根据所述当前训练周期的前一个训练周期的环境状态向量和所述前一个训练周期的动作输出,确定所述当前训练周期的环境状态向量。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京市商汤科技开发有限公司,未经北京市商汤科技开发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201810885459.2/1.html,转载请声明来源钻瓜专利网。





