English

OminiControl: Minimal and Universal Control for Diffusion Transformer

Computer Vision and Pattern Recognition 2025-07-08 v6 Artificial Intelligence Machine Learning

Abstract

We present OminiControl, a novel approach that rethinks how image conditions are integrated into Diffusion Transformer (DiT) architectures. Current image conditioning methods either introduce substantial parameter overhead or handle only specific control tasks effectively, limiting their practical versatility. OminiControl addresses these limitations through three key innovations: (1) a minimal architectural design that leverages the DiT's own VAE encoder and transformer blocks, requiring just 0.1% additional parameters; (2) a unified sequence processing strategy that combines condition tokens with image tokens for flexible token interactions; and (3) a dynamic position encoding mechanism that adapts to both spatially-aligned and non-aligned control tasks. Our extensive experiments show that this streamlined approach not only matches but surpasses the performance of specialized methods across multiple conditioning tasks. To overcome data limitations in subject-driven generation, we also introduce Subjects200K, a large-scale dataset of identity-consistent image pairs synthesized using DiT models themselves. This work demonstrates that effective image control can be achieved without architectural complexity, opening new possibilities for efficient and versatile image generation systems.

Keywords

Cite

@article{arxiv.2411.15098,
  title  = {OminiControl: Minimal and Universal Control for Diffusion Transformer},
  author = {Zhenxiong Tan and Songhua Liu and Xingyi Yang and Qiaochu Xue and Xinchao Wang},
  journal= {arXiv preprint arXiv:2411.15098},
  year   = {2025}
}

Comments

Accepted to ICCV 2025

R2 v1 2026-06-28T20:09:16.174Z