Представлен AcFlow для контроля диффузионных трансформеров

Исследователи представили AcFlow — инструмент управления генерацией изображений в моделях Diffusion Transformers (DiT) на этапе инференса. Метод модифицирует промежуточные активации токенов с помощью обучаемого поля скоростей, сохраняя базовую модель замороженной.

Разработка позволяет плавно регулировать интенсивность стиля и подавлять нежелательные визуальные концепции, с чем обычные текстовые промпты часто не справляются. В тестах AcFlow показал баланс стиля и контента на уровне 0,5365/0,2860 по сравнению с 0,4397/0,2684 у лучших существующих аналогов.

Код проекта опубликован разработчиками в открытом доступе на платформе GitHub.