《计算机应用研究》|Application Research of Computers

基于灵敏度分析的FPGM剪枝算法研究

Research on fpgm pruning based on sensitivity analysis

免费全文下载 (已被下载 次)  
获取PDF全文
作者 冉光金,李震,李良荣
机构 贵州大学 大数据与信息工程学院
统计 摘要被查看 次,已被下载
摘要 针对等比例剪枝导致的重要卷积层剪枝过度、残留大量冗余参数以及精度损失较大的问题,在FPGM剪枝策略基础上融入灵敏度分析进行网络剪枝。算法采用精度反馈来分析每一层卷积层的重要性,控制单层剪枝比例分析每一层不同剪枝比例对精度损失的影响,获取各个卷积层的灵敏度;结合FPGM策略分析卷积层内卷积核的重要程度按灵敏度的剪枝比例剪掉不重要的卷积核,达到对神经网络进行有效剪枝的目的。实验结果表明,所提方法在MobileNet-v1和ResNet50上剪枝率为50%的情况下,精确度仅下降1.56%和0.11%;所提方法在精度损失一致下,ResNet50上具有更高剪枝率和更低计算量。
关键词 深度学习;网络剪枝;图像分类;灵敏度
基金项目 国家自然科学基金资助项目(61361012)
本文URL http://www.arocmag.com/article/02-2022-01-038.html
收稿日期
修回日期
页码 -
中图分类号 TP312
文献标志码