2608.16101

Total: 1

#1 EMS Coreset: An Efficient Expectation-Maximization Algorithm for Sinkhorn Coreset [PDF] [Copy] [Kimi] [REL]

Authors: Haoyun Yin, Chuanhui Liu, Xiao Wang

Coresets distill large datasets into small, representative subsets for efficient downstream learning. Yet Optimal Transport (OT)-based selection typically requires intensive computation of transport plans, limiting scalability. We introduce a scalable Sinkhorn coreset method that permits closed-form updates of the entropically regularized OT coupling by allowing non-uniform coreset weights. This produces centroids that generalize k-means via soft assignments. We establish asymptotic consistency of the selected measure and Lipschitz stability to data perturbations, providing accuracy and robustness guarantees. Across synthetic and real-world benchmarks, the proposed method achieves competitive or improved approximation quality while substantially reducing runtime compared to Wasserstein- and standard Sinkhorn-based coreset selection, especially at large scale.

Subjects: Machine Learning , Machine Learning

Publish: 2026-08-17 04:46:34 UTC