一种自动神经网络剪枝方法
    2.
    发明公开

    公开(公告)号:CN115392454A

    公开(公告)日:2022-11-25

    申请号:CN202211006937.0

    申请日:2022-08-22

    Abstract: 本发明涉及一种自动神经网络剪枝方法,属于神经网络领域。本发明在卷积层后添加BN层,在BN层加入缩放因子γ并对缩放因子γ加L1正则化使之稀疏;融合卷积层Filter与BN层,从而使缩放因子γ的稀疏化作用于卷积层Filter;使用融合后的卷积层Filter权重的绝对值作为衡量其重要性的手段,结合对特征输出的可重建性的影响,获得全局剪枝方案;对裁减之后的网络重新训练。本发明基于BN层,在BN层加入缩放因子γ并对之加L1正则化使之稀疏,然后裁剪缩放因子γ值小的部分对应权重。解决了传统基于权重的剪枝方法不容易平衡不同层剪枝比例的问题,提升剪枝效率。

Patent Agency Ranking