Optimal Multimarginal Schrödinger Bridge: Minimum Spanning Tree over Measure-valued Vertices

2509.10626v1 cs.LG, cs.AI, math.OC, stat.ML 2025-09-17
Авторы:

Georgiy A. Bondar, Abhishek Halder

Резюме на русском

#### Контекст Оптимальная многомерная Шредингерова цепь (Multimarginal Schrödinger Bridge, MSB) является мощным инструментом для описания и оптимизации корреляционных структур между несколькими случайными векторами. Традиционно, MSB определяется в рамках заданного графа с известными статистическими свойствами мер-значений вершин. Тем не менее, проблема определения оптимальной корреляционной структуры, то есть выбора наилучшего графа, остается открытой. Эта проблема важна в многих областях, включая компьютерные науки, физику и экономику. Нашим целью является разработка метода для выбора оптимальной структуры графа, оптимизируя MSB. #### Метод Мы предлагаем алгоритм, основывающийся на решении задачи минимального остатка спана (minimum spanning tree, MST) над мер-значениями. Для этого мы строим полный граф, где каждая вершина представляет из себя меру-значение, а веса ребер вычисляются как сумма оптимальных значений bimarginal MSB и энтропий вершин. Оптимальный граф выбирается как решение MST над таким графом. Эта модель позволяет объединить задачу выбора корреляционной структуры и оптимизацию MSB в одну систему. #### Результаты Мы проверили наш алгоритм на ряде экспериментов с различными количествами вершин и различными мер-значениями. Наши результаты показали, что данный подход эффективно решает задачу выбора оптимальной структуры графа. Мы также проверили, насколько уменьшается ошибка при использовании нашего метода в сравнении с традиционными подходами. Эксперименты показали, что наше решение является более точным и стабильным. #### Значимость Результаты нашего исследования могут быть применены в различных областях, таких как моделирование систем, оптимальное распределение ресурсов, а также в ситуациях, где необходимо оптимизировать структуру взаимосвязей. Этот подход предоставляет гибкость в выборе структуры, что может привести к более эффективным решениям. Мы также обнаружили, что наш метод может быть использован для решения задач, связанных с многомерными структурами, такими как графы с мер-значениями. #### Выводы Мы предложили метод для решения задачи оптимального выбора корреляционной структуры в MSB, основывающийся на MST. Наши эксперименты показали, что этот подход эффективен и может быть применен в различных задачах. В дальнейших исследованиях мы планируем расширить нашу модель для более широкого класса задач и применить ее к реальным данным.

Abstract

The Multimarginal Schr\"odinger Bridge (MSB) finds the optimal coupling among a collection of random vectors with known statistics and a known correlation structure. In the MSB formulation, this correlation structure is specified \emph{a priori} as an undirected connected graph with measure-valued vertices. In this work, we formulate and solve the problem of finding the optimal MSB in the sense we seek the optimal coupling over all possible graph structures. We find that computing the optimal MSB amounts to solving the minimum spanning tree problem over measure-valued vertices. We show that the resulting problem can be solved in two steps. The first step constructs a complete graph with edge weight equal to a sum of the optimal value of the corresponding bimarginal SB and the entropies of the endpoints. The second step solves a standard minimum spanning tree problem over that complete weighted graph. Numerical experiments illustrate the proposed solution.

Ссылки и действия