Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Diff2Flow: Training Flow Matching Models via Diffusion Model Alignment

About

Diffusion models have revolutionized generative tasks through high-fidelity outputs, yet flow matching (FM) offers faster inference and empirical performance gains. However, current foundation FM models are computationally prohibitive for finetuning, while diffusion models like Stable Diffusion benefit from efficient architectures and ecosystem support. This work addresses the critical challenge of efficiently transferring knowledge from pre-trained diffusion models to flow matching. We propose Diff2Flow, a novel framework that systematically bridges diffusion and FM paradigms by rescaling timesteps, aligning interpolants, and deriving FM-compatible velocity fields from diffusion predictions. This alignment enables direct and efficient FM finetuning of diffusion priors with no extra computation overhead. Our experiments demonstrate that Diff2Flow outperforms na\"ive FM and diffusion finetuning particularly under parameter-efficient constraints, while achieving superior or competitive performance across diverse downstream tasks compared to state-of-the-art methods. We will release our code at https://github.com/CompVis/diff2flow.

Johannes Schusterbauer, Ming Gui, Frank Fundel, Bj\"orn Ommer• 2025

Related benchmarks

TaskDatasetResultRank
Image GenerationImageNet 256x256 50k images (val)
FID7.45
42
Class-to-image generationImageNet 256x256 50k images 2012 (test)
FID7.45
28
Showing 2 of 2 rows

Other info

Code

Follow for update