用注意力熵动态优化扩散模型训练,提升对齐效率与效果
AEGPO: Adaptive Entropy-Guided Policy Optimization for Diffusion Models
- 基于注意力熵变化和峰值,动态识别高价值样本与关键去噪步骤
- 实验显示收敛速度更快,文本到图像生成对齐效果优于标准GRPO
- 适合关注高效强化学习对齐扩散模型的研究者与实践者
基于人类反馈的强化学习(RLHF)在对齐扩散模型和流模型方面展现出潜力,但如GRPO等策略优化方法存在采样策略低效且静态的问题。这些方法对所有提示和去噪步骤一视同仁,忽略了样本学习价值的显著差异以及关键探索时刻的动态特性。本文深入分析了GRPO训练中的内部注意力动态,发现注意力熵可作为双重信号代理:其相对变化ΔEntropy反映当前策略与基础策略的偏差,是样本学习价值的稳健指标;而绝对注意力熵峰值Entropy(t)则量化注意力分散程度,有效识别高价值探索的关键时间步。据此提出自适应熵引导策略优化(AEGPO),在全局层面利用ΔEntropy动态分配采样预算,优先处理高价值提示;在局部层面利用Entropy(t)峰值指导探索,仅在关键高分散时间步进行优化,而非均匀分布于所有去噪步骤。通过聚焦最具信息量的样本与时刻,AEGPO实现了更高效的策略优化。在文本到图像生成任务上的实验表明,AEGPO显著加速收敛,并在对齐性能上优于标准GRPO变体。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) shows promise for aligning diffusion and flow models, yet policy optimization methods such as GRPO suffer from inefficient and static sampling strategies. These methods treat all prompts and denoising steps uniformly, ignoring substantial variations in sample learning value as well as the dynamic nature of critical exploration moments. To address this issue, we conduct a detailed analysis of the internal attention dynamics during GRPO training and uncover a key insight: attention entropy can serve as a powerful dual-signal proxy. First, across different samples, the relative change in attention entropy (ΔEntropy), which reflects the divergence between the current policy and the base policy, acts as a robust indicator of sample learning value. Second, during the denoising process, the peaks of absolute attention entropy (Entropy(t)), which quantify attention dispersion, effectively identify critical timesteps where high-value exploration occurs. Building on this observation, we propose Adaptive Entropy-Guided Policy Optimization (AEGPO), a novel dual-signal, dual-level adaptive optimization strategy. At the global level, AEGPO uses ΔEntropy to dynamically allocate rollout budgets, prioritizing prompts with higher learning value. At the local level, it exploits the peaks of Entropy(t) to guide exploration selectively at critical high-dispersion timesteps rather than uniformly across all denoising steps. By focusing computation on the most informative samples and the most critical moments, AEGPO enables more efficient and effective policy optimization. Experiments on text-to-image generation tasks demonstrate that AEGPO significantly accelerates convergence and achieves superior alignment performance compared to standard GRPO variants.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。