让扩散模型的生成控制动态变化,提升文本生成质量与可控性
Guidance Is Not a Hyperparameter: Learning Dynamic Control in Diffusion Language Models

- 用强化学习让引导尺度随生成阶段动态调整
- 在三个任务中实现更好可控性与质量平衡
- 适合关注文本生成可控性的研究者
Classifier-Free Guidance(CFG)是控制扩散生成模型的常用机制,但其引导尺度通常作为固定超参数使用。这种静态设计导致在自然语言处理领域,不同任务和生成阶段的最佳引导程度难以兼顾,影响生成效果。本文将引导尺度选择建模为序列决策问题,提出通过强化学习学习动态引导轨迹。具体地,将引导尺度视为基于扩散状态在每一步选择的离散控制动作,并采用近端策略优化(PPO)在任务级奖励下优化策略。在三个使用离散扩散语言模型的受控文本生成任务上的实验表明,自适应引导策略在可控性与生成质量之间始终优于固定尺度方法。对学习策略的进一步分析揭示了不同任务下具有区分性和可解释性的引导轨迹,证明将引导视为动态控制过程而非静态设计更为重要。
原文摘要 · Abstract (English)
Classifier-Free Guidance (CFG) is a widely used mechanism for controlling diffusion-based generative models, yet its guidance scale is typically treated as a fixed hyperparameter throughout generation. This static design yields a suboptimal controllability and quality tradeoff, as the optimal degree of guidance varies across tasks and across different stages of the diffusion process, especially in NLP domain. We recast CFG scale selection as a sequential decision-making problem and propose to learn dynamic guidance trajectories via reinforcement learning. Specifically, we model the guidance scale as a discrete control action selected at each generation step based on the evolving diffusion state, and optimize a policy using Proximal Policy Optimization (PPO) under task-level rewards. Experiments on three controlled NLP generation tasks using discrete diffusion language models demonstrate that adaptive guidance consistently achieves a better balance between controllability and generation quality than fixed-scale strategies. Further analysis of the learned policies reveals distinct and interpretable guidance trajectories across tasks, underscoring the importance of treating guidance as a dynamic control process rather than a static design choice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。