Generalized Fine-Tuning of Diffusion Models via Stochastic Control and FBSDEs

Zirui Wang, Lu Wang

We propose a generalized fine-tuning framework for diffusion models from the perspective of stochastic control. Beyond entropy-regularized formulations, we introduce a general runn…