arXiv:cs.LG· Dwaipayan Haldar, Indranil Patra, Azhar Yousuf, Chandra Sekhar Seelamantula·· 10 小时前
SpectrumQuant:面向 LLM 量化的频谱对齐方法
SpectrumQuant: Spectrum Alignment for LLM Quantization
AI 导读
SpectrumQuant 是一种闭式旋转方法,通过从校准数据估计激活协方差来构建协方差感知的正交变换,用于 LLM 训练后量化。在 Llama 和 Mistral 模型上的实验显示,其效果与 SpinQuant、DartQuant 等学习式旋转方法相当,但所需 GPU 显存不到后者的三分之一,耗时也显著更少。
正文
Abstract:Orthogonal transforms have emerged as an effective tool for low-bit quantization of large language models (LLMs), as they reshape activation distributions without changing the underlying model function. Existing approaches typically rely either on fixed data-agnostic transforms, such as Hadamard rotations, or on calibration-time optimization of learned rotations. We study rotation selection from first principles and show that maximizing covariance leads to spectral alignment and provides a principled criterion for constructing quantization-friendly orthogonal transforms. Based on this insight, we propose SpectrumQuant, a closed-form rotation method that estimates activation covariance from calibration data and uses it to construct a covariance-aware orthogonal transform for post-training quantization. Our formulation characterizes quantization-friendly rotations as compositions of alignment and mixing components, while recovering Hadamard matrices as a special case. Interestingly, a class of butterfly matrices belongs to this category. In the realm of LLM quantization, SpectrumQuant bridges the gap between data-agnostic structured rotations and learned rotations, providing an effective and efficient data-aware alternative for post-training quantization. Experiments on Llama and Mistral models show that SpectrumQuant is competitive with state-of-the-art rotation strategies while requiring less than one-third of the GPU memory and significantly less time than learned rotations approaches like SpinQuant and DartQuant.
| Subjects: | Machine Learning (cs.LG); Machine Learning (stat.ML) |
| Cite as: | arXiv:2610.20303 [cs.LG] |
| (or arXiv:2610.20303v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2610.20303 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Dwaipayan Haldar [view email]
[v1]
Wed, 14 Oct 2026 14:54:16 UTC (9,460 KB)
来源:arXiv:cs.LG · arxiv.org