[发明专利]对机器学习模型训练的方法和装置在审
| 申请号: | 202111340905.X | 申请日: | 2021-11-12 |
| 公开(公告)号: | CN113988316A | 公开(公告)日: | 2022-01-28 |
| 发明(设计)人: | 姜仟艺;宋祺;张睿 | 申请(专利权)人: | 北京三快在线科技有限公司 |
| 主分类号: | G06N20/00 | 分类号: | G06N20/00;G06K9/62 |
| 代理公司: | 北京三高永信知识产权代理有限责任公司 11138 | 代理人: | 谢冬寒 |
| 地址: | 100080 北京市海*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 机器 学习 模型 训练 方法 装置 | ||
1.一种对机器学习模型训练的方法,其特征在于,所述方法包括:
获取第一样本图像、第二样本图像和所述第二样本图像对应的基准识别结果;
对所述第一样本图像进行图像变化处理得到第三样本图像;
基于待训练的图像字符识别模型和所述第一样本图像,确定第一识别结果,基于所述待训练的图像字符识别模型和所述第二样本图像,确定第二识别结果,基于所述待训练的图像字符识别模型和所述第三样本图像,确定第三识别结果;
基于所述第一识别结果、所述第二识别结果、所述第三识别结果和所述基准识别结果,对所述待训练的图像字符识别模型进行参数调整。
2.根据权利要求1所述的方法,其特征在于,所述基于所述第一识别结果、所述第二识别结果、所述第三识别结果和所述基准识别结果,对所述待训练的图像字符识别模型进行参数调整,包括:
基于所述第二识别结果和所述基准识别结果确定第一损失值,基于所述第三识别结果和所述第一识别结果确定第二损失值;
基于所述第一损失值和所述第二损失值,对所述待训练的图像字符识别模型进行参数调整。
3.根据权利要求2所述的方法,其特征在于,所述基于所述第一损失值和所述第二损失值,对所述待训练的图像字符识别模型进行参数调整,包括:
基于第一权重值和第二权重值,对所述第一损失值和所述第二损失值进行加权求和处理得到第三损失值;
基于所述第三损失值,对所述待训练的图像字符识别模型进行参数调整。
4.根据权利要求3所述的方法,其特征在于,所述第一权重值小于所述第二权重值。
5.根据权利要求1所述的方法,其特征在于,所述基于待训练的图像字符识别模型和所述第一样本图像,确定第一识别结果,包括:
获取在所述图像字符识别模型的历史训练过程中记录的至少一组模型参数;
基于待训练的图像字符识别模型和所述至少一组模型参数对应的图像字符识别模型,分别对所述第一样本图像进行字符识别,得到多个待选识别结果和每个待选识别结果中每个字符对应的置信度;
基于所述多个待选识别结果和每个待选识别结果中每个字符对应的置信度,在所述多个待选识别结果中选取第一识别结果。
6.根据权利要求5所述的方法,其特征在于,所述基于所述多个待选识别结果和每个待选识别结果中每个字符对应的置信度,在所述多个待选识别结果中选取第一识别结果,包括:
在所述多个待选识别结果中,确定出现次数满足目标条件的目标待选识别结果,如果所述目标待选识别结果对应的最低的置信度大于置信度阈值,则将所述目标待选识别结果确定为第一识别结果。
7.根据权利要求6所述的方法,其特征在于,所述确定出现次数满足目标条件的目标待选识别结果,包括:
确定出现次数大于次数阈值的目标待选识别结果;或者,
确定出现次数大于其他待选识别结果出现次数的目标待选识别结果;或者,
确定出现次数在待选识别结果总数中占比大于比例阈值的目标待选识别结果。
8.一种对机器学习模型训练的装置,其特征在于,所述装置包括:
获取模块,用于获取第一样本图像、第二样本图像和所述第二样本图像对应的基准识别结果;
处理模块,用于对所述第一样本图像进行图像变化处理得到第三样本图像;
确定模块,用于基于待训练的图像字符识别模型和所述第一样本图像,确定第一识别结果,基于所述待训练的图像字符识别模型和所述第二样本图像,确定第二识别结果,基于所述待训练的图像字符识别模型和所述第三样本图像,确定第三识别结果;
调参模块,用于基于所述第一识别结果、所述第二识别结果、所述第三识别结果和所述基准识别结果,对所述待训练的图像字符识别模型进行参数调整。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京三快在线科技有限公司,未经北京三快在线科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111340905.X/1.html,转载请声明来源钻瓜专利网。





