[控场AI]
· 5 分钟阅读· 2,843 字

Split and Bridge:免联合训练的多模态扩散生成新范式

Split and Bridge:免联合训练的多模态扩散生成新范式

SnB 在采样阶段桥接预训练单模态扩散模型,实现无需联合训练的多模态生成。

《Split and Bridge》提出一种自下而上的多模态生成框架 SnB:不联合训练统一模型,而是复用预训练单模态扩散模型,在采样时让各模态从同一噪声出发共享一段扩散轨迹,到达分裂点后分叉,并由轻量桥接模块维持跨模态一致性。该设计即插即用、可扩展、适配成本低,在 PolyMNIST、CelebAMask-HQ、PIE-Bench++、ROCO 等基准上取得质量与一致性的强平衡,并能零样本延伸至交错协同生成、图像编辑和风格耦合/解耦。论文将其定位为对自上而下联合训练范式的实用补充,代码已开源。

多模态生成领域长期被一种「自上而下」的思路主导:从零开始联合训练一个大模型,让它自己学出一个共享的隐空间。这种范式效果不错,但代价也很明显——训练成本高、扩展性差,而且每换一种模态组合就得重新来过。最近一篇名为《Split and Bridge: Multimodal Generation via Diffusion Bridging》的工作提出了另一种思路:既然单模态的扩散模型已经足够强大,能不能在采样阶段把它们「桥接」起来,而不是重新训练一个统一模型?

Split and Bridge 论文来源截图

从「学一个共享空间」到「走一条共享轨迹」

SnB(Split and Bridge)的核心洞察是:多模态数据(比如图像和分割掩码)本质上描述的是同一个底层结构的不同侧面。与其让模型费力学一个统一的隐变量,不如让不同模态的扩散过程先共享一段轨迹,在某个「分裂点」之后再各自分叉,进入模态专属的生成路径。

具体来说,采样时所有模态从同一个噪声起点出发,沿着共享的扩散轨迹前进到分裂点;此后由一个「桥接模块」(bridge module)引导,各模态转向自己的轨迹继续去噪。桥接模块的作用是维持分叉后各模态之间的一致性,避免它们各走各的、最后生成互不匹配的结果。

这个设计的巧妙之处在于「底向上」:不改变任何预训练单模态模型的结构和权重,只在采样阶段做文章。论文将其定位为对现有自上而下多模态方法的实用补充,而非替代。

扩散模型(Diffusion Model) 是一类生成模型,其核心思想是通过逐步加噪将数据破坏为纯噪声,再训练一个网络学习逆向的去噪过程,从而从噪声中逐步恢复出数据样本。采样即从随机噪声出发,沿学到的去噪轨迹迭代生成目标。在多模态场景中,每个模态可拥有独立的扩散过程;SnB 的关键操作是让不同模态在采样初期共享同一段去噪轨迹,这要求各模态的扩散时间步与噪声调度能够对齐。共享轨迹相当于为不同模态提供了一个临时的「公共坐标系」,使它们在分叉前已具备一定的结构对应性,从而降低后续桥接引导的难度。

桥接模块(Bridge Module) 是 SnB 中维持跨模态一致性的关键组件。当各模态在分裂点后进入专属去噪路径时,它们可能逐渐偏离彼此对应的结构关系;桥接模块的作用是在每个去噪步骤中,将其他模态的当前状态信息注入本模态的生成过程,形成一种跨模态的约束或引导。它通常是一个轻量网络,参数规模远小于底层的单模态扩散模型,因此训练成本低、可快速适配新模态组合。桥接模块的设计使 SnB 无需改动预训练模型权重,仅通过采样阶段的模块化引导即可实现模态间的对齐。

即插即用的工程价值

SnB 最直接的吸引力是工程上的轻量。由于不需要联合训练,研究者可以直接复用社区里那些已经训练好的强力单模态扩散模型,以「即插即用」的方式组合使用。这在实践中意味着几件事:

  • 可扩展性:新增一种模态时,只需接入对应的预训练模型和一个桥接模块,无需重训整个系统。
  • 适配成本低:换数据集、换任务时,底层的单模态能力不用重新学习。
  • 门槛下降:没有大规模联合训练资源的研究者也能做多模态生成实验。

论文强调这种复用「无需联合训练」(without joint training),这也是 SnB 与主流方案最本质的区别。

基准测试上的质量与一致性平衡

在实验部分,SnB 在多个标准基准上验证了效果,包括 PolyMNIST 和 CelebAMask-HQ,以及更复杂的 PIE-Bench++ 和 ROCO。论文的结论是:SnB 在样本质量与跨模态一致性之间取得了「强平衡」(strong balance)。

这里需要留意的是,多模态生成的一个经典难题正是质量与一致性的权衡——单独看每个模态都很逼真,但放在一起就对不上。SnB 通过共享轨迹加桥接引导的方式,试图在分叉后仍然锁住这种对应关系。论文没有在摘要中给出具体的量化对比数字,但其在多个复杂度递增的基准上都报告了有竞争力的表现,覆盖面从简单的合成数据一直延伸到真实场景数据集。

PolyMNIST 是多模态生成领域的经典合成基准,由 MNIST 手写数字经不同变换(如不同背景、字体风格)生成多个模态视图,用于检验模型能否在生成各模态的同时保持语义一致。CelebAMask-HQ 则是真实人脸数据集,提供人脸图像及其对应的语义分割掩码,常用于图像与掩码的联合生成与编辑任务。PIE-Bench++ 和 ROCO 进一步提升了复杂度:前者面向图像编辑评测,考察编辑前后的一致性;后者包含医学图像与文本描述配对,涉及跨模态语义对齐。这些基准从合成到真实、从简单配对到复杂语义,构成了递增的评测链条,用以全面衡量 SnB 的生成质量与跨模态一致性。

零样本下游任务的自然延伸

SnB 的另一个亮点是它能自然延伸到若干零样本下游任务:

交错协同生成(interleaved co-generation):同时生成相互对应的多个模态,比如一张图和它的分割掩码。

图像编辑:利用桥接机制在保持其他模态约束的前提下修改某一模态的内容。

风格耦合与解耦(style coupling/decoupling):控制不同模态之间风格的关联程度——既可以绑定,也可以分离。

这些任务不需要额外训练,直接受益于 SnB 的采样框架。这也从侧面说明,「共享轨迹 + 分叉」的结构本身就蕴含了跨模态对齐的能力,而不仅仅是生成逼真样本。

零样本(Zero-shot) 在此指模型未经针对特定下游任务的额外训练,仅凭已有的生成框架和预训练权重即可完成该任务。SnB 之所以能零样本延伸至交错协同生成、图像编辑和风格耦合/解耦,是因为其采样框架本身就在共享轨迹与桥接引导中编码了跨模态对应关系:改变某一模态的初始噪声或中间状态,桥接模块会自然地将变化传导至其他模态,从而支持条件生成与编辑。这种「结构即能力」的特性减少了对任务特定微调数据的依赖,也降低了部署新任务的门槛。

自下而上范式的意义

SnB 的更大价值可能在于它提出了一个值得认真对待的问题:多模态生成是否一定要从零联合训练?在预训练模型越来越强、越来越容易获取的当下,采样阶段的组合式方法提供了一条成本更低、灵活性更高的路径。论文作者明确将 SnB 定位为「补充」而非「取代」——自上而下的联合训练在需要深度模态融合的场景仍有优势,而 SnB 则在复用、扩展和快速适配方面占优。

代码已在 GitHub 开源(AhmadArrabi/SnB),对于想尝试多模态扩散生成又缺乏大规模训练资源的研究者来说,这是一个值得关注的方向。

#Split and Bridge#多模态生成#扩散模型#免联合训练#桥接模块#零样本任务#图像编辑
分享:

相关推荐