在审

图像分类网络生成方法、装置、电子设备及存储介质

王耀威、黄文柯、王握、白鑫贝、汤左淦
鹏城实验室

摘要

本申请公开了图像分类网络生成方法、装置、电子设备及存储介质,涉及计算机技术领域。首先根据图像分类网络的网络参数生成搜索空间,从搜索空间中获取至少一个子网络,利用图像数据集和初始教师网络对子网络进行知识蒸馏,得到学生网络。然后根据学生网络的性能参数,从学生网络中选取教师网络,并将教师网络添加至教师网络队列中。利用教师网络队列对子网络进行知识蒸馏,直至得到满足预设性能参数的教师网络,将教师网络作为目标图像分类网络。由此基于初始教师网络和动态更新的教师网络队列,通过知识蒸馏技术使得子网络可以从多维度进行学习,从而极大地提升了模型的泛化能力,得到满足预设性能参数的目标图像分类网络。

暂无引用专利