传统熵约束失效,新方法通过感知熵提升图像生成多样性。
When Policy Entropy Constraint Fails: Preserving Diversity in Flow-based RLHF via Perceptual Entropy

- 引入感知熵捕捉视觉空间多样性,替代失效的策略熵
- 在多个模型和奖励机制下,多样性得分提升20倍以上
- 适合需要高质量多样图像生成的研究者和开发者
RLHF被广泛用于对齐流匹配文本到图像模型与人类偏好,但微调后常导致严重多样性崩溃。传统上认为策略熵与多样性正相关,因而采用熵正则化。然而我们发现,在流模型中该直觉失效:策略熵保持恒定,即使感知多样性已崩溃。理论与实证分析表明,恒定熵源于固定的预设噪声调度,而多样性崩溃由策略梯度的模式追逐特性驱动。因此,策略熵无法阻止模型收敛至感知空间中的狭窄高奖励区域。为此,我们提出感知熵,可有效捕捉感知空间中的多样性并保持标准熵的性质。基于此,我们设计两种熵正则化策略:感知熵约束与生成空间感知约束,显著提升感知多样性与质量。在两个基础模型、神经与规则奖励、三个感知空间上的实验均显示一致改进:PEC取得0.734的综合评分(基线为0.366);互补设置下多样性平均达0.989(基线仅0.047)。项目页面公开可用。
原文摘要 · Abstract (English)
RLHF is widely used to align flow-matching text-to-image models with human preferences, but often leads to severe diversity collapse after fine-tuning. In RL, diversity is often assumed to correlate with policy entropy, motivating entropy regularization. However, we show this intuition breaks in flow models: policy entropy remains constant, even while perceptual diversity collapses. We explain this mismatch both theoretically and empirically: the constant entropy arises from the fixed, pre-defined noise schedule, while the diversity collapse is driven by the mode-seeking nature of policy gradients. As a result, policy entropy fails to prevent the model from converging to a narrow high-reward region in the perceptual space. To this end, we introduce perceptual entropy that captures diversity in a perceptual space and maintains the property of standard entropy. Building upon this insight, we propose two entropy-regularized strategies, Perceptual Entropy Constraint and Perceptual Constraints on Generation Space, to preserve perceptual diversity and improve the quality. Experiments across two base models, neural and rule-based rewards, and three perceptual spaces demonstrate consistent gains in the quality-diversity trade-off; PEC achieves the best overall score of 0.734 (vs. baseline's 0.366); a complementary setting of PEC further reaches a diversity average of 0.989 (vs. baseline's 0.047). Our project page (https://xiaofeng-tan.github.io/projects/PEC) is publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。