[发明专利]训练多标签分类模型的方法和装置在审
申请号: | 201711187395.0 | 申请日: | 2017-11-24 |
公开(公告)号: | CN109840530A | 公开(公告)日: | 2019-06-04 |
发明(设计)人: | 刘晓阳;胡晓林;王月红;曹忆南 | 申请(专利权)人: | 华为技术有限公司;清华大学 |
主分类号: | G06K9/62 | 分类号: | G06K9/62 |
代理公司: | 北京龙双利达知识产权代理有限公司 11329 | 代理人: | 王君;肖鹂 |
地址: | 518129 广东*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: |
本申请提供了训练多标签分类模型的方法和装置,能够动态学习图像特征,使特征提取网络更适应任务需求,并且多标签分类效果好。该方法包括:在训练数据集中确定n个样本和与所述n个样本对应的标签矩阵Yc*n,所述标签矩阵Yc*n中的元素yi*j表示第i个样本是否包含第j个标签指示的对象,c表示与样本相关的标签的个数;利用特征提取网络提取所述n个样本的特征矩阵Xd*n;利用特征映射网络获取所述特征矩阵Xd*n的预测标签矩阵 |
||
搜索关键词: | 标签矩阵 样本 标签分类 方法和装置 标签指示 特征矩阵 特征提取 预测 动态学习 权值参数 任务需求 图像特征 网络提取 训练数据 映射矩阵 映射网络 元素表示 置信度 标签 更新 申请 网络 | ||
【主权项】:
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和所述预测标签矩阵
对所述权值参数Z、所述特征映射矩阵Mc*d进行更新,训练所述多标签分类模型;其中,n、c、i、j和d均为正整数,且i的取值范围为1至n,j的取值范围为1至c。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于华为技术有限公司;清华大学,未经华为技术有限公司;清华大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201711187395.0/,转载请声明来源钻瓜专利网。