A Mean-Field Framework for Inference-Time Distributional Control of Diffusion Models

Samuel Howard, Nikolas Nüsken

Diffusion models are increasingly used as controllable samplers, whose generations can be steered at inference time according to a chosen reward function. While such rewards are ty…