1 江南大学轻工过程先进控制教育部重点实验室, 江苏 无锡 214122
2 无锡信捷电气股份有限公司, 江苏 无锡214122
针对卷积神经网络巨大的计算量和存储量导致其难以应用于嵌入式终端设备的难题,提出了一种基于灰色关联分析的模型裁剪方法。利用基于灰色关联分析的裁剪方法处理经过数据训练后的权重模型文件,获得每个卷积核重要性的量化表示;每次裁剪从模型中删除量化结果值最小的卷积核,从而减少计算量,加快推理速度;对于新产生的模型,通过迭代训练来弥补其性能上的损失。实验结果表明,相比APoZ法、L1法,所提方法在同一推理速度提升下精度提高了5.3%和10.4%,在VGG-16模型上取得了相对于初始模型2.7倍的加速效果,存储量压缩为原来的1/13.5。
图像处理 模型裁剪 深度学习 卷积神经网络 灰色关联分析 模型加速 激光与光电子学进展
2020, 57(4): 041011