菜单

关于 🐙 GitHub
arXiv 提交日期: 2026-06-11
📄 Abstract - TWLA: Achieving Ternary Weights and Low-Bit Activations for LLMs via Post-Training Quantization

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 <this https URL.

顶级标签: llm model training
详细标签: post-training quantization ternarization low-bit activations weight compression inference acceleration 或 搜索:

TWLA:通过训练后量化实现大语言模型的三值权重与低位激活 / TWLA: Achieving Ternary Weights and Low-Bit Activations for LLMs via Post-Training Quantization


1️⃣ 一句话总结

本文提出了一种名为TWLA的新型训练后量化框架,通过巧妙的数学变换和优化策略,成功将大语言模型的权重压缩到1.58位、激活值量化到4位,在保持高精度的同时大幅提升推理速度,解决了此前极低比特量化中激活值难以压缩的难题。

源自 arXiv: 2606.13054