模型压缩

模型压缩
PaddleSlim是一个模型压缩工具库,包含模型剪裁、定点量化、知识蒸馏、超参搜索和模型结构搜索等一系列模型压缩策略。
对于业务用户,PaddleSlim提供完整的模型压缩解决方案,可用于图像分类、检测、分割等各种类型的视觉场景。 同时也在持续探索NLP领域模型的压缩方案。另外,PaddleSlim提供且在不断完善各种压缩策略在经典开源任务的benchmark, 以便业务用户参考。
对于模型压缩算法研究者或开发者,PaddleSlim提供各种压缩策略的底层辅助接口,方便用户复现、调研和使用最新论文方法。 PaddleSlim会从底层能力、技术咨询合作和业务场景等角度支持开发者进行模型压缩策略相关的创新工作。
功能
• 模型剪裁
o 卷积通道均匀剪裁
o 基于敏感度的卷积通道剪裁
o 基于进化算法的自动剪裁
• 定点量化
o 在线量化训练(training aware)
o 离线量化(post training)
• 知识蒸馏
o 支持单进程知识蒸馏
o 支持多进程分布式知识蒸馏
• 神经网络结构自动搜索(NAS)
o 支持基于进化算法的轻量神经网络结构自动搜索
o 支持One-Shot网络结构自动搜索
o 支持 FLOPS / 硬件延时约束
o 支持多平台模型延时评估
o 支持用户自定义搜索算法和搜索空间
安装
依赖:
Paddle >= 1.7.0
pip install paddleslim -i https://pypi.org/simple
使用
• 快速开始:通过简单示例介绍如何快速使用PaddleSlim。
• 进阶教程:PaddleSlim高阶教程。
• 模型库:各个压缩策略在图像分类、目标检测和图像语义分割模型上的实验结论,包括模型精度、预测速度和可供下载的预训练模型。
• API文档
• Paddle检测库:介绍如何在检测库中使用PaddleSlim。
• Paddle分割库:介绍如何在分割库中使用PaddleSlim。
• PaddleLite:介绍如何使用预测库PaddleLite部署PaddleSlim产出的模型。
部分压缩策略效果
分类模型
数据: ImageNet2012; 模型: MobileNetV1;
在这里插入图片描述

数据:Pascal VOC;模型:MobileNet-V1-YOLOv3
在这里插入图片描述

数据:COCO;模型:MobileNet-V1-YOLOv3
在这里插入图片描述

搜索
数据:ImageNet2012; 模型:MobileNetV2
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值