arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.04651cs.LG

SAFE:基于熵感知预测控制的稳定对齐微调用于人类反馈的强化学习(RLHF)

SAFE: Stable Alignment Finetuning with Entropy-Aware Predictive Control for Reinforcement Learning from Human Feedback (RLHF)

Dipan Maity

AI总结:

SAFE通过熵感知预测控制提升RLHF稳定性,实现更高的训练奖励和更稳定的优化性能。

摘要:

近来的文献将近端策略优化(PPO)定位为强化学习从人类反馈(RLHF)部分的标准方法。PPO在经验上表现良好,但其动机具有启发式性质,并且在LM-RLHF中处理KL散度约束的方式是随意的,导致奖励震荡、熵崩溃、价值函数漂移和突然的策略发散,需要频繁重启和大量的超参数调整。在本文中,我们开发了一种新的纯在线策略演员-评论家RL方法,用于LM-RLHF设置。我们提出了SAFE(具有熵感知控制的稳定对齐微调),一种新的RLHF算法,结合了双软最小批评器用于悲观价值估计,以及一个新的多层稳定框架,结合熵门控KL调节和PID控制的自适应阈值。与标准PPO的对称KL惩罚不同,SAFE区分高熵探索和低熵模式崩溃,并根据奖励速度动态调整惩罚。在3B参数模型上的实验表明,SAFE在训练平均奖励方面比PPO高5.15%(0.725 vs 0.689),奖励崩溃可忽略,KL控制优于PPO。我们的方法增加了最小的计算开销,并提供了一个可解释、抗崩溃的RLHF框架,该框架在保持积极学习速度的同时,确保了适合生产部署的稳定长周期优化。代码可在https://github.com/ryyzn9/SAFE上获得。

英文摘要:

Proximal Policy Optimization (PPO) has been positioned by recent literature as the canonical method for the RL part of Reinforcement Learning from Human Feedback (RLHF). PPO performs well empirically but has a heuristic motivation and handles the KL-divergence constraint used in LM-RLHF in an ad-hoc manner and suffers form reward oscillations, entropy collapse, value function drift, and sudden policy divergence that require frequent restarts and extensive hyperparameter tuning. In this paper, we develop a new pure on policy actor-critic RL method for the LM-RLHF setting. We present SAFE (Stable Alignment Finetuning with Entropy-aware control),a novel RLHF algorithm that combines a Double Soft-Min Critic for pessimistic value estimation with a new multi-layer stabilization framework combining entropy-gated KL regulation, and PID-controlled adaptive thresholds. Unlike standard PPO's symmetric KL penalties, SAFE distinguishes high-entropy exploration from low-entropy mode collapse and adjusts penalties dynamically based on reward velocity. Experiments on a 3B parameter model show SAFE achieves +5.15\% training-average reward than PPO (0.725 vs 0.689), negligible reward crashes, and superior KL control than ppo . Our method adds minimal computational overhead and provides an interpretable, crash-resistant RLHF framework that maintains aggressive learning speed while ensuring stable long-horizon optimization suitable for production deployment. Code is available at https://github.com/ryyzn9/SAFE

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑