paper-with-me

Papers

What Makes Low-Bit Quantization-Aware Training Work for Reasoning LLMs? A Systematic Study

2026-01-21 · Keyu Lv, Manyi Zhang, Xiaobo Xia, Jingchen Ni, Shannan Yan, Xianzhi Yu, Lu Hou, Chun Yuan, Haoli Bai arxiv

Reasoning models excel at complex tasks such as coding and mathematics, yet their inference is often slow and token-inefficient. To improve the inference efficiency, post-training quantization (PTQ) usually comes with the cost of large accuracy drops, especially for reasoning tasks under low-bit settings. In this study, we present a systematic empirical study of quantization-aware training (QAT) for reasoning models. Our key findings include: (1) Knowledge distillation is a robust objective for reasoning models trained via either supervised fine-tuning or reinforcement learning; (2) PTQ provides a strong initialization for QAT, improving accuracy while reducing training cost; (3) Reinforcement learning remains feasible for quantized models given a viable cold start and yields additional gains; and (4) Aligning the PTQ calibration domain with the QAT training domain accelerates convergence and often improves the final accuracy. Finally, we consolidate these findings into an optimized workflow (Reasoning-QAT), and show that it consistently outperforms state-of-the-art PTQ methods across multiple LLM backbones and reasoning datasets. For instance, on Qwen3-0.6B, it surpasses GPTQ by 44.53% on MATH-500 and consistently recovers performance in the 2-bit regime.

📄 PDF Abstract BibTeX arXiv:2601.14888

Code (0)

등록된 구현이 없습니다.

Tasks

Knowledge DistillationReinforcement Learning

Similar Papers 제목 키워드 기반

Attention-aware Post-training Quantization without Backpropagation

2024-06-19 · Junhan Kim, Ho-young Kim, Eulrang Cho, Chungman Lee 외

Quantization is a promising solution for deploying large-scale language models (LLMs) on resource-constrained devices. Existing quantization approaches, however, rely on gradient-based optimization, regardless of it bein…

Quantization

PreQuant: A Task-agnostic Quantization Approach for Pre-trained Language Models

2023-05-30 · Zhuocheng Gong, Jiahao Liu, Qifan Wang, Yang Yang 외

While transformer-based pre-trained language models (PLMs) have dominated a number of NLP applications, these models are heavy to deploy and expensive to use. Therefore, effectively compressing large-scale PLMs becomes a…

parameter-efficient fine-tuningQuantization

Automated Log-Scale Quantization for Low-Cost Deep Neural Networks

2021-06-19 · CVPR 2021 1 · Sangyun Oh, Hyeonuk Sim, Sugil Lee, Jongeun Lee

Quantization plays an important role in deep neural network (DNN) hardware. In particular, logarithmic quantization has multiple advantages for DNN hardware implementations, and its weakness in terms of lower perform…

Image EnhancementQuantizationSemantic Segmentation

Texture Vector-Quantization and Reconstruction Aware Prediction for Generative Super-Resolution

2025-09-28 · Qifan Li, Jiale Zou, Jinhua Zhang, Wei Long 외 arxiv

Vector-quantized based models have recently demonstrated strong potential for visual prior modeling. However, existing VQ-based methods simply encode visual features with nearest codebook items and train index predictor …

LoopQ: Quantization for Recursive Transformers

2026-05-08 · Rui Fang, Hsi-Wen Chen, Ming-Syan Chen arxiv

Looped language models (LoopLMs) improve parameter efficiency by recursively reusing Transformer blocks, enabling deeper computation under a fixed model size. However, this reuse makes LoopLMs more fragile under post-tra…