STLA: Spatiotemporal Lookahead Alignment for Post-Training Quantization
Zuqi Zhang, Chenghe Sun, Xiangyi Chu, Wei-Han Yu, Ka-Fai Un, Rui Martins, Pui-In Mak, Jiawei Xu
摘要
Adaptive rounding techniques in Post-Training Quantization (PTQ) enable the efficient deployment of Large Language Models (LLMs) with low resource and data dependencies. While learning-based rounding methods are accurate yet costly, compensation-based approaches offer a highly efficient alternative. However, synergizing these two to realize their full potential is hindered by spatiotemporal misalignment in the decoupled paradigm. Key challenges include temporal parameter conflict, the invalidation of the initial Round-to-Nearest (RTN) assumption, and spatially-inconsistent optimization objectives. This paper introduces STLA, a novel rounding-optimized PTQ framework that achieves both fast and accurate LLM quantization. STLA resolves temporal inconsistency through cluster-wise integrated rounding optimization, which collocates the learning and compensation phases. STLA achieves spatial alignment through a unified global objective derived from the Schur Complement, enabling the solver to look ahead and align local rounding decisions with the optimal future compensation of remaining weights. Furthermore, we propose a Hessian-guided clustering strategy that exploits both diagonal and off-diagonal information to maximize intra-cluster error cancellation. Extensive experiments demonstrate that STLA establishes a new state-of-the-art for low-bit PTQ while maintaining high computational efficiency. The code is available at https://github.com/AI2C-Lab/STLA.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper10
- Measuring Massive Multitask Language UnderstandingDan Hendrycks, Collin Burns, Steven Basart, Andy Zou 等ICLR 2021 · 被引用 7,905 次
- Learned Step Size quantizationSteven K. Esser, Jeffrey L. McKinstry, Deepika Bablani, Rathinakumar Appuswamy 等ICLR 2020 · 被引用 1,037 次
- Up or Down? Adaptive Rounding for Post-Training QuantizationMarkus Nagel, Rana Ali Amjad, Mart van Baalen, Christos Louizos 等ICML 2020 · 被引用 816 次
- QuIP: 2-Bit Quantization of Large Language Models With GuaranteesJerry Chee, Yaohui Cai, Volodymyr Kuleshov, Christopher De SaNeurIPS 2023 · 被引用 503 次
- Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and PruningElias Frantar, Dan AlistarhNeurIPS 2022 · 被引用 440 次
相关 Paper
- VLM-PTQ: Efficient Post-Training Quantization for Large Vision-Language ModelsJuncan Deng, Kejie HuangCVPR 2026 · 被引用 2 次
- PT-LLM: Post-Training Ternarization for Large Language ModelsXianglong Yan, Chengzhu Bao, Zhiteng Li, Tianao Zhang 等ICLR 2026 · 被引用 4 次
- OAC: Output-adaptive Calibration for Accurate Post-training QuantizationAli Edalati, Alireza Ghaffari, Mahsa Ghazvini Nejad, Lu Hou 等AAAI 2025 · 被引用 8 次
- LRQuant: Learnable and Robust Post-Training Quantization for Large Language ModelsJiaqi Zhao, Miao Zhang, Chao Zeng, Ming Wang 等ACL 2024
- Rethinking Residual Errors in Compensation-based LLM QuantizationShuaiting Li, Juncan Deng, Kedong Xu, Rongtao Deng 等ICLR 2026 · 被引用 4 次
