《表5 模型使用知识蒸馏与未使用知识蒸馏的对比》
试验发现,当教师模型为Resnet101时,学生模型Resnet18的识别准确率最高,准确率为95.8%。将它与传统的resnnet50和Resnet101模型的准确率等进行比较,汇总结果如表5所示。通过表5可以看到,在参数方面,Resnet50模型的参数量是Resnet18模型的2.19倍,Resnet101模型的参数量是Resnet18模型的3.81倍。参数量越多意味着模型越大,部署到相应设备上就越困难;在训练时间上,蒸馏后的Resnet18的总训练时间为11.3 h,Resnet50的总训练时间是Resnet18的1.71倍,Resnet101的总训练时间是Resnet18的2.77倍;在准确率方面,蒸馏后的Resnet18模型的准确率为95.8%,Resnet50模型的准确率为96.3%,Resnet101模型的准确率为96.9%。通过对比可以发现在知识蒸馏后,Resnet18这一小模型的准确率可以接近Resnet50和Resnet101这些大模型的准确率。从模型训练时间和模型部署上考虑,知识蒸馏是很有必要的,通过牺牲小部分准确率可以大大缩短模型训练时间和降低模型部署成本。
图表编号 | XD00192675800 严禁用于非法目的 |
---|---|
绘制时间 | 2020.09.01 |
作者 | 倪建功、李娟、邓立苗、韩仲志 |
绘制单位 | 青岛农业大学理学与信息科学学院、青岛农业大学理学与信息科学学院、青岛农业大学理学与信息科学学院、青岛农业大学理学与信息科学学院 |
更多格式 | 高清、无水印(增值服务) |