摘要
本发明提供了一种基于知识蒸馏的模型压缩方法及系统,通过继承已经训练成熟的教师模型的软标签,并使用软标签定义学生模型的蒸馏损失函数,使得学生模型得以压缩,在此基础上,通过随机抽取样本混合生成新的样本扩展训练集,缓解数据不足问题,同时提升模型对噪声和分布偏移的鲁棒性,通过动态权重调整,使学生模型迅速拟合教师模型知识,在训练后期自主优化,避免过拟合软标签。通过在未标注数据占比30%的测试集上,在使用真实标签的准确率为78.3%基础上,使用伪目标使准确率提升11.2%,通过使用训练成熟的学生模型进行部署,省略中间层特征对齐,GPU显存占用从3.2GB降至1.8GB,满足边缘设备部署需求。