paper-with-me

홈 › Papers

OminiControl2: Efficient Conditioning for Diffusion Transformers

2025-03-11 · Zhenxiong Tan, Qiaochu Xue, Xingyi Yang, Songhua Liu, Xinchao Wang

Fine-grained control of text-to-image diffusion transformer models (DiT) remains a critical challenge for practical deployment. While recent advances such as OminiControl and others have enabled a controllable generation of diverse control signals, these methods face significant computational inefficiency when handling long conditional inputs. We present OminiControl2, an efficient framework that achieves efficient image-conditional image generation. OminiControl2 introduces two key innovations: (1) a dynamic compression strategy that streamlines conditional inputs by preserving only the most semantically relevant tokens during generation, and (2) a conditional feature reuse mechanism that computes condition token features only once and reuses them across denoising steps. These architectural improvements preserve the original framework's parameter efficiency and multi-modal versatility while dramatically reducing computational costs. Our experiments demonstrate that OminiControl2 reduces conditional processing overhead by over 90% compared to its predecessor, achieving an overall 5.9$\times$ speedup in multi-conditional generation scenarios. This efficiency enables the practical implementation of complex, multi-modal control for high-quality image synthesis with DiT models.

📄 PDF Abstract BibTeX arXiv:2503.08280

Code (1)

Yuanshi9815/OminiControl 공식 구현 pytorch

Tasks

Conditional Image GenerationDenoisingImage Generation

Methods 이 논문이 사용한 방법론

Diffusion Diffusion models generate samples by gradually removing noise from a signal, and their training objective can be expressed as a reweighted variational lower-bound…

Similar Papers 제목 키워드 기반

OminiControl: Minimal and Universal Control for Diffusion Transformer

2024-11-22 · Zhenxiong Tan, Songhua Liu, Xingyi Yang, Qiaochu Xue 외

In this paper, we introduce OminiControl, a highly versatile and parameter-efficient framework that integrates image conditions into pre-trained Diffusion Transformer (DiT) models. At its core, OminiControl leverages a p…

Gated Condition Injection without Multimodal Attention: Towards Controllable Linear-Attention Transformers

2026-03-29 · Yuhe Liu, Zhenxiong Tan, Yujia Hu, Songhua Liu 외 arxiv

Recent advances in diffusion-based controllable visual generation have led to remarkable improvements in image quality. However, these powerful models are typically deployed on cloud servers due to their large computatio…

Rethinking Global Text Conditioning in Diffusion Transformers

2026-02-09 · Nikita Starodubcev, Daniil Pakhomov, Zongze Wu, Ilya Drobyshevskiy 외 arxiv

Diffusion transformers typically incorporate textual information via attention layers and a modulation mechanism using a pooled text embedding. Nevertheless, recent approaches discard modulation-based text conditioning a…

Video GenerationImage Editing

Image-to-Image Translation with Diffusion Transformers and CLIP-Based Image Conditioning

2025-05-21 · Qiang Zhu, Kuan Lu, Menghao Huo, Yuxiao Li

Image-to-image translation aims to learn a mapping between a source and a target domain, enabling tasks such as style transfer, appearance transformation, and domain adaptation. In this work, we explore a diffusion-based…

DenoisingDomain AdaptationImage-to-Image TranslationStyle Transfer+1

GenTron: Diffusion Transformers for Image and Video Generation

2023-12-07 · CVPR 2024 1 · Shoufa Chen, Mengmeng Xu, Jiawei Ren, Yuren Cong 외

In this study, we explore Transformer-based diffusion models for image and video generation. Despite the dominance of Transformer architectures in various fields due to their flexibility and scalability, the visual gener…

Text-to-Video GenerationVideo Generation