arXiv:2501.05450cs.CVcs.DC2025-01CVPR被引 14

将扩散模型训练分散到多个独立计算集群,降低对高带宽网络依赖。

Decentralized Diffusion Models

论文配图:Decentralized Diffusion Models
图 1 · 摘自论文原文
  • 分训多专家模型,各节点完全隔离,仅通过轻量路由集成
  • 240亿参数模型仅用8个GPU节点,一周内完成训练
  • 适合资源有限但需高质量生成的团队或个人研究者

大规模AI模型训练需在数千个GPU间同步梯度,依赖集中式高带宽网络,导致成本高、能耗大。本文提出去中心化扩散模型框架,通过将数据集分片并行训练多个独立专家模型,消除对集中式网络的依赖。推理时由轻量级路由器集成专家输出。实验证明,该集成方案等价于单一大模型在全量数据上的训练目标。该方法可将训练负载分布至多个“计算孤岛”,显著降低基础设施成本,并提升对局部GPU故障的容错能力。研究人员可利用按需获取的小型GPU节点实现高效训练。我们在ImageNet和LAION Aesthetics数据集上验证,去中心化扩散模型在相同算力下性能优于标准模型。最终,我们成功扩展至240亿参数规模,仅用8个独立GPU节点即在一周内完成训练。

原文摘要 · Abstract (English)

Large-scale AI model training divides work across thousands of GPUs, then synchronizes gradients across them at each step. This incurs a significant network burden that only centralized, monolithic clusters can support, driving up infrastructure costs and straining power systems. We propose Decentralized Diffusion Models, a scalable framework for distributing diffusion model training across independent clusters or datacenters by eliminating the dependence on a centralized, high-bandwidth networking fabric. Our method trains a set of expert diffusion models over partitions of the dataset, each in full isolation from one another. At inference time, the experts ensemble through a lightweight router. We show that the ensemble collectively optimizes the same objective as a single model trained over the whole dataset. This means we can divide the training burden among a number of "compute islands," lowering infrastructure costs and improving resilience to localized GPU failures. Decentralized diffusion models empower researchers to take advantage of smaller, more cost-effective and more readily available compute like on-demand GPU nodes rather than central integrated systems. We conduct extensive experiments on ImageNet and LAION Aesthetics, showing that decentralized diffusion models FLOP-for-FLOP outperform standard diffusion models. We finally scale our approach to 24 billion parameters, demonstrating that high-quality diffusion models can now be trained with just eight individual GPU nodes in less than a week.

扩散模型去中心化高效训练小规模部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。