用控制理论统一解释扩散模型可控生成,提升对齐效果与效率。
Diffusion Controller: Framework, Algorithms and Parameterization
- 将反向扩散采样视为状态仅控制问题,基于最优性条件推导出可训练策略。
- 在稳定扩散上实现偏好对齐胜率提升,质量与效率优于灰盒基线和LoRA。
- 提出轻量级侧网络参数化,支持冻结主干的灰盒微调,适配高效可控生成场景。
可控扩散生成常依赖各种启发式方法,缺乏统一理解。本文提出扩散控制器(DiffCon),从控制论视角将反向扩散采样建模为广义线性可解马尔可夫决策过程中的状态仅随机控制问题。控制通过重加权预训练反向转移核实现,平衡终端目标与f-散度成本。由此推导出实用的强化学习方法:(i) f-散度正则化策略梯度更新,含类似PPO的规则;(ii) 正则化决定的奖励加权回归目标,在KL散度下保证最小化器保持。该框架进一步给出原则性模型形式:最优得分可分解为固定预训练基线加轻量控制修正,从而提出基于中间去噪输出的侧网络参数化,支持冻结主干的灰盒适应。在Stable Diffusion v1.4上的实验表明,无论监督还是奖励驱动微调,均一致提升偏好对齐胜率,并改善质量-效率权衡,优于灰盒基线甚至参数高效的白盒适配器LoRA。
原文摘要 · Abstract (English)
Controllable diffusion generation often relies on various heuristics that are seemingly disconnected without a unified understanding. We bridge this gap with Diffusion Controller (DiffCon), a unified control-theoretic view that casts reverse diffusion sampling as state-only stochastic control within (generalized) linearly-solvable Markov Decision Processes (LS-MDPs). Under this framework, control acts by reweighting the pretrained reverse-time transition kernels, balancing terminal objectives against an $f$-divergence cost. From the resulting optimality conditions, we derive practical reinforcement learning methods for diffusion fine-tuning: (i) f-divergence-regularized policy-gradient updates, including a PPO-style rule, and (ii) a regularizer-determined reward-weighted regression objective with a minimizer-preservation guarantee under the Kullback-Leibler (KL) divergence. The LS-MDP framework further implies a principled model form: the optimal score decomposes into a fixed pretrained baseline plus a lightweight control correction, motivating a side-network parameterization conditioned on exposed intermediate denoising outputs, enabling effective gray-box adaptation with a frozen backbone. Experiments on Stable Diffusion v1.4 across supervised and reward-driven finetuning show consistent gains in preference-alignment win rates and improved quality-efficiency trade-offs versus gray-box baselines and even the parameter-efficient white-box adapter LoRA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。