应用于边缘设备的深度神经网络模型分级压缩方法及装置

摘要

本说明书提供一种应用于边缘设备的深度神经网络模型分级压缩方法及装置，包括确定预设的教师模型，根据边缘设备的资源配置确定学生模型的网络层数；根据两种模型的网络层数，确定压缩比例，当根据压缩比例确定需要设置助教模型时，根据两种模型的网络层数，确定预备助教模型的网络层数；基于提前停止知识蒸馏方法利用教师模型对预备助教模型进行训练，得到训练后的预备助教模型；基于提前停止知识蒸馏方法利用训练后的预备助教模型对学生模型进行训练，得到利用预备助教模型训练后的学生模型，若该学生模型的准确度提高，将预备助教模型作为助教模型。本说明书训练得到的学生模型，准确度高，结构精简，适于边缘设备安装应用。