神经网络模型的量化简介（工程版）

news2024/11/25 7:03:13

1.量化简介

模型量化（Model Quantization）是深度学习中一种优化技术，旨在减少模型的计算和存储需求，同时尽量保持模型的性能。具体来说，模型量化通过将模型的权重和激活值从高精度（通常是32位浮点数）缩减到较低精度（如16位、8位甚至更低的整数），从而减少计算复杂度和内存占用

模型量化具有以下优点：

减少存储需求：低精度表示可以显著减少模型的存储空间。
加速推理速度：低精度计算通常比高精度计算更快，特别是在支持低精度运算的硬件上。
降低功耗：减少计算复杂度和内存访问次数，可以降低功耗，这对于移动和嵌入式设备尤为重要。

2.概念介绍

按量化参数分类，可分为静态量化和动态量化，区别这两种方式主要是是否提供校准集，动态量化更简单，更灵活，但是推理开销较大，会内嵌一些动态参数，特别注意，动态量化对于TensorRT的engine生成较复杂，因为TensorRT不支持DynamicQuantizeLinear操作。
按量化时间分类，可分为训练后量化（PTQ）和量化感知训练（QAT），区别这两种方式主要是是否需要对模型进行训练，也就是需要搭配troch、tensorflow等框架来训练模型，QAT方式更加复杂，精度也更高，常规的使用建议直接采用PTQ方式，待PTQ量化模型性能较差，再采用QAT方式进行优化。

2.1 动态量化（Dynamic Quantization）（量化方式）

描述：

动态量化在模型推理时对部分权重和激活值进行量化。具体来说，模型的权重在推理前保持为浮点数，在推理时将其动态转换为低精度（如8位整数）进行计算。激活值在推理过程中也会动态地从浮点数转换为低精度值。

优点：

简便性：无需校准数据集，适用于推理阶段的快速量化。
灵活性：适用于多种模型和场景，不需要重新训练模型。

缺点：

推理开销：在推理过程中进行动态量化，可能增加一些计算开销。
精度损失：虽然动态量化可以减少一些精度损失，但在某些情况下仍可能影响模型性能。

2.2. 静态量化（Static Quantization）（量化方式）

描述：

静态量化的目标是求取量化比例因子，主要通过对称量化、非对称量化方式来求，而找到最大值或阈值的方法有MinMax、KLD、ADMM、EQ等方法。

优点：

高效推理：预先量化的模型在推理时效率更高，无需额外的动态计算。
较好的精度：通过校准数据集可以较好地保持模型精度。

缺点：

需要校准数据集：需要一个代表性的校准数据集来计算量化参数。
实施复杂：量化过程可能较为复杂，需要对模型进行详细的分析和调整。

2.3. 量化感知训练（Quantization-Aware Training, QAT）（量化时间，通过结合静态量化）

描述：

量化感知训练（QAT）是一种在模型训练过程中引入量化操作的技术。通过在训练阶段模拟量化误差，模型能够学习并适应量化带来的精度损失，从而在量化后依然保持较高的准确性。QAT通常结合静态量化方法，在训练时将权重和激活值量化为低精度（如8位整数），并在反向传播时考虑量化误差。

优点：

最小化精度损失：在训练过程中模拟量化误差，使模型适应量化后的精度损失，通常能保持较高的准确性。
高效推理：量化后的模型在推理时效率很高。

缺点：

训练复杂度增加：需要在训练过程中引入量化感知机制，增加训练复杂度和时间。
计算资源需求高：训练过程需要更多的计算资源和时间。

2.4.后训练量化（Post-Training Quantization, PTQ）

描述：

在模型训练完成后进行量化，不需要重新训练模型。
KL散度方法：通过计算原Float32与量化后的参数分布，得到最优

优点：

简单快速，适合快速部署。

缺点：

可能会有一定的精度损失，尤其是在量化敏感的模型中。

2.5 总结

动态量化适合需要快速部署和灵活应用的场景，但在推理时可能增加计算开销。
静态量化适合推理效率要求高的场景，但需要校准数据集和复杂的量化过程。
后训练量化适合已经训练好的模型，希望快速量化以便部署的场景，但可能会导致精度下降。
量化感知训练适合对精度要求高的场景，但增加了训练复杂度和资源需求。

3. QAT与PTQ区别

训练中量化：QAT，精度高，复杂；训练后量化：PTQ，精度略低，简单。
训练后量化 PTQ ：它是使用一批校准数据对训练好的模型进行校准, 将训练过的FP32网络直接转换为定点计算的网络，过程中无需对原始模型进行任何训练。只对几个超参数调整就可完成量化过程, 且过程简单快速, 无需训练, 因此此方法已被广泛应用于大量的端侧和云侧部署场景，优先推荐您尝试PTQ方法来查看是否满足您的部署精度和性能要求。
量化感知训练 QAT ：它是将训练过的模型量化后又再进行重训练。由于定点数值无法用于反向梯度计算，实际操作过程是在某些op前插入伪量化节点（fake quantization nodes），用于在训练时获取流经该op的数据的截断值，便于在部署量化模型时对节点进行量化时使用。我们需要在训练中通过不断优化精度来获取最佳的量化参数。由于它需要对模型进行训练, 对操作人员技术要求较高。

4. QAT/PTQ量化与静态/动态量化的区别

PTQ（Post-Training Quantization，训练后量化）和QAT（Quantization-Aware Training，量化感知训练）主要是量化方法，而静态量化和动态量化是量化策略。

训练后量化（PTQ）

描述

PTQ是在模型训练完成后，对模型进行量化的一种方法。它不需要在训练过程中考虑量化误差，而是在训练结束后，通过分析训练数据或校准数据，对模型的权重和激活值进行量化。

静态量化 vs 动态量化

静态量化：PTQ可以用于静态量化，即在量化时使用一组校准数据来确定量化参数（如缩放因子和零点），并在推理时使用这些预先计算的量化参数。
动态量化：PTQ也可以用于动态量化，即在推理时根据输入数据动态地确定量化参数。这种方法通常用于激活值的量化，而权重通常在量化后保持静态。

量化感知训练（QAT）

描述

QAT是在训练过程中引入量化操作的一种方法。通过在训练阶段模拟量化误差，模型能够学习并适应量化带来的精度损失，从而在量化后依然保持较高的准确性。

静态量化 vs 动态量化

静态量化：QAT通常用于静态量化，即在训练过程中模拟量化误差，并在训练完成后使用固定的量化参数进行推理。由于模型已经适应了这些量化参数，推理时可以直接使用预先计算的量化参数，达到高效推理的效果。
动态量化：QAT一般不用于动态量化，因为动态量化主要是在推理时动态确定量化参数，而QAT的核心在于训练过程中模拟和适应量化误差。