UniCombine: marco de generación controlable con combinación arbitraria de indicaciones de texto, mapeo espacial e imágenes de sujeto 133

小编 ·

UniCombine: marco de generación controlable con combinación arbitraria de indicaciones de texto, mapeo espacial e imágenes de sujeto

UniCombine es un marco de generación controlable multi-condicional lanzado conjuntamente por la Universidad de Fudan, Tencent Youtu Lab y otras instituciones. Basado en la arquitectura Diffusion Transformer (DiT), puede manejar tareas de generación combinando condiciones de control arbitrarias como indicaciones de texto, mapeos espaciales e imágenes de sujeto.

UniCombine: marco de generación controlable con combinación arbitraria de indicaciones de texto, mapeo espacial e imágenes de sujeto

¿Qué es UniCombine?

UniCombine es un marco de generación controlable multicondición presentado conjuntamente por la Universidad de Fudan, el Tencent Youtu Lab y otras instituciones. Basado en la arquitectura del Diffusion Transformer (DiT), es capaz de manejar tareas de generación combinadas de condiciones de control arbitrarias, como indicaciones de texto, mapas espaciales e imágenes del sujeto.

Funciones principales de UniCombine

UniCombine puede manejar combinaciones arbitrarias de condiciones, sin limitarse a indicaciones de texto (text prompts), mapas espaciales (spatial maps) e imágenes del sujeto (subject images), y garantiza que los resultados generados sean coherentes con todas las condiciones de entrada. Por ejemplo, puede realizar las siguientes tareas:

  • Inserción del sujeto (Subject-Insertion): insertar un sujeto específico en una imagen de fondo.

  • Alineación espacial del sujeto (Subject-Spatial): ajustar la disposición del sujeto según un mapa espacial.

  • Generación con múltiples condiciones espaciales (Multi-Spatial): generar imágenes combinando varias condiciones espaciales.

Arquitectura técnica de UniCombine

La arquitectura de UniCombine se basa en MMDiT (Masked Multi-Modal Diffusion Transformer) e introduce los siguientes mecanismos:

  • Mecanismo de atención MMDiT condicional (Conditional MMDiT Attention): se utiliza para procesar secuencias de entrada multicondición y fusionar eficazmente las características de las distintas condiciones.

  • Módulo de conmutación LoRA (LoRA Switching Module): gestiona múltiples ramas de condiciones y activa dinámicamente módulos LoRA condicionales preentrenados según el tipo de condición.

  • Módulo Denoising-LoRA entrenable (Denoising-LoRA Module): en la versión entrenada, el rendimiento se mejora aún más mediante el entrenamiento de este módulo.

Marco UniCombine: admite la generación de combinaciones arbitrarias de indicaciones de texto, mapeo espacial e imágenes del sujeto.webp

Características y ventajas de UniCombine

  • Versatilidad: capaz de manejar combinaciones arbitrarias de condiciones, aplicable a diversas tareas de generación.

  • Flexibilidad: ofrece dos versiones, una sin entrenamiento (training-free) y otra basada en entrenamiento (training-based).

  • Alto rendimiento: sobresale en tareas de generación multicondición, alcanzando un rendimiento de vanguardia.

Instalación y uso de UniCombine

El código de UniCombine es de código abierto y los pasos de instalación son los siguientes:

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

Además, debido a las limitaciones de la biblioteca diffusers, es necesario actualizar el código manualmente.

Escenarios de aplicación de UniCombine

  • Exhibición de productos y generación de copys: generar imágenes de productos en múltiples estilos y copys coincidentes a partir de texto e imágenes del sujeto.

  • Recomendación personalizada: generar páginas de exhibición de productos personalizadas combinando las preferencias del usuario.

  • Prueba virtual: generar efectos de prueba combinando el avatar virtual del usuario, mejorando la experiencia de compra.

  • Publicidad y contenido de recomendación: generar imágenes, textos y guiones de vídeos, aumentando la riqueza del contenido.

  • Búsqueda multimodal: optimizar la precisión de los resultados de búsqueda combinando las características del producto.

Repositorio de GitHub:https://github.com/Xuan-World/UniCombinearXiv

Artículo:https://arxiv.org/html/2503.09277v1