Enhancing Vector Quantization with Distributional Matching: A Theoretical and Empirical Study