RelaCtrl: Relevance-Guided Efficient Control for Diffusion Transformers

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Cao, Ke, Wang, Jing, Ma, Ao, Feng, Jiasong, He, Xuanhua, Ling, Run, Liu, Haowei, Lu, Jian, Feng, Wei, Wang, Haozhe, Pei, Hongjuan, Shao, Yihua, Zhang, Zhanjie, Zhang, Jie
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910033256644608
author Cao, Ke
Wang, Jing
Ma, Ao
Feng, Jiasong
He, Xuanhua
Ling, Run
Liu, Haowei
Lu, Jian
Feng, Wei
Wang, Haozhe
Pei, Hongjuan
Shao, Yihua
Zhang, Zhanjie
Zhang, Jie
author_facet Cao, Ke
Wang, Jing
Ma, Ao
Feng, Jiasong
He, Xuanhua
Ling, Run
Liu, Haowei
Lu, Jian
Feng, Wei
Wang, Haozhe
Pei, Hongjuan
Shao, Yihua
Zhang, Zhanjie
Zhang, Jie
contents The Diffusion Transformer plays a pivotal role in advancing text-to-image and text-to-video generation, owing primarily to its inherent scalability. However, existing controlled diffusion transformer methods incur significant parameter and computational overheads and suffer from inefficient resource allocation due to their failure to account for the varying relevance of control information across different transformer layers. To address this, we propose the Relevance-Guided Efficient Controllable Generation framework, RelaCtrl, enabling efficient and resource-optimized integration of control signals into the Diffusion Transformer. First, we evaluate the relevance of each layer in the Diffusion Transformer to the control information by assessing the "ControlNet Relevance Score"-i.e., the impact of skipping each control layer on both the quality of generation and the control effectiveness during inference. Based on the strength of the relevance, we then tailor the positioning, parameter scale, and modeling capacity of the control layers to reduce unnecessary parameters and redundant computations. Additionally, to further improve efficiency, we replace the self-attention and FFN in the commonly used copy block with the carefully designed Two-Dimensional Shuffle Mixer (TDSM), enabling efficient implementation of both the token mixer and channel mixer. Both qualitative and quantitative experimental results demonstrate that our approach achieves superior performance with only 15% of the parameters and computational complexity compared to PixArt-delta.
format Preprint
id arxiv_https___arxiv_org_abs_2502_14377
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle RelaCtrl: Relevance-Guided Efficient Control for Diffusion Transformers
Cao, Ke
Wang, Jing
Ma, Ao
Feng, Jiasong
He, Xuanhua
Ling, Run
Liu, Haowei
Lu, Jian
Feng, Wei
Wang, Haozhe
Pei, Hongjuan
Shao, Yihua
Zhang, Zhanjie
Zhang, Jie
Computer Vision and Pattern Recognition
The Diffusion Transformer plays a pivotal role in advancing text-to-image and text-to-video generation, owing primarily to its inherent scalability. However, existing controlled diffusion transformer methods incur significant parameter and computational overheads and suffer from inefficient resource allocation due to their failure to account for the varying relevance of control information across different transformer layers. To address this, we propose the Relevance-Guided Efficient Controllable Generation framework, RelaCtrl, enabling efficient and resource-optimized integration of control signals into the Diffusion Transformer. First, we evaluate the relevance of each layer in the Diffusion Transformer to the control information by assessing the "ControlNet Relevance Score"-i.e., the impact of skipping each control layer on both the quality of generation and the control effectiveness during inference. Based on the strength of the relevance, we then tailor the positioning, parameter scale, and modeling capacity of the control layers to reduce unnecessary parameters and redundant computations. Additionally, to further improve efficiency, we replace the self-attention and FFN in the commonly used copy block with the carefully designed Two-Dimensional Shuffle Mixer (TDSM), enabling efficient implementation of both the token mixer and channel mixer. Both qualitative and quantitative experimental results demonstrate that our approach achieves superior performance with only 15% of the parameters and computational complexity compared to PixArt-delta.
title RelaCtrl: Relevance-Guided Efficient Control for Diffusion Transformers
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2502.14377