mlsanityData to 7 Oct 2026

Regret-Aware Policy Optimization: Environment-Level Memory for Replay Suppression under Delayed Harm

arXiv 2604.07428 ↗First posted 8 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Safety in reinforcement learning (RL) is typically enforced through objective shaping while keeping environment dynamics stationary with respect to observable state-action pairs. Under delayed harm, this can lead to replay: after a washout period, reintroducing the same stimulus under matched observable conditions reproduces a similar harmful cascade. We introduce the Replay Suppression Diagnostic (RSD), a controlled exposure-decay-replay protocol that isolates this failure mode under frozen-policy evaluation. We show that, under stationary observable transition kernels, replay cannot be structurally suppressed without inducing a persistent shift in replay-time action distributions. Motivated by platform-mediated systems, we propose Regret-Aware Policy Optimization (RAPO), which augments the environment with persistent harm-trace and scar fields and applies a bounded, mass-preserving transition reweighting to reduce reachability of historically harmful regions. On graph diffusion tasks (50-1000 nodes), RAPO suppresses replay, reducing re-amplification gain (RAG) from 0.98 to 0.33 on 250-node graphs while retaining 82% of task return. Disabling transition deformation only during replay restores re-amplification (RAG 0.91), isolating environment-level deformation as the causal mechanism.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 17 references that are in the map. First posted on arXiv 8 Apr 2026.

Builds on

11 of 17 references.

References · 11

  1. 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
  2. 2017Constrained Policy Optimizationcited by 928
  3. 2015Deep Recurrent Q-Learning for Partially Observable MDPscited by 457
  4. 2017Robust Adversarial Reinforcement Learningcited by 439
  5. 2017Safe Reinforcement Learning via Shieldingcited by 427
  6. 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 400
  7. 2017Safe Model-based Reinforcement Learning with Stability Guaranteescited by 347
  8. 2015Risk-Constrained Reinforcement Learning with Percentile Risk Criteriacited by 320
  9. 2016Control Barrier Function Based Quadratic Programs for Safety Critical Systemscited by 261
  10. 2018Top-K Off-Policy Correction for a REINFORCE Recommender Systemcited by 181
  11. 2018RUDDER: Return Decomposition for Delayed Rewardscited by 162

Built on by

Not cited yet.