A Particle-Flow Algorithm for Free-Support Wasserstein Barycenters
2509.11435v2
stat.ML, cs.LG, stat.CO
2025-09-17
Авторы:
Kisung You
Резюме на русском
## Контекст
В поле видимости статистических моделей и анализа данных, рассматривается понятие **Wasserstein barycenter**, которое представляет собой расширение Евклидового среднего до множества вероятностных мер. Оно определяется как минимум задачи усреднения весовой суммы квадратов 2-Wasserstein-дисторбансов. Эта концепция находит приложение в задачах кластеризации, статистического обучения и работе с изображениями. Однако существующие методы, такие как регуляризация с помощью энтропии, ограничивают точность и стабильность решений. Мотивирует разработку новых подходов, которые были бы более строгими и эффективными.
## Метод
Разработан **новый подход**, основанный на геометрической модели шаровых пространств, для вычисления Wasserstein barycenter. Без использования энтропии, метод использует **формальную римановскую геометрию** Wasserstein-пространства. Атомы барицентра, как частицы, двигаются по траекториям, возникающим из **адвекции в среднем по транспортным планам**. В случае отсутствия **Monge-переносов**, применяются **barycentric projections**. Этот подход позволяет сохранить геометрические особенности входных данных с минимальными вычислительными затратами.
## Результаты
Использованы синтетические и реальные данные для проверки алгоритма. На синтетических данных показана четкость решений и точность алгоритма. На реальных данных (кластеризация изображений, байесовский анализ, агрегация постероритетов) получены результаты, которые демонстрируют **высокую точность** и **устойчивость** к погрешностям в данных. Алгоритм показал **гармоническое сочетание точности и производительности**, становясь конкурентоспособным в сравнении с традиционными методами.
## Значимость
Полученный метод находит применение в **статистическом анализе**, **машинном обучении**, **графическом дизайне**. Он обеспечивает **высокую точность**, **устойчивость** и **уменьшает вычислительные затраты**. Благодаря принципиально новому подходу, алгоритм может использоваться в **больших масштабах** и в **реальном времени**, увеличивая эффективность в решении задач усреднения вероятностных мер.
## Выводы
Разработанный **particle-flow-алгоритм** представляет собой **новый стандарт** для вычисления Wasserstein barycenter. Он комбинирует **точность**, **эффективность** и **теоретическую строгость**. Будущие исследования будут сконцентрированы на расширении применимости и оптимизации алгоритма для **больших данных** и **реального времени**.
Abstract
The Wasserstein barycenter extends the Euclidean mean to the space of
probability measures by minimizing the weighted sum of squared 2-Wasserstein
distances. We develop a free-support algorithm for computing Wasserstein
barycenters that avoids entropic regularization and instead follows the formal
Riemannian geometry of Wasserstein space. In our approach, barycenter atoms
evolve as particles advected by averaged optimal-transport displacements, with
barycentric projections of optimal transport plans used in place of Monge maps
when the latter do not exist. This yields a geometry-aware particle-flow update
that preserves sharp features of the Wasserstein barycenter while remaining
computationally tractable. We establish theoretical guarantees, including
consistency of barycentric projections, monotone descent and convergence to
stationary points, stability with respect to perturbations of the inputs, and
resolution consistency as the number of atoms increases. Empirical studies on
averaging probability distributions, Bayesian posterior aggregation, image
prototypes and classification, and large-scale clustering demonstrate accuracy
and scalability of the proposed particle-flow approach, positioning it as a
principled alternative to both linear programming and regularized solvers.