← 返回 2026-02-19

面向扩散模型的高效文本引导卷积适配器 Efficient Text-Guided Convolutional Adapter for the Diffusion Model

Aryan Das, Koushik Biswas, Swalpa Kumar Roy, Badri Narayana Patro, Vinay Kumar Verma 📅 2026-02-16 👍 8 2026-07-13 08:35
图像生成 扩散模型 条件生成 跨注意力 适配器

提出Nexus适配器,通过跨注意力融合文本与结构,实现高效条件生成

前置知识

扩散模型(Diffusion Models)

扩散模型是一类生成模型,通过逐步向数据添加高斯噪声(前向过程),然后学习逐步去噪(反向过程)来生成新样本。在本文中,模型使用预训练的 Latent Diffusion Model (LDM) 作为生成骨干,通过 UNet 架构预测噪声残差 $\epsilon_\theta(X_t, t, c)$,损失函数为 $\mathcal{L} = \mathbb{E}_{t,X_0,\epsilon}[\|\epsilon - \epsilon_\theta(X_t, t)\|^2]$。在推理时,模型从纯高斯噪声出发,经过 T 步(本文设为 35 步)去噪采样,最终生成图像。

本文提出的 Nexus 适配器是在预训练扩散模型基础上添加的条件引导模块,理解扩散模型的基本原理(前向/反向过程、UNet 架构、交叉注意力机制)是理解本文方法的基础。

交叉注意力(Cross-Attention)

交叉注意力是一种注意力机制,允许一个序列的查询(Query)与另一个序列的键值对(Key-Value)进行交互。公式为 $\text{CrossAttention}(Q, K, V) = \text{SoftMax}\left(\frac{QK^\top}{\sqrt{d}}\right) \cdot V$,其中 $d$ 是嵌入维度。在扩散模型中,交叉注意力用于将文本条件 $c$(CLIP 编码的文本嵌入)作为键值对注入 UNet 的每一层,实现文本语义对生成过程的引导。

Nexus 适配器的核心创新就是在适配器内部引入交叉注意力机制,使适配器能够感知输入文本提示,实现多模态条件融合。理解交叉注意力的工作原理对把握本文的创新点至关重要。

ControlNet 和 T2I-Adapter

ControlNet 是一种为扩散模型添加空间条件控制的方法,它在冻结的 UNet 骨干旁引入一个可训练的控制网络分支,参数量几乎与基础模型编码器相当(361.28M 参数)。T2I-Adapter 则采用轻量级适配器模块注入结构信息,参数量为 77.37M。两者都是将结构条件(如边缘图、深度图、分割图)注入扩散过程的代表性工作,但它们共同的局限是:适配器不感知输入文本提示,仅独立处理结构输入。

本文提出的 Nexus 适配器正是针对 ControlNet 和 T2I-Adapter 的局限性进行改进,理解这些基线方法的设计思路和不足,才能理解 Nexus 适配器的创新意义。

CLIP 文本编码器

CLIP(Contrastive Language-Image Pre-training)是一种视觉-语言预训练模型,其文本编码器将自然语言文本映射到高维语义空间。在本文中,CLIP 文本编码器被冻结使用,将输入文本提示 $P$ 编码为全局语义嵌入 $T = T_{\text{CLIP}}(P) \in \mathbb{R}^{n \times d}$,其中 $n$ 是 token 数量,$d$ 是嵌入维度。这些文本嵌入既用于 UNet 的交叉注意力,也被 Nexus 适配器的交叉注意力层复用。

Nexus 适配器的关键设计是复用冻结的 CLIP 文本嵌入,在适配器内部通过交叉注意力实现文本语义对结构特征的调制。理解 CLIP 编码器的作用有助于把握适配器如何实现文本-结构融合。

深度可分离卷积(Depthwise Separable Convolution)

深度可分离卷积将标准卷积分解为两步:深度卷积(Depthwise Convolution,对每个输入通道独立应用卷积核)和逐点卷积(Pointwise Convolution,使用 $1 \times 1$ 卷积进行通道融合)。相比标准 $3 \times 3$ 卷积,深度可分离卷积将参数量和计算量大幅降低。在本文中,Nexus Slim 适配器采用分组数为 2 的深度卷积($\text{Conv}_{D}^{3 \times 3}$),显著减少参数量。

Nexus Slim 适配器的核心效率设计就是采用深度可分离卷积替代标准卷积,理解这一技术对把握 Nexus Slim 如何实现参数效率至关重要。

研究动机

当前条件图像生成方法存在三个核心问题。首先,参数效率低下:ControlNet 和 ControlNet++ 需要 361.28M 参数和 116.61 GFlops,几乎与基础 UNet 编码器相当,这在资源受限环境中难以部署。CtrLoRA 虽然使用 LoRA 将可训练参数降至 37M,但 361.28M 的冻结参数仍然活跃,GPU 内存占用依然很高,Flops 达到 135.15G。其次,适配器与文本提示脱节:现有方法如 T2I-Adapter 的适配器仅接收结构信号(边缘图、深度图、分割图)作为输入,完全不感知文本提示的语义内容,导致适配器只能优化结构保留,无法利用文本语义进行更有效的条件引导。第三,训练成本高昂:ControlNet 训练 200K 步需要约 129 小时,推理每张图需 38ms;CtrLoRA 训练时间更长达 150 小时,推理每张图 45ms。这些问题使得现有方法在实际应用中面临参数冗余、语义不一致和计算开销大的困境。

本文的目标是本文旨在提出一种既能保留结构条件信息、又能感知文本语义提示的轻量级适配器框架。具体目标包括:第一,设计参数高效的适配器架构,使其总参数量和计算量显著低于 ControlNet 和 T2I-Adapter;第二,在适配器内部引入跨注意力机制,使适配器能够同时利用结构输入和文本提示进行条件引导,实现文本-结构的多模态融合;第三,保持或超越现有基线的生成质量和语义对齐能力,同时大幅降低训练和推理成本。最终目标是提供一种实用的、可部署的条件生成方案,适用于资源受限的实际应用场景。

与已有工作不同的是,本文的独特切入角度在于:现有方法将适配器视为独立于文本提示的结构条件处理器,而本文首次提出「条件聚焦对齐」(conditional-focused alignment)的概念,在适配器内部通过交叉注意力机制同时关注文本提示和结构输入。这一视角的转变带来了根本性的架构差异:ControlNet 等方法在推理时每一步去噪都依赖适配器的逐步预测,存在「去噪耦合」(denoising-coupled)问题,中间预测的误导可能级联传播;而 Nexus 适配器提供的是「全局、恒定的条件引导」(global, constant guidance),在生成过程中保持一致的语义先验,消除了去噪依赖性,实现了更稳定的文本-结构融合。这种从逐步条件到全局引导的范式转换,是本文区别于所有现有工作的核心创新点。

核心方法

本文的方法整体思路可以概括为:在冻结的预训练 Stable Diffusion 骨干之上,设计一个轻量级的 Nexus 适配器,该适配器同时接收结构条件输入(边缘图、深度图、分割图等)和文本提示,通过内部的跨注意力机制实现多模态融合,然后将融合后的特征注入 UNet 骨干。技术路线分为三个层次:第一层是条件图像处理,适配器对输入的结构条件图进行像素反洗牌(Pixel Unshuffle)下采样至 $64 \times 64$,然后通过 4 个层级变换块进行多尺度特征提取,每一层通过步长卷积实现空间下采样;第二层是语义对齐,复用冻结的 CLIP 文本嵌入,在每个适配器块内部通过交叉注意力机制让视觉特征对文本语义进行注意;第三层是特征融合,将适配器输出的多尺度条件特征与 UNet 编码器的对应层激活进行逐元素加法融合,引导去噪过程。整个框架中,Stable Diffusion 骨干保持冻结,仅训练适配器的少量参数。

Nexus 适配器的核心创新是「提示引导的条件对齐」(Prompt-Driven Conditional Guidance),这一设计与现有方法有本质区别。传统方法如 T2I-Adapter 的适配器仅接收结构输入,不感知文本提示,因此只能优化结构保留,无法利用语义上下文;ControlNet 虽然在 UNet 中注入文本信息,但其适配器分支本身不直接处理文本嵌入。Nexus 适配器首次在适配器内部引入跨注意力机制,使适配器能够同时感知结构输入和文本语义:在每个 Nexus 块中,视觉特征经过卷积变换后作为查询(Query),CLIP 文本嵌入作为键值对(Key-Value),通过交叉注意力实现文本语义对空间特征的自适应调制。这种设计使适配器在保留结构信息的同时,能够根据文本语义动态调整特征表示,实现更精细的文本-结构对齐。与 ControlNet 的逐步去噪依赖不同,Nexus 适配器提供的是全局恒定的条件引导,在整个生成过程中保持一致的语义先验,消除了中间预测级联传播的风险。

方法步骤详情

Nexus 适配器的完整工作流程如下。第一步是条件图像预处理:给定条件图像 $I_c \in \mathbb{R}^{3 \times H \times W}$,首先通过像素反洗牌操作将其空间分辨率下采样至 $64 \times 64$,得到初始张量 $X_L^0 \in \mathbb{R}^{C_0 \times 64 \times 64}$。第二步是多尺度特征提取:通过 $K=4$ 个层级变换块 $A_k$ 逐层提取特征,$E_k = A_k(X_L^{k-1})$,每层之间通过步长卷积 $\phi(\cdot)$ 进行空间下采样:$X_L^k = \phi(E_k)$。前三层通道维度倍增($C_k = 2 \cdot C_{k-1}$),空间尺寸减半;第四层通道维度回缩至 $C_3$,空间继续减半。第三步是块内语义对齐:每个 Nexus 块(Prime 或 Slim)接收输入特征 $X \in \mathbb{R}^{C \times H \times W}$ 和 CLIP 文本嵌入 $T \in \mathbb{R}^{n \times d}$,先通过卷积变换(Prime 用标准 $3 \times 3$ 卷积,Slim 用深度可分离卷积)提取视觉特征,再通过交叉注意力机制将文本语义注入视觉特征。以 Prime 块为例:$P_1 = \text{ReLU}(\text{Conv}_{3 \times 3}(X))$,$P_2 = \text{Conv}_{1 \times 1}(P_1)$,$P_3 = \text{ReLU}(\text{Conv}_{3 \times 3}(P_2))$,$P_4 = \text{Conv}_{1 \times 1}(P_3)$,归一化后 $F_P^n = \text{Norm}(P_4)$,然后 $F_P^{attn} = \text{CrossAttention}(F_P^n, T, T)$,$F_P = F_P^n + F_P^{attn}$。第四步是特征融合:适配器输出的多尺度特征 $E_k$ 与 UNet 编码器对应层的激活 $U_k$ 逐元素加法融合:$\hat{U}_k = U_k + E_k$,引导去噪生成。

技术新颖性

Nexus 适配器的技术新颖性体现在三个方面。第一,首次在适配器内部实现文本-结构联合条件引导:现有方法(ControlNet、T2I-Adapter、CtrLoRA)的适配器均独立于文本提示,仅处理结构信号;Nexus 适配器通过在每个块内引入交叉注意力机制,使适配器能够同时感知文本和结构,实现多模态条件融合,这是该领域的首创。第二,全局条件引导范式:ControlNet 等方法在推理时每一步去噪都依赖适配器的逐步预测,存在去噪耦合问题;Nexus 适配器将条件特征一次性融合到 UNet 中,在整个去噪过程中提供恒定的语义-结构先验,这种从逐步引导到全局引导的范式转换是架构层面的根本创新。第三,高效卷积设计:Nexus Slim 适配器采用深度可分离卷积(分组数为 2)替代标准卷积,在参数量降至 59.29M 的同时保持竞争力;Nexus Prime 适配器虽使用标准卷积,但总参数量仅 85.82M,比 T2I-Adapter 的 77.37M 略多但性能显著提升,比 ControlNet 的 361.28M 大幅减少。这种在参数效率和性能之间的灵活权衡设计,为不同资源约束场景提供了实用选择。

Architecture of our method
Figure 1: Architecture of our method
Qualitative ablation on conflicting prompts and conditional images
Figure 4: Qualitative ablation on conflicting prompts and conditional images

实验结果

本文在 COCO 2017 数据集上对四种条件模态(Canny 边缘、深度图、素描、分割图)进行了全面实验,结果表明 Nexus 适配器在效率和性能上均优于现有基线。在 CLIP Score 指标上(越高越好),Nexus Prime 在 Canny(27.33)、Depth(27.68)、Sketch(27.66)三个任务上均取得最高分,仅在 Segmentation(27.03)上略低于 ControlNet++(27.19);Nexus Slim 虽然得分低于 Prime,但在大多数任务上超越 T2I-Adapter。在 FID 指标上(越低越好),Nexus Prime 同样在 Canny(22.56)、Depth(23.91)、Sketch(24.73)三个任务上取得最优,仅在 Segmentation(25.78)上略高于 ControlNet++(25.63)。在计算效率方面,Nexus Slim 以 23.77 GFlops 和 59.29M 参数成为所有方法中最高效的,比 T2I-Adapter(29.97 GFlops,77.37M)分别低 20.7% 和 23.4%;Nexus Prime 为 33.32 GFlops 和 85.82M 参数,仍远低于 ControlNet(116.61 GFlops,361.28M)。在训练效率上,Nexus Slim 仅需约 26 小时训练(vs ControlNet 的 129 小时),推理速度约 7ms/张(vs ControlNet 的 38ms/张);Nexus Prime 训练约 37 小时,推理约 11ms/张。消融实验进一步验证了设计选择:增加分组数至 8 虽然降低参数但显著损害性能(FID 从 27.14 升至 30.49);4 块配置(完整模型)显著优于 2 块和 3 块配置;在无文本提示的条件下,Nexus 适配器表现出最小的性能退化,验证了全局条件引导的鲁棒性。

Model complexity measured by Flops and number of trainable parameters
Table 1: Model complexity measured by Flops and number of trainable parameters
Comparison of CLIP Scores across conditioning tasks
Table 2: Comparison of CLIP Scores across conditioning tasks
Comparison of FID scores across different conditioning tasks
Table 3: Comparison of FID scores across different conditioning tasks
Ablation on grouped convolution configuration in Nexus Slim Adapter
Table 4: Ablation on grouped convolution configuration in Nexus Slim Adapter
Ablation study comparing models under no-text-prompt settings
Table 5: Ablation study comparing models under no-text-prompt settings
Evaluation of models on CUB-200 dataset
Table 6: Evaluation of models on CUB-200 dataset
Estimated training and inference time across models
Table 7: Estimated training and inference time across models
Ablation on the number of adapter blocks
Table 8: Ablation on the number of adapter blocks
Qualitative comparison across four conditioning types
Figure 2: Qualitative comparison across four conditioning types
Qualitative comparison with and without prompts
Figure 3: Qualitative comparison with and without prompts
查看结构化数据
任务指标本文基线提升
Canny 边缘条件生成 CLIP Score (↑) Nexus Prime: 27.33 ControlNet: 27.11, ControlNet++: 26.79 相比 ControlNet 提升 0.22,相比 ControlNet++ 提升 0.54
Depth 深度图条件生成 CLIP Score (↑) Nexus Prime: 27.68 ControlNet++: 27.26, ControlNet: 27.21 相比 ControlNet++ 提升 0.42,相比 ControlNet 提升 0.47
Sketch 素描条件生成 CLIP Score (↑) Nexus Prime: 27.66 ControlNet++: 27.24, ControlNet: 27.08 相比 ControlNet++ 提升 0.42,相比 ControlNet 提升 0.58
Canny 边缘条件生成 FID (↓) Nexus Prime: 22.56 CtrLoRA: 22.89, ControlNet: 23.14 相比 CtrLoRA 降低 0.33,相比 ControlNet 降低 0.58
Depth 深度图条件生成 FID (↓) Nexus Prime: 23.91 ControlNet++: 25.49, Nexus Slim: 25.30 相比 ControlNet++ 降低 1.58,相比 Nexus Slim 降低 1.39
Sketch 素描条件生成 FID (↓) Nexus Prime: 24.73 ControlNet++: 24.86, UniCon: 24.91 相比 ControlNet++ 降低 0.13,相比 UniCon 降低 0.18
计算效率 参数量 (M) Nexus Slim: 59.29, Nexus Prime: 85.82 ControlNet: 361.28, T2I-Adapter: 77.37 Nexus Slim 比 ControlNet 减少 83.6%,比 T2I-Adapter 减少 23.4%
计算效率 GFlops Nexus Slim: 23.77, Nexus Prime: 33.32 ControlNet: 116.61, T2I-Adapter: 29.97 Nexus Slim 比 ControlNet 减少 79.6%,比 T2I-Adapter 减少 20.7%
训练效率 训练时间 (小时/200K步) Nexus Slim: ≈26, Nexus Prime: ≈37 ControlNet: ≈129, CtrLoRA: ≈150 Nexus Slim 比 ControlNet 减少 79.8%,比 CtrLoRA 减少 82.7%
推理效率 推理时间 (ms/张) Nexus Slim: ≈7, Nexus Prime: ≈11 ControlNet: ≈38, CtrLoRA: ≈45 Nexus Slim 比 ControlNet 减少 81.6%,比 CtrLoRA 减少 84.4%

局限与改进

本文存在以下局限性。首先,在 Segmentation 任务上,Nexus Prime 的 CLIP Score(27.03)和 FID(25.78)均略逊于 ControlNet++(CLIP 27.19,FID 25.63),ControlNet++ 通过循环一致性损失(cyclic consistency loss)在区域特定学习上具有一定优势,而 Nexus 适配器在分割任务上未能完全超越这一设计。其次,实验仅在 COCO 2017 数据集上进行评估,缺乏在其他数据集(如 LAION、CUB-200 以外的细粒度数据集)上的泛化性验证,虽然补充材料中包含了 CUB-200 的实验,但覆盖面仍然有限。第三,本文的方法依赖于冻结的 Stable Diffusion v1.5 骨干,未探索在更新的扩散架构(如 SDXL、Stable Diffusion 3、DiT 等)上的适用性,这限制了方法的通用性。第四,Nexus 适配器的跨注意力机制增加了每个块的计算开销,虽然总量仍低于 ControlNet,但相比仅用卷积的 T2I-Adapter,Nexus Prime 的 GFlops 略高(33.32 vs 29.97),在极端资源受限场景下可能不够理想。第五,论文在定性比较中未提供用户研究(User Study)的结果,仅依赖自动指标和视觉对比,可能存在评估偏差。

独立分析的弱点

本文存在几个值得关注的弱点。第一,Nexus Slim 在 Segmentation 任务上的 CLIP Score 仅 25.89,明显低于其他方法,这表明深度可分离卷积在处理分割图这类需要精细区域理解的条件时存在表达能力不足的问题,可以考虑引入动态卷积或条件感知的分组策略来增强 Slim 块的区域建模能力。第二,适配器的 4 块层级设计是固定的,未探索自适应深度或动态块选择机制,不同条件类型(边缘 vs 分割)可能需要不同深度的处理,引入早退(early exit)机制或条件感知的块选择可能进一步提升效率。第三,交叉注意力仅在适配器块内部使用,文本嵌入 $T$ 是全局语义表示,未利用位置信息或局部文本-视觉对齐,引入细粒度的文本-空间注意力可能改善细节生成质量。第四,实验中每个条件模态独立训练,未探索多条件联合训练或条件泛化能力,这限制了模型在实际应用中的灵活性。第五,论文未提供与最新方法如 ControlNet++ 的参数效率对比时的内存占用数据,仅比较了 FLOPs 和参数量,而实际部署中 GPU 内存往往是更关键的约束。

未来方向

基于本文成果,未来研究可以从以下方向展开。首先,将 Nexus 适配器扩展到更新的扩散架构,如 SDXL、Stable Diffusion 3 和 DiT(Diffusion Transformer),验证跨注意力融合策略在不同骨干架构上的通用性,特别是在 Transformer 架构中的适配器设计。其次,探索多条件融合的 Nexus 适配器,允许单个模型同时处理多种条件类型(如同时使用边缘图和深度图),通过条件类型嵌入或路由机制实现灵活的条件组合。第三,引入提示工程优化,研究不同类型的文本提示(详细描述 vs 简短关键词)对适配器条件引导效果的影响,可能通过提示增强或提示校准策略提升细粒度控制能力。第四,将 Nexus 适配器的思想扩展到视频生成和编辑任务,探索在时序维度上的条件融合机制。第五,研究适配器的可控性与可解释性,通过可视化交叉注意力权重分析适配器如何利用文本语义调制结构特征,为模型设计提供更深入的理论指导。第六,探索适配器的压缩与蒸馏技术,进一步降低推理开销,使方法更适合实时应用和边缘部署。

复现评估

本文在可复现性方面表现良好。论文提供了开源代码仓库(https://github.com/arya-domain/Nexus-Adapters),这为复现提供了重要基础。数据方面,所有实验基于 COCO 2017 数据集,这是一个公开广泛使用的标准数据集,条件图(Canny 边缘、MiDaS 深度图、边缘预测模型素描、COCO-Stuff 分割)的生成方法在论文中有明确说明。算力需求方面,论文明确指出实验在 2 块 NVIDIA A100 80GB GPU 上进行,Nexus Slim 训练约 26 小时,Nexus Prime 训练约 37 小时,这一算力需求对大多数研究机构是可接受的。训练细节方面,论文提供了完整的超参数设置:AdamW 优化器、学习率 $5 \times 10^{-6}$、batch size 2 + 梯度累积 4、混合精度训练(fp16)、200K 训练步数、35 步去噪采样、分类器自由引导权重 7.5。主要复现难度在于:需要预训练的 Stable Diffusion v1.5 权重和 CLIP 文本编码器;分组卷积的具体实现细节(分组数为 2 的深度卷积)需要仔细核对代码;像素反洗牌操作的具体实现可能因框架而异。总体而言,结合开源代码和详细的技术说明,本文的复现难度为中等偏低。