arXiv:2510.14503cs.LG2025-10被引 1

通过可逆性信号与状态回滚,提升强化学习的安全性与稳定性。

Learning to Undo: Rollback-Augmented Reinforcement Learning with Reversibility Signals

  • 引入状态可逆性度量Phi,动态调整时序差分更新的惩罚项。
  • 在CliffWalking中减少99.8%的坠落事故,奖励提升55%。
  • 适合需要高安全性的决策场景,如自动驾驶、机器人控制。

本文提出一种可逆学习框架,以增强基于价值的强化学习代理在部分不可逆环境中的鲁棒性与效率,缓解价值过估计问题与不稳定性。该框架包含两个互补机制:一个基于经验的转移可逆性度量Φ(s,a),以及一种选择性状态回滚操作。我们引入一个在线的每状态动作估计量Φ,用于量化在固定时间窗口K内返回先前状态的可能性。该度量用于动态调整时序差分更新中的惩罚项,将可逆性感知直接融入价值函数。系统还包含选择性回滚算子:当某动作的预期回报显著低于其瞬时估值且超过预设阈值时,代理将被惩罚并回退至前一状态,从而中断次优高风险轨迹,避免灾难性行为。结合可逆性感知评估与定向回滚,该方法显著提升了安全性、性能与稳定性。在CliffWalking v0环境中,灾难性坠落减少超99.8%,平均回合回报提升55%;在Taxi v3环境中,非法动作抑制≥99.9%,累积奖励提升65.7%,同时大幅降低奖励方差。消融实验表明,回滚机制是安全与性能提升的关键所在,为安全可靠的序列决策迈出了坚实一步。

原文摘要 · Abstract (English)

This paper proposes a reversible learning framework to improve the robustness and efficiency of value based Reinforcement Learning agents, addressing vulnerability to value overestimation and instability in partially irreversible environments. The framework has two complementary core mechanisms: an empirically derived transition reversibility measure called Phi of s and a, and a selective state rollback operation. We introduce an online per state action estimator called Phi that quantifies the likelihood of returning to a prior state within a fixed horizon K. This measure is used to adjust the penalty term during temporal difference updates dynamically, integrating reversibility awareness directly into the value function. The system also includes a selective rollback operator. When an action yields an expected return markedly lower than its instantaneous estimated value and violates a predefined threshold, the agent is penalized and returns to the preceding state rather than progressing. This interrupts sub optimal high risk trajectories and avoids catastrophic steps. By combining reversibility aware evaluation with targeted rollback, the method improves safety, performance, and stability. In the CliffWalking v0 domain, the framework reduced catastrophic falls by over 99.8 percent and yielded a 55 percent increase in mean episode return. In the Taxi v3 domain, it suppressed illegal actions by greater than or equal to 99.9 percent and achieved a 65.7 percent improvement in cumulative reward, while also sharply reducing reward variance in both environments. Ablation studies confirm that the rollback mechanism is the critical component underlying these safety and performance gains, marking a robust step toward safe and reliable sequential decision making.

强化学习可逆性安全决策

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。