Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Sampling

## Контекст Boltzmann-распределения широко используются в различных научных областях, включая физику, химию и биологию, для описания систем с множеством степеней свободы. Однако, сэмплинг из таких распределений, особенно в высокомерных пространствах, представляется вызовом из-за сложности энергетических ландшафтов. Существующие подходы, такие как генеративные модели и методы типа Metropolis-Hastings, либо требуют больших объемов выборочных данных, либо ограничиваются локальными оптимизациями, не полностью использующими выразительную силу современных архитектур, таких как Continuous Normalizing Flows (CNF). Этот факт подчеркивает необходимость развития эффективных и скейлируемых методов для точного сэмплинга в высокомерных системах. ## Метод Мы предлагаем **Energy-Weighted Flow Matching (EWFM)**, новую методологию для сэмплинга из Boltzmann-распределений $\mu_{\text{target}}(x) \propto \exp(-E(x)/T)$ с помощью Continuous Normalizing Flows. Целью EWFM является устранение необходимости в больших выборках, используя только энергетические оценки. Основная идея заключается в переформулировке задачи условного flow matching через импортс-сэмплинг, что позволяет применять произвольные распределения-предложения в качестве процесса обучения. Для улучшения удобочитаемости и эффективности мы разрабатываем два алгоритма: **итеративный EWFM (iEWFM)**, который прогрессивно улучшает предложения в процессе обучения, и **анелированный EWFM (aEWFM)**, который включает аннелирование температуры для сложных энергетических ландшафтов. ## Результаты Мы проверяем EWFM на нескольких системах, включая 55-узловые кластеры Lennard-Jones. Наши результаты показывают, что метод EWFM демонстрирует высокую точность сэмплинга, подходящую за стандартными подходами, но с потреблением до 1000 раз меньше вычислительных ресурсов. Особенно заметен эффект на сложных ландшафтах энергии, где алгоритмы EWFM показывают существенное преимущество в скорости и точности. Эти результаты подтверждают высокую эффективность EWFM в области моделирования высокомерных систем. ## Значимость Предложенный подход имеет значительное значение в многочисленных областях, включая молекулярное моделирование, теорию графов и генетическое кодирование. Он позволяет избежать ограничений существующих методов, обеспечивая более точное и стремительное сэмплирование. Это не только повышает эффективность вычислений, но и расширяет возможности в области высокопроизводительных вычислений. ## Выводы Мы представляем EWFM — новый подход к сэмплингу из Boltzmann-распределений с помощью Continuous Normalizing Flows. Он успешно решает проблемы эф