[发明专利]基于超参优化的深度强化学习模型的训练方法、装置在审
申请号: | 202011621981.3 | 申请日: | 2020-12-31 |
公开(公告)号: | CN113723615A | 公开(公告)日: | 2021-11-30 |
发明(设计)人: | 张玥;尹泽夏;霍雨森;王小波;郑宇 | 申请(专利权)人: | 京东城市(北京)数字科技有限公司 |
主分类号: | G06N20/00 | 分类号: | G06N20/00 |
代理公司: | 北京清亦华知识产权代理事务所(普通合伙) 11201 | 代理人: | 张大威 |
地址: | 100086 北京市海淀区知*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 优化 深度 强化 学习 模型 训练 方法 装置 | ||
1.一种基于超参优化的深度强化学习模型的训练方法,其特征在于,所述方法包括:
获取多个初始超参数组合,和多个第一深度强化学习模型;
采用所述初始超参数组合中的多个超参数训练所述多个第一深度强化学习模型,以得到与所述多个第一深度强化学习模型分别对应的训练评价指标;
根据所述训练评价指标从多个第一深度强化学习模型之中筛选出第二深度强化学习模型;
采用与所述第二深度强化学习模型对应的多个目标超参数对所述初始超参数组合进行优化处理,以形成目标超参数组合;以及
采用所述目标超参数组合之中的多个超参数训练所述第二深度强化学习模型,以得到目标深度强化学习模型。
2.如权利要求1所述的方法,其特征在于,所述采用与所述第二深度强化学习模型对应的多个目标超参数对所述初始超参数组合进行优化处理,以形成目标超参数组合,包括:
确定所述初始超参数组合属于的超参数集合;
将所述多个目标超参数补充添加至所述超参数集合之中,以得到目标超参数集合;以及
从所述目标超参数集合之中选取至少部分超参数,并根据所述至少部分超参数形成所述目标超参数组合。
3.如权利要求1所述的方法,其特征在于,还包括:
在训练所述第二深度强化学习模型的次数达到设定迭代次数时,重新采用与所述第二深度强化学习模型对应的多个目标超参数对所述初始超参数组合进行优化处理。
4.如权利要求1所述的方法,其特征在于,所述采用所述目标超参数组合之中的多个超参数训练所述第二深度强化学习模型,以得到目标深度强化学习模型,包括:
采用所述目标超参数组合之中的多个超参数迭代训练所述第二深度强化学习模型,以得到所述第二深度强化学习模型输出的预测值;
如果所述预测值和标定值之间的损失值满足损失阈值,则将训练得到的所述第二深度强化学习模型作为所述目标深度强化学习模型。
5.如权利要求1所述的方法,其特征在于,所述第二深度强化学习模型的数量为设定个数,所述根据所述训练评价指标从多个第一深度强化学习模型之中筛选出第二深度强化学习模型,包括:
根据所述训练评价指标,对所述多个第一深度强化学习模型进行排序;
将排序在前的所述设定个数的第一深度强化学习模型作为所述第二深度强化学习模型。
6.如权利要求1所述的方法,其特征在于,所述训练所述多个第一深度强化学习模型,包括:
采用并行训练的方式训练所述多个第一深度强化学习模型。
7.如权利要求5所述的方法,其特征在于,还包括:
确定对所述第一深度强化学习模型进行训练的指标性能需求;
根据所述指标性能需求自适应地调整所述设定个数。
8.一种基于超参优化的深度强化学习模型的训练装置,其特征在于,所述装置包括:
第一获取模块,用于获取多个初始超参数组合,和多个第一深度强化学习模型;
第一训练模块,用于采用所述初始超参数组合中的多个超参数训练所述多个第一深度强化学习模型,以得到与所述多个第一深度强化学习模型分别对应的训练评价指标;
第一筛选模块,用于根据所述训练评价指标从多个第一深度强化学习模型之中筛选出第二深度强化学习模型;
第一处理模块,用于采用与所述第二深度强化学习模型对应的多个目标超参数对所述初始超参数组合进行优化处理,以形成目标超参数组合;以及
第二训练模块,用于采用所述目标超参数组合之中的多个超参数训练所述第二深度强化学习模型,以得到目标深度强化学习模型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于京东城市(北京)数字科技有限公司,未经京东城市(北京)数字科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202011621981.3/1.html,转载请声明来源钻瓜专利网。