aimet

不重训模型,把已训练网络压到 4bit 还能保住精度

开源 free 模型训练
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 2728
维护状态 活跃
是否开源 是
定价模式 free

项目预览

项目数据

分类模型训练
开发团队qualcomm
所属国家
定价模式free
价格说明开源免费,遵循许可证使用
访问状态
是否开源是
开源协议NOASSERTION
主要语言Python
技术栈/模型auto-ml,compression,deep-learning,deep-neural-networks,machine-learning,network-compression,network-quantization,open-source,opensource,pruning,quantization
GitHub 星标★ 2728
30天Star增速
HF 下载量
上线时间2020-04-21 00:00:00
代码更新—
仓库体积—
本站数据更新—
维护状态活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-10-17
浏览次数0

项目介绍

AIMET(AI Model Efficiency Toolkit)是高通开源的一套面向已训练神经网络模型的量化与压缩工具库。它解决的核心问题是:模型在移动端、边缘设备上部署时体积过大、推理过慢、功耗过高,而直接训练小模型又损失精度。AIMET 不重新训练模型,而是在已有浮点模型基础上做后处理优化,提供跨层均衡(CLE)、偏差校正(BC)、自适应舍入(AdaRound)、量化感知训练(QAT)、结构化/非结构化剪枝、奇异值分解压缩等一整套技术,并配套可视化诊断工具,帮助开发者把模型压到 8bit、4bit 甚至更低,同时尽量保持精度。它支持 PyTorch、TensorFlow、ONNX 等主流框架,可与高通 AI Engine 等推理栈配合,是端侧 AI 部署链路中较完整的模型压缩方案。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

正在加载星标数据…

技术标签

使用教程

—

难度:进阶 约 10 分钟 部署方式:库/依赖 3 步

环境要求

  • Python 环境(能使用 pip 安装包)
  • 已训练的浮点模型(PyTorch 或 ONNX 格式)
  • 按需选择框架:PyTorch 对应 aimet-torch,ONNX 对应 aimet-onnx
  • 网络可访问 PyPI 官方源

安装与启动步骤

  1. 1安装 PyTorch 版

    README 说明 aimet-torch 已发布到 PyPI,直接在目标环境用 pip 安装即可。

    pip install aimet-torch
  2. 2安装 ONNX 版

    若模型是 ONNX 格式,则安装 aims-onnx 包;两包按需装,不必都装。

    pip install aimet-onnx
  3. 3查阅官方 Quick Start

    README 指向 Quick Start 页面,具体用法、版本要求以该文档为准,建议先通读再动手。

如何确认成功

执行 pip show aimet-torch(或 aimet-onnx),能正常显示包名与版本号即说明安装成功。

常见问题

Q:AIMET 需要重新训练模型吗?

A:不需要。它是在已训练好的浮点模型上做量化与压缩后处理,通过跨层均衡、偏差校正、AdaRound、QAT 等技术尽量保持精度。

Q:该装 aimet-torch 还是 aimet-onnx?

A:取决于你的模型格式:PyTorch 模型装 aimet-torch,ONNX 模型装 aimet-onnx,两者都以独立包的形式发布在 PyPI 上。

Q:安装或使用中遇到问题去哪里求助?

A:README 提供了 GitHub Discussions 与 Qualcomm AI Hub Slack 两个社区渠道,可在其中提问交流。

Q:支持哪些深度学习框架?

A:项目定位支持 PyTorch、TensorFlow、ONNX 等主流框架,README 中明确给出 PyPI 包的是 PyTorch 与 ONNX 两条线。

注意事项

  • README 仅说明包已上架 PyPI 并给出 Quick Start 链接,实际安装步骤与版本约束请以官方 Quick Start 文档为准。
  • 生产环境建议使用独立虚拟环境(如 venv/conda)安装,避免与已有依赖冲突。
  • 量化压缩会改变模型数值精度,务必在自有数据集上验证精度与推理性能后再部署。

核心亮点

  • 提供 CLE、BC、AdaRound、QAT、SVD、剪枝等完整量化压缩算法链,而非单一技术
  • 支持 PyTorch/TensorFlow/ONNX 多框架,并附量化诊断与可视化工具,便于定位精度损失
  • 由高通维护,面向真实端侧部署场景,工程落地性较强

不足之处

  • 文档与示例偏工程化,新手理解量化流程门槛较高
  • 部分高级特性与高通自家推理栈绑定较紧,跨平台通用性受限

适用场景

  • 将云端训练好的大模型压缩后部署到手机、IoT 等边缘设备
  • 在保持精度的前提下把模型量化到 8bit/4bit 以降低推理延迟和内存占用
  • 对已有模型做剪枝和 SVD 压缩,减小模型体积便于分发和加载

替代项目

TensorRT、NNCF、neural-compressor

上一篇:Olive

下一篇:model-optimization

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 464 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 63828 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1317 2026-08-13