Distributional Matching for Vector Quantization: A Unified Theoretical and Empirical Framework
arXiv, Under Review
TL;DR Matching feature and codebook distributions provides a unified approach to reducing vector-quantization instability and codebook collapse, with Wasserstein and maximum mean discrepancy objectives supported by theoretical analysis and visual-tokenization experiments.