Исследователи представили новый подход к обучению прямых стратегий в Generative Flow Networks (GFlowNets) через призму информационной геометрии. Метод позволяет оптимизировать процесс сэмплирования траекторий, рассматривая стратегию как распределение вероятностей на пространстве путей. Это повышает эффективность обучения моделей, работающих с дискретными и смешанными данными, где требуется аппроксимация целевой плотности на основе заданных функций вознаграждения.

Традиционные подходы к GFlowNets часто сталкиваются с трудностями при масштабировании и стабильности обучения из-за сложности ландшафта целевых распределений. Авторы работы предлагают использовать геометрические свойства пространства параметров для более точного обновления стратегии. Такой подход минимизирует расхождение между текущей стратегией и оптимальным распределением, что ускоряет сходимость модели в задачах генеративного моделирования.

Математическая формулировка через информационную геометрию позволяет лучше контролировать процесс исследования пространства состояний. Это особенно важно для задач, где целевая плотность известна лишь с точностью до нормировочной константы, что типично для байесовского вывода и генерации молекулярных структур. Использование метрики Фишера в данном контексте обеспечивает более плавное и устойчивое обновление весов нейронной сети.

Ключевые факты

  • Предложен метод обучения GFlowNets, основанный на информационной геометрии индуцированного сэмплера траекторий.
  • Подход улучшает аппроксимацию целевых плотностей для дискретных и смешанных объектов.
  • Метод позволяет более эффективно использовать функции вознаграждения для обучения генеративных моделей.
  • Исследование направлено на повышение стабильности и скорости сходимости при работе с ненормированными распределениями.