一种基于模型蒸馏的图像分类方法、装置、存储介质及终端专利查询专利号|摘要-满商公司网

一种基于模型蒸馏的图像分类方法、装置、存储介质及终端

基本信息

申请号	CN202110499675.5	申请日	-
公开（公告）号	CN113408571A	公开（公告）日	2021-09-17
申请公布号	CN113408571A	申请公布日	2021-09-17
分类号	G06K9/62（2006.01）I;G06N3/04（2006.01）I	分类	计算；推算；计数;
发明人	廖丹萍	申请（专利权）人	浙江视觉智能创新中心有限公司
代理机构	北京辰权知识产权代理有限公司	代理人	李小朋
地址	311215浙江省杭州市萧山区宁围街道钱江世纪公园C区1幢1单元
法律状态	-

摘要

本发明公开了一种基于模型蒸馏的图像分类方法，该方法包括：获取待分类目标图像并输入预先训练的学生模型中；预先训练的学生模型基于模型蒸馏法训练生成，模型蒸馏法采用主成分分析算法对预先训练的教师模型输出的特征进行特征降维，学生模型的模型特征通道数小于教师模型的模型特征通道数；输出目标图像对应的多个类别概率值；基于多个类别概率值判定待分类目标图像的目标类别。因此，本申请实施例通过采用主成分分析算法对教师模型输出的特征进行特征降维，并约束学生模型的输出特征与降维后的特征一致，使得学生模型也能学到和教师模型相似区分度的特征，由于学生模型结构简单以及参数少，从而提升了硬件平台的运行速度，提高了图像分类效率。