about
Decentralized Diffusion Models (arxiv.org)
1 point by dvrp on Jan 15, 2025 | hide | past | pdf | 1 comment on HN

In plain words: Instead of syncing GPUs over one fast network, it trains separate image models on data splits independently, then blends their outputs with a small router. They beat a standard single model at equal compute, and scaled to 24 billion parameters on independent GPU nodes.

Abstract

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.

David McAllister, Matthew Tancik, Jiaming Song, Angjoo Kanazawa
arXiv:2501.05450 · cs.CV, cs.DC, cs.LG · submitted Jan 9, 2025 · updated Jan 10, 2025
abstract · pdf · html · Project webpage: https://decentralizeddiffusion.github.io/

add comment on HN

The first author of the paper posted about it here: https://x.com/davidrmcall/status/1879215125573587247.