In plain words: Instead of generating each compressed token one at a time, it predicts a whole group at once, so sampling speed stays the same even as more compression layers are added. It beats the usual step-by-step models on image generation and text-to-speech without slowing down.
Abstract
We introduce ResGen, an efficient Residual Vector Quantization (RVQ)-based generative model for high-fidelity generation with fast sampling. RVQ improves data fidelity by increasing the number of quantization steps, referred to as depth, but deeper quantization typically increases inference steps in generative models. To address this, ResGen directly predicts the vector embedding of collective tokens rather than individual ones, ensuring that inference steps remain independent of RVQ depth. Additionally, we formulate token masking and multi-token prediction within a probabilistic framework using discrete diffusion and variational inference. We validate the efficacy and generalizability of the proposed method on two challenging tasks across different modalities: conditional image generation on ImageNet 256x256 and zero-shot text-to-speech synthesis. Experimental results demonstrate that ResGen outperforms autoregressive counterparts in both tasks, delivering superior performance without compromising sampling speed. Furthermore, as we scale the depth of RVQ, our generative models exhibit enhanced generation fidelity or faster sampling speeds compared to similarly sized baseline models.
Jaehyeon Kim, Taehong Moon, Keon Lee, Jaewoong Cho
arXiv:2412.10208 · cs.LG · submitted Dec 13, 2024 · updated Jun 2, 2025
abstract · pdf · html · ICML 2025