Исследователи проанализировали внутреннюю механику дискретных диффузионных моделей, ответив на вопрос, что именно оптимизирует нейросеть в процессе обучения. Авторы доказали, что денойзер, отношение оценок (score ratio) и предикторы мостов являются математически эквивалентными объектами, представленными в разных координатах. Некорректный выбор системы координат при интерпретации сети приводит к изменению процесса обучения и генерации данных.

Работа предлагает строгий вывод нижней границы доказательства (ELBO) для цепей Маркова в непрерывном времени (CTMC) применительно к любым процессам зашумления. Авторы показывают, что выбор параметризации напрямую влияет на динамику обучения и качество итоговых выборок. Исследование помогает лучше понять теоретические основы генеративных моделей, работающих с дискретными данными, такими как текст или категориальные признаки.

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

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

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