English

Rethinking Multi-Condition DiTs: Eliminating Redundant Attention via Position-Alignment and Keyword-Scoping

Computer Vision and Pattern Recognition 2026-02-09 v1 Artificial Intelligence Multimedia

Abstract

While modern text-to-image models excel at prompt-based generation, they often lack the fine-grained control necessary for specific user requirements like spatial layouts or subject appearances. Multi-condition control addresses this, yet its integration into Diffusion Transformers (DiTs) is bottlenecked by the conventional ``concatenate-and-attend'' strategy, which suffers from quadratic computational and memory overhead as the number of conditions scales. Our analysis reveals that much of this cross-modal interaction is spatially or semantically redundant. To this end, we propose Position-aligned and Keyword-scoped Attention (PKA), a highly efficient framework designed to eliminate these redundancies. Specifically, Position-Aligned Attention (PAA) linearizes spatial control by enforcing localized patch alignment, while Keyword-Scoped Attention (KSA) prunes irrelevant subject-driven interactions via semantic-aware masking. To facilitate efficient learning, we further introduce a Conditional Sensitivity-Aware Sampling (CSAS) strategy that reweights the training objective towards critical denoising phases, drastically accelerating convergence and enhancing conditional fidelity. Empirically, PKA delivers a 10.0×\times inference speedup and a 5.1×\times VRAM saving, providing a scalable and resource-friendly solution for high-fidelity multi-conditioned generation.

Keywords

Cite

@article{arxiv.2602.06850,
  title  = {Rethinking Multi-Condition DiTs: Eliminating Redundant Attention via Position-Alignment and Keyword-Scoping},
  author = {Chao Zhou and Tianyi Wei and Yiling Chen and Wenbo Zhou and Nenghai Yu},
  journal= {arXiv preprint arXiv:2602.06850},
  year   = {2026}
}
R2 v1 2026-07-01T10:24:43.660Z