发现生成模型权重指数集中,实现无损压缩与加速
To Compress or Not? Pushing the Frontier of Lossless GenAI Model Weights Compression with Exponent Concentration
- 利用指数熵低的统计特性,设计无损压缩编码方案
- 在671B模型上实现26.9%内存节省与177.1%吞吐提升
- 适合追求极致效率的生成式AI部署场景
生成式AI模型参数量达千亿级,低精度计算成为高效部署的关键。本文揭示生成模型权重中指数集中现象:不同架构与模态下指数熵普遍偏低。该现象源于随机梯度下降诱导的α-稳定分布,我们证明了指数熵的紧致理论边界,推导出接近FP4.67的理论压缩极限,进而提出实用的FP8格式。基于此,我们构建了指数集中型FP8(ECF8)框架,采用熵感知编码与GPU优化解码。在高达6710亿参数的LLM与DiT模型上,实验显示内存最多节省26.9%,吞吐提升达177.1%,且计算完全无损,输出结果零偏差。研究确立指数集中为训练模型的统计规律,为FP8时代的无损低精度浮点设计提供了理论指导。
原文摘要 · Abstract (English)
The scaling of Generative AI (GenAI) models into the hundreds of billions of parameters makes low-precision computation indispensable for efficient deployment. We argue that the fundamental solution lies in developing low-precision floating-point formats, which inherently provide numerical stability, memory savings, and hardware efficiency without dequantization overhead. In this paper, we present a theoretical and empirical study of an exponent concentration phenomenon in GenAI weights: exponents consistently exhibit low entropy across architectures and modalities. We show that this arises naturally from $α$-stable distributions induced by stochastic gradient descent, and we prove tight bounds on the entropy of exponents. Our analysis establishes a theoretical compression limit near FP4.67, which motivates the design of a practical FP8 format. Building on these insights, we propose Exponent-Concentrated FP8 (ECF8), a lossless compression framework with entropy-aware encoding and GPU-optimized decoding. Experiments on LLMs and DiTs up to 671B parameters demonstrate up to 26.9% memory savings and 177.1% throughput acceleration, with perfectly lossless computations, i.e., no deviation in model outputs. Our results establish exponent concentration as a statistical law of trained models and open a principled path for lossless low-precision floating-point design in the FP8 era.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。