SWIFT halves ImageNet-256 diffusion training via weak denoiser
TL;DR
- SWIFT trains diffusion models alongside a frozen weak denoiser used only to extrapolate training targets, then discarded, so inference cost is unchanged.
- On one ImageNet-256 configuration, Swift surpasses the baseline's best FID using roughly half the total training compute, including guide preparation.
- On the RAEv2 representation, the paper reports FID 1.15 without inference-time guidance and 0.99 with guidance, with CIFAR-10 and ImageNet-256 experiments.
A new arXiv preprint proposes a diffusion training recipe called Swift that claims to reach the same image-quality ceiling using roughly half the compute. The method keeps a frozen weak denoiser alongside the main model and, for each noisy input, pushes the clean regression target away from the weak model's prediction, "amplifying components that remain underfit," the paper reports. The weak guide is discarded after training, so inference overhead is unchanged.
The paper, by Yanjin He, Cuifeng Shen, Wanzhou Lei and five collaborators, states that in "one ImageNet-256 configuration, Swift surpasses the baseline's best FID using roughly half the total training compute, including guide preparation and training overhead." On the RAEv2 representation the authors report FID 1.15 without inference-time guidance and 0.99 with guidance.
The claim rests on a spectral analysis of finite-time population gradient flow in a reproducing kernel Hilbert space, with experiments on CIFAR-10 and ImageNet-256 across model scales. The abstract does not name the specific baseline model the half-compute figure is measured against, publish absolute GPU-hour counts, or describe how the early-stopped weak-guide checkpoint is chosen, so the headline number is a single-paper assertion pending replication.
Originally reported by paper
Read the original article →Original headline: SWIFT Halves Diffusion Training Compute to Match Best FID Using a Frozen Weak Denoiser