用轻量级适配器让扩散模型更好理解文本和结构输入
Efficient Text-Guided Convolutional Adapter for the Diffusion Model
- 通过跨注意力机制融合文本与结构输入,实现多模态条件生成
- 仅增加800万参数即超越基线,轻量版减少1800万参数仍达顶尖性能
- 适合需要高效可控图像生成的开发者与研究者
我们提出Nexus Adapters,一种新型文本引导的高效适配器,用于结构保持的条件图像生成(SPCG)。现有方法在使用基模型进行提示条件化、适配器处理结构输入(如草图或深度图)时,存在效率低下问题,常需与基架构相当的参数量,而扩散模型训练成本高,参数翻倍极不经济。且适配器对输入提示无感知,仅优化结构输入。为此,我们设计了两个高效适配器:Nexus Prime与Slim,均受提示与结构输入共同引导。每个Nexus Block引入交叉注意力机制,实现丰富的多模态条件化,使适配器同时理解提示与结构。实验表明,Nexus Prime仅增加800万参数即显著提升性能,优于基线T2I-Adapter;Nexus Slim则比T2I-Adapter减少1800万参数,仍达到当前最优结果。代码已开源。
原文摘要 · Abstract (English)
We introduce the Nexus Adapters, novel text-guided efficient adapters to the diffusion-based framework for the Structure Preserving Conditional Generation (SPCG). Recently, structure-preserving methods have achieved promising results in conditional image generation by using a base model for prompt conditioning and an adapter for structure input, such as sketches or depth maps. These approaches are highly inefficient and sometimes require equal parameters in the adapter compared to the base architecture. It is not always possible to train the model since the diffusion model is itself costly, and doubling the parameter is highly inefficient. In these approaches, the adapter is not aware of the input prompt; therefore, it is optimal only for the structural input but not for the input prompt. To overcome the above challenges, we proposed two efficient adapters, Nexus Prime and Slim, which are guided by prompts and structural inputs. Each Nexus Block incorporates cross-attention mechanisms to enable rich multimodal conditioning. Therefore, the proposed adapter has a better understanding of the input prompt while preserving the structure. We conducted extensive experiments on the proposed models and demonstrated that the Nexus Prime adapter significantly enhances performance, requiring only 8M additional parameters compared to the baseline, T2I-Adapter. Furthermore, we also introduced a lightweight Nexus Slim adapter with 18M fewer parameters than the T2I-Adapter, which still achieved state-of-the-art results. Code: https://github.com/arya-domain/Nexus-Adapters
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。