[发明专利]分类模型的训练方法,装置和计算机可读存储介质在审

专利信息
申请号: 202210674953.0 申请日: 2022-06-15
公开(公告)号: CN114997318A 公开(公告)日: 2022-09-02
发明(设计)人: 杨一博;王睿彬 申请(专利权)人: 北京沃东天骏信息技术有限公司;北京京东世纪贸易有限公司
主分类号: G06K9/62 分类号: G06K9/62;G06V10/74;G06V10/764
代理公司: 中国贸促会专利商标事务所有限公司 11038 代理人: 孙玉;许蓓
地址: 100176 北京市大兴区北京经济*** 国省代码: 北京;11
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 分类 模型 训练 方法 装置 计算机 可读 存储 介质
【权利要求书】:

1.一种分类模型的训练方法,包括:

将用于训练的样本划分为多个子批,其中,每个子批包括一个或多个样本;

将每个子批的样本输入分类模型,确定所述每个子批对应的分类误差相对于所述分类模型的初始化参数的梯度向量;

根据所述每个子批对应的梯度向量,确定目标函数的值,其中,所述目标函数与所述分类模型的分类损失负相关;

在所述目标函数的值未达到最大值的情况下,根据所述目标函数调整所述分类模型的初始化参数;

直至所述目标函数的值达到最大值,将所述分类模型的初始化参数作为优化后的初始化参数,并对具有所述优化后的初始化参数的所述分类模型进行训练。

2.根据权利要求1所述的训练方法,其中,所述目标函数包括第一子函数和第二子函数中至少一个,所述根据所述每个子批对应的梯度向量,确定目标函数的值包括以下至少一项:

根据所述每个子批对应的梯度向量,确定所述每个子批对应的梯度向量的模长,根据所述每个子批对应的梯度向量的模长确定所述第一子函数的值;

根据所述每个子批对应的梯度向量,确定每两个子批对应的梯度向量的相似度,根据所述每两个子批对应的梯度向量的相似度确定所述第二子函数的值。

3.根据权利要求2所述的训练方法,其中,所述根据所述每个子批对应的梯度向量的模长确定所述第一子函数的值包括:

根据所述每个子批对应的梯度向量的模长确定平均模长,作为所述第一子函数的值。

4.根据权利要求2所述的训练方法,其中,所述根据所述每个子批对应的梯度向量,确定每两个子批对应的梯度向量的相似度,根据所述每两个子批对应的梯度向量的相似度确定所述第二子函数的值包括:

根据所述每个子批对应的梯度向量,确定每两个子批对应的梯度向量的余弦相似度;

根据所述每两个子批对应的梯度向量的余弦相似度确定平均余弦相似度,作为所述第二子函数的值。

5.根据权利要求2所述的训练方法,其中,所述目标函数还包括所述每个子批对应的梯度向量的模长中的最大模长不超过预设值的条件。

6.根据权利要求2所述的训练方法,其中,所述目标函数采用以下方法确定:

对所述分类模型的分类损失函数进行优化曲面分析,确定评价指标函数,其中,所述评价指标函数为确定所述分类模型的分类损失的上限的函数;

根据所述评价指标函数确定所述目标函数。

7.根据权利要求6所述的训练方法,其中,所述评价指标函数包括确定每个子批对应梯度向量的模长的最大模长的第三子函数,确定每两个子批对应的梯度向量之间的相似度的第四子函数,以及确定每个子批对应的优化后的初始化参数的最大模长和最小模长的比值的第五子函数,所述根据所述评价指标函数确定所述目标函数包括:

根据所述第三子函数确定所述目标函数中所述每个子批对应的梯度向量的模长中的最大模长不超过预设值的条件;

根据所述第四子函数确定所述第二子函数;

通过利用所述初始化参数的一阶梯度和每个子批对应的梯度向量来逼近每个子批对应的优化后的初始化参数,将所述第五子函数转换为所述第一子函数。

8.根据权利要求4所述的训练方法,其中,所述每两个子批对应的梯度向量的余弦相似度为所述每两个子批对应的梯度向量的点乘与所述每两个子批对应的梯度向量的模长的比值。

9.根据权利要求1所述的训练方法,其中,所述根据所述目标函数调整所述分类模型的初始化参数包括:

确定所述目标函数相对于上一周期的调整向量的梯度值;

根据上一周期的调整向量,所述梯度值和学习率,确定当前周期的调整向量;

根据当前周期的调整向量与上一周期所述分类模型的初始化参数,确定当前周期所述分类模型的初始化参数。

10.根据权利要求9所述的训练方法,其中,将当前周期的调整向量与上一周期所述分类模型的初始化参数的乘积,作为当前周期所述分类模型的初始化参数。

下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京沃东天骏信息技术有限公司;北京京东世纪贸易有限公司,未经北京沃东天骏信息技术有限公司;北京京东世纪贸易有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服

本文链接:http://www.vipzhuanli.com/pat/books/202210674953.0/1.html,转载请声明来源钻瓜专利网。

×

专利文献下载

说明:

1、专利原文基于中国国家知识产权局专利说明书;

2、支持发明专利 、实用新型专利、外观设计专利(升级中);

3、专利数据每周两次同步更新,支持Adobe PDF格式;

4、内容包括专利技术的结构示意图流程工艺图技术构造图

5、已全新升级为极速版,下载速度显著提升!欢迎使用!

请您登陆后,进行下载,点击【登陆】 【注册】

关于我们 寻求报道 投稿须知 广告合作 版权声明 网站地图 友情链接 企业标识 联系我们

钻瓜专利网在线咨询

周一至周五 9:00-18:00

咨询在线客服咨询在线客服
tel code back_top