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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Yuhe, Tan, Zhenxiong, Hu, Yujia, Liu, Songhua, Wang, Xinchao
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911551201476608
author Liu, Yuhe
Tan, Zhenxiong
Hu, Yujia
Liu, Songhua
Wang, Xinchao
author_facet Liu, Yuhe
Tan, Zhenxiong
Hu, Yujia
Liu, Songhua
Wang, Xinchao
contents 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 computational demands, raising serious concerns about user data privacy. To enable secure and efficient on-device generation, we explore in this paper controllable diffusion models built upon linear attention architectures, which offer superior scalability and efficiency, even on edge devices. Yet, our experiments reveal that existing controllable generation frameworks, such as ControlNet and OminiControl, either lack the flexibility to support multiple heterogeneous condition types or suffer from slow convergence on such linear-attention models. To address these limitations, we propose a novel controllable diffusion framework tailored for linear attention backbones like SANA. The core of our method lies in a unified gated conditioning module working in a dual-path pipeline, which effectively integrates multi-type conditional inputs, such as spatially aligned and non-aligned cues. Extensive experiments on multiple tasks and benchmarks demonstrate that our approach achieves state-of-the-art controllable generation performance based on linear-attention models, surpassing existing methods in terms of fidelity and controllability.
format Preprint
id arxiv_https___arxiv_org_abs_2603_27666
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Gated Condition Injection without Multimodal Attention: Towards Controllable Linear-Attention Transformers
Liu, Yuhe
Tan, Zhenxiong
Hu, Yujia
Liu, Songhua
Wang, Xinchao
Computer Vision and Pattern Recognition
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 computational demands, raising serious concerns about user data privacy. To enable secure and efficient on-device generation, we explore in this paper controllable diffusion models built upon linear attention architectures, which offer superior scalability and efficiency, even on edge devices. Yet, our experiments reveal that existing controllable generation frameworks, such as ControlNet and OminiControl, either lack the flexibility to support multiple heterogeneous condition types or suffer from slow convergence on such linear-attention models. To address these limitations, we propose a novel controllable diffusion framework tailored for linear attention backbones like SANA. The core of our method lies in a unified gated conditioning module working in a dual-path pipeline, which effectively integrates multi-type conditional inputs, such as spatially aligned and non-aligned cues. Extensive experiments on multiple tasks and benchmarks demonstrate that our approach achieves state-of-the-art controllable generation performance based on linear-attention models, surpassing existing methods in terms of fidelity and controllability.
title Gated Condition Injection without Multimodal Attention: Towards Controllable Linear-Attention Transformers
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2603.27666