arXiv:2606.13054cs.LGcs.AI2026-06中稿 · ICML被引 4

让大模型权重三值化、激活四比特化,实现高效推理

TWLA: Achieving Ternary Weights and Low-Bit Activations for LLMs via Post-Training Quantization

论文配图:TWLA: Achieving Ternary Weights and Low-Bit Activations for LLMs via Post-Training Quantization
图 1 · 摘自论文原文
  • 通过两阶段优化实现权重三值化,降低存储和计算开销
  • 在W1.58A4下保持高精度,显著加速推理
  • 适合追求低资源部署的大模型应用

大语言模型虽具备强大语言处理能力,但其内存与计算成本阻碍了实际部署。三值化作为压缩技术展现出巨大潜力,能显著减少模型尺寸与推理复杂度。然而现有方法难以应对激活分布的长尾特性,导致激活仍需高精度,限制了端到端推理加速。为此,本文提出TWLA——一种后训练量化框架,可在保持高精度的前提下实现1.58比特权重压缩与4比特激活量化。TWLA包含三个组件:(1) 欧几里得到流形的非对称三值量化器(E2M-ATQ),通过从欧氏初始化到流形重定位的两阶段优化最小化层输出误差;(2) 克罗内克正交三模态整形(KOTMS),利用克罗内克结构的正交旋转将权重重塑为三值友好型三模态分布,共享旋转有效抑制激活异常值;(3) 层间感知激活混合精度(ILA-AMP),在比特分配中显式引入相邻层二阶交互代价,联合优化由共享正交变换引发的各层激活量化增益差异,防止少数弱层引发级联失效。大量实验表明,TWLA在W1.58A4配置下仍保持高精度,并带来显著推理加速。代码已开源于https://github.com/Kishon-zzx/TWLA。

原文摘要 · Abstract (English)

Large language models (LLMs) exhibit exceptional general language processing capabilities, but their memory and compute costs hinder deployment. Ternarization has emerged as a promising compression technique, offering significant reductions in model size and inference complexity. However, existing methods struggle with heavy-tailed activation distributions and therefore keep activations in high precision, fundamentally limiting end-to-end inference acceleration. To overcome this limitation, we propose TWLA, a post-training quantization (PTQ) framework that achieves 1.58-bit weight compression and 4-bit activation quantization while maintaining high accuracy. TWLA comprises three components: (1) Euclidean-to-Manifold Asymmetric Ternary Quantizer (E2M-ATQ) minimizes layer-output error under weight ternarization via a two-stage optimization from Euclidean initialization to manifold relocation; (2) Kronecker Orthogonal Tri-Modal Shaping (KOTMS) applies a Kronecker-structured orthogonal rotation to reshape weights into ternary-friendly tri-modal distributions, while the shared rotation statistically suppresses activation outliers; and (3) Inter-Layer Aware Activation Mixed Precision (ILA-AMP) explicitly introduces adjacent-layer second-order interaction costs in bit allocation and jointly optimizes for the layer-wise disparity of activation quantization gains induced by the shared orthogonal transform, preventing cascades triggered by a few weak layers. Extensive experiments demonstrate that TWLA maintains high accuracy under W1.58A4, while delivering significant inference acceleration. The code is available at https://github.com/Kishon-zzx/TWLA.

大模型压缩量化推理加速

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