English

MoBiQuant: Mixture-of-Bits Quantization for Token-Adaptive Any-Precision LLM

Machine Learning 2026-05-26 v2 Artificial Intelligence Computation and Language

Abstract

Dynamic runtime latency and memory constraints necessitate flexible large language model (LLM) deployment, where an LLM can be inferred with various quantization precisions based on available computational resources. Recent work on such any-precision quantization either relies on hardware-inefficient vector quantization or induces additional scaling factors when switching between bit-widths. Meanwhile, existing post-training quantization (PTQ) methods calibrated for a fixed low precision show poor generalizability under runtime precision change. In this work, we attribute the source of poor generalization across bit-widths to a precision-dependent \textit{outlier migration} phenomenon where the distribution of PTQ-sensitive tokens changes across precisions. Motivated by this observation, we propose \texttt{MoBiQuant}, a novel any-precision Mixture-of-Bits quantization framework that adjusts weight precision for flexible LLM inference based on token sensitivity. Specifically, we propose a many-in-one recursive residual quantization that can iteratively reconstruct higher-precision weights at runtime and mitigates \textit{outlier migration} with a token-aware router to dynamically select the optimal inference precision of each token.Extensive experiments show that \texttt{MoBiQuant} matches or surpasses frontier single-precision PTQ while exhibiting strong elasticity, achieving significant memory savings and throughput gains of up to 1.34×1.34\times over state-of-the-art any-precision methods.

Keywords

Cite

@article{arxiv.2602.20191,
  title  = {MoBiQuant: Mixture-of-Bits Quantization for Token-Adaptive Any-Precision LLM},
  author = {Dongwei Wang and Jinhee Kim and Seokho Han and Denis Gudovskiy and Yohei Nakata and Tomoyuki Okuno and KhayTze Peong and Kang Eun Jeon and Jong Hwan Ko and Yiran Chen and Huanrui Yang},
  journal= {arXiv preprint arXiv:2602.20191},
  year   = {2026}
}

Comments

20 pages, 10 figures

R2 v1 2026-07-01T10:48:29.950Z