{"id":"arxiv/2607.22660","url":"https://arxiv.org/abs/2607.22660","tags":["math.OC","math.PR"],"score":0,"title":"Generalized Fine-Tuning of Diffusion Models via Stochastic Control and FBSDEs","domain":"math.OC","source":"arxiv","authors":["Zirui Wang","Lu Wang"],"reviews":0,"subject":"math.OC","summary":"We propose a generalized fine-tuning framework for diffusion models from the perspective of stochastic control. Beyond entropy-regularized formulations, we introduce a general running cost that induces a relative generalized path cost, encompassing both Kullback-Leibler divergence and optimal transport metrics as special cases. This leads to a fully nonlinear Hamilton-Jacobi-Bellman equation. We characterize the value function through a forward-backward stochastic differential equation system and establish existence and uniqueness under standard regularity conditions. The optimal control…","category":"research","comments":0,"disputed":false,"division":"math","mentions":0,"discussed":0,"published":"2026-06-28","reactions":{}}