Исследователи проанализировали внутреннюю механику дискретных диффузионных моделей, ответив на вопрос, что именно оптимизирует нейросеть в процессе обучения. Авторы доказали, что денойзер, отношение оценок (score ratio) и предикторы мостов являются математически эквивалентными объектами, представленными в разных координатах. Некорректный выбор системы координат при интерпретации сети приводит к изменению процесса обучения и генерации данных.
Работа предлагает строгий вывод нижней границы доказательства (ELBO) для цепей Маркова в непрерывном времени (CTMC) применительно к любым процессам зашумления. Авторы показывают, что выбор параметризации напрямую влияет на динамику обучения и качество итоговых выборок. Исследование помогает лучше понять теоретические основы генеративных моделей, работающих с дискретными данными, такими как текст или категориальные признаки.
Математический аппарат, представленный в статье, позволяет унифицировать подходы к проектированию диффузионных моделей. Вместо разрозненных методов обучения авторы предлагают единую структуру, основанную на анализе скоростей переходов (jump rates). Это дает возможность более точно настраивать архитектуры моделей, избегая ошибок в интерпретации градиентов и вероятностных распределений.
Ключевые факты
- Исследование фокусируется на эквивалентности денойзеров, отношений оценок и предикторов мостов в дискретных моделях.
- Предложен строгий вывод ELBO для цепей Маркова в непрерывном времени (CTMC) для произвольных процессов зашумления.
- Доказано, что выбор системы координат при чтении нейронной сети критически влияет на процесс обучения и сэмплирования.
- Работа направлена на унификацию теоретической базы генеративных моделей для дискретных данных.