arXiv:2606.25087cs.CV2026-06

通过学习低损失子空间,实现无需训练的高效神经网络量化

Neural Network Quantization by Learning Low-Loss Subspaces

论文配图:Neural Network Quantization by Learning Low-Loss Subspaces
图 1 · 摘自论文原文
  • 在权重重塑空间中学习量化友好路径,避免传统方法依赖梯度估计
  • 所获子空间中点直接量化后性能媲美量化感知训练
  • 适用于追求低延迟部署的模型压缩场景

神经网络量化旨在寻找参数的离散表示,以尽可能保持全精度模型的性能。施加离散约束会将参数从优化极小值点扰动,通常导致性能下降。近期研究表明,低损失的全精度解并非孤立存在,而是构成损失景观中连续的低损失子空间,该子空间内采样的模型多样且保持高精度。这引发一个关键问题:能否构建一个位于全精度模型低损失子空间内的量化模型,从而自动维持性能?本文通过学习权重空间中量化感知的线性路径来解决该问题,优化以最小化损失。我们证明,所得子空间的中点在设计上即为量化友好,其直接量化即可获得与量化感知训练相当的性能。该方法提供了权重量化的全新视角,与传统方法不同,既不依赖直通估计器,也不在训练中显式进行离散化。

原文摘要 · Abstract (English)

Neural network quantization aims to find a discrete representation of parameters that preserves the performance of a full-precision (FP) model as faithfully as possible. Enforcing discrete constraints perturbs parameters away from a well-optimized minimum, generally resulting in performance degradation. Recent studies indicate that low-loss FP solutions are not isolated, but instead belong to connected low-loss subspaces of the loss landscape, where the loss maintains nearly the same minimum value. Models sampled from these subspaces are diverse and retain high accuracy. This raises the question: can a quantized model be constructed to lie within a low-loss subspace of the FP model, thereby automatically preserving performance? We address this question by learning quantization-aware linear paths in weight space optimized to minimize loss. We demonstrate that the midpoint of the resulting subspace is, by design, quantization-friendly and that its direct quantization yields performance comparable to that of quantization-aware training. The proposed procedure offers a novel perspective on weight quantization and, in contrast to conventional methods, neither relies on the straight-through estimator nor involves explicit discretization during training.

神经网络量化低损失子空间模型压缩无梯度量化

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