Skip to content
arXiv stat.ML · Papers

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

arXiv:2608.08770v1 Announce Type: new Abstract: 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 typically defined on individual samples, for many applications it is desirable to steer accord