• 模型剪枝算法——L1正则化BN层的γ因子


    ICCV在2017年刊登了一篇经典论文《 Learning Efficient Convolutional Networks through Network Slimming》。在神经网络的卷积操作之后会得到多个特征图,通过策略突出重要的特征达到对网络瘦身的目的。在该论文中使用的剪枝策略就是稀疏化BN层中的缩放因子 \gamma
    BatchNorm的本质是使输入数据标准化,关于0对称,数据分布到一个量级中,在训练的时候有利于加速收敛
    BatchNorm本来公式:
    \hat{x} = \frac{x^{k}-E[x^{k}]}{\sqrt{var[x^{k}]}}
    在实际应用时,引入了两个可训练的参数 \gamma 、 \beta。后文会详解介绍。
    为什么说输入数据分布不均匀,网络分布不容易收敛,以sigmoid为例进行介绍。sigmoid函数在神经网络中常用来做激活函数,为了将非线性引入神经网络中,使得神经网络具有更加复杂的决策边界。
                    ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​         \phi (z) = \frac{1}{1+e^{-z}}
    如sigmoid函数图像所示,输入数据在红框范围内,函数梯度较大,反向传播收敛更快。在红框外,梯度小参数更新慢,甚至有梯度消失的情况。
    因此加入BN层能够很好的将数据分布规范化到均值为0,方差为1的标准正态分布。提高了激活函数的灵敏度,加速训练。
    但是这样一来又引入了新的问题,我们观察红框内的函数形状类似线性函数。为了保持非线性,因此在BN中加入可训练的参数 \gamma\beta来呈现非线性。 (此处不理解为什么呈现的是非线性)
            ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​                  yk=rkx^k+γk" role="presentation" style="position: relative;">yk=rkx^k+γk
    改进后的BN公式:
    神经网络中网络层连接顺序:conv->BN->激活层
    其中,卷积层的每个通道都会对应一个缩放因子\gamma,我们对\gamma小的值进行prunning,得到稀疏的网络层。
    如何将重要的特征(通道)的\gamma值提高?  为什么重要的特征(通道)的\gamma值高?——使用L1正则化能对\gamma进行稀疏作用。
    我们先来回顾一下L1、L2正则化。
    通常L1正则化用来稀疏与特征选择。目标函数通常由损失函数(此处为MSE)和正则化函数组成,L1正则项表示如下。传入的参数 \theta经过L1正则化可以达到稀疏的效果。
            ​​​​​​​        ​​​​​​​        ​​​​​​​         J(\theta ) = \frac{1}{m}\sum_{i=1}^{m}h_{\theta }(x^{i}-y^{i})^{2} + \lambda \sum_{n}^{i=1}\left | \theta_{j} \right |
    L1正则化函数图像以及它的求导函数sign(θ)的图像如下。L1在反向传播,梯度更新的时候梯度下降的步长衡为1,在参数更新的时候很多参数都学成了0,因此能达到稀疏的目的。

    L2正则化用来平滑特征,防止过拟合。目标函数携带L2正则项表示:

            ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​        J(\theta ) = \frac{1}{m}\sum_{i=1}^{m}h_{\theta }(x^{i}-y^{i})^{2} + \lambda \sum_{n}^{i=1}\theta_{j}^{2}

    L2正则化函数及求导函数的图像:

    L2求导为θ,当参数特别大时,参数更新的梯度也大,当参数特别小时,参数更新的梯度也小。因此产生平滑特征的效果。L2可以每个参数都变小但是不至于变成0,这样可以减少模型的复杂度,防止模型拟合数据中的噪声。
    因此可以利用L1正则化对参数\gamma进行稀疏作用。
            ​​​​​​​        ​​​​​​​        ​​​​​​​        ​​​​​​​       

     
  • 相关阅读:
    C++11新特性-原始字面量
    分库分表实战之从根上带你吃透MySQL的索引
    E. Restoring the Permutation
    求最长的只包含两种类型的连续子数组的最大长度
    浅谈数据治理中的智能数据目录
    基于Three.js实现三维空间中的箭头移动动画
    你了解TCP协议吗(二)?
    [21天学习挑战赛——内核笔记](一)——设备树的概述(硬件、目标、效果、文件类型)
    药物从研发到上市需要经历哪些流程?||新药研发
    MyBatis中文网
  • 原文地址:https://blog.csdn.net/t765833631/article/details/134286655