WTF: simulation-free RL fine-tuning for flow maps cuts training compute by up to 280x

yeewhye · x · 2026-10-15

A NeurIPS 2026 Spotlight paper introduces Wasserstein-Tilted Flow Maps (WTF), enabling simulation-free RL fine-tuning for diffusion and flow models. Instead of expensive rollouts, WTF replaces KL regularization with a transport-based alternative that moves each sample toward higher reward. One step beats 50 steps of the base model, with up to 280x less training compute.

Related event: WTF Flow Maps Enable Simulation-Free RL Fine-Tuning, 280x Cheaper(3 posts)→

Original post →

More from Research

Research channel →