mlsanityData to 14 Oct 2026

When Does Backpropagating Through Policy Memory Matter? Physical Credit, Optimizer Updates, and Observability

arXiv 2609.33169 ↗First posted 27 Sep 2026cs.LG cs.ROcited by 0

Not on the map. Why?

Abstract (arXiv)

Policies with memory can learn along two backward paths: through the physical states their actions produce and through the representations they store. Transformer-XL and truncated backpropagation through time cut the second path at stored history while keeping its values. We ask when this cut matters. Holding the forward computation fixed and varying only derivative edges, we measure parameter gradients, the updates the optimizer applies, and continued training in a Transformer vessel-trajectory model and a quadrotor tracking policy. In the vessel model, detaching the key-value cache shrank the gradient to about a tenth of its norm, with little rotation, when gradients flowed through all earlier physical states, but barely changed it under one-step physical credit. In this strongly clipped regime the optimizer, not the gradient, set how far updates differed: global-norm clipping removed most of the gradient difference between memory-cut graphs, whereas AdamW turned a 2% gradient difference between two placements of the cut into update differences of up to 31% at the step where the placement was switched. In a quadrotor trained from initialization with 0.20 m/s velocity noise, removing memory raised tracking error by 43% and cutting memory gradients raised it by 32%; at low noise the cut's mean cost exceeded the value of memory. Two-step truncation segments gave no measurable gain, although with hidden velocity a two-step window captured most of the value of memory; eight-step segments removed half to three quarters of the cost. Switching the cut on only for the last fifth of training understated its cost about threefold at 0.20-0.30 m/s, but not at low noise or with hidden velocity. These results suggest measuring the cost of a memory cut by training with it from initialization, and comparing backward graphs by the updates the optimizer applies rather than by raw gradients.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 25 references that are in the map. First posted on arXiv 27 Sep 2026.

Builds on

23 of 25 references.

References · 23

  1. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  2. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  3. 2019Stabilizing Transformers for Reinforcement Learningcited by 270
  4. 2022Recurrent Memory Transformercited by 203
  5. 2022Accelerated Policy Learning with Parallel Differentiable Simulationcited by 104
  6. 2022Do Differentiable Simulators Give Better Policy Gradients?cited by 89
  7. 2021Gradients are Not All You Needcited by 86
  8. 2015Memory-based control with recurrent neural networkscited by 78
  9. 2023When Do Transformers Shine in RL? Decoupling Memory from Credit Assignmentcited by 63
  10. 2024Back to Newton's Laws: Learning Vision-based Agile Flight via Differentiable Physicscited by 45
  11. 2017Unbiasing Truncated Backpropagation Through Timecited by 37
  12. 2018Sparse Attentive Backtracking: Temporal CreditAssignment Through Remindingcited by 34
11 more
  1. 2019Adaptively Truncating Backpropagation Through Time to Control Gradient Biascited by 22
  2. 2022Training Efficient Controllers via Analytic Policy Gradientcited by 22
  3. 2024Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulationcited by 20
  4. 2024Stabilizing Reinforcement Learning in Differentiable Multiphysics Simulationcited by 20
  5. 2024Learning Quadrotor Control From Visual Features Using Differentiable Simulationcited by 19
  6. 2025Memo: Training Memory-Efficient Embodied Agents with Reinforcement Learningcited by 9
  7. 2026Stored in Optimizer State, Valued by Later Training: A Causal Account of Subliminal Trait Transfercited by 3
  8. 2024Autonomous Vehicle Controllers From End-to-End Differentiable Simulationcited by 1
  9. 2026SKETCH: Semantic Key-Point Conditioning for Long-Horizon Vessel Trajectory Predictioncited by 1
  10. 2026Finite-Horizon Input-Output Dynamics of Minibatch Perturbations in AdamWcited by 1
  11. 2026Delayed Optimizer-State Transport Shapes Short-Horizon Training Decisionscited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Backpropagating Through Simulation: Analytic Policy Gradients for Sample and Learning Efficient Differentiable Continuous Control
  2. 2026Memory-State Critic for Asymmetric Actor-Critic with Application to Vision-Based Pursuit-Evasion
  3. 2025Privileged observations enable rapid and reliable policy discovery directly in the physical world
  4. 2026Physical Time as a Missing Dimension of Robot Policy Memory Dynamics
  5. 2026Reparameterization Flow Policy Optimization
  6. 2026Linear Recurrent Memory Suffices to Distil a World-Model Policy for Robot Air Hockey
  7. 2025Leveling the Playing Field: Carefully Comparing Classical and Learned Controllers for Quadrotor Trajectory Tracking
  8. 2024Residual Policy Learning for Perceptive Quadruped Control Using Differentiable Simulation