[发明专利]分类模型的训练方法,装置和计算机可读存储介质在审
申请号: | 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所述的训练方法,其中,将当前周期的调整向量与上一周期所述分类模型的初始化参数的乘积,作为当前周期所述分类模型的初始化参数。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京沃东天骏信息技术有限公司;北京京东世纪贸易有限公司,未经北京沃东天骏信息技术有限公司;北京京东世纪贸易有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202210674953.0/1.html,转载请声明来源钻瓜专利网。
- 上一篇:数据处理方法、设备及存储介质
- 下一篇:一种快装式送布装置及装配方法