UniCombine: фреймворк для генерации с произвольной комбинацией условий — текстовых подсказок, пространственных карт и изображений объектов 132

小编 ·

UniCombine: фреймворк для генерации с произвольной комбинацией условий — текстовых подсказок, пространственных карт и изображений объектов

UniCombine — это фреймворк управляемой генерации с несколькими условиями, совместно представленный Фуданьским университетом, лабораторией Tencent Youtu и другими учреждениями. Основанный на архитектуре Diffusion Transformer (DiT), он способен обрабатывать задачи генерации с произвольными комбинациями условий управления, такими как текстовые подсказки, пространственные карты и изображения объектов.

UniCombine: фреймворк для генерации с произвольной комбинацией условий — текстовых подсказок, пространственных карт и изображений объектов

Что такое UniCombine?

UniCombine — это фреймворк управляемой генерации с множеством условий, совместно разработанный Фуданьским университетом, лабораторией Tencent YouTu и другими организациями. Он основан на архитектуре Diffusion Transformer (DiT) и способен обрабатывать задачи комбинированной генерации с произвольными условиями управления, такими как текстовые подсказки, пространственные карты и изображения субъектов.

Основные функции UniCombine

UniCombine может обрабатывать любые комбинации условий, включая текстовые подсказки (text prompts), пространственные карты (spatial maps) и изображения субъектов (subject images), обеспечивая при этом согласованность результатов генерации со всеми входными условиями. Например, он может выполнять следующие задачи:

  • Вставка субъекта (Subject-Insertion): вставка указанного субъекта в фоновое изображение.

  • Пространственное выравнивание субъекта (Subject-Spatial): настройка расположения субъекта в соответствии с пространственной картой.

  • Генерация с множеством пространственных условий (Multi-Spatial): генерация изображений с использованием нескольких пространственных условий.

Техническая архитектура UniCombine

Архитектура UniCombine основана на MMDiT (Masked Multi-Modal Diffusion Transformer) и включает следующие механизмы:

  • Механизм внимания Conditional MMDiT (Conditional MMDiT Attention): используется для обработки последовательностей с множеством условий и обеспечивает эффективное слияние признаков из разных условий.

  • Модуль переключения LoRA (LoRA Switching Module): управляет несколькими ветвями условий и динамически активирует предварительно обученные условные модули LoRA в зависимости от типа условия.

  • Обучаемый модуль шумоподавления LoRA (Denoising-LoRA Module): в обучающей версии дополнительно повышает производительность за счёт обучения этого модуля.

UniCombine框架:支持文本提示、空间映射和主体图像的任意组合生成.webp

Особенности и преимущества UniCombine

  • Универсальность: способен работать с произвольными комбинациями условий, подходит для множества задач генерации.

  • Гибкость: предоставляет две версии — без обучения (training-free) и с обучением (training-based).

  • Высокая производительность: демонстрирует выдающиеся результаты в задачах генерации с множеством условий, достигая самых современных показателей.

Установка и использование UniCombine

Исходный код UniCombine открыт, шаги по установке следующие:

conda create -n unicombine python=3.12
conda activate unicombine
pip install -r requirements.txt

Кроме того, из-за ограничений библиотеки diffusers необходимо вручную обновить код.

Сценарии применения UniCombine

  • Демонстрация товаров и генерация текстов: генерация изображений товаров в различных стилях и соответствующих текстов на основе текста и изображений субъектов.

  • Персональные рекомендации: создание индивидуальных страниц демонстрации товаров с учётом предпочтений пользователей.

  • Виртуальная примерка: генерация эффектов примерки с использованием виртуального образа пользователя для улучшения качества покупок.

  • Реклама и вовлекающий контент: генерация изображений и видеосценариев для повышения разнообразия контента.

  • Мультимодальный поиск: оптимизация точности результатов поиска на основе характеристик товаров.

Репозиторий на GitHub:https://github.com/Xuan-World/UniCombinearXiv 

Статья:https://arxiv.org/html/2503.09277v1