首页
/ AIMET 2.6.0版本发布:量化工具链的重大升级

AIMET 2.6.0版本发布:量化工具链的重大升级

2025-06-20 00:23:19作者:龚格成

AIMET(AI Model Efficiency Toolkit)是由高通创新中心开发的AI模型效率工具包,专注于神经网络模型的量化和优化。该项目提供了跨框架(PyTorch、TensorFlow、ONNX)的量化工具,帮助开发者在不显著损失模型精度的情况下,大幅提升模型在边缘设备上的推理效率。

核心特性更新

ONNX运行时增强

2.6.0版本对ONNX支持进行了多项重要改进。最值得注意的是新增了直接向QuantizationSimModel构造函数传递ONNX Runtime执行提供者(EPs)的能力,这使得开发者可以更灵活地控制量化模型的执行环境。

在性能优化方面,新版本显著降低了顺序MSE(最小二乘法)过程中的CPU和GPU内存使用量,这对于处理大型模型尤为重要。同时修复了自动混合精度(AMP)功能中的多个问题,包括生成不兼容量化器配置的问题以及与动态卷积操作的兼容性问题。

PyTorch量化新特性

PyTorch方面最引人注目的是新增了对float8量化的模拟支持。float8是一种新兴的低精度数据类型,能够在保持合理模型精度的同时,进一步减少内存占用和计算开销。

实验性功能方面,新增了aimet_torch.onnx.export API,允许将QuantizationSimModel导出为ONNX QDQ(Quantize-Dequantize)图,这为跨框架部署提供了更多可能性。

重要改进与修复

跨框架一致性

2.6.0版本特别注重了不同框架间量化行为的一致性。现在ONNX的对称编码计算方式已与aimet_torch保持一致,减少了开发者在跨框架迁移时的适配成本。

PyTorch方面,修复了Resize层输入输出编码对齐的问题,并增加了对LeakyReLU层的超组融合处理,这些改进使得量化过程更加稳定可靠。

用户体验优化

在错误处理方面,修复了当torch.onnx.export失败时抛出的AttributeError问题,同时改进了对deepspeed导入失败情况的处理,使得工具链更加健壮。

文档方面更新了LoRA(Low-Rank Adaptation)用户指南,帮助开发者更好地理解和使用这一模型适配技术。

向后兼容性说明

ONNX接口方面,2.6.0版本正式弃用了QuantizationSimModel.__init__中的多个参数,包括use_cudadevicerounding_modeuse_symmetric_encodings。开发者应检查现有代码并相应调整。

技术意义与应用价值

AIMET 2.6.0的发布标志着该工具链在以下几个方面的进步:

  1. 精度与效率的平衡:float8量化的引入为超低精度推理开辟了新途径,特别适合资源受限的边缘设备。

  2. 框架互通性:增强的ONNX支持使得PyTorch和ONNX模型之间的转换更加顺畅,有利于模型从训练到部署的完整流程。

  3. 内存优化:减少MSE过程中的内存消耗,使得在普通硬件上量化大型模型成为可能。

  4. 开发者体验:错误修复和文档更新降低了使用门槛,使更多开发者能够受益于模型量化技术。

对于从事边缘AI部署的工程师来说,2.6.0版本提供了更强大、更稳定的工具集,能够帮助他们在保持模型性能的同时,显著提升推理效率,是值得升级的重要版本。

登录后查看全文
热门项目推荐

项目优选

收起
kernelkernel
deepin linux kernel
C
22
6
docsdocs
OpenHarmony documentation | OpenHarmony开发者文档
Dockerfile
168
2.05 K
openHiTLS-examplesopenHiTLS-examples
本仓将为广大高校开发者提供开源实践和创新开发平台,收集和展示openHiTLS示例代码及创新应用,欢迎大家投稿,让全世界看到您的精巧密码实现设计,也让更多人通过您的优秀成果,理解、喜爱上密码技术。
C
92
599
nop-entropynop-entropy
Nop Platform 2.0是基于可逆计算理论实现的采用面向语言编程范式的新一代低代码开发平台,包含基于全新原理从零开始研发的GraphQL引擎、ORM引擎、工作流引擎、报表引擎、规则引擎、批处理引引擎等完整设计。nop-entropy是它的后端部分,采用java语言实现,可选择集成Spring框架或者Quarkus框架。中小企业可以免费商用
Java
8
0
ohos_react_nativeohos_react_native
React Native鸿蒙化仓库
C++
199
279
RuoYi-Vue3RuoYi-Vue3
🎉 (RuoYi)官方仓库 基于SpringBoot,Spring Security,JWT,Vue3 & Vite、Element Plus 的前后端分离权限管理系统
Vue
954
563
金融AI编程实战金融AI编程实战
为非计算机科班出身 (例如财经类高校金融学院) 同学量身定制,新手友好,让学生以亲身实践开源开发的方式,学会使用计算机自动化自己的科研/创新工作。案例以量化投资为主线,涉及 Bash、Python、SQL、BI、AI 等全技术栈,培养面向未来的数智化人才 (如数据工程师、数据分析师、数据科学家、数据决策者、量化投资人)。
Python
78
71
leetcodeleetcode
🔥LeetCode solutions in any programming language | 多种编程语言实现 LeetCode、《剑指 Offer(第 2 版)》、《程序员面试金典(第 6 版)》题解
Java
60
17
apintoapinto
基于golang开发的网关。具有各种插件,可以自行扩展,即插即用。此外,它可以快速帮助企业管理API服务,提高API服务的稳定性和安全性。
Go
22
0