Суть метода: от высокого к низкому, а затем обратно
MrFlow (Multi-Resolution Flow Matching) — это метод ускорения генерации изображений, не требующий дообучения моделей. Алгоритм работает по принципу «снизу вверх»: сначала создается изображение низкого разрешения, затем оно масштабируется в пиксельном пространстве с помощью Real-ESRGAN, и наконец, выполняется короткая стадия уточнения (refinement) на высоком разрешении. Это позволяет перенести основную вычислительную нагрузку с дорогих шагов диффузии на высоком разрешении на более дешевые шаги на низком.
Ключевое преимущество подхода — отсутствие необходимости в кастомных ядрах, дообучении апскейлера или изменении весов базовой модели. Реализация построена на стандартных библиотеках PyTorch и Diffusers.
Результаты бенчмарков: от 8x до 25x ускорения
Метод демонстрирует стабильно высокие показатели ускорения (end-to-end speedup) на различных архитектурах, включая FLUX.1-dev, FLUX.2 Klein, Qwen-Image и Z-Image. Особый интерес вызывает работа с новыми моделями, такими как Krea-2, где сообщество уже зафиксировало успешные результаты.
| Бэкенд (Backbone) | Настройка (Steps) | Ускорение (Speedup) | Примечание |
|---|---|---|---|
| FLUX.1-dev | 12 + 1 | 8.25x | Стандартная настройка |
| Qwen-Image | 12 + 1 | 10.3x | Высокое качество при 1024px |
| FLUX.2 Klein Base 9B | 12 + 1 | 8.79x | База и non-base варианты |
| Z-Image-Turbo | 8 + 1 | 21.0x | Модель с дистилляцией временных шагов |
| Qwen-Image + Pi-Flow | 4 + 1 | до 25x | Комбинация с дистиллированными весами |
Практическое применение и ComfyUI
Для пользователей ComfyUI выпущен официальный плагин ComfyUI-MrFlow. Он позволяет интегрировать MrFlow в существующие рабочие процессы без написания кода. Плагин включает готовые ноды, примеры JSON-конфигураций и вспомогательные графы для моделей семейства Qwen.
Исследователи также рекомендуют экспериментировать с параметрами уточнения. Использование direct sigma в диапазоне 0.16–0.20 вместо стандартных значений часто улучшает качество генерации, особенно при наличии текста на изображении. Для вывода в разрешении 2048px рекомендуется использовать апскейлер 4x Foolhardy Remacri.
Почему это важно
MrFlow решает главную проблему диффузионных моделей — высокую стоимость инференса. Позволяя использовать более мощные модели (например, Krea-2 или FLUX.2) в реальном времени или с меньшими затратами GPU, метод делает передовые генеративные модели доступнее для локального использования и коммерческих пайплайнов. Код и статья доступны на arXiv и GitHub.
Источник: Github ↗