Search papers, labs, and topics across Lattice.
The paper introduces VP-VAE, a novel VQ-VAE variant that decouples representation learning from codebook optimization by replacing the discrete codebook with adaptive latent perturbations generated via Metropolis-Hastings sampling. This approach avoids codebook collapse and training instability common in standard VQ-VAEs. The authors also derive FSP, a simplified version of VP-VAE under the assumption of uniform latent variables, offering a theoretical explanation and practical improvement for fixed quantizers. Experiments on image and audio datasets demonstrate improved reconstruction fidelity and more balanced token usage compared to standard VQ-VAE approaches.
Ditch the codebook: VP-VAE achieves stable VQ-VAE training by perturbing latent vectors instead of relying on explicit vector quantization.
Vector Quantized Variational Autoencoders (VQ-VAEs) are fundamental to modern generative modeling, yet they often suffer from training instability and "codebook collapse" due to the inherent coupling of representation learning and discrete codebook optimization. In this paper, we propose VP-VAE (Vector Perturbation VAE), a novel paradigm that decouples representation learning from discretization by eliminating the need for an explicit codebook during training. Our key insight is that, from the neural network's viewpoint, performing quantization primarily manifests as injecting a structured perturbation in latent space. Accordingly, VP-VAE replaces the non-differentiable quantizer with distribution-consistent and scale-adaptive latent perturbations generated via Metropolis--Hastings sampling. This design enables stable training without a codebook while making the model robust to inference-time quantization error. Moreover, under the assumption of approximately uniform latent variables, we derive FSP (Finite Scalar Perturbation), a lightweight variant of VP-VAE that provides a unified theoretical explanation and a practical improvement for FSQ-style fixed quantizers. Extensive experiments on image and audio benchmarks demonstrate that VP-VAE and FSP improve reconstruction fidelity and achieve substantially more balanced token usage, while avoiding the instability inherent to coupled codebook training.
