Drifting Field Policy: A One-Step Generative Policy via Wasserstein Gradient Flow

Juil Koo, Mingue Park, Jiwon Choi +2

We propose Drifting Field Policy (DFP), a non-ODE one-step generative policy built on the drifting model paradigm. We frame the policy update as a reverse-KL Wasserstein-2 gradient…