deeplearning算法优化道理

fj520 · 发表于 2023-6-19 16:17

deeplearning算法优化道理

目录

· 量化道理介绍
· 剪裁道理介绍
· 蒸馏道理介绍
· 轻量级模型布局搜索道理介绍
1. Quantization Aware Training量化介绍

1.1 布景

近年来，定点量化使用更少的比特数（如8-bit、3-bit、2-bit等）暗示神经网络的权重和激活已被验证是有效的。定点量化的长处包罗低内存带宽、低功耗、低计算资源占用以及低模型存储需求等。
低精度定点数操作的硬件面积大小及能耗比高精度浮点数要少几个数量级。使用定点量化可带来4倍的模型压缩、4倍的内存带宽提升，以及更高效的cache操作(很多硬件设备，内存访谒是主要能耗)。除此之外，计算速度也会更快(凡是具有2x-3x的性能提升)。由表2可知，在很多场景下，定点量化操作对精度并不会造成损掉。此外，定点量化对神经网络于嵌入式设备上的揣度来说是极其重要的。
目前，学术界主要将量化分为两大类：Post Training Quantization和Quantization Aware Training。Post Training Quantization是指使用KL散度、滑动平均等方式确定量化参数且不需要从头训练的定点量化方式。Quantization Aware Training是在训练过程中对量化进行建模以确定量化参数，它与Post Training Quantization模式对比可以提供更高的预测精度。
1.2 量化道理

1.2.1 量化方式

目前，存在着许多方式可以将浮点数量化成定点数。例如：
r=min(max(x,a),b)r=min(max(x,a),b)
s=fracb−an−1s=fracb−an−1
q=leftlfloorfracr−asrightrceilq=leftlfloorfracr−asrightrceil
式中，$x$是待量化的浮点值，$[a, b]$是量化范围，$a$是待量化浮点数中的最小值， $b$ 是待量化浮点数中的最大值。$left lfloor right rceil$ 暗示将成果四舍五入到比来的整数。如果量化级别为$k$，则$n$为$2^k$。例如，若$k$为8，则$n$为256。$q$是量化得到的整数。 PaddleSlim框架中选择的量化方式为最大绝对值量化(max-abs)，具体描述如下：
M=max(abs(x))M=max(abs(x))
q=leftlfloorfracxM∗(n−1)rightrceilq=leftlfloorfracxM∗(n−1)rightrceil
式中，$x$是待被量化的浮点值，$M$是待量化浮点数中的绝对值最大值。$left lfloor right rceil$暗示将成果四舍五入到比来的整数。对于8bit量化，PaddleSlim采用int8_t，即$n=2^7=128$。$q$是量化得到的整数。无论是min-max量化还是max-abs量化，他们都可以暗示为如下形式： $q = scale * r + b$ 此中min-max和max-abs被称为量化参数或者量化比例或者量化范围。
1.2.2 量化训练
1.2.2.1 前向传布
前向传布过程采用模拟量化的方式，具体描述如下：
基于模拟量化训练的前向过程可被描述为以下四个部门： 1) 输入和权重均被量化成8-bit整数。 2) 在8-bit整数上执行矩阵乘法或者卷积操作。 3) 反量化矩阵乘法或者卷积操作的输出成果为32-bit浮点型数据。 4) 在32-bit浮点型数据上执行偏置加法操作。此处，偏置并未被量化。对于通用矩阵乘法(GEMM)，输入$X$和权重$W$的量化操作可被表述为如下过程：
Xq=leftlfloorfracXXm∗(n−1)rightrceilXq=leftlfloorfracXXm∗(n−1)rightrceil
Wq=leftlfloorfracWWm∗(n−1)rightrceilWq=leftlfloorfracWWm∗(n−1)rightrceil
执行通用矩阵乘法： $$ Y_q = X_q * Wq
对量化乘积成果$Yq$进行反量化:对量化乘积成果$Yq$进行反量化:
begin{align} Y{dq} = frac{Y_q}{(n - 1) * (n - 1)} * X_m * W_m =frac{X_q * W_q}{(n - 1) * (n - 1)} * X_m * W_m =(frac{X_q}{n - 1} * X_m) * (frac{W_q}{n - 1} * W_m) end{align} $$ 上述公式表白反量化操作可以被移动到GEMM之前，即先对$Xq$和$Wq$执行反量化操作再做GEMM操作。因此，前向传布的工作流亦可暗示为如下方式：
训练过程中，PaddleSlim使用等价工作流。在设计中，量化Pass在IrGraph中插入量化op和反量化op。因为在持续的量化、反量化操作之后输入仍然为32-bit浮点型数据。因此，PaddleSlim量化训练框架所采用的量化方式被称为模拟量化。
1.2.2.2 反向传布
权重更新所需的梯度值可以由量化后的权重和量化后的激活求得。反向传布过程中的所有输入和输出均为32-bit浮点型数据。注意，梯度更新操作需要在原始权重长进行，即计算出的梯度将被加到原始权重上而非量化后或反量化后的权重上。
因此，量化Pass也会改变相应反向算子的某些输入。
1.2.2.3 确定量化比例系数
存在着两种策略可以计算求取量化比例系数，即动态策略和静态策略。动态策略会在每次迭代过程中计算量化比例系数的值。静态策略则对分歧的输入采用不异的量化比例系数。对于权重而言，在训练过程中采用动态策略。换句话说，在每次迭代过程中量化比例系数均会被从头计算得到直至训练过程结束。对于激活而言，可以选择动态策略也可以选择静态策略。若选择使用静态策略，则量化比例系数会在训练过程中被评估求得，且在揣渡过程中被使用(分歧的输入均保持不变)。静态策略中的量化比例系数可于训练过程中通过如下三种方式进行评估：
1. 在一个窗口中计算激活最大绝对值的平均值。
2. 在一个窗口中计算激活最大绝对值的最大值。
3. 在一个窗口中计算激活最大绝对值的滑动平均值，计算公式如下：
Vt=(1−k)∗V+k∗Vt−1Vt=(1−k)∗V+k∗Vt−1
式中，$V$ 是当前batch的最大绝对值， $Vt$是滑动平均值。$k$是一个因子，例如其值可取为0.9。
1.2.4 训练后量化
训练后量化是基于采样数据，采用KL散度等方式计算量化比例因子的方式。对比量化训练，训练后量化不需要从头训练，可以快速得到量化模型。
训练后量化的方针是求取量化比例因子，主要有两种方式：非饱和量化方式 ( No Saturation) 和饱和量化方式 (Saturation)。非饱和量化方式计算FP32类型Tensor中绝对值的最大值abs_max，将其映射为127，则量化比例因子等于abs_max/127。饱和量化方式使用KL散度计算一个合适的阈值T (0<T<mab_max)，将其映射为127，则量化比例因子等于T/127。一般而言，对于待量化op的权重Tensor，采用非饱和量化方式，对于待量化op的激活Tensor（包罗输入和输出），采用饱和量化方式。
训练后量化的实现法式如下：
· 加载预训练的FP32模型，配置DataLoader；
· 读取样本数据，执行模型的前向推理，保留待量化op激活Tensor的数值；
· 基于激活Tensor的采样数据，使用饱和量化方式计算它的量化比例因子；
· 模型权重Tensor数据一直保持不变，使用非饱和方式计算它每个通道的绝对值最大值，作为每个通道的量化比例因子；
· 将FP32模型转成INT8模型，进行保留。
2. 卷积核剪裁道理
该策略参考paper: Pruning Filters for Efficient ConvNets
该策略通过减少卷积层中卷积核的数量，来减小模型大小和降低模型计算复杂度。
2.1 剪裁卷积核
剪裁注意事项1 剪裁一个conv layer的filter，需要改削后续conv layer的filter。剪掉Xi的一个filter，会导致$X{i+1}$少一个channel, $X{i+1}$对应的filter在input_channel纬度上也要减1。
剪裁注意事项2
剪裁完$Xi$之后，按照注意事项1从$X{i+1}$的filter中删除了一行，在计算$X_{i+1}$的filters的l1_norm的时候，有两种选择：算上被删除的一行：independent pruning 减去被删除的一行：greedy pruning
剪裁注意事项3 在对ResNet等复杂网络剪裁的时候，还要考虑到后当前卷积层的改削对上一层卷积层的影响。在对residual block剪裁时，$X{i+1}$层如何剪裁取决于project shortcut的剪裁成果，因为要保证project shortcut的output和$X{i+1}$的output能被正确的concat.
2.2 Uniform剪裁卷积网络
每层剪裁一样比例的卷积核。在剪裁一个卷积核之前，按l1_norm对filter从高到低排序，越靠后的filter越不重要，优先剪掉靠后的filter.
2.3 基于敏感度剪裁卷积网络
按照每个卷积层敏感度的分歧，剪掉分歧比例的卷积核。
两个假设
· 在一个conv layer的parameter内部，按l1_norm对filter从高到低排序，越靠后的filter越不重要。
· 两个layer剪裁不异的比例的filters，称对模型精度影响更大的layer的敏感度相对高。
剪裁filter的指导原则
· layer的剪裁比例与其敏感度成反比
· 优先剪裁layer内l1_norm相对低的filter
敏感度的理解
横坐标是将filter剪裁掉的比例，竖坐标是精度的损掉，每条彩色虚线暗示的是网络中的一个卷积层。以分歧的剪裁比例单独剪裁一个卷积层，并不雅察看其在验证数据集上的精度损掉，并绘出虚线。虚线上升较慢的，对应的卷积层相对不敏感，优先剪不敏感的卷积层的filter.
选择最优的剪裁率组合
折线拟合为曲线，每在竖坐标轴上拔取一个精度损掉值，就在横坐标轴上对应着一组剪裁率。用户给定一个模型整体的剪裁率，通过移动实线来找到一组满足条件的且合法的剪裁率。
迭代剪裁
考虑到多个卷积层间的相关性，一个卷积层的改削可能会影响其它卷积层的敏感度，采纳了多次剪裁的策略，法式如下：
· step1: 统计各卷积层的敏感度信息
· step2: 按照当前统计的敏感度信息，对每个卷积层剪掉少量filter, 并统计FLOPS，如果FLOPS已满足要求，进入step4，否则进行step3。
· step3: 对网络进行简单的fine-tune，进入step1
· step4: fine-tune训练至收敛
3. 蒸馏
一般情况下，模型参数量越多，布局越复杂，其性能越好，但参数也越允余，运算量和资源消耗也越大；模型蒸馏是将复杂网络中的有用信息将复杂网络中的有用信息提取出来提取出来，迁移到一个更小的网络中去，在的东西包中，撑持两种蒸馏的方式。
第一种是传统的蒸馏方式（参考论文：Distilling the Knowledge in a Neural Network）
使用复杂的网络作为teacher模型去监督训练一个参数量和运算量更少的student模型。teacher模型可以是一个或者多个提前训练好的高性能模型。student模型的训练有两个方针：一个是原始的方针函数，为student模型输出的类别概率和label的交叉熵，记为hard-target；另一个是student模型输出的类别概率和teacher模型输出的类别概率的交叉熵，记为soft target，这两个loss加权后得到最终的训练loss，共同监督studuent模型的训练。第二种是基于FSP的蒸馏方式（参考论文：A Gift from Knowledge Distillation: Fast Optimization, Network Minimization and Transfer Learning）对比传统的蒸馏方式直接用小模型去拟合大模型的输出，该方式用小模型去拟合大模型分歧层特征之间的转换关系，其用一个FSP矩阵（特征的内积）来暗示分歧层特征之间的关系，大模型和小模型分歧层之间分袂获得多个FSP矩阵，然后使用L2 loss让小模型的对应层FSP矩阵和大模型对应层的FSP矩阵尽量一致，具体。这种方式的优势，通俗的解释是，比如将蒸馏类比成teacher（大模型）教student（小模型）解决一个问题，传统的蒸馏是直接告诉小模型问题的答案，让小模型学习，而学习FSP矩阵是让小模型学习解决问题的中间过程和方式，因此其学到的信息更多。
由于小模型和大模型之间通过L2 loss进行监督，必需保证两个FSP矩阵的维度必需不异，而FSP矩阵的维度为M*N，此中M、N分袂为输入和输出特征的channel数，因此大模型和小模型的FSP矩阵需要一一对应。
4. 轻量级模型布局搜索
深度学习模型在很多任务上都取得了不错的效果，网络布局的好坏对最终模型的效果有非常重要的影响。手工设计网络需要非常丰硕的经验和众多测验考试，而且众多的超参数和网络布局参数会发生爆炸性的组合，常规的random search几乎不成行，因此比来几年自动模型搜索技术（Neural Architecture Search）成为研究热点。区别于传统NAS，专注在搜索精度高而且速度快的模型布局，将该功能统称为Light-NAS.
4.1 搜索策略
搜索策略定义了使用怎样的算法可以快速、准确找到最优的网络布局参数配置。常见的搜索方式包罗：强化学习、贝叶斯优化、进化算法、基于梯度的算法。当前的实现以模拟退火算法为主。
4.1.1 模拟退火
模拟退火算法来源于固体退火道理，将固体加温至充实高，再让其徐徐冷却，加温时，固体内部粒子随温升变为无序状，内能增大，而徐徐冷却时粒子渐趋有序，在每个温度都达到平衡态，最后在常温时达到基态，内能减为最小。
鉴于物理中固体物质的退火过程与一般组合优化问题之间的相似性，将其用于网络布局的搜索。
使用模拟退火算法搜索模型的过程如下:
Tk=T0∗thetakTk=T0∗thetak
begin{equation} P(r_k) = begin{cases} e^{frac{(r_k-r)}{T_k}} & r_k < r1 & r_k>=r end{cases} end{equation}
在第k次迭代，搜到的网络为$N_k$, 对$N_k$训练若干epoch后，在测试集上得到reward为$r_k$, 以概率$P(r_k)$接受$r_k$，即执行$r=r_k$。$r$在搜索过程起始时被初始化为0. $T_0$为初始化温度，$theta$为温度衰减系数，$T_k$为第k次迭代的温度。
在的NAS任务中，区别于RL每次从头生成一个完整的网络，将网络布局映射成一段编码，第一次随机初始化，然后每次随机改削编码中的一部门（对应于网络布局的一部门）生成一个新的编码，然后将这个编码再映射回网络布局，通过在训练集上训练必然的epochs后的精度以及网络延时融合获得reward，来指导退火算法的收敛。
4.2 搜索空间
搜索空间定义了优化问题的变量，变量规模决定了搜索算法的难度和搜索时间。因此为了加快搜索速度，定义一个合理的搜索空间至关重要。在Light-NAS中，为了加速搜索速度，将一个网络划分为多个block，先手动按链状层级布局堆叠c，再使用搜索算法自动搜索每个block内部的布局。
因为要搜索出在移动端运行速度快的模型，参考了MobileNetV2中的Linear Bottlenecks和Inverted residuals布局，搜索每一个Inverted residuals中的具体参数，包罗kernelsize、channel扩张倍数、反复次数、channels number。4.3 模型延时评估
搜索过程撑持 FLOPS 约束和模型延时约束。而基于 Android/iOS 移动端、开发板等硬件平台，迭代搜索过程中不竭测试模型的延时不仅消耗时间而且非常不便利，因此开发了模型延时评估器来评估搜索得到模型的延时。通过延时评估器评估得到的延时与模型实际测试的延时波动偏差小于 10%。
延时评估器分为配置硬件延时评估器和评估模型延时两个阶段，配置硬件延时评估器只需要执行一次，而评估模型延时则在搜索过程中不竭评估搜索得到的模型延时。
· 配置硬件延时评估器
1. 获取搜索空间中所有不反复的 op 及其参数
2. 获取每组 op 及其参数的延时
· 评估模型延时
1. 获取给定模型的所有 op 及其参数
2. 按照给定模型的所有 op 及参数，操作延时评估器去估计模型的延时
5. 参考文献
1. High-Performance Hardware for Machine Learning
2. Quantizing deep convolutional networks for efficient inference: A whitepaper
3. Pruning Filters for Efficient ConvNets
4. Distilling the Knowledge in a Neural Network
5. A Gift from Knowledge Distillation: Fast Optimization, Network Minimization and Transfer Learning

		自动登录	找回密码
密码			立即注册