文章      动态     相关文章     最新文章     手机版动态     相关动态     |   首页|会员中心|保存桌面|手机浏览

l8g5bv

http://fabua.ksxb.net/coml8g5bv/

相关列表
文章列表
  • 暂无文章
推荐文章
联系方式
  • 联系人:汪女士
  • 电话:18732873755
模型压缩部署概述天翼云手机端「模型压缩部署概述」
发布时间:2025-02-22        浏览次数:0        返回列表
  • 一,模型在线部署
    • 1.1,深度学习项目开发流程
    • 1.2,模型训练和推理的不同
  • 二,手机端CPU推理框架的优化
  • 三,不同硬件平台量化方式总结
  • 参考资料

深度学习和计算机视觉方向除了算法训练/研究,还有两个重要的方向: 模型压缩(模型优化、量化)、模型部署(模型转换、后端功能SDK开发)。所谓模型部署,即将算法研究员训练出的模型部署到具体的端边云芯片平台上,并完成特定业务的视频结构化应用开发。

现阶段的平台主要分为云平台(如英伟达 )、手机移动端平台( 系列芯片)和其他嵌入式端侧平台(海思 、安霸 、地平线 、英伟达 等芯片)。对于模型部署/移植/优化工程师来说,虽然模型优化、量化等是更有挑战性和技术性的知识,但是对于新手的我们往往是在做解决模型无法在端侧部署的问题,包括但不限于:实现新 、修改不兼容的属性、修改不兼容的权重形状、学习不同芯片平台的推理部署框架等。对于模型转换来说,现在行业主流是使用 和 模型作为中间模型。

在高校做深度学习 应用一般是这样一个过程,比如使用 框架训练出一个模型,然后直接使用 框架做推理()完成功能验证,但是在工业界这是不可能的,因为这样模型推理速度很慢,一般我们必须有专门的深度学习推理加速框架去做模型推理()。以 GPU 云平台推理框架 为例,简单描述模型训练推理过程就是:训练好网络模型(权重参数数据类型为 )输入 ,然后 做解析优化,并进行在线推理和输出结果。两种不同的模型训练推理过程对比如下图所示:

训练和推理.jpg

前面的描述较为简单,实际在工业届,理想的深度学习项目开发流程应该分为三个步骤: 模型离线训练、模型压缩和模型在线部署,后面两个步骤互有交叉,具体详情如下:

  1. 模型离线训练:实时性低,数据离线且更新不频繁, 较大,消耗大量 GPU 资源。

    • 设计开发模型网络结构;
    • 准备数据集并进行数据预处理、 等操作;
    • 深度学习框架训练模型:数据增强、超参数调整、优化器选择、训练策略调整(多尺度训练)、、模型融合等;
    • 模型测试。
  2. 模型优化压缩:主要涉及模型优化、模型转换、模型量化和模型编译优化,这些过程很多都在高性能计算推理框架中集成了,各个芯片厂商也提供了相应的工具链和推理库来完成模型优化压缩。实际开发中,在不同的平台选择不同的推理加速引擎框架,比如 平台选择 ,手机移动端()选择 , 芯片平台,如海思3519、地平线X3、安霸CV22等则直接在厂商给出的工具链进行模型的优化()和压缩。

    • 模型优化 :主要指计算图优化。首先对计算图进行分析并应用一系列与硬件无关的优化策略,从而在逻辑上降低运行时的开销,常见的类似优化策略其包括:算子融合( 融合)、算子替换、常数折叠、公共子表达式消除等。
    • 模型转换 :、、模型->(需要踩坑非常多,、、 三者之间的算子要注意兼容)。注意 一般用作训练框架和推理框架之间转换的中间模型格式。
    • 模型量化 :主要指训练后量化(Post-training quantization );权重、激活使用不同的量化位宽,如速度最快的量化方式 、速度和精度平衡的量化方式 。
    • 模型编译优化(编译优化+ 指令生成+内存优化):模型编译针对不同的硬件平台有不同优化方法,与前面的和硬件无关的模型层面的优化不同。平台存在 方法;而 平台算子是通过特定二进制指令实现,其编译优化方法包括,卷积层的拆分、卷积核权重数据重排、 算子调优等。
  3. 模型部署/SDK输出: 针对视频级应用需要输出功能接口的SDK。实时性要求高,数据线上且更新频繁, 为 1。主要需要完成多模型的集成、模型输入的预处理、非DL算法模块的开发、 各个模块 的串联,以及最后 接口()的输出。

    • 板端框架模型推理: :。不同的 芯片/不同的公司有着不同的推理框架,但是模型的推理流程大致是一样的。包括:输入图像数据预处理、加载模型文件并解析、填充输入图像和模型权重数据到相应地址、模型推理、释放模型资源。这里主要需要学习不同的模型部署和推理框架。
    • pipeline 应用开发: 在实际的深度学习项目开发过程中,模型推理只是其中的基础功能,具体的我们还需要实现多模型的集成、模型输入前处理、以及非 算法模块的开发: 包括检测模块、跟踪模块、选帧模块、关联模块和业务算法模块等,并将各模块串联成一个 ,从而完成视频结构化应用的开发。
    • SDK集成: 在完成了具体业务 的算法开发后,一般就需要输出 接口的 给到下层的业务侧(前后端)人员调用了。这里主要涉及 接口的转换、 多线程/多通道等sample的开发、以及大量的单元、性能、精度、稳定性测试。
    • 芯片平台板端推理 ,不同的 芯片有着不同的 库代码,但是模型运行流程类似。

不同平台的模型的编译优化是不同的,比如 和一般 的区别在于后端模型编译上, 是编译生成 ( 函数), 是编译生成二进制指令;前端的计算图优化没有本质区别,基本通用。

所以综上所述,深度学习项目开发流程可以大致总结为三个步骤: 模型离线训练模型优化压缩模型部署/SDK输出,后两个步骤互有交叉。前面 个步骤在 上完成,最后一个步骤开发的代码是需要在在 芯片系统上运行的。最后以视差模型在海思 平台的部署为例,其模型部署工作流程如下:

视差模型在海思3519平台部署工作流程.png

为了更好进行模型优化和部署的工作,需要总结一下模型推理()和训练()的不同:

  1. 网络权重值固定,只有前向传播(),无需反向传播,因此:
    • 模型权值和结构固定,可以做计算图优化,比如算子融合等;
    • 输入输出大小固定,可以做 优化,比如 重排和 重排。
  2. 会很小(比如 ),存在 的问题。
  3. 可以使用低精度的技术,训练阶段要进行反向传播,每次梯度的更新是很微小的,需要相对较高的精度比如 来处理数据。但是推理阶段,对精度要求没那么高,现在很多论文都表明使用低精度如 或者 数据类型来做推理,也不会带来很大的精度损失。

对于 和软件工程师来说,在手机 端做模型推理框架的优化,可以从上到下考虑:

  1. 算法层优化:最上面就是算法层,如可以用winograd从数学上减少乘法的数量(仅在大channel尺寸下有效);
  2. 框架优化:推理框架可以实现内存池、多线程等策略;
  3. 硬件层优化:主要包括: 适应不同的硬件架构特性、和优化、内存数据重排、 汇编优化等。
芯片厂商 芯片型号 支持方式 支持精度 量化方式/范围 量化工具
华为 Hisi系列3519A/3559A/3516C等 整网编译 int16/int8 非线性(对数) 量化 nnie_mapper
Ambarella CV22/CV25 整网编译 int8/int16 支持权重激活选择不同的位宽量化、自动搜索最优的量化策略 工具链CNNGen 的 Parsers
Nvidia 全系列GPGPU 整网编译/CUDA C fp32/fp16/int8/int4/int1 : 非对称 KL 散度 + 量化 框架
Qualcomm 全系列 SoC 整网编译 fp32/fp16/int8 非对称最大最小值量化 + 量化 AIMET 模型量化压缩工具
Rockchips RV1108/RV1109/RV1126等 整网编译 非对称量化/混合量化 RKNN Toolkit2