REVIEW 3 major objections 5 minor 1 cited by
Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churn
T0 review · 3 major / 5 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read Reducing churn — the uncontrolled change in a network's outputs for data outside the training batch — restores plasticity in continual RL, and a simple regularizer beats six prior methods across 24 benchmark settings.
desk verdict A broad, believable empirical case that churn reduction helps continual RL, burdened by a theory section whose NTK mechanism doesn't line up with the PPO experiments. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The machinery is the empirical Neural Tangent Kernel matrix $N_\theta(i,j) = \nabla_\theta f_\theta(x_i)^\top \nabla_\theta f_\theta(x_j)$, whose rank serves as the paper's formal indicator of plasticity, together with the first-order churn identity $C_f(\bar{x}, \theta, \Delta\theta) \approx \nabla_\theta f_\theta(\bar{x})^\top \Delta\theta$. Combining these yields the vector form $C_f(\theta, \Delta\theta) \approx -\eta N_\theta S G_L$, showing that the NTK determines churn independently of the loss; substituting this into the error recursion gives the update matrix $I - \eta N_{\theta_t} S_i$ that carries the stability argument. C-CHAIN is the algorithm that minimizes churn on a reference batch disjoint from the training batch, and its gradient decomposition into a kernel-suppression term and a projection term is what supports the two-efficacy claim.
What would settle it
Train a continual RL agent while counterfactually forcing the empirical NTK to keep high rank without reducing churn (for example, injecting orthogonal gradient directions that stabilize rank but leave out-of-batch outputs free to move); if plasticity still degrades, churn is not the operative cause, and if plasticity is preserved, churn is incidental to rank. A cheaper test: compute the $②$ term for the PPO clipped objective and check whether it equals $-\bar{g}^\top g \cdot g$; if not, the step-size-adjustment interpretation is miscalibrated for the main experiments.
Extended reading notes
Core claim
The central claim is that churn and plasticity loss are two views of the same NTK collapse. The error dynamics $E_i(\theta_{t+1}) \approx (I - \eta N_{\theta_t} S_i) E_i(\theta_t)$ show that when all pairs of data points develop correlated gradients, the NTK matrix $N_\theta$ loses rank, out-of-batch outputs swing uncontrollably with every update, and learning becomes unstable exactly as tasks keep changing. Churn reduction breaks the cycle: C-CHAIN's loss $L_f^{\mathrm{cr}}(\theta) = \tfrac{1}{2}\mathbb{E}_{\bar{x} \in B_{\mathrm{ref}}}[C_f(\bar{x}, \theta, \Delta\theta)^2]$ pushes against churn on reference data, and its gradient splits into two effects — a decorrelation term that suppresses off-diagonal NTK entries and a projection term that rescales the regular training gradient by the kernel between reference and training data. The paper demonstrates the rank-preserving effect empirically and shows that the two components add up to the full benefit, with the decorrelation (orthogonal) component carrying most of the weight.
Load-bearing premise
The two-effect analysis is derived for a TD/Q-learning loss, but the main experiments train with PPO's clipped policy objective, for which the key gradient identity that produces the step-size-adjustment term does not hold exactly; if that interpretation does not transfer, half the explanatory story is unsupported in the settings where the method is actually evaluated.
Editorial extensions
If this is right
- If churn is a causal lever, NTK rank collapse is preventable by regularization alone, without resets, weight clipping, or parameter-space constraints.
- Because C-CHAIN only needs a reference batch from current data and never needs to know when a task switched, it can be layered onto existing RL algorithms (the paper demonstrates PPO and DoubleDQN) with a single relative-scale hyperparameter.
- The decomposition predicts that the decorrelation effect dominates, so cheaper approximations to the Hessian or kernel term could yield most of the benefit.
- Better-than-oracle results on continual control suggest churn reduction preserves useful transfer across related tasks rather than merely blocking interference.
- The same recipe transfers to continuous control and to a value-based agent, indicating the mechanism is not specific to one objective or action space.
Reading between the lines
- The theory's two-effect derivation uses a TD/Q-learning loss, while the main experiments run PPO's clipped objective; if the projection interpretation fails to transfer, the step-size-adjustment half of the story weakens while the decorrelation half may still hold — an asymmetry worth testing directly.
- A natural stress test: if churn is the operative cause, then a counterfactual that keeps the NTK full-rank while allowing churn (or reduces churn while keeping rank low) should separate the two mechanisms; the paper's own MNIST results, where C-CHAIN helps less than in RL, hint that churn's role is amplified when the data distribution is shaped by the agent's own policy.
- Because churn reduction and parameter-near-initialization methods act on different objects (kernel off-diagonals versus parameter distance), combining C-CHAIN with weight clipping or L2-init-style regularization might compound gains, a hypothesis the paper does not test.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper studies loss of plasticity in continual reinforcement learning through the lens of churn, defined as network output variability on out-of-batch data induced by mini-batch training. The authors use the empirical NTK to argue that rank collapse and churn exacerbate each other, propose a regularizer C-CHAIN that minimizes churn on a reference batch, and show that it improves average performance over vanilla PPO and six prior methods on 24 continual RL environments spanning Gym Control, ProcGen, DMC, and MinAtar. They also report reliable-metric confidence intervals and an NTK analysis intended to support the causal mechanism.
Significance. If the causal story holds, the paper makes a valuable contribution: it identifies churn as a controllable and measurable factor in plasticity loss, proposes a simple add-on regularizer that works across multiple RL algorithms and environments, and provides reproducible code. The empirical results are strong: C-CHAIN achieves the best aggregate scores in both Gym Control and ProcGen, with non-overlapping confidence intervals versus the best baseline on median, IQM, and optimality gap. The two-effect decomposition in Section 4.3 is a useful conceptual framework even if the derivation is heuristic. However, the theory-experiment gap discussed below means the causal mechanism is not yet established.
major comments (3)
- [Section 4.3, Eq. 12] The two-effect decomposition is derived for a scalar-output network trained on a TD or squared loss, with the specific identity ∇_{Q_θ} L_Q = -(Q_θ(x) - target) leading to Eq. 12. The main experiments use PPO's clipped surrogate objective for a vector-valued stochastic policy; for that loss, ∇_θ(∇_f L) is not -g, and the parameter update is not of the assumed form Δθ = -η E_x[∇_θ f_θ(x) ∇_f L]. Therefore the claimed gradient-decorrelation and step-size adjustment effects are not shown to apply to the actual experimental setting. The paper should either extend the derivation to the PPO objective, present the analysis as a heuristic motivating the method, or test the two-effect decomposition on the TD/DQN setting where it applies.
- [Appendix B and Eq. 2] The empirical NTK in Figures 6 and 7 is computed from gradients of the PPO objective (g_i = ∇_θ L_PPO), not from gradients of the network output f_θ(x) as defined in Eq. 2. Rank collapse of this loss-gradient Gram matrix is a different object from the theoretical NTK, so the measurements do not directly support the claim that churn reduction prevents rank collapse of the NTK in Eq. 2. The authors should recompute the empirical NTK using output gradients, or explicitly justify and clearly label the loss-gradient Gram matrix as a proxy, adjusting the wording in Section 5.2 accordingly.
- [Section 4.2, Eq. 7] The vicious cycle between rank decrease and churn exacerbation is asserted rather than derived. Equation 7 is a linear recurrence for the error with a fixed N_θ; it does not by itself imply that churn lowers the rank of N_θ or that lower rank increases churn. The text should either provide a formal argument or explicitly state this as a hypothesis supported by the empirical observations.
minor comments (5)
- [Figure 1] The caption and surrounding text do not clearly explain the visual notation (e.g., which shaded regions correspond to training versus reference data); consider adding a legend or a more explicit textual description.
- [Eq. 5] The notation 'G_L is the gradient matrix of ∇_{f_θ} L(θ)' is confusing, and the matrix dimensions in the product N_θ S G_L are not made precise; please clarify the dimensions and the definition of G_L.
- [Section 4.3, after Eq. 12] The sentence 'This term could either dampen or accelerate the regular gradient g ∇_f L_f which depends on the sign of the kernel...' has an ambiguous antecedent for 'which depends'; rephrase to indicate clearly what depends on the sign.
- [Section A.1] For MountainCar, the observation noise scale σ = 0.02 is chosen manually after finding σ = 2.0 too large; this per-environment design choice appears only in the appendix and should be mentioned in the main text when describing the setup.
- [Abstract and Conclusion] Given the theory-experiment gap noted in the major comments, the abstract and conclusion use 'demonstrate' for the causal claims; consider softening to 'suggest' or 'provide evidence for' until the mismatch is resolved.
Circularity Check
No significant circularity: C-CHAIN's churn objective is definitional, but the plasticity and NTK-rank claims are independently derived and empirically tested.
full rationale
The derivation chain is self-contained: churn is defined in Eq. 3, expressed through the NTK in Eqs. 4-5, used to derive the continual-learning error dynamics in Eq. 7, and then minimized by the C-CHAIN loss in Eq. 8. The paper's causal claims—that reducing churn suppresses NTK off-diagonal entries, maintains approximate rank, and improves continual RL performance—are not obtained by fitting the target variables; they are checked against held-out measurements of rank and off-diagonal magnitude (Figs. 6-7) and against aggregate returns on 24 continual RL settings. The only self-citation with author overlap, Tang and Berseth 2024, supplies the churn definition and the original single-task churn-reduction idea; it does not by itself justify the continual-RL NTK-rank mechanism, which is derived in Eqs. 10-12 and tested independently. The reader's flagged concerns—that Eq. 12 is derived for TD/Q-learning while the main experiments use PPO, and that Appendix B's empirical NTK is a loss-gradient Gram matrix rather than the output-gradient NTK of Eq. 2—are correctness/validity risks, not circularity: they do not make any prediction equal to an input by construction. Score 1 reflects the minor non-load-bearing self-citation; no circular step was found.
Assumptions & free parameters
free parameters (2)
- β (target relative loss for auto λ) =
1e3 to 1e5 (Gym/ProcGen), 0.5/0.05 (DMC), 0.01 (MinAtar)
- σ noise scale for MountainCar =
0.02 instead of 2.0
assumptions (4)
- domain assumption First-order Taylor expansion of churn (Eq 3), dropping O(||Δθ||²)
- domain assumption Low-dimensional empirical objective and simplicity bias cause NTK rank decrease (Section 4.2)
- ad hoc to paper The two-effects decomposition (①②) derived for TD loss extends to PPO's clipped objective
- ad hoc to paper The empirical NTK computed from PPO loss gradients (Appendix B) represents the NTK of the network output in Eq 2
Cite this review
Pith. "Pith review of Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churn." pith.science (2026). https://pith.science/paper/DMP34KVZ
@misc{pith2026250600592,
author = {Pith},
title = {Pith review of: Mitigating Plasticity Loss in Continual Reinforcement Learning by Reducing Churn},
year = {2026},
howpublished = {\url{https://pith.science/paper/DMP34KVZ}},
note = {Machine review of arXiv:2506.00592}
}
read the original abstract
Plasticity, or the ability of an agent to adapt to new tasks, environments, or distributions, is crucial for continual learning. In this paper, we study the loss of plasticity in deep continual RL from the lens of churn: network output variability for out-of-batch data induced by mini-batch training. We demonstrate that (1) the loss of plasticity is accompanied by the exacerbation of churn due to the gradual rank decrease of the Neural Tangent Kernel (NTK) matrix; (2) reducing churn helps prevent rank collapse and adjusts the step size of regular RL gradients adaptively. Moreover, we introduce Continual Churn Approximated Reduction (C-CHAIN) and demonstrate it improves learning performance and outperforms baselines in a diverse range of continual learning environments on OpenAI Gym Control, ProcGen, DeepMind Control Suite, and MinAtar benchmarks.
Figures
Figures from the paper (9 more)
Forward citations
Cited by 1 Pith paper
-
Calibrated Partial Resets: Preventing Policy Collapse in Continual Reinforcement Learning
Utility-scaled partial neuron resets prevent policy collapse in long-horizon continual RL while matching or beating binary-reset and uniform-decay baselines on several benchmarks.
Reference graph
Works this paper leans on
-
[1]
in terms of accuracy with mean and standard error over three seeds. Algorithm RandomLabel-MNIST Permuted-MNIST Vanilla 0.1501±0.0030 0.6430±0.0029 C-CHAIN 0.2482±0.0141 0.6797±0.0042 L2 Init 0.8607±0.0021 0.8039±0.0027 Weight Clipping 0.4304±0.0052 0.8281±0.0001 is then computed by following the definition in Equation 2: Nθ(i, j) =g⊤ i gj for i, j∈ {1,2, ...
work page 2020
-
[5]
Dohare, S., Hernandez-Garcia, J. F., Rahman, P., Sutton, R. S., and Mahmood, A. R. Maintaining plasticity in deep continual learning.arXiv preprint, arXiv:2306.13812,
-
[6]
Ellis, B., Jackson, M. T., Lupu, A., Goldie, A. D., Fel- lows, M., Whiteson, S., and Foerster, J. Adam on local time: Addressing nonstationarity in rl with relative adam timesteps.arXiv preprint, arXiv:2412.17113,
-
[7]
Elsayed, M. and Mahmood, A. R. Addressing loss of plas- ticity and catastrophic forgetting in continual learning. arXiv preprint, arXiv:2404.00781,
-
[8]
Elsayed, M., Farrahi, H., Dangel, F., and Mahmood, A. R. Revisiting scalable hessian diagonal approximations for applications in reinforcement learning. InICML, 2024a. Elsayed, M., Lan, Q., Lyle, C., and Mahmood, A. R. Weight clipping for deep continual and reinforcement learning. RLJ, 5:2198–2217, 2024b. 10 Mitigating Plasticity Loss in Continual RL by R...
work page 2024
-
[10]
Rolnick, D., Ahuja, A., Schwarz, J., Lillicrap, T
1109/MCSE.2007.58. Rolnick, D., Ahuja, A., Schwarz, J., Lillicrap, T. P., and Wayne, G. Experience replay for continual learning. In NeurIPS, pp. 348–358,
work page 2007
-
[11]
V ., Pas- canu, R., Martens, J., and Dabney, W
Lyle, C., Zheng, Z., Khetarpal, K., Hasselt, H. V ., Pas- canu, R., Martens, J., and Dabney, W. Disentangling the causes of plasticity loss in neural networks.arXiv preprint, arXiv:2402.18762,
-
[12]
Ma, G., Li, L., Zhang, S., Liu, Z., Wang, Z., Chen, Y ., Shen, L., Wang, X., and Tao, D. Revisiting plasticity in visual reinforcement learning: Data, modules and training stages.arXiv preprint, arXiv:2310.07418,
Show all 26 references
-
[16]
Proximal policy optimization algorithms
Schulman, J., Wolski, F., Dhariwal, P., Radford, A., and Klimov, O. Proximal policy optimization algorithms. arXiv preprint, arXiv:1707.06347,
-
[17]
P., and Riedmiller, M
Tassa, Y ., Doron, Y ., Muldal, A., Erez, T., Li, Y ., de Las Casas, D., Budden, D., Abdolmaleki, A., Merel, J., Lefrancq, A., Lillicrap, T. P., and Riedmiller, M. A. Deep- mind control suite.arXiv preprint, arXiv:1801.00690,
-
[18]
Deep reinforcement learning and the deadly triad.arXiv preprint, arXiv:1812.02648,
van Hasselt, H., Doron, Y ., Strub, F., Hessel, M., Sonnerat, N., and Modayil, J. Deep reinforcement learning and the deadly triad.arXiv preprint, arXiv:1812.02648,
-
[19]
Continual learning for large language models: A survey
Wu, T., Luo, L., Li, Y .-F., Pan, S., Vu, T.-T., and Haffari, G. Continual learning for large language models: A survey. arXiv preprint, arXiv:2402.01364,
-
[20]
and Tian, T
Young, K. and Tian, T. Minatar: An atari-inspired testbed for more efficient reinforcement learning experiments. arXiv preprint, arXiv:1903.03176,
1903 arXiv
-
[21]
Experimental Details A.1
12 Mitigating Plasticity Loss in Continual RL by Reducing Churn A. Experimental Details A.1. Continual RL Environment SetupsWe follow the setups in (Muppidi et al., 2024). For Gym Control, we use four environments: CartPole-v1, Acrobot-v1, LunarLander-v2 and MountainCar-v0. Fo...
2024
-
[22]
Therefore, we useσ= 0.02for MountainCar-v0
and we found σ= 2.0 is too large for this environment to be learnable. Therefore, we useσ= 0.02for MountainCar-v0. For ProcGen, we use all sixteen environments in the suite, while only four (i.e., Starpilot, Fruitbot, Chaser, Dodgeball) were adopted in (Muppidi et al., 2024). ...
2024
-
[23]
The values of conventional hyperparameters are taken from the recommended values inCleanRL
Hyperparameters of PPO and C-CHAIN used in continual DMC environments. The values of conventional hyperparameters are taken from the recommended values inCleanRL. PPO Hyperparameters Learning Rate 3e−4 Training Interval 2048 steps Discount Factor (γ) 0.99 GAE Parameter (λ) 0.9...
-
[24]
The values of conventional hyperpa- rameters are taken from the recommended values in (Young & Tian, 2019)
Hyperparameters of DoubleDQN and C-CHAIN used in continual MinAtar environments. The values of conventional hyperpa- rameters are taken from the recommended values in (Young & Tian, 2019). DoubleDQN Hyperparameters Learning Rate 3e−4 Training Interval 1 step Discount Factor (γ...
2019
-
[25]
We can observe that C-CHAIN improves the vanilla agent but does not perform on par with L2 Init and Weight Clipping. This shows that the efficacy of C-CHAIN is relatively limited in the two continual supervised learning environments, especially in contrast to its superiority i...
2024
-
[2013]
URL http://www.amazon.com/exec/obidos/ redirect?tag=citeulike07-20&path= ASIN/1449319793
ISBN 9789351100065. URL http://www.amazon.com/exec/obidos/ redirect?tag=citeulike07-20&path= ASIN/1449319793. Mnih, V ., Badia, A. P., Mirza, M., Graves, A., Lillicrap, T. P., Harley, T., Silver, D., and Kavukcuoglu, K. Asyn- chronous methods for deep reinforcement learning. I...
1928
-
[2016]
Kumar, A., Agarwal, R., Ghosh, D., and Levine, S
doi: 10.3233/978-1-61499-649-1-87. Kumar, A., Agarwal, R., Ghosh, D., and Levine, S. Im- plicit under-parameterization inhibits data-efficient deep reinforcement learning. InICLR,
-
[2019]
The phenomenon of policy churn.arXiv preprint, arXiv:2206.00730,
Schaul, T., Barreto, A., Quan, J., and Ostrovski, G. The phenomenon of policy churn.arXiv preprint, arXiv:2206.00730,
-
[2020]
L., Pascanu, R., and Clopath, C
Berariu, T., Czarnecki, W., De, S., Bornschein, J., Smith, S. L., Pascanu, R., and Clopath, C. A study on the plastic- ity of neural networks.arXiv preprint, arXiv:2106.00042,
-
[2021]
Openai gym
Brockman, G., Cheung, V ., Pettersson, L., Schneider, J., Schulman, J., Tang, J., and Zaremba, W. Openai gym. arXiv preprint, arXiv:1606.01540,
-
[2022]
J., Liu, Y ., and Van Roy, B
Kumar, S., Marklund, H., Rao, A., Zhu, Y ., Jeon, H. J., Liu, Y ., and Van Roy, B. Continual learning as computation- ally constrained reinforcement learning.arXiv preprint, arXiv:2307.04345, 2023a. Kumar, S., Marklund, H., and Roy, B. V . Maintaining plas- ticity in continual...
-
[2023]
Towards charac- terizing divergence in deep q-learning.arXiv preprint, arXiv:1903.08894,
Achiam, J., Knight, E., and Abbeel, P. Towards charac- terizing divergence in deep q-learning.arXiv preprint, arXiv:1903.08894,
1903 arXiv
-
[2024]
Lever- aging procedural generation to benchmark reinforcement learning
Cobbe, K., Hesse, C., Hilton, J., and Schulman, J. Lever- aging procedural generation to benchmark reinforcement learning. InICML, volume 119, pp. 2048–2056,
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.