arXiv:2608.06916cs.LG2026-08

提出端到端混合精度协同设计框架,实现边缘AI模型高效部署

MiCoPro: End-to-End Mixed Precision HW/SW Co-design with HW-aware Proxy Model

  • 用硬件感知代理模型快速搜索最优混合精度配置
  • 在延迟约束下实现40%加速,精度损失小于3%
  • 支持从PyTorch直接生成裸金属C代码,适配多种边缘硬件

低比特量化神经网络在边缘设备上已证明可实现高效存储与计算。为在保持精度的同时最大化加速效果,逐层混合精度量化(MPQ)成为主流方案。然而,现有方法在探索MPQ方案时灵活性与效率不足,难以准确理解不同量化策略对训练后量化与量化感知训练的影响。此外,缺乏端到端的MPQ模型优化与部署框架。为此,我们提出MiCo框架,一个面向边缘AI应用的完整MPQ探索与部署方案。该框架采用新型优化算法,在严格延迟约束下搜索精度最优的量化配置。进一步扩展为MiCoPro,引入鲁棒的硬件感知代理(HAP)模型,提升预测准确性与硬件适应性。通过目标特定的延迟建模,MiCoPro实现从PyTorch模型到裸金属C代码的快速探索与直接部署。我们在BitFusion加速器和扩展SIMD的RISC-V处理器上验证了框架的通用性,实现最高40%的延迟降低,精度下降小于3%。

原文摘要 · Abstract (English)

Quantized Neural Networks~(QNN) with low-bitwidth data have proven promising in efficient storage and computation on edge devices. To mitigate accuracy degradation while maximizing speedup, layer-wise mixed-precision quantization~(MPQ) becomes a popular solution. However, existing algorithms for exploring MPQ schemes are limited in flexibility and efficiency. Comprehending the complex impacts of different MPQ schemes on post-training quantization and quantization-aware training results is a challenge for conventional methods. Furthermore, an end-to-end framework for the optimization and deployment of MPQ models is missing in existing work. To address these challenges, we propose the MiCo framework, a holistic MPQ exploration and deployment framework for edge AI applications. The framework adopts a novel optimization algorithm to search for accuracy-optimal quantization configurations under strict latency constraints. We further extended the framework to MiCoPro, which introduces a robust Hardware-Aware Proxy (HAP) model to enhance prediction accuracy and hardware versatility. By leveraging target-specific latency modeling, MiCoPro enables rapid exploration and direct deployment from PyTorch models to bare-metal C code. We demonstrate the versatility of our framework on both the BitFusion accelerator and SIMD-extended RISC-V processors, achieving up to 40\% of latency reduction with less than 3\% of accuracy drop.

混合精度边缘计算协同设计量化部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。