[发明专利]训练多标签分类模型的方法和装置有效

专利信息
申请号: 201711187818.9 申请日: 2017-11-24
公开(公告)号: CN109840531B 公开(公告)日: 2023-08-25
发明(设计)人: 刘晓阳;胡晓林;王月红;曹忆南 申请(专利权)人: 华为技术有限公司;清华大学
主分类号: G06V10/774 分类号: G06V10/774;G06V10/764;G06V10/82;G06N3/0464
代理公司: 北京龙双利达知识产权代理有限公司 11329 代理人: 王君;肖鹂
地址: 518129 广东*** 国省代码: 广东;44
权利要求书: 查看更多 说明书: 查看更多
摘要: 本申请提供了一种训练多标签分类模型的方法和装置,能够动态学习图像特征,使特征提取网络更适应任务需求,并且多标签分类效果好。该方法包括:从训练数据集中确定n个样本和与所述n个样本对应的标签矩阵Yc*n,所述标签矩阵Yc*n中的元素yi*j表示第i个样本是否包含第j个标签指示的对象,c表示与样本相关的标签的个数;利用特征提取网络提取所述n个样本的特征矩阵Xd*n;利用第一映射网络获取所述特征矩阵Xd*n的预测标签矩阵利用第二映射网络获取所述标签矩阵Yc*n的低秩标签矩阵根据所述标签矩阵Yc*n、所述预测标签矩阵和所述低秩标签矩阵对所述权值参数Z、所述特征映射矩阵Mc*d和所述低秩标签相关性矩阵S进行更新,训练所述多标签分类模型。
搜索关键词: 训练 标签 分类 模型 方法 装置
【主权项】:
1.一种训练多标签分类模型的方法,其特征在于,包括:从训练数据集中确定n个样本和与所述n个样本对应的标签矩阵Yc*n,所述标签矩阵Yc*n中的元素yi*j表示第i个样本是否包含第j个标签指示的对象,c表示与所述训练数据集中的样本相关的标签的个数;利用特征提取网络提取所述n个样本的特征矩阵Xd*n,其中,所述特征提取网络具有权值参数Z,d表示所述特征矩阵Xd*n的特征维度;利用第一映射网络获取所述特征矩阵Xd*n的预测标签矩阵所述预测标签矩阵中的元素表示第i个样本包含第j个标签指示的对象的置信度,其中,所述第一映射网络的权值矩阵为特征映射矩阵Mc*d;利用第二映射网络获取所述标签矩阵Yc*n的低秩标签矩阵其中,所述第二映射网络的权值矩阵为低秩标签相关性矩阵S,所述低秩标签相关性矩阵S用于描述所述c个标签之间的关系;根据所述标签矩阵Yc*n、所述预测标签矩阵和所述低秩标签矩阵对所述权值参数Z、所述特征映射矩阵Mc*d和所述低秩标签相关性矩阵S进行更新,训练所述多标签分类模型;其中,n、c、i、j和d均为正整数,且i的取值范围为1至n,j的取值范围为1至c。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

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

本文链接:http://www.vipzhuanli.com/patent/201711187818.9/,转载请声明来源钻瓜专利网。

同类专利
专利分类
×

专利文献下载

说明:

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

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

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

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

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

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

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

钻瓜专利网在线咨询

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

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