让视觉自回归模型实现高效精准的文本生成图像控制。
ScaleWeaver: Weaving Efficient Controllable T2I Generation with Multi-Scale Reference Attention
- 引入参考注意力模块,减少冗余计算,稳定控制信号注入。
- 仅用少量参数实现高质量可控生成,推理效率优于扩散模型。
- 适合追求高效可控图像生成的研究者与应用开发者。
基于视觉自回归(VAR)模型的文本到图像生成近年来在生成质量和推理效率方面取得显著进展。尽管扩散模型已探索多种控制机制,但VAR范式中的精确灵活控制仍研究不足。本文提出ScaleWeaver框架,通过参数高效微调,在先进VAR模型上实现高保真、可控制的图像生成。核心是改进的MMDiT块,集成提出的参考注意力模块,高效融入条件信息。不同于MM注意力,该模块舍弃图像→条件的无用注意力,降低计算开销并稳定控制注入。同时,通过策略性参数复用,仅引入少量参数即可处理控制信息,并使用零初始化线性投影确保控制信号有效融合而不破坏基础模型生成能力。大量实验表明,ScaleWeaver在保持高质量生成的同时实现精准控制,且效率优于基于扩散的方法,为视觉自回归范式下的可控文本到图像生成提供实用有效方案。代码与模型将公开。
原文摘要 · Abstract (English)
Text-to-image generation with visual autoregressive~(VAR) models has recently achieved impressive advances in generation fidelity and inference efficiency. While control mechanisms have been explored for diffusion models, enabling precise and flexible control within VAR paradigm remains underexplored. To bridge this critical gap, in this paper, we introduce ScaleWeaver, a novel framework designed to achieve high-fidelity, controllable generation upon advanced VAR models through parameter-efficient fine-tuning. The core module in ScaleWeaver is the improved MMDiT block with the proposed Reference Attention module, which efficiently and effectively incorporates conditional information. Different from MM Attention, the proposed Reference Attention module discards the unnecessary attention from image$\rightarrow$condition, reducing computational cost while stabilizing control injection. Besides, it strategically emphasizes parameter reuse, leveraging the capability of the VAR backbone itself with a few introduced parameters to process control information, and equipping a zero-initialized linear projection to ensure that control signals are incorporated effectively without disrupting the generative capability of the base model. Extensive experiments show that ScaleWeaver delivers high-quality generation and precise control while attaining superior efficiency over diffusion-based methods, making ScaleWeaver a practical and effective solution for controllable text-to-image generation within the visual autoregressive paradigm. Code and models will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。