[发明专利]模型训练、质量确定方法、装置、电子设备及存储介质在审

专利信息
申请号: 202011008060.X 申请日: 2020-09-23
公开(公告)号: CN112149733A 公开(公告)日: 2020-12-29
发明(设计)人: 鲁方波;汪贤;成超;樊鸿飞;蔡媛 申请(专利权)人: 北京金山云网络技术有限公司
主分类号: G06K9/62 分类号: G06K9/62
代理公司: 北京华夏泰和知识产权代理有限公司 11662 代理人: 杜欣
地址: 100085 北京市海淀*** 国省代码: 北京;11
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 模型 训练 质量 确定 方法 装置 电子设备 存储 介质
【权利要求书】:

1.一种模型训练方法,其特征在于,包括:

获取第一数据集及多个不同的预设模型,所述第一数据集包括多个第一数据;

将所述第一数据集分别输入多个预设模型,利用多个所述预设模型分别预测所述第一数据集中每个第一数据对应的第一标签;

针对每个所述预设模型,基于多个预设模型中除自身外的其它预设模型预测的第一标签及与所述第一标签对应的第一数据,构建所述预设模型对应的伪标签数据集;

根据所述伪标签数据集及用于预先训练所述预设模型的训练集构建所述预设模型对应的第二数据集;

利用所述第二数据集对所述预设模型进行训练,直至所述预设模型训练结束,得到多个质量确定模型,以用于确定图像数据的质量。

2.根据权利要求1所述的模型训练方法,其特征在于,所述针对每个所述预设模型,基于多个预设模型中除自身外的其它预设模型预测的第一标签及与所述第一标签对应的第一数据,构建所述预设模型对应的伪标签数据集,包括:

针对每个所述预设模型,获取多个预设模型中除所述预设模型自身之外的其它预设模型预测的第一标签;

将获取到的第一标签中满足预设条件的第一标签确定为伪标签;

将第一数据集中与所述伪标签对应的第一数据确定为伪标签数据;

基于所述伪标签及所述伪标签数据构建所述预设模型对应的伪标签数据集。

3.根据权利要求2所述的模型训练方法,其特征在于,所述将获取到的第一标签中满足预设条件的第一标签确定为伪标签,包括:

按照预设算法及每个预设模型预测的第一标签计算每个预设模型对应的参考标签;

针对每个所述预设模型,将多个预设模型中除所述预设模型之外的预设模型分别确定为参考模型;

针对每个所述预设模型对应的参考标签,计算任意两个所述参考模型所对应的参考标签之间的差值的绝对值,得到与所述参考标签对应的第一差异数据;

计算所述预设模型所对应的参考标签与任一参考模型所对应的参考标签之间的差值的绝对值,得到与所述参考标签对应的第二差异数据;

若任一参考标签对应的第一差异数据小于第一预设阈值,且该参考标签对应的第二差异数据大于或者等于第二预设阈值,将所述参考标签确定为伪标签。

4.根据权利要求3所述的模型训练方法,其特征在于,所述按照预设算法及每个预设模型预测的第一标签计算每个预设模型对应的参考标签,包括:

计算每个预设模型预测的第一标签的平均值,得到所述预设模型对应的参考标签。

5.根据权利要求1所述的模型训练方法,其特征在于,所述方法还包括:

判断任一所述预设模型对应的伪标签数据集与其它预设模型对应的伪标签数据集的重合度是否超过预设阈值;

若任一所述重合度超过预设阈值,确定所述预设模型训练结束;

或者,判断任一所述预设模型重新训练的迭代次数是否达到预设次数;

若任一所述预设模型重新训练的迭代次数达到预设次数,确定所述预设模型训练结束。

6.根据权利要求1所述的模型训练方法,其特征在于,所述方法还包括:

获取第二数据集,所述第二数据集包括:第二数据及第二标签;

利用所述第二数据集分别对多个预设初始模型进行训练,得到多个预设模型。

7.一种质量确定方法,其特征在于,包括:

在多个质量确定模型中选择一个或多个质量确定模型,所述质量确定模型是通过以下方式训练得到的:获取第一数据集及多个不同的预设模型,所述第一数据集包括多个第一数据,将所述第一数据集分别输入多个预设模型,利用多个所述预设模型分别预测所述第一数据集中每个第一数据对应的第一标签,针对每个所述预设模型,基于多个预设模型中除自身外的其它预设模型预测的第一标签及与所述第一标签对应的第一数据,构建所述预设模型对应的伪标签数据集,根据所述伪标签数据集及用于预先训练所述预设模型的训练集构建所述预设模型对应的第二数据集,利用所述第二数据集对所述预设模型进行训练,直至所述预设模型训练结束;

将待确定质量的图像数据输入一个或多个质量确定模型中,得到所述图像数据的质量确定结果。

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

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

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

×

专利文献下载

说明:

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

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

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

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

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

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

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

钻瓜专利网在线咨询

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

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