Исследователи предлагают сэмплер экстраполяции кумулятивной интенсивности с временной репараметризацией (TR-CIE) для улучшения качества выборки при дискретном согласовании потоков в условиях ограниченного числа вычислений функции. Метод сочетает репараметризацию времени на основе расписания с правилом обновления через экстраполяцию кумулятивной интенсивности, чтобы смягчить жёсткость и повысить точность аппроксимации.
- TR-CIE перемасштабирует временную сетку в соответствии с расписанием шума, поглощая члены роста, зависящие от расписания, при стандартных факторизованных параметризациях скорости DFM.
- Сэмплер использует кэшированные выходные данные модели с предыдущих шагов в качестве члена истории для аппроксимации пошаговых кумулятивных интенсивностей на неравномерных временных сетках.
- Он требует только одного вычисления функции на шаг и не вводит дополнительных вычислений модели по сравнению со стандартным τ-скачком.
- Теоретический анализ оценивает локальную ошибку аппроксимации кумулятивных интенсивностей и устанавливает результаты сходимости.
- Эксперименты на синтетических задачах, генерации текста и бенчмарках text-to-image демонстрируют улучшенное качество выборки при ограниченном NFE.
Этот подход решает проблему отсутствия эффективных методов сэмплирования для дискретного согласования потоков, предоставляя теоретически обоснованный метод, который повышает производительность без увеличения вычислительных накладных расходов.