MMODELYST
Papers/LoopQ: Quantization for Recursive Transformers
PAP

LoopQ: Quantization for Recursive Transformers

May 8, 2026

arXiv
Abstract

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-training quantization (PTQ). We present the first systematic study of quantization in LoopLMs and identify three challenges: distribution shift across roles, state reuse across loop transitions, and recursive error accumulation. To address these challenges, we propose LoopQ, a loop-aware PTQ framework that preserves a shared quantized backbone while introducing lightweight adaptations. LoopQ combines activation scaling, selective transformation, cross-loop state alignment, and trajectory-aware optimization to reduce distributional mismatch within loops and error accumulation across loops. Experiments across seven benchmarks show that, under W4A4 quantization, LoopQ improves average downstream accuracy by 68.8% and reduces average perplexity by 87.7% compared with the strongest static PTQ baseline.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Rui Fang, Hsi-Wen Chen, Ming-Syan Chen
Your notes (browser-local)
saved
arXiv:2605.16343