一种模型压缩方法、装置、电子设备和存储介质
摘要:
本申请公开了一种模型压缩方法、装置、电子设备和存储介质;该方法涉及人工智能领域中的机器学习方向,通过待压缩的学生网络和教师网络分别对检测样本提取特征,得到所述检测样本的第一特征和第二特征;计算第一特征和第二特征的特征差异;基于所述特征差异,对所述学生网络进行调整;对调整后的学生网络进行裁剪;对裁剪后的学生网络进行预设迭代压缩处理,以得到满足预设压缩条件的学生网络;本申请可以通过对学生网络进行裁剪和迭代压缩处理,实现对学生网络的压缩,同时还使用教师网络对学生网络的性能进行调整,以补偿压缩带来的学生网络性能和精度下降,使得在压缩模型的同时保证压缩后模型的性能。
0/0