专利名称:训练多标签分类模型的方法和装置专利类型:发明专利
发明人:刘晓阳,胡晓林,王月红,曹忆南申请号:CN201711187818.9申请日:20171124公开号:CN109840531A公开日:20190604
摘要:本申请提供了一种训练多标签分类模型的方法和装置,能够动态学习图像特征,使特征提取网络更适应任务需求,并且多标签分类效果好。该方法包括:从训练数据集中确定n个样本和与所述n个样本对应的标签矩阵Y,所述标签矩阵Y中的元素y表示第i个样本是否包含第j个标签指示的对象,c表示与样本相关的标签的个数;利用特征提取网络提取所述n个样本的特征矩阵X;利用第一映射网络获取所述特征矩阵X的预测标签矩阵利用第二映射网络获取所述标签矩阵Y的低秩标签矩阵根据所述标签矩阵Y、所述预测标签矩阵和所述低秩标签矩阵对所述权值参数Z、所述特征映射矩阵M和所述低秩标签相关性矩阵S进行更新,训练所述多标签分类模型。
申请人:华为技术有限公司,清华大学
地址:518129 广东省深圳市龙岗区坂田华为总部办公楼
国籍:CN
代理机构:北京龙双利达知识产权代理有限公司
更多信息请下载全文后查看
因篇幅问题不能全部显示,请点此查看更多更全内容