AI RESEARCH

Weighted quantization using MMD: From mean field to mean shift via gradient flows

arXiv CS.LG

ArXi:2502.10600v3 Announce Type: replace-cross Approximating a probability distribution using a set of particles is a fundamental problem in machine learning and statistics, with applications including clustering and quantization. Formally, we seek a weighted mixture of Dirac measures that best approximates the target distribution. While much existing work relies on the Wasserstein distance to quantify approximation errors, maximum mean discrepancy (MMD) has received comparatively less attention, especially when allowing for variable particle weights.