ALER: Adaptive Learnable Experience Rewriting for Reinforcement Learning
cs.LG, cs.AI
Submitted: 2026-09-30
Updated: 2026-09-30
Comments: 28 pages, 12 figures, 18 tables
Project page: https://quartz-admirer.github.io/ALER-Adaptive-Learnable-Experience-Rewriting
License: http://creativecommons.org/licenses/by/4.0/
The gist: In partially observable reinforcement learning (RL), a later observation can make stored information obsolete or change what it implies for the next decision.
Terminology
Abstract
In partially observable reinforcement learning (RL), a later observation can make stored information obsolete or change what it implies for the next decision. Memory architectures and benchmarks for RL mostly test retention, the ability to keep information unchanged until it is needed. We formalize two further requirements. Rewriting sets the decision-relevant content to a value independent of the old one, and experience fusion transforms the old content by a rule that a later observation specifies. For tasks built from such updates, we count the memory states that a solution needs, and several baselines reach their lowest success rates on compositions that need more states. We introduce ALER (Adaptive Learnable Experience Rewriting), an agent that pairs an LSTM with a slot memory. An independently addressed Gumbel-Softmax write that concentrates its weight on one slot overwrites that slot, and a learned gate fuses the retrieved content with the recurrent state before the policy and value heads. We also introduce Rune-Mazes, three environments in which rune observations invert, cancel, reset, or repeat updates of a hidden cue under vector and pixel observations. Against seven baselines, ALER reaches a success rate of at least 0.82 in all sixteen Endless T-Maze configurations and at least 0.99 on all five Rune T-Maze compositions, and it has the highest mean success rate on four-branch Rune Multi-Corridor with an Invert rune. On pixel-based Rune MiniGrid Memory, it has a higher mean success rate than PPO-LSTM in eight of ten configurations. Project page: https://quartz-admirer.github.io/ALER-Adaptive-Learnable-Experience-Rewriting/.
Sources
- DeepMind Lab
- Deep Transformer Q-Networks for Partially Observable Reinforcement Learning
- Streaming Reinforcement Learning under Partial Observability with Real-Time Recurrent Learning
- Neural Turing Machines
- Memory Augmented Neural Networks with Wormhole Connections
- Memory-based control with recurrent neural networks
- Stable Hadamard Memory: Revitalizing Memory-Augmented Agents for Reinforcement Learning
- WorldMemArena: Evaluating Multimodal Agent Memory Through Action-World Interaction
- DeepMind Control Suite
- Unsupervised Predictive Memory in a Goal-Directed Agent
- Remember When It Matters: Proactive Memory Agent for Long-Horizon Agents
- Reinforcement Learning Neural Turing Machines - Revised
- Learn to Memorize: Optimizing LLM-based Agents with Adaptive Memory Framework
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks