Исследователи представили метод One-Sided Quantile Coupling, повышающий эффективность обучения генеративных моделей на базе Flow Matching. Новый подход оптимизирует сопряжение между исходным распределением и целевыми данными, решая проблему вычислительной сложности традиционных методов оптимального транспорта. Это позволяет значительно ускорить сходимость модели и улучшить качество генерации при меньших затратах ресурсов.
Flow Matching обучается путем регрессии векторного поля вероятностного пути, соединяющего простой шум и сложные данные. Качество итоговой генерации напрямую зависит от того, как именно сопоставляются пары «источник-цель». Ранее для этого использовались процедуры мини-батч-транспорта или алгоритмы назначения, вычислительная стоимость которых быстро возрастала при увеличении объема данных, что ограничивало масштабируемость процесса обучения.
Предложенный метод использует квантильное сопряжение, которое позволяет эффективно выравнивать распределения без необходимости выполнения дорогостоящих операций глобального поиска соответствий. Такой подход обеспечивает более стабильное обучение и позволяет модели лучше улавливать структуру целевых данных, сохраняя при этом непрерывную временную динамику, характерную для Flow Matching. Метод демонстрирует высокую эффективность в задачах генерации изображений и других высокоразмерных данных.
Ключевые факты
- Метод One-Sided Quantile Coupling снижает вычислительную сложность сопряжения выборок по сравнению с классическими алгоритмами оптимального транспорта.
- Подход оптимизирует регрессию векторного поля, что напрямую влияет на точность аппроксимации вероятностного пути.
- Техника позволяет масштабировать обучение генеративных моделей на большие наборы данных без экспоненциального роста затрат на вычисления.
- Использование квантильного сопряжения обеспечивает более качественную сходимость модели к целевому распределению данных.