生成模型与扩散 · CVPR2026 ·2026.06.03

异构去中心化扩散模型

Heterogeneous Decentralized Diffusion Models

Zhiying Jiang、Raihan Seraj、Marcos Villagra、Bidhan Roy

与你有关的部分

把 DDPM 与流匹配两种目标在去中心化多专家训练中混用、推理时统一,并给出 ImageNet-DDPM 到流匹配的权重转换方案,为算力受限的视觉生成研究提供了可复用的低成本训练路径。

原文依据 a heterogeneous decentralized training paradigm that allows experts to use different objectives (DDPM and Flow Matching), unified at inference time without any retraining
可迁移到 文本到图像生成扩散模型训练图像生成

摘要译文

由模型翻译 · 术语统一 · 请对照原文核对关键结论

训练前沿规模的扩散模型往往需要大量计算资源,且这些资源集中在紧耦合的集群中,使得只有资源雄厚的机构才能参与。尽管去中心化扩散模型(DDM)能够孤立地训练多个专家,但现有方法需要 1176 GPU 天,且所有专家必须采用同质的训练目标。我们提出一个高效框架,在支持异构训练目标的同时大幅降低资源需求。我们的方法结合了三项关键贡献:(1) 一种异构去中心化训练范式,允许专家使用不同的目标(DDPM 与流匹配),并在推理时统一,无需任何重新训练;(2) 从 ImageNet-DDPM 到流匹配目标的预训练检查点转换,加速收敛,并使得无需针对特定目标进行预训练即可完成初始化;(3) PixArt-α 高效的 AdaLN-Single 架构,在保持质量的同时减少参数量。在 LAION-Aesthetics 上的实验表明,相对于此前 DDM 工作报告的训练规模,我们的方法将计算量降低 16 倍、数据量降低 14 倍。在对齐的推理设置下,我们的异构配置取得了比同质基线更好的弗雷歇初始距离和更高的提示内多样性。通过消除同步要求并支持 DDPM/流匹配混合目标,我们的框架使得仅需 24–48GB 显存的单 GPU 贡献者也能参与去中心化生成模型训练。

英文原文摘要 1546 字符

Training frontier-scale diffusion models often requires substantial computational resources concentrated in tightly-coupled clusters, limiting participation to well-resourced institutions. While Decentralized Diffusion Models (DDM) enable training multiple experts in isolation, existing approaches require 1176 GPU-days and homogeneous training objectives across all experts. We present an efficient framework that dramatically reduces resource requirements while supporting heterogeneous training objectives. Our approach combines three key contributions: (1) a heterogeneous decentralized training paradigm that allows experts to use different objectives (DDPM and Flow Matching), unified at inference time without any retraining; (2) pretrained checkpoint conversion from ImageNet-DDPM to Flow Matching objectives, accelerating convergence and enabling initialization without objective-specific pretraining; and (3) PixArt-\alpha's efficient AdaLN-Single architecture, reducing parameters while maintaining quality. Experiments on LAION-Aesthetics show that, relative to the training scale reported for prior DDM work, our approach reduces the compute by 16xand data by 14x. Under aligned inference settings, our heterogeneous configuration achieves better FID and higher intra-prompt diversity than the homogeneous baseline. By eliminating synchronization requirements and enabling mixed DDPM/FM objectives, our framework makes decentralized generative model training accessible to contributors with single GPUs requiring only 24--48GB VRAM.

标签

会议 CVPR2026
页码 2391-2400
Bibkey
来源 CVF Open Access