Continuous First, Discrete Later: VQ-VAEs Without Dimensional Collapse

arXiv:2605.06870 · cs.LG · Submitted 2026-05-07 · Read on arXiv

cs.LG

Submitted: 2026-05-07

Updated: 2026-09-27

Code: https://github.com/xyz-zy/vqvae_latent_span_collapse

Terminology

• Quantization — 20× in this paper · explained in 12 episodes
• Core — 7× in this paper · explained in 2 episodes
• Gradient Flow — 6× in this paper · explained in 2 episodes
• Generative Model — 3× in this paper · explained in 2 episodes
• Latent Space — 3× in this paper · explained in 7 episodes
• Representation Learning — 3× in this paper · explained in 2 episodes
• Tokenization — 3× in this paper · explained in 2 episodes
• Ablation — 2× in this paper · explained in 1 episode
• Adam — 2× in this paper · explained in 2 episodes
• Generative Models — 2× in this paper · explained in 2 episodes
• In International Conference on Learning Representations — 2× in this paper
• Institutional Review Board — 2× in this paper
• Adaptation — 1× in this paper · explained in 1 episode
• Benchmark — 1× in this paper · explained in 2 episodes
• Computer Vision and Pattern Recognition — 1× in this paper
• Consistency — 1× in this paper · explained in 3 episodes
• Fidelity — 1× in this paper · explained in 2 episodes
• Fine-Tuning — 1× in this paper · explained in 5 episodes
• Foundation Model — 1× in this paper · explained in 3 episodes
• Generative Modeling — 1× in this paper · explained in 3 episodes
• International Conference on Computer Vision — 1× in this paper
• Transformer — 1× in this paper · explained in 2 episodes
• Variational Bayes — 1× in this paper · explained in 2 episodes

Sources

• Moshi: a speech-text foundation model for real-time dialogue
• Jukebox: A Generative Model for Music
• WavTokenizer: an Efficient Acoustic Discrete Codec Tokenizer for Audio Language Modeling
• Bridging Continuous and Discrete Tokens for Autoregressive Visual Generation
• Early Quantization Shrinks Codebook: A Simple Fix for Diversity-Preserving Tokenization
• Representation Collapsing Problems in Vector Quantization
• Addressing Representation Collapse in Vector Quantized Models with One Linear Layer

Related papers

• Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
• AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
• Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
• Convergence issues in Relational Concept Analysis based on AOC-posets
• Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
• Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks