English

Taming the Entropy Cliff: Variable Codebook Size Quantization for Autoregressive Visual Generation

Computer Vision and Pattern Recognition 2026-05-08 v1 Artificial Intelligence Machine Learning

Abstract

Most discrete visual tokenizers rely on a default design: every position in the sequence shares the same codebook. Researchers try to scale the codebook size KK to get better reconstruction performance. Such a constant-codebook design hits a fundamental information-theoretic limit. We observe that the per-position conditional entropy of the training set decays so quickly along the sequence that, after a few positions, the conditional distribution becomes essentially deterministic. On ImageNet with K=16384K=16384, this happens within only 2 out of 256 positions, turning the remaining 254 into a memorization problem. We call this phenomenon the Entropy Cliff and formalize it with a simple expression: t=log2N/log2Kt^{*} = \lceil \log_2 N / \log_2 K \rceil. Interestingly, this phenomenon is not observed in language, as its natural structure keeps the effective entropy per position well below the codebook capacity. To address this, we propose Variable Codebook Size Quantization (VCQ), where the codebook size KtK_t grows monotonically along the sequence from Kmin=2K_{\min}=2 to KmaxK_{\max}, leaving the loss function, parameter count, and AR training procedure unchanged. With a vanilla autoregressive Transformer and standard next-token prediction, a base version of VCQ reduces gFID w/o CFG from 27.98 to 14.80 on ImageNet 256×256256\times256 over the baseline. Scaled up, it reaches gFID 1.71 with 684M autoregressive parameters, without any extra training techniques such as semantic regularization or causal alignment. The extreme information bottleneck at Kmin=2K_{\min}=2 naturally induces a coarse-to-fine semantic hierarchy: a linear probe on only the first 10 tokens reaches 43.8% top-1 accuracy on ImageNet, compared to 27.1% for uniform codebooks. Ultimately, these results show that what matters is not only the total capacity of the codebook, but also how that capacity is distributed and organized.

Keywords

Cite

@article{arxiv.2605.06207,
  title  = {Taming the Entropy Cliff: Variable Codebook Size Quantization for Autoregressive Visual Generation},
  author = {Bowen Zheng and Weijian Luo and Guang Yang and Colin Zhang and Tianyang Hu},
  journal= {arXiv preprint arXiv:2605.06207},
  year   = {2026}
}
R2 v1 2026-07-01T12:54:58.854Z