UNIC-Adapter: Unified Image-instruction Adapter with Multi-modal Transformer for Image Generation

TL;DR

This paper introduces UNIC-Adapter, a unified multi-modal transformer framework that enables flexible, controllable image generation from diverse inputs without needing multiple specialized models.

Contribution

The paper presents a novel unified adapter built on a multi-modal diffusion transformer, allowing controllable image synthesis across various conditions within a single model.

Findings

Effective control over pixel-level layouts and styles

Versatile performance across multiple image generation tasks

Outperforms specialized models in controllability

Abstract

Recently, text-to-image generation models have achieved remarkable advancements, particularly with diffusion models facilitating high-quality image synthesis from textual descriptions. However, these models often struggle with achieving precise control over pixel-level layouts, object appearances, and global styles when using text prompts alone. To mitigate this issue, previous works introduce conditional images as auxiliary inputs for image generation, enhancing control but typically necessitating specialized models tailored to different types of reference inputs. In this paper, we explore a new approach to unify controllable generation within a single framework. Specifically, we propose the unified image-instruction adapter (UNIC-Adapter) built on the Multi-Modal-Diffusion Transformer architecture, to enable flexible and controllable generation across diverse conditions without the need for multiple specialized models. Our UNIC-Adapter effectively extracts multi-modal instruction information by incorporating both conditional images and task instructions, injecting this information into the image generation process through a cross-attention mechanism enhanced by Rotary Position Embedding. Experimental results across a variety of tasks, including pixel-level spatial control, subject-driven image generation, and style-image-based image synthesis, demonstrate the effectiveness of our UNIC-Adapter in unified controllable image generation.