Mixtures Closest To A Given Measure: A Semidefinite Programming Approach
Abstract
Lay Summary
Many real-world datasets are made of several hidden groups: for example, a collection of images, measurements, or signals may come from several different sources. A common way to model such data is to fit a mixture of simple probability distributions, but two hard questions are how many components to use and what their parameters should be. Existing methods often rely on local search or on first choosing a fixed number of components, which can make the result sensitive to initialization. We propose a different approach: instead of directly guessing the components, we use a small set of statistical summaries of the data, called moments, and formulate the search for the closest mixture as an optimization problem that can be solved systematically. Our method can compare distributions using either Wasserstein distance, which measures how much probability mass must be moved, or total variation distance, which measures how different two distributions are overall. Under suitable mathematical conditions, the resulting sequence of optimization problems converges to the best mixture, and in some cases the mixture can be recovered exactly. In clustering experiments, the method can estimate the number of clusters and provide strong initial centers for standard algorithms such as k-means and EM. This makes mixture learning more principled and less dependent on fragile initialization choices.