arXiv:2411.05845cs.LG2024-11被引 2

用高低精度分离提升模型推理效率,兼顾性能与可靠性。

Neural Precision Polarization: Simplifying Neural Network Inference with Dual-Level Precision

  • 核心思路:多数权重用低精度,关键路径保留高精度补偿误差。
  • 实测达464 TOPS/W能效,支持边缘设备部署。
  • 适合做低功耗芯片加速或模拟计算的模型优化场景。

本文提出一种神经网络推理中的精度极化方案,仅使用极低和极高两种精度级别:大部分网络权重和激活值采用低精度表示,而关键路径则保留高精度以实现误差补偿。该方法使不同精度层级可独立优化,显著降低内存与计算开销,同时保持模型准确率。训练阶段在云端完成浮点模型训练,随后下载至边缘设备,直接将权重与激活量化至如NF4或INT8等目标精度。为缓解量化带来的精度损失,引入基于低秩近似的替代路径,通过敏感度指标在少量数据上训练,恢复因量化及工艺波动(如模拟加速实现时)导致的性能下降。仿真结果表明,结合秩8误差恢复路径与位平面级存内计算,系统可实现约464 TOPS/W的能效与可靠性表现。

原文摘要 · Abstract (English)

We introduce a precision polarization scheme for DNN inference that utilizes only very low and very high precision levels, assigning low precision to the majority of network weights and activations while reserving high precision paths for targeted error compensation. This separation allows for distinct optimization of each precision level, thereby reducing memory and computation demands without compromising model accuracy. In the discussed approach, a floating-point model can be trained in the cloud and then downloaded to an edge device, where network weights and activations are directly quantized to meet the edge devices' desired level, such as NF4 or INT8. To address accuracy loss from quantization, surrogate paths are introduced, leveraging low-rank approximations on a layer-by-layer basis. These paths are trained with a sensitivity-based metric on minimal training data to recover accuracy loss under quantization as well as due to process variability, such as when the main prediction path is implemented using analog acceleration. Our simulation results show that neural precision polarization enables approximately 464 TOPS per Watt MAC efficiency and reliability by integrating rank-8 error recovery paths with highly efficient, though potentially unreliable, bit plane-wise compute-in-memory processing.

模型压缩边缘计算能效优化量化推理

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