基于灰色关联分析的卷积神经网络模型裁剪方法

摘要：

针对卷积神经网络巨大的计算量和存储量导致其难以应用于嵌入式终端设备的难题,提出了一种基于灰色关联分析的模型裁剪方法.利用基于灰色关联分析的裁剪方法处理经过数据训练后的权重模型文件,获得每个卷积核重要性的量化表示;每次裁剪从模型中删除量化结果值最小的卷积核,从而减少计算量,加快推理速度;对于新产生的模型,通过迭代训练来弥补其性能上的损失.实验结果表明,相比APoZ法、L1法,所提方法在同一推理速度提升下精度提高了5.3％和10.4％,在VGG-16模型上取得了相对于初始模型2.7倍的加速效果,存储量压缩为原来的1/13.5.

关键词：图像处理、模型裁剪、深度学习、卷积神经网络、灰色关联分析、模型加速

所属期刊栏目：57

分类号：TP183(自动化基础理论)

资助基金：江苏高校优势学科建设工程项目;江苏省产学研前瞻性联合研究项目;江苏省科技成果转化专项

在线出版日期：2020-05-21（万方平台首次上网日期，不代表论文的发表时间）

页数：共7页

页码：127-133