English

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

Machine Learning 2026-04-10 v1 Artificial Intelligence

Abstract

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.

Keywords

Cite

@article{arxiv.2604.07428,
  title  = {Regret-Aware Policy Optimization: Environment-Level Memory for Replay Suppression under Delayed Harm},
  author = {Prakul Sunil Hiremath},
  journal= {arXiv preprint arXiv:2604.07428},
  year   = {2026}
}

Comments

18 pages, 3 figures. Includes theoretical analysis and experiments on graph diffusion environments

R2 v1 2026-07-01T11:59:51.965Z