Balancing Plasticity and Stability with Fast and Slow Successor Features

arXiv cs.LG Papers

Summary

This paper investigates the stability-plasticity dilemma in reinforcement learning under gradual non-stationarity, finding that stabilizing successor features via synaptic consolidation across multiple timescales outperforms plasticity-focused methods.

arXiv:2605.26357v1 Announce Type: new Abstract: A hallmark of intelligence is the ability to adapt in non-stationary environments, yet deep Reinforcement Learning (RL) agents often struggle in such settings. Prior studies introduce non-stationarity through abrupt shifts in features or dynamics, whereas real-world environments often evolve gradually through continual drift. This distinction has important implications for the "stability-plasticity dilemma" in RL, as abrupt task changes may demand more plasticity than naturalistic settings. To address this, we modify existing 3D Miniworld and MuJoCo environments to incorporate naturalistic, continual non-stationarity, and use them to examine how stability and adaptation affect performance under continuous environmental change. We find that methods favoring stability, such as synaptic consolidation, outperform approaches focused on plasticity, such as parameters resetting. Motivated by this result, and prior evidence that Successor Features (SFs) reduce interference, we investigate whether SFs are better consolidation targets than Q-values. Across both environments, applying neuro-inspired synaptic consolidation to SFs yields superior performance on continually changing settings. Moreover, consolidation is most effective when SFs are stabilized across multiple timescales, which capture complementary aspects of gradual environmental change. Together, these results suggest that stability is more critical in continual learning when changes are gradual, and that multi-timescale consolidation of predictive representations is an effective approach.
Original Article
View Cached Full Text

Cached at: 05/27/26, 09:09 AM

# Balancing Plasticity and Stability with Fast and Slow Successor Features
Source: [https://arxiv.org/html/2605.26357](https://arxiv.org/html/2605.26357)
###### Abstract

A hallmark of intelligence is the ability to adapt in non\-stationary environments, yet deep Reinforcement Learning \(RL\) agents often struggle in such settings\. Prior studies introduce non\-stationarity through abrupt shifts in features or dynamics, whereas real\-world environments often evolve gradually through continual drift\. This distinction has important implications for the “stability\-plasticity dilemma” in RL, as abrupt task changes may demand more plasticity than naturalistic settings\. To address this, we modify existing 3D Miniworld and MuJoCo environments to incorporate naturalistic, continual non\-stationarity, and use them to examine how stability and adaptation affect performance under continuous environmental change\. We find that methods favoring stability, such as synaptic consolidation, outperform approaches focused on plasticity, such as parameters resetting\. Motivated by this result, and prior evidence that Successor Features \(SFs\) reduce interference, we investigate whether SFs are better consolidation targets than Q\-values\. Across both environments, applying neuro\-inspired synaptic consolidation to SFs yields superior performance on continually changing settings\. Moreover, consolidation is most effective when SFs are stabilized across multiple timescales, which capture complementary aspects of gradual environmental change\. Together, these results suggest that stability is more critical in continual learning when changes are gradual, and that multi\-timescale consolidation of predictive representations is an effective approach\.

Continual Reinforcement Learning, Successor Features, Plasticity, Stability, Synaptic Consolidation

## 1Introduction

Events in the real world are often constantly evolving\. Humans and animals must therefore adapt in environments where the underlying dynamics shift naturally and continually\. In contrast, many continual learning studies in Artificial Intelligence \(AI\) focus on abrupt, task\-boundary changes, where the features or dynamics across tasks differ substantially\. Standard RL techniques, such as Q\-learning, struggle under such conditions and often suffer from catastrophic forgetting\(McCloskey & Cohen,[1989](https://arxiv.org/html/2605.26357#bib.bib33); French,[1999](https://arxiv.org/html/2605.26357#bib.bib16)\)\. Developing methods that enable deep RL agents to learn effectively in naturalistic, continually changing environments remain a major goal in AI research\(Khetarpal et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib24); Abel et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib2); Silver & Sutton,[2025](https://arxiv.org/html/2605.26357#bib.bib42)\)\.

While early work in supervised continual learning emphasized stability \(i\.e\., the ability to retain previously acquired knowledge and prevent catastrophic forgetting\)\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27); Zenke et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib54)\), RL poses unique challenges, since changing policies alter the samples an agent encounters, compounding environmental non\-stationarity\. In RL, Atari\(Bellemare et al\.,[2013](https://arxiv.org/html/2605.26357#bib.bib5)\)has emerged as one of the standard testbeds for sequential task learning, where stability\-focused methods such as Elastic Weight Consolidation \(EWC\)\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27)\)and replay\(Rolnick et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib39)\)became dominant strategies\. More recently, studies have shifted towards the complementary issue of plasticity \(i\.e\., the capacity to rapidly adapt to new experiences\), using either sequential Atari tasks\(Abbas et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib1)\)or artificial tasks in MuJoCo created by randomly sampling friction coefficients per task\(Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)\. Stability has mostly been studied in sequential multi\-task settings and plasticity in single\-task dynamics, but real\-world environments rarely fit either case, as naturalistic, continual non\-stationarity appears as a single task while still creating a stream of shifting sub\-tasks\.

Despite these advances, it remains unclear how stability and plasticity trade off in more real\-world\-like environments that undergo naturalistic, continual non\-stationarity, where agents must adapt to naturalistic and continual changes without explicit task boundaries\. A natural way to study this problem is to develop environments with naturalistic, continually evolving dynamics, and compare algorithms that are task agnostic, and either enhance plasticity \(e\.g\., parameter resets\(Nikishin et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib34),[2023](https://arxiv.org/html/2605.26357#bib.bib35); Sokar et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib44); Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14); Lee et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib29)\)\) or preserve stability \(e\.g\., consolidation that either protects important parameters\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27)\)or allow learning across multiple timescales\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21),[2019](https://arxiv.org/html/2605.26357#bib.bib22); Anand & Precup,[2023](https://arxiv.org/html/2605.26357#bib.bib3)\)\)\. While other approaches exist, such as replay\-based methods\(Riemer et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib38); Rolnick et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib39); Caccia et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib11)\), they are less suited to our setting since their benefits rely on storing and mixing past and recent samples, which is problematic when no clear task boundaries exist\. More broadly, prior approaches tackle the stability\-plasticity trade\-off at the level of Q\-values or policies, leaving the role of representations largely understudied\.

In this paper, we investigate whether predictive representations can offer a principled solution to the plasticity–stability dilemma under naturalistic, continual non\-stationarity\. We focus on Successor Features \(SFs\)\(Barreto et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib4); Borsa et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib9); Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\), which capture predictive structure and enable transfer across tasks with shared dynamics, and study whether they can simultaneously support rapid adaptation and resistance to interference\. We evaluate this question in two environments with continuously evolving dynamics: a slippery four rooms environment, where actions are occasionally replaced, and MuJoCo control tasks, where the embodiment’s mass varies over time\. These sources of non\-stationarity reflect realistic changes in action outcomes \(e\.g\., wet or icy ground\) and body dynamics\. Non\-stationarity is induced via continuous stochastic drift processes, including noisy sinusoidal dynamics\(Xie et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib51)\), as well as its non\-periodic variant and Ornstein–Uhlenbeck \(OU\) drift\.

In summary, our main contributions in this paper are:

1. 1\.A naturalistic continual non\-stationarity evaluation protocol\.We introduce a continual RL setup with smooth, continuous non\-stationarity and no explicit task boundaries, instantiated using periodic, or non\-periodic stochastic sine functions or OU dynamics, in both navigation and continuous control domains\.
2. 2\.A controlled diagnosis of the plasticity–stability trade\-off\.By systematically comparing mechanisms that inject plasticity with those that preserve stability, we provide evidence that performance degradation under continuous non\-stationarity is primarily driven by instability rather than insufficient plasticity\.
3. 3\.A novel integration of SFs with multi\-timescale synaptic consolidation\.We propose a principled framework that combines predictive representations \(SFs\) with synaptic consolidation across multiple timescales, enabling stable learning under continuous non\-stationarity while preserving adaptability\.
4. 4\.Interpretability of predictive representations across timescales\.We use cross\-attention over SFs learned at different consolidation timescales as a diagnostic tool to quantify their relative contributions, providing new insights into how stability and plasticity are distributed over temporal dimensions\.

![Refer to caption](https://arxiv.org/html/2605.26357v1/x1.png)Figure 1:Motivating stability\-plasticity tradeoffs in naturalistic, continually non\-stationary RL where the environment evolves gradually, rather than abruptly\. To illustrate, we show\(a\)the Humanoid walking forward task and\(b\)an example of the noisy sine function used to generate smooth changes in its mass\.\(c\)Average episode return plot and\(d\)Area under the curve \(AUC\) show that stability\-preserving methods \(EWC, SC\) outperform purely plastic ones \(CBP, P\-last\), with further gains from consolidating SFs \(SF\+SC, purple\) rather than Q\-values \(TD3\+SC, green\)\. Plasticity injection for TD3\+P\-last \(yellow\) was performed halfway through the training\.
## 2Related work

Our work builds upon prior studies of stability or plasticity in RL\. Early work to mitigate forgetting emphasized stability, introducing methods that use importance measures such as Fisher information to protect parameters critical for previous tasks\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27); Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41)\), manipulating replay mechanisms\(Rolnick et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib39); Riemer et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib38); Kaplanis et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib23); Caccia et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib11)\), augmenting architectures\(Powers et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib37)\), or employing consolidation systems that maintain multiple sets of parameters updated at different timescales\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21),[2019](https://arxiv.org/html/2605.26357#bib.bib22); Anand & Precup,[2023](https://arxiv.org/html/2605.26357#bib.bib3)\)\. Several approaches explicitly rely on task information, for example by using task boundaries to trigger consolidation\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27)\)or distillation phases\(Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41)\)\. Other approaches that are described as task\-agnostic, but nevertheless rely on auxiliary mechanisms such as recency tracking by separating “new” or “replay” samples\(Rolnick et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib39)\), or the use of drift detection mechanisms that trigger architecture adaptations\(Powers et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib37)\)\. These assumptions are problematic in environments that evolve naturally and continually, where there are no discrete task boundaries to detect and the very notion of “new” versus “old” experiences becomes ill\-defined\.

More recently, studies in continual RL have shifted attention to the problem of loss of plasticity\. By analyzing neural activities, effective rank of the representations, and gradient dynamics during training, proposed mitigation strategies have focused on modifying the activation functions or optimizers\(Ben\-Iwhiwhu et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib6); Abbas et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib1)\), regularizing the parameters using weight decay or normalization\(Lyle et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib31)\), and, more commonly, injecting plasticity by resetting subsets of network parameters such as the last few layers\(Nikishin et al\.,[2022](https://arxiv.org/html/2605.26357#bib.bib34),[2023](https://arxiv.org/html/2605.26357#bib.bib35)\)or the ones that are least active\(Sokar et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib44); Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)\. However, most of these approaches have been evaluated only in discrete or single\-task settings, or under non\-stationarity that is abrupt rather than naturally and continually\.

Among these prior approaches, our study is most closely related to consolidation\-based approaches\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27); Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41); Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21)\)and to recent efforts examining loss of plasticity in deep RL\(Nikishin et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib35); Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)as they do not require explicit or implicit task statistics\. However, they have yet to be evaluated under naturalistic, continually evolving settings, and remain limited to discrete tasks or single\-task settings\. Moreover, whether such approaches are effective when applied to learned representations, rather than Q\-values or policies, remain unclear\. In this work, we address these limitations by analyzing stability and plasticity under naturalistic, continual changes, and by proposing a synaptic consolidation system with SFs, that consolidates representations across multiple timescales\.

## 3Preliminaries

### 3\.1Reinforcement Learning under Continuous Non\-Stationarity

A Markov Decision Process \(MDP\) defined by the tuple\(𝒮,𝒜,p,r,γ\)\(\\mathcal\{S\},\\mathcal\{A\},p,r,\\gamma\), where𝒮\\mathcal\{S\}and𝒜\\mathcal\{A\}denote the state and action spaces,p​\(s′∣s,a\)p\(s^\{\\prime\}\\mid s,a\)is the transition function,r:𝒮→ℝr:\\mathcal\{S\}\\rightarrow\\mathbb\{R\}is the reward function, andγ∈\[0,1\)\\gamma\\in\[0,1\)is the discount factor\(Sutton & Barto,[2018](https://arxiv.org/html/2605.26357#bib.bib45)\)\.

At each time steptt, the agent observesSt∈𝒮S\_\{t\}\\in\\mathcal\{S\}, selects an actionAt∼π\(⋅∣St\)A\_\{t\}\\sim\\pi\(\\cdot\\mid S\_\{t\}\), transitions toSt\+1∼p\(⋅∣St,At\)S\_\{t\+1\}\\sim p\(\\cdot\\mid S\_\{t\},A\_\{t\}\), and receives rewardRt\+1R\_\{t\+1\}\.

Standard MDPs assume stationary dynamics, i\.e\., a fixed transition functionp​\(s′∣s,a\)p\(s^\{\\prime\}\\mid s,a\)\. However, real\-world environments are often non\-stationary\. Prior work in continual reinforcement learning typically models such non\-stationarity as a sequence of discrete tasks with abrupt changes\.

In contrast, we considercontinuous non\-stationarity, where the environment evolves gradually over time\. We introduce a time\-varying latent parameterωt∈Ω\\omega\_\{t\}\\in\\Omegathat modulates the transition dynamics, yielding a sequence of MDPs:

ℳt=\(𝒮,𝒜,pωt,r,γ\),\\mathcal\{M\}\_\{t\}=\(\\mathcal\{S\},\\mathcal\{A\},p\_\{\\omega\_\{t\}\},r,\\gamma\),\(1\)wherepωt​\(s′∣s,a\)≡p​\(s′∣s,a;ωt\)p\_\{\\omega\_\{t\}\}\(s^\{\\prime\}\\mid s,a\)\\equiv p\(s^\{\\prime\}\\mid s,a;\\omega\_\{t\}\)varies smoothly withωt\\omega\_\{t\}\.

We assumeωt\\omega\_\{t\}evolves according to a continuous stochastic process \(e\.g\., a noisy sine wave\), resulting in gradual changes in dynamics rather than abrupt task switches\. This setting capturesnaturalistic non\-stationarity, where the agent must continually adapt to drifting conditions while retaining prior knowledge\. The latent variableωt\\omega\_\{t\}is not observed by the agent, and its evolution may revisit similar values over time, leading to recurring dynamics \(Figure[1](https://arxiv.org/html/2605.26357#S1.F1)b\)\.

### 3\.2Successor Features

Successor Features \(SFs\) provide a decomposition of the state\-action value function into reward parameters and predictive representations of future feature occupancy:

Q​\(St,At,𝒘\)=ψ​\(St,At,𝒘\)⊤​𝒘,Q\(S\_\{t\},A\_\{t\},\\boldsymbol\{w\}\)=\\psi\(S\_\{t\},A\_\{t\},\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\},\(2\)whereψ∈ℝn\\psi\\in\\mathbb\{R\}^\{n\}captures the expected discounted occupancy of features, and𝒘∈ℝn\\boldsymbol\{w\}\\in\\mathbb\{R\}^\{n\}parameterizes the reward function\(Borsa et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib9)\)\.

Canonically, SFs for a state\-action pair\(s,a\)\(s,a\)under a policyπ\\piare defined as:

ψπ​\(s,a\)=𝔼π​\[∑i=t∞γi−t​ϕ​\(Si\+1\)∣St=s,At=a\],\\psi^\{\\pi\}\(s,a\)=\\mathbb\{E\}^\{\\pi\}\\left\[\\sum\_\{i=t\}^\{\\infty\}\\gamma^\{i\-t\}\\phi\(S\_\{i\+1\}\)\\mid S\_\{t\}=s,A\_\{t\}=a\\right\],\(3\)whereϕ∈ℝn\\phi\\in\\mathbb\{R\}^\{n\}denotes basis features\(Barreto et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib4)\)\. The reward can be expressed as a linear function of these features:

Rt\+1=ϕ​\(St\+1\)⊤​𝒘\.R\_\{t\+1\}=\\phi\(S\_\{t\+1\}\)^\{\\top\}\\boldsymbol\{w\}\.\(4\)
Prior work has primarily leveraged SFsψ\\psifortransfer learning under stationary dynamics, where the transition function,p​\(s′∣s,a\)p\(s^\{\\prime\}\\mid s,a\), remains fixed and only the reward parameters𝒘\\boldsymbol\{w\}change across tasks\. In such settings, SFs enable efficient generalization by reusing learned predictive representations across different reward functions\(Barreto et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib4); Borsa et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib9)\)\.

In contrast, this work considers environments withnon\-stationary transition dynamics, where the underlying dynamics evolve continuously over time, induced by the time\-varying latent parameterωt\\omega\_\{t\}\. In our settings, both basis featuresϕ\\phiand SFsψ\\psimust adapt to changing dynamics,p​\(s′∣s,a;ωt\)p\(s^\{\\prime\}\\mid s,a;\\omega\_\{t\}\), potentially leading to instability and interference\. This raises the key question of whether SFs themselves alone can remain effective under naturalistic, continuously changing dynamics, whether they suffer from losses in stability or plasticity, and how such limitations, if exist, can be mitigated\.

To study this, we build on Simple SFs\(Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\), which learn SFs directly during interaction without auxiliary losses or pre\-training\. Bothψ\\psiand𝒘\\boldsymbol\{w\}are learned jointly via:

Lw=12​‖Rt\+1−ϕ¯​\(St\+1\)⊤​𝒘‖2,L\_\{w\}=\\frac\{1\}\{2\}\\left\\\|R\_\{t\+1\}\-\\overline\{\\phi\}\(S\_\{t\+1\}\)^\{\\top\}\\boldsymbol\{w\}\\right\\\|^\{2\},\(5\)Lψ=12​‖y^−ψ​\(St,At,𝒘\)⊤​𝒘‖2,L\_\{\\psi\}=\\frac\{1\}\{2\}\\left\\\|\\hat\{y\}\-\\psi\(S\_\{t\},A\_\{t\},\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}\\right\\\|^\{2\},\(6\)whereϕ¯​\(St\+1\)\\overline\{\\phi\}\(S\_\{t\+1\}\)is the L2\-normalized feature representation treated as constant via a stop\-gradient operator\. The bootstrapped target is:

y^=Rt\+1\+γ​maxa′⁡ψ​\(St\+1,a′,𝒘\)⊤​𝒘\.\\hat\{y\}=R\_\{t\+1\}\+\\gamma\\;\\max\_\{a^\{\\prime\}\}\\;\\psi\(S\_\{t\+1\},a^\{\\prime\},\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}\.\(7\)
![Refer to caption](https://arxiv.org/html/2605.26357v1/x2.png)Figure 2:a:Neuro\-inspired synaptic consolidation model adapted from\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)\. The visible variable,u1u\_\{1\}, represents the synaptic efficacyvv, while downstream hidden variablesu2,u3,…u\_\{2\},u\_\{3\},\.\.\.interact bidirectionally across timescales, with beaker capacitiesC1<C2<…,<CKC\_\{1\}<C\_\{2\}<\.\.\.,<C\_\{K\}and tube widths representing flow strengthg1,2\>g2,3\>…,\>gK,K\+1g\_\{1,2\}\>g\_\{2,3\}\>\.\.\.,\>g\_\{K,K\+1\}controlling the rate of interaction between the variables\. Together, the beaker sizes \(CkC\_\{k\}\) and flow strength \(gk,k\+1g\_\{k,k\+1\}\) govern the effective timescales of plasticity and stability\.b:The synaptic efficacyvvis replaced by the parameters of SFs, thus allowing SFs to be learned across different timescales\.c:Our architectural design\. See section[4](https://arxiv.org/html/2605.26357#S4)for more details on training the system\.#### 3\.2\.1Role of the Reward Parameters

In our setting, the latent variableωt\\omega\_\{t\}induces time\-varying transition dynamics, but does not define a sequence of discrete tasks with distinct reward functions\. Instead, the reward remains a function of the stateStS\_\{t\}through the basis featuresϕ\\phi\(Eq\.[4](https://arxiv.org/html/2605.26357#S3.E4)\)\.

Accordingly,𝒘\\boldsymbol\{w\}should be interpreted as a vector of reward parameters that linearly combines basis featuresϕ\\phito predict rewards, rather than as a task identifier\. While𝒘\\boldsymbol\{w\}is learned online, it does not track the non\-stationarity induced byωt\\omega\_\{t\}\. Instead, adaptation to changing dynamics is primarily captured by the basis featuresϕ\\phiand the SFsψ\\psi\. FollowingBorsa et al\. \([2018](https://arxiv.org/html/2605.26357#bib.bib9)\), we condition SFs on the reward parameter𝒘\\boldsymbol\{w\}, i\.e\.,ψ​\(s,a,𝒘\)\\psi\(s,a,\\boldsymbol\{w\}\), ensuring consistency with the decompositionQ​\(s,a,𝒘\)=ψ​\(s,a,𝒘\)⊤​𝒘Q\(s,a,\\boldsymbol\{w\}\)=\\psi\(s,a,\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}\.

### 3\.3Neuro\-inspired Synaptic Consolidation Mechanism

In this work, we revisit the Synaptic Consolidation mechanism \(SC\)\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)which has been previously adapted to deep RL\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21),[2019](https://arxiv.org/html/2605.26357#bib.bib22)\)\. Despite these adaptations, SC remains far less studied than Systems Consolidation\(McClelland et al\.,[1995](https://arxiv.org/html/2605.26357#bib.bib32)\)in AI\.

We revisit SC because \(i\) it can be learned without explicit or implicit task statistics, \(ii\) it generalizes beyond dual fast/slow schemes\(Anand & Precup,[2023](https://arxiv.org/html/2605.26357#bib.bib3); Lee et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib29)\)by supporting multiple timescales, \(iii\) its linear\-chain formulation provides a principled balance of stability and plasticity without ad\-hoc mechanisms, and \(iv\) it has already been shown to be effective in deep RL\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21),[2019](https://arxiv.org/html/2605.26357#bib.bib22)\)\. In summary, the multi\-timescale mechanism promotes both rapid adaptation and stability\. Fast timescale components enable behaviour consistent withfunctional plasticity, defined as the observable ability of an agent to adapt under changing dynamics\.

To make this concrete, we now outline the SC mechanism — originally proposed to model how synaptic strength stabilizes over time\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)— which can be understood as a chain ofKKinteracting variables,u1,u2,…,uKu\_\{1\},u\_\{2\},\\ldots,u\_\{K\}, each associated with a capacityCk∈ℤ\+C\_\{k\}\\in\\mathbb\{Z\}\{\+\}\. The first variableu1u\_\{1\}, corresponds to visible synaptic efficacyvv, i\.e\., the strength of the connection between two neurons, and is the most plastic component \(see Figure[2](https://arxiv.org/html/2605.26357#S3.F2)a\. for a schematic of this system\)\. Its dynamics are:

C1​du1d​t=d​vd​t\+g1,2​\(u2−u1\),for k = 1\\displaystyle C\_\{1\}\\frac\{d\_\{u\_\{1\}\}\}\{dt\}=\\frac\{dv\}\{dt\}\+g\_\{1,2\}\(u\_\{2\}\-u\_\{1\}\),\\quad\\text\{for k = 1\}\(8\)whereg1,2∈ℝg\_\{1,2\}\\in\\mathbb\{R\}determines the flow strength betweenu1u\_\{1\}andu2u\_\{2\}\.

Interior variablesuku\_\{k\}\(for k = 2,3, …, K\-1\), interact bidirectionally with their two neighbors:

Ck​dukd​t=gk−1,k​\(uk−1−uk\)\+gk,k\+1​\(uk\+1−uk\)\\displaystyle C\_\{k\}\\frac\{d\_\{u\_\{k\}\}\}\{dt\}=g\_\{k\-1,k\}\(u\_\{k\-1\}\-u\_\{k\}\)\+g\_\{k,k\+1\}\(u\_\{k\+1\}\-u\_\{k\}\)\(9\)withgk−1,k,gk,k\+1∈ℝg\_\{k\-1,k\},g\_\{k,k\+1\}\\in\\mathbb\{R\}\. Finally, the last variableuKu\_\{K\}, has no downstream neighbor\. Thus, settinguK\+1←0u\_\{K\+1\}\\leftarrow 0produces a natural leak term that induces decay:

CK​duKd​t=gK−1,K​\(uK−1−uK\)\+gK,K\+1​\(−uK\)\\displaystyle C\_\{K\}\\frac\{d\_\{u\_\{K\}\}\}\{dt\}=g\_\{K\-1,K\}\(u\_\{K\-1\}\-u\_\{K\}\)\+g\_\{K,K\+1\}\(\-u\_\{K\}\)\(10\)Together, the capacityCkC\_\{k\}and the flow strengthgk,k\+1g\_\{k,k\+1\}define the continuous timescales of plasticity and stability of each variableuku\_\{k\}\. To implement these dynamics in RL, which operates in discrete steps, we discretize them with Euler’s method\.

![Refer to caption](https://arxiv.org/html/2605.26357v1/x3.png)Figure 3:Results from Slippery Four Rooms with naturalistic, continual evolving slip dynamics that randomly replace actions\.\(a\):Average return across two sequential tasks \(Task 1 and 2\), each repeated twice \(Exposure 1 and 2\)\. In DQN\+P\-last \(yellow\), plasticity injection is applied midway through training by randomly re\-initializing the last layer’s parameters\.\(b\):Steps to reach a predefined performance threshold \(fewer is better\)\. Overall, stability\-preserving methods \(EWC, SC\) outperform plastic ones \(P\-last\), with further gains from combining SFs with SC \(SF\+SC\)\.![Refer to caption](https://arxiv.org/html/2605.26357v1/x4.png)Figure 4:Slippery Four Rooms

## 4Learning Successor Features with Synaptic Consolidation

In line with prior work which discretize synaptic consolidation by replacing synaptic efficacyvvwith the parameters of a Q\-value function\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21)\)or a policy\(Kaplanis et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib22)\),*we instead apply synaptic consolidation to the parameters of the SFs\.*Specifically, each variableuku\_\{k\}is mapped to the corresponding SF parametersθk∈ℝn\\theta\_\{k\}\\in\\mathbb\{R\}^\{n\}, yieldingψuk=ψθuk∈ℝn\\psi\_\{u\_\{k\}\}=\\psi\_\{\\theta\_\{u\_\{k\}\}\}\\in\\mathbb\{R\}^\{n\}, whereψ\\psidenotes the SFs\. For brevity, we will writeψuk\\psi\_\{u\_\{k\}\}to denoteψθuk\\psi\_\{\\theta\_\{u\_\{k\}\}\}\. We next derive the learning rules using Euler’s method\.

Letηk=Δ​t/Ck\\eta\_\{k\}=\\Delta t/C\_\{k\}\. There will be two learning phases \(t\+12t\+\\frac\{1\}\{2\}andt\+1t\+1\) for the most plastic variable, SFψu1\\psi\_\{u\_\{1\}\}\. At phaset\+12t\+\\frac\{1\}\{2\}, SFψu1\\psi\_\{u\_\{1\}\}is learned via optimizing Q\-SF\-TD lossLψL\_\{\\psi\}\(Eq\.[6](https://arxiv.org/html/2605.26357#S3.E6)\):

ψu1t\+12=ψu1t−α​∇ψu1Lψu1\\displaystyle\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}=\\psi\_\{u\_\{1\}\}^\{t\}\-\\alpha\\nabla\_\{\\psi\_\{u\_\{1\}\}\}L\_\{\\psi\_\{u\_\{1\}\}\}\(11\)whereα∈ℝ\\alpha\\in\\mathbb\{R\}is the learning rate\. At the second phase,t\+1t\+1, we updateψu1\\psi\_\{u\_\{1\}\}and the rest of the SFs variable \(ψu2,ψu3,…,ψuK\\psi\_\{u\_\{2\}\},\\psi\_\{u\_\{3\}\},\\ldots,\\psi\_\{u\_\{K\}\}\) using the Euler update\. For the first variableψu1\\psi\_\{u\_\{1\}\}:

ψu1t\+1=ψu1t\+12\+η1​\[g1,2​\(ψu2t−ψu1t\+12\)\]\\displaystyle\\psi\_\{u\_\{1\}\}^\{t\+1\}=\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\+\\eta\_\{1\}\\big\[g\_\{1,2\}\\big\(\\psi\_\{u\_\{2\}\}^\{t\}\-\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\\big\)\\big\]\(12\)For the interior variablesk=2,…,K−1k=2,\\ldots,K\-1:

ψukt\+1\\displaystyle\\psi\_\{u\_\{k\}\}^\{t\+1\}=ψukt\+ηk\[gk−1,k\(ψuk−1t−ψukt\)\\displaystyle=\\psi\_\{u\_\{k\}\}^\{t\}\+\\eta\_\{k\}\\big\[g\_\{k\-1,k\}\(\\psi\_\{u\_\{k\-1\}\}^\{t\}\-\\psi\_\{u\_\{k\}\}^\{t\}\)\+gk,k\+1\(ψuk\+1t−ψukt\)\]\\displaystyle\+g\_\{k,k\+1\}\(\\psi\_\{u\_\{k\+1\}\}^\{t\}\-\\psi\_\{u\_\{k\}\}^\{t\}\)\\big\]\(13\)For the last variableKK:

ψuKt\+1\\displaystyle\\psi\_\{u\_\{K\}\}^\{t\+1\}=ψuKt\+ηK\[gK−1,K\(ψuK−1t−ψuKt\)\\displaystyle=\\psi\_\{u\_\{K\}\}^\{t\}\+\\eta\_\{K\}\\big\[g\_\{K\-1,K\}\(\\psi\_\{u\_\{K\-1\}\}^\{t\}\-\\psi\_\{u\_\{K\}\}^\{t\}\)−gK,K\+1\(ψuKt\)\]\\displaystyle\-g\_\{K,K\+1\}\(\\psi\_\{u\_\{K\}\}^\{t\}\)\\big\]\(14\)
We provide a pseudocode of the algorithm in Appendix[C](https://arxiv.org/html/2605.26357#A3)\. It is important that these updates \(Eqs\.[12](https://arxiv.org/html/2605.26357#S4.E12),[13](https://arxiv.org/html/2605.26357#S4.E13)and[14](https://arxiv.org/html/2605.26357#S4.E14)\) are performed using Stochastic Gradient Descent \(SGD\) rather than adaptive approaches like Adaptive Moment Estimation \(Adam,\(Kingma & Ba,[2014](https://arxiv.org/html/2605.26357#bib.bib25)\)\), which do not preserve the timescales information\. We provide a proof sketch in Appendix[B](https://arxiv.org/html/2605.26357#A2)supporting this claim\.

## 5Experimental results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x5.png)Figure 5:Results from MuJoCo, where agent embodiments undergo continuous mass changes during training\.\(a\):Average episode return\. Plasticity injection for TD3\+P\-last \(yellow\) was performed halfway through the training\.\(b\):Area under the Curve \(AUC\) of returns in \(a\)\. Together, the plots show that stability\-preserving methods \(EWC and SC\) outperform plastic ones \(CBP and P\-last\), with further gains achieved by combining synaptic consolidation with SFs \(SF\+SC\)\. Results for Humanoid can be seen in Figure[1](https://arxiv.org/html/2605.26357#S1.F1)\(c\-d\)\.In this study, we consider two environments\. The first is a slippery variant of the 3D Four Rooms environments, adapted fromChua et al\. \([2024](https://arxiv.org/html/2605.26357#bib.bib13)\), which mimics conditions such as walking on wet or icy surfaces\. In this environment, the “slippery” event refers to the agent’s action being randomly replaced by an alternative, based on a probability value sampled from a noisy sine function to simulate continual dynamics shifts \(Figure[10](https://arxiv.org/html/2605.26357#A5.F10)in Appendix[E](https://arxiv.org/html/2605.26357#A5)\)\.

The agent alternates between two tasks, where in the first task it receives a reward of \+1 for reaching the green box and \-1 for reaching the yellow box, and in the second task the rewards are reversed\. The agent cycles through this two\-task sequence twice and only receives egocentric pixel observations \(Figure[4](https://arxiv.org/html/2605.26357#S3.F4)\)\.

The second environment is the MuJoCo suite\(Todorov et al\.,[2012](https://arxiv.org/html/2605.26357#bib.bib46)\), using the DeepMind Control Suite \(DMC\)\(Tunyasuvunakool et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib47)\), which provides an accessible framework for modifying dynamics of the embodiments\. We focus on four embodiments, Half\-cheetah, Walker, Quadruped and Humanoid, ordered by increasing complexity in terms of their observation and action spaces\. The agents are rewarded for walking or running forward\. To simulate the continual dynamics shifts, at every ten episodes, we perturbed the embodiment’s mass by sampling from a noisy sine function\. See Appendix[D](https://arxiv.org/html/2605.26357#A4)for code availability\.

For base models, we use Double Deep Q\-Network \(DQN\)\(Van Hasselt et al\.,[2016](https://arxiv.org/html/2605.26357#bib.bib48)\)for Slippery Four Rooms environment and the Deterministic Policy Gradient algorithm\(Silver et al\.,[2014](https://arxiv.org/html/2605.26357#bib.bib43)\)with twin critics for MuJoCo \(TD3\)\(Fujimoto et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib17)\)\. For SFs, we use Simple SFs\(Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\)which can be learned without auxiliary losses\. We selected these models due to their flexibility, which makes extensions with plasticity injections or synaptic consolidation mechanisms rather straightforward\.

We compared against baselines that do not require task statistics\. For plasticity, we reset subsets of parameters—last layer \(DQN\+P\-last, TD3\+P\-last\)\(Nikishin et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib35)\)or least used via continual backprop \(TD3\+CBP\)\(Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\), the latter more effective in MuJoCo\. For stability, we used online Elastic Weight Consolidation \(EWC\)\(Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41)\)for Q\-values and SFs \(DQN\+EWC, TD3\+EWC, SF\+EWC\)\. We also included synaptic consolidation \(SC, Figure[2](https://arxiv.org/html/2605.26357#S3.F2)\) with Q\-value \(DQN\+SC, TD3\+SC\) and SF variants \(SF\+SC\)\. Results are averaged over 5 seeds\.

Our experiments are designed to address the following questions:

1. 1\.When agents undergo naturalistic, continual non\-stationary shifts, is the primary bottleneck one of plasticity or stability? If stability is the limiting factor, which consolidation mechanism is more effective? EWC or SC?
2. 2\.Second, is it more effective to consolidate parameters of Q\-value or Successor Features?
3. 3\.Third, will SC remain effective under continual non\-periodic or stochastic drifts?

### 5\.1Stability as the Bottleneck: Synaptic Consolidation outperforms EWC

![Refer to caption](https://arxiv.org/html/2605.26357v1/x6.png)Figure 6:Quantification of mass changes for the Humanoid embodiment\. We consider three levels of mass dynamics variation: mild \(25%\), moderate \(50%\), and severe \(100%\), corresponding to the maximum change allowed before the physical simulation becomes unstable\. Across these settings, plasticity\-preserving methods \(CBP, P\-last\) are less effective than approaches incorporating synaptic consolidation \(SC\)\. Under moderate—and especially severe—conditions, where the agent experiences substantial changes in dynamics, applying SC to Successor Features \(SF \+ SC\) yields the best performance\. In contrast, under mild changes, the benefits of SC are limited, as the environment remains close to stationary\. For other results, please refer to Appendix[T](https://arxiv.org/html/2605.26357#A20)\.![Refer to caption](https://arxiv.org/html/2605.26357v1/x7.png)Figure 7:Analysis of timescales in MuJoCo using our model, SF \+ SC\. More consolidation variables \(6–9\) improve learning efficiency, highlighting the benefit of slower timescales\. Zero variables correspond to the Simple SF agent\. See Appendix[M](https://arxiv.org/html/2605.26357#A13)for results in the Slippery Four Rooms environment\.To study the question of plasticity versus stability, we first evaluated the performance in the slippery Four Rooms environment using DQN, along with its plasticity injection variants \(P\-last\) and stability\-preserving variants \(EWC and SC\)\. The results in Figure[3](https://arxiv.org/html/2605.26357#S3.F3)show that stability\-preserving models \(DQN \+ EWC and DQN \+ SC\) consistently outperformed the plasticity\-injection model \(DQN \+ P\-last\), with synaptic consolidation \(DQN \+ SC\) achieving higher learning efficiency\.

We next evaluated performance in the MuJoCo suite using TD3, together with its two plasticity\-injection variants \(P\-last and CBP\) and stability\-preserving variants \(EWC and SC\)\. The results in Figure[5](https://arxiv.org/html/2605.26357#S5.F5)show that stability\-preserving models \(TD3 \+ EWC and TD3 \+ SC\) consistently outperformed the plasticity\-injection model \(TD3 \+ P\-last and TD3 \+ CBP\)\. Once again, synaptic consolidation \(TD3 \+ SC\) achieved higher learning efficiency, particularly in the more complex embodiments, Humanoid \(Figure[1](https://arxiv.org/html/2605.26357#S1.F1)\) and Quadruped \(Figure[5](https://arxiv.org/html/2605.26357#S5.F5)a and b\)\.

Both sets of results indicate thatstability is the primary bottleneckas agents lacking stability fail to learn effectively in the naturalistic, continual non\-stationary environments\. The results further showed that the synaptic consolidation \(SC\) mechanism is more effective than EWC\.

Next, we asked, what exactly should be stabilized: the Q\-value parameters themselves, or the parameters of the underlying representations such as SFs?

### 5\.2What should be stabilized: Q\-values or Successor Features?

As many stability preserving methods focus on stabilizing parameters of Q\-value functions, we investigate if SFs could be a better target for consolidation\. To address this, we evaluated a SFs variant combined with synaptic consolidation \(SF \+ SC\) in both slippery Four Rooms and the MuJoCo suite\. The results for both slippery Four Rooms \(Figure[3](https://arxiv.org/html/2605.26357#S3.F3)\) and the MuJoCo suite \(Figures[1](https://arxiv.org/html/2605.26357#S1.F1)and[5](https://arxiv.org/html/2605.26357#S5.F5)\) showed that SF \+ SC consistently improves performance compared to Q\-value based consolidation\. We also compared with EWC \(Appendix[K](https://arxiv.org/html/2605.26357#A11)\), which revealed that while SC is more effective than EWC overall, only the combination of SFs with SC yields consistently strong performance\.

![Refer to caption](https://arxiv.org/html/2605.26357v1/x8.png)Figure 8:Cross\-Attention analysis of individual consolidations, replacing memory recall via backflow\.\(a\):Implementation design\.\(b\-c\):Attention probabilities over consolidation variables, where higher probability indicates greater contribution to learning\. Full results are provided in Appendix[O](https://arxiv.org/html/2605.26357#A15)\.
### 5\.3Quantifying Non\-Stationarity via Mass Perturbations

To systematically study how the magnitude of environmental change affects learning dynamics, we parameterize non\-stationarity through controlled perturbations of body mass\. Specifically, we consider three regimes of variation—mild \(25%\), moderate \(50%\), and severe \(100%\)—defined relative to the maximum perturbation before the MuJoCo simulation becomes unstable\.111Unless otherwise stated, the main results presented correspond to the severe \(100%\) setting\.

Across Humanoid \(Figure[6](https://arxiv.org/html/2605.26357#S5.F6)\) and other embodiments, we observe a clear dependence on the degree of non\-stationarity\. Under mild variation, where the transition dynamics remain close to stationary, applying synaptic consolidation \(SC\) to Q\-values is most effective\. In contrast, as the magnitude of mass perturbation increases, SC applied to Successor Features \(SFs\) becomes increasingly advantageous, particularly under moderate and severe regimes where the environment undergoes substantial and continuous changes\.

### 5\.4Robustness to Non\-Periodic and Stochastic Drift

To test robustness beyond periodic drift, we replace the noisy sine modulation with either a non\-periodic sine function \(Appendix[E\.4](https://arxiv.org/html/2605.26357#A5.SS4)\) or Ornstein–Uhlenbeck \(OU\) process \(Appendix[E\.6](https://arxiv.org/html/2605.26357#A5.SS6)\)\. Results from mass changes induced by a non\-periodic sine function \(Appendix[H](https://arxiv.org/html/2605.26357#A8)\) and OU process \(Appendix[I](https://arxiv.org/html/2605.26357#A9)\) show that applying SC to SFs continues to improve learning performance\.

## 6Analyzing Multi\-Timescale Contributions

To gain insights into why combining SFs and SC yields an effective model, we perform ablation studies by varying the number of consolidation variables, and we complement this with a cross\-attention\(Dosovitskiy et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib15)\)analysis to analyze the relative contributions of individual variables\.

### 6\.1Do fast or slow timescale variables matter more for learning?

In this analysis, we varied over the number of consolidation variables \(3, 6, or 9\), with fewer variables yielding greater plasticity, and more variables yielding greater stability\. The aim was to assess whether the inclusion of slower timescale variables improve policy learning\. Figure[7](https://arxiv.org/html/2605.26357#S5.F7)illustrates the results using the embodiments from MuJoCo \(see Appendix[M](https://arxiv.org/html/2605.26357#A13)for more results\)\. We found that using six or more timescale variables leads to better learning performance\. Zero variables correspond to the Simple SF agent\. For the slippery four rooms environment \(Figure[29](https://arxiv.org/html/2605.26357#A13.F29)in Appendix[6\.1](https://arxiv.org/html/2605.26357#S6.SS1)\), we observed a similar trend: using six or nine consolidation variables leads to better performance\. Together, these results demonstrated that preserving stability through the use of slow timescales is crucial in our settings\.

### 6\.2What does cross\-attention reveal about the contributions of consolidation variables?

While varying over the number of consolidation variables provides a coarse measure of their utility, it does not reveal which specific variables contribute most\. At the same time, using a cross\-attention mechanism prevents the need for information to propagate gradually via the flow strength \(gk,k\+1g\_\{k,k\+1\}\), instead providing an instant readout from all the consolidation variables \(see Figure[8](https://arxiv.org/html/2605.26357#S5.F8)a\)\.

We adapt the cross\-attention mechanism by letting the reward weight vectorwwserves as the query, while the SF consolidation variables \(excluding the most plasticS​Fu1SF\_\{u\_\{1\}\}\) serve as keys and values\. The softmax probabilities from the cross\-attention mechanism \(Figure[8](https://arxiv.org/html/2605.26357#S5.F8)b and c\) showed that the faster timescale variables, particularlyS​Fu2SF\_\{u\_\{2\}\}andS​Fu3SF\_\{u\_\{3\}\}, received the highest attention\. However, slower variables also contributed\. Together, these findings suggest that fast timescales drive most learning, while slower ones provide complementary stability\.

### 6\.3Can Larger Networks Replace Multi\-Timescale Consolidation?

![Refer to caption](https://arxiv.org/html/2605.26357v1/x9.png)Figure 9:Capacity analysis using Humanoid\. X\-axis shows the number of parameters, while the y\-axis shows performance measured by area under the curve \(AUC\)\. Increasing the parameter count of TD3 and its variants did not consistently improve performance compared to SF \+ SC \(star\), suggesting the contribution of consolidating SFs beyond network capacity scaling alone\.Since each consolidation variable introduces an additional set of parameters, improvements from synaptic consolidation could potentially be attributed to increased model capacity rather than the consolidation dynamics themselves\. To control for this possibility, we compare against enlarged baseline networks \(TD3, TD3 \+ CBP, TD3 \+ EWC, TD3 \+ P\-last\) with parameter counts exceeding those of the SF\-based models \(SF, SF \+ EWC, SF \+ SC\)\.

The results in Figure[9](https://arxiv.org/html/2605.26357#S6.F9)show that, despite substantially increasing their capacity, TD3 and its variants fail to match the performance of SF \+ SC, which achieves the strongest performance while using significantly fewer parameters\. This suggests that the gains do not arise solely from increased model size, but rather from the combination of predictive representations and multi\-timescale consolidation dynamics\. From a scalability perspective, these findings further indicate that our approach is parameter\-efficient and capable of achieving strong continual learning performance\.

## 7Discussion

In this work, we introduced naturalistic continual non\-stationarity benchmarks to study the stability–plasticity trade\-off in deep RL\. All our experimental results suggest that stability is often the primary bottleneck, and that maintaining plasticity \(e\.g\., reset\-based methods\) is insufficient\. It is important to note that we do not directly measure plasticity, and evaluate adaptation through learning performance\.

We find that combining SC with SFs yields a multi\-timescale learning system in which slower components preserve stable predictive structure while faster components enable rapid adaptation\. Cross\-attention analysis indicates that different timescales contribute differently to behavior\. Importantly, this stability–plasticity balance does not emerge from policy constraints alone\. Compared with Proximal Policy Optimization \(PPO\)\(Schulman et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib40)\), which relies on trust\-region updates to stabilize learning, SFs combined with SC exhibited substantially greater robustness under continuous non\-stationarity \(Appendix[S](https://arxiv.org/html/2605.26357#A19)\)\.

However, several limitations remain\. First, the method is restricted to SGD\-based updates, as combining it with adaptive optimizers like Adam can lead to instability\. Second, it introduces computational overhead due to non\-parallelizable analytical updates, which grows with the number of timescales\. More extreme settings such as multi\-agents remain interesting directions for future research\.

## 8Acknowledgments

We would like to express our deepest gratitude to Christos Kaplanis for his tremendous support throughout this project\. From its earliest stages, Christos generously shared his expertise, insights, and experience on applying synaptic consolidation mechanisms to value and policy networks, building on his prior work\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21),[2019](https://arxiv.org/html/2605.26357#bib.bib22)\)\. His guidance and discussions played an important role in shaping many of the ideas explored in this paper, and we are sincerely grateful for his encouragement and collaboration throughout this journey\.

We would also like to thank Guillaume Lajoie, Razvan Pascanu, Claudia Clopath, Rui Ponte Costa, Marcus Benna, and Stefano Fusi for insightful discussions on continual reinforcement learning and the role synaptic consolidation can play in supporting continual adaptation and stability\.

We would also like to thank Paul Masset, Isabeau Prémont\-schwarz, Nanda Harishankar Krishna, Roy Henha Eyono, Hafez Ghaemi, and Maren Wehrheim for their thoughtful feedback and for reviewing earlier drafts of the manuscript\.

We are also grateful for the wonderful research community at Mila222[https://mila\.quebec/en](https://mila.quebec/en), McGill University and in Montréal for fostering an inspiring and collaborative environment that motivated us to pursue ambitious and interdisciplinary research at the intersection of Neuroscience and Artificial Intelligence \(NeuroAI\)\.

Raymond Chua was supported by the DeepMind Graduate Award and UNIQUE Excellence Scholarship \(PhD\)\. We extend our gratitude to the FRQNT Strategic Clusters Program \(2020\-RS4\-265502 \- Centre UNIQUE \- Quebec Neuro\-AI Research Center\)\.

Blake A\. Richards was supported by NSERC \(Discovery Grant RGPIN\-2020\- 05105, RGPIN\-2018\-04821; Discovery Accelerator Supplement: RGPAS\-2020\-00031\), CIFAR \(Canada AI Chair; Learning in Machine and Brains Fellowship\), and by funds provided by the National Science Foundation and DoD OUSD \(R&E\) under Cooperative Agreement DBI\-2229929 \(The NSF AI Institute for Artificial and Natural Intelligence\)\.

This research was also enabled by computational resources provided by Calcul Québec333[https://www\.calculquebec\.ca/](https://www.calculquebec.ca/)and the Digital Research Alliance of Canada444[https://alliancecan\.ca/en](https://alliancecan.ca/en)\. The authors acknowledge the material support of NVIDIA in the form of computational resources\.

Last but not least, we are also grateful to the anonymous reviewers whose insightful comments and suggestions significantly enhanced the quality of this manuscript\.

## 9Impact statement

Our studies primarily revolve around navigation tasks and embodiments simulations, with the techniques developed being most pertinent to the field of robotics\. Given this specific focus, the broader societal implications of our work are likely to be quite limited\.

## References

- Abbas et al\. \(2023\)Abbas, Z\., Zhao, R\., Modayil, J\., White, A\., and Machado, M\. C\.Loss of plasticity in continual deep reinforcement learning\.In*Conference on lifelong learning agents*, pp\. 620–636\. PMLR, 2023\.
- Abel et al\. \(2023\)Abel, D\., Barreto, A\., Van Roy, B\., Precup, D\., van Hasselt, H\. P\., and Singh, S\.A definition of continual reinforcement learning\.*Advances in Neural Information Processing Systems*, 36:50377–50407, 2023\.
- Anand & Precup \(2023\)Anand, N\. and Precup, D\.Prediction and control in continual reinforcement learning\.*Advances in Neural Information Processing Systems*, 36:63779–63817, 2023\.
- Barreto et al\. \(2017\)Barreto, A\., Dabney, W\., Munos, R\., Hunt, J\. J\., Schaul, T\., van Hasselt, H\. P\., and Silver, D\.Successor features for transfer in reinforcement learning\.*Advances in neural information processing systems*, 30, 2017\.
- Bellemare et al\. \(2013\)Bellemare, M\. G\., Naddaf, Y\., Veness, J\., and Bowling, M\.The arcade learning environment: An evaluation platform for general agents\.*Journal of artificial intelligence research*, 47:253–279, 2013\.
- Ben\-Iwhiwhu et al\. \(2022\)Ben\-Iwhiwhu, E\., Nath, S\., Pilly, P\. K\., Kolouri, S\., and Soltoggio, A\.Lifelong reinforcement learning with modulating masks\.*arXiv preprint arXiv:2212\.11110*, 2022\.
- Benna & Fusi \(2016\)Benna, M\. K\. and Fusi, S\.Computational principles of synaptic memory consolidation\.*Nature neuroscience*, 19\(12\):1697–1706, 2016\.
- Biewald \(2020\)Biewald, L\.Experiment tracking with weights and biases, 2020\.URL[https://www\.wandb\.com/](https://www.wandb.com/)\.Software available from wandb\.com\.
- Borsa et al\. \(2018\)Borsa, D\., Barreto, A\., Quan, J\., Mankowitz, D\., Munos, R\., Van Hasselt, H\., Silver, D\., and Schaul, T\.Universal successor features approximators\.*arXiv preprint arXiv:1812\.07626*, 2018\.
- Bradbury et al\. \(2018\)Bradbury, J\., Frostig, R\., Hawkins, P\., Johnson, M\. J\., Leary, C\., Maclaurin, D\., Necula, G\., Paszke, A\., VanderPlas, J\., Wanderman\-Milne, S\., and Zhang, Q\.JAX: composable transformations of Python\+NumPy programs, 2018\.URL[http://github\.com/google/jax](http://github.com/google/jax)\.
- Caccia et al\. \(2023\)Caccia, M\., Mueller, J\., Kim, T\., Charlin, L\., and Fakoor, R\.Task\-agnostic continual reinforcement learning: Gaining insights and overcoming challenges\.In*Conference on Lifelong Learning Agents*, pp\. 89–119\. PMLR, 2023\.
- Chevalier\-Boisvert et al\. \(2023\)Chevalier\-Boisvert, M\., Dai, B\., Towers, M\., de Lazcano, R\., Willems, L\., Lahlou, S\., Pal, S\., Castro, P\. S\., and Terry, J\.Minigrid & miniworld: Modular & customizable reinforcement learning environments for goal\-oriented tasks\.*CoRR*, abs/2306\.13831, 2023\.
- Chua et al\. \(2024\)Chua, R\., Ghosh, A\., Kaplanis, C\., Richards, B\. A\., and Precup, D\.Learning successor features the simple way\.*Advances in Neural Information Processing Systems*, 37:49957–50030, 2024\.
- Dohare et al\. \(2024\)Dohare, S\., Hernandez\-Garcia, J\. F\., Lan, Q\., Rahman, P\., Mahmood, A\. R\., and Sutton, R\. S\.Loss of plasticity in deep continual learning\.*Nature*, 632\(8026\):768–774, 2024\.
- Dosovitskiy et al\. \(2020\)Dosovitskiy, A\., Beyer, L\., Kolesnikov, A\., Weissenborn, D\., Zhai, X\., Unterthiner, T\., Dehghani, M\., Minderer, M\., Heigold, G\., Gelly, S\., et al\.An image is worth 16x16 words: Transformers for image recognition at scale\.*arXiv preprint arXiv:2010\.11929*, 2020\.
- French \(1999\)French, R\. M\.Catastrophic forgetting in connectionist networks\.*Trends in cognitive sciences*, 3\(4\):128–135, 1999\.
- Fujimoto et al\. \(2018\)Fujimoto, S\., van Hoof, H\., and Meger, D\.Addressing function approximation error in actor\-critic methods, 2018\.
- Godwin\* et al\. \(2020\)Godwin\*, J\., Keck\*, T\., Battaglia, P\., Bapst, V\., Kipf, T\., Li, Y\., Stachenfeld, K\., Veličković, P\., and Sanchez\-Gonzalez, A\.Jraph: A library for graph neural networks in jax\., 2020\.URL[http://github\.com/deepmind/jraph](http://github.com/deepmind/jraph)\.
- Heek et al\. \(2024\)Heek, J\., Levskaya, A\., Oliver, A\., Ritter, M\., Rondepierre, B\., Steiner, A\., and van Zee, M\.Flax: A neural network library and ecosystem for JAX, 2024\.URL[http://github\.com/google/flax](http://github.com/google/flax)\.
- Hunter \(2007\)Hunter, J\. D\.Matplotlib: A 2d graphics environment\.*Computing in Science & Engineering*, 9\(3\):90–95, 2007\.doi:10\.1109/MCSE\.2007\.55\.
- Kaplanis et al\. \(2018\)Kaplanis, C\., Shanahan, M\., and Clopath, C\.Continual reinforcement learning with complex synapses\.In*International Conference on Machine Learning*, pp\. 2497–2506\. PMLR, 2018\.
- Kaplanis et al\. \(2019\)Kaplanis, C\., Shanahan, M\., and Clopath, C\.Policy consolidation for continual reinforcement learning\.*arXiv preprint arXiv:1902\.00255*, 2019\.
- Kaplanis et al\. \(2020\)Kaplanis, C\., Clopath, C\., and Shanahan, M\.Continual reinforcement learning with multi\-timescale replay \(2020\)\.*DOI: https://doi\. org/10\.48550/arXiv*, 2020\.
- Khetarpal et al\. \(2022\)Khetarpal, K\., Riemer, M\., Rish, I\., and Precup, D\.Towards continual reinforcement learning: A review and perspectives\.*Journal of Artificial Intelligence Research*, 75:1401–1476, 2022\.
- Kingma & Ba \(2014\)Kingma, D\. P\. and Ba, J\.Adam: A method for stochastic optimization\.*arXiv preprint arXiv:1412\.6980*, 2014\.
- Kingma & Welling \(2013\)Kingma, D\. P\. and Welling, M\.Auto\-encoding variational bayes\.*arXiv preprint arXiv:1312\.6114*, 2013\.
- Kirkpatrick et al\. \(2017\)Kirkpatrick, J\., Pascanu, R\., Rabinowitz, N\., Veness, J\., Desjardins, G\., Rusu, A\. A\., Milan, K\., Quan, J\., Ramalho, T\., Grabska\-Barwinska, A\., et al\.Overcoming catastrophic forgetting in neural networks\.*Proceedings of the national academy of sciences*, 114\(13\):3521–3526, 2017\.
- Kluyver et al\. \(2016\)Kluyver, T\., Ragan\-Kelley, B\., Pérez, F\., Granger, B\., Bussonnier, M\., Frederic, J\., Kelley, K\., Hamrick, J\., Grout, J\., Corlay, S\., Ivanov, P\., Avila, D\., Abdalla, S\., and Willing, C\.Jupyter notebooks – a publishing format for reproducible computational workflows\.In Loizides, F\. and Schmidt, B\. \(eds\.\),*Positioning and Power in Academic Publishing: Players, Agents and Agendas*, pp\. 87 – 90\. IOS Press, 2016\.
- Lee et al\. \(2024\)Lee, H\., Cho, H\., Kim, H\., Kim, D\., Min, D\., Choo, J\., and Lyle, C\.Slow and steady wins the race: Maintaining plasticity with hare and tortoise networks\.*arXiv preprint arXiv:2406\.02596*, 2024\.
- Lillicrap et al\. \(2015\)Lillicrap, T\. P\., Hunt, J\. J\., Pritzel, A\., Heess, N\., Erez, T\., Tassa, Y\., Silver, D\., and Wierstra, D\.Continuous control with deep reinforcement learning\.*arXiv preprint arXiv:1509\.02971*, 2015\.
- Lyle et al\. \(2024\)Lyle, C\., Zheng, Z\., Khetarpal, K\., van Hasselt, H\., Pascanu, R\., Martens, J\., and Dabney, W\.Disentangling the causes of plasticity loss in neural networks\.*arXiv preprint arXiv:2402\.18762*, 2024\.
- McClelland et al\. \(1995\)McClelland, J\. L\., McNaughton, B\. L\., and O’Reilly, R\. C\.Why there are complementary learning systems in the hippocampus and neocortex: insights from the successes and failures of connectionist models of learning and memory\.*Psychological review*, 102\(3\):419, 1995\.
- McCloskey & Cohen \(1989\)McCloskey, M\. and Cohen, N\. J\.Catastrophic interference in connectionist networks: The sequential learning problem\.In*Psychology of learning and motivation*, volume 24, pp\. 109–165\. Elsevier, 1989\.
- Nikishin et al\. \(2022\)Nikishin, E\., Schwarzer, M\., D’Oro, P\., Bacon, P\.\-L\., and Courville, A\.The primacy bias in deep reinforcement learning\.In*International conference on machine learning*, pp\. 16828–16847\. PMLR, 2022\.
- Nikishin et al\. \(2023\)Nikishin, E\., Oh, J\., Ostrovski, G\., Lyle, C\., Pascanu, R\., Dabney, W\., and Barreto, A\.Deep reinforcement learning with plasticity injection\.*Advances in Neural Information Processing Systems*, 36:37142–37159, 2023\.
- Paszke et al\. \(2019\)Paszke, A\., Gross, S\., Massa, F\., Lerer, A\., Bradbury, J\., Chanan, G\., Killeen, T\., Lin, Z\., Gimelshein, N\., Antiga, L\., et al\.Pytorch: An imperative style, high\-performance deep learning library\.*Advances in neural information processing systems*, 32, 2019\.
- Powers et al\. \(2022\)Powers, S\., Xing, E\., and Gupta, A\.Self\-activating neural ensembles for continual reinforcement learning\.In*Conference on Lifelong Learning Agents*, pp\. 683–704\. PMLR, 2022\.
- Riemer et al\. \(2018\)Riemer, M\., Cases, I\., Ajemian, R\., Liu, M\., Rish, I\., Tu, Y\., and Tesauro, G\.Learning to learn without forgetting by maximizing transfer and minimizing interference\.*arXiv preprint arXiv:1810\.11910*, 2018\.
- Rolnick et al\. \(2019\)Rolnick, D\., Ahuja, A\., Schwarz, J\., Lillicrap, T\., and Wayne, G\.Experience replay for continual learning\.*Advances in neural information processing systems*, 32, 2019\.
- Schulman et al\. \(2017\)Schulman, J\., Wolski, F\., Dhariwal, P\., Radford, A\., and Klimov, O\.Proximal policy optimization algorithms\.*arXiv preprint arXiv:1707\.06347*, 2017\.
- Schwarz et al\. \(2018\)Schwarz, J\., Czarnecki, W\., Luketina, J\., Grabska\-Barwinska, A\., Teh, Y\. W\., Pascanu, R\., and Hadsell, R\.Progress & compress: A scalable framework for continual learning\.In*International conference on machine learning*, pp\. 4528–4537\. PMLR, 2018\.
- Silver & Sutton \(2025\)Silver, D\. and Sutton, R\. S\.Welcome to the era of experience\.*Google AI*, 1, 2025\.
- Silver et al\. \(2014\)Silver, D\., Lever, G\., Heess, N\., Degris, T\., Wierstra, D\., and Riedmiller, M\.Deterministic policy gradient algorithms\.In*International conference on machine learning*, pp\. 387–395\. Pmlr, 2014\.
- Sokar et al\. \(2023\)Sokar, G\., Agarwal, R\., Castro, P\. S\., and Evci, U\.The dormant neuron phenomenon in deep reinforcement learning\.In*International Conference on Machine Learning*, pp\. 32145–32168\. PMLR, 2023\.
- Sutton & Barto \(2018\)Sutton, R\. S\. and Barto, A\. G\.*Reinforcement learning: An introduction*\.MIT press, 2018\.
- Todorov et al\. \(2012\)Todorov, E\., Erez, T\., and Tassa, Y\.Mujoco: A physics engine for model\-based control\.In*2012 IEEE/RSJ international conference on intelligent robots and systems*, pp\. 5026–5033\. IEEE, 2012\.
- Tunyasuvunakool et al\. \(2020\)Tunyasuvunakool, S\., Muldal, A\., Doron, Y\., Liu, S\., Bohez, S\., Merel, J\., Erez, T\., Lillicrap, T\., Heess, N\., and Tassa, Y\.dm\_control: Software and tasks for continuous control\.*Software Impacts*, 6:100022, 2020\.
- Van Hasselt et al\. \(2016\)Van Hasselt, H\., Guez, A\., and Silver, D\.Deep reinforcement learning with double q\-learning\.In*Proceedings of the AAAI conference on artificial intelligence*, 2016\.
- Van Rossum & Drake \(2009\)Van Rossum, G\. and Drake, F\. L\.*Python 3 Reference Manual*\.CreateSpace, Scotts Valley, CA, 2009\.ISBN 1441412697\.
- Waskom \(2021\)Waskom, M\. L\.seaborn: statistical data visualization\.*Journal of Open Source Software*, 6\(60\):3021, 2021\.doi:10\.21105/joss\.03021\.URL[https://doi\.org/10\.21105/joss\.03021](https://doi.org/10.21105/joss.03021)\.
- Xie et al\. \(2020\)Xie, A\., Harrison, J\., and Finn, C\.Deep reinforcement learning amidst lifelong non\-stationarity\.*arXiv preprint arXiv:2006\.10701*, 2020\.
- Yadan \(2019\)Yadan, O\.Hydra \- a framework for elegantly configuring complex applications\.Github, 2019\.URL[https://github\.com/facebookresearch/hydra](https://github.com/facebookresearch/hydra)\.
- Yarats et al\. \(2021\)Yarats, D\., Fergus, R\., Lazaric, A\., and Pinto, L\.Mastering visual continuous control: Improved data\-augmented reinforcement learning\.*arXiv preprint arXiv:2107\.09645*, 2021\.
- Zenke et al\. \(2017\)Zenke, F\., Poole, B\., and Ganguli, S\.Continual learning through synaptic intelligence\.In*International conference on machine learning*, pp\. 3987–3995\. PMLR, 2017\.

## Appendix AAppendix

This supplementary section provides detailed insights and additional information that supports the findings and methodology discussed in the main paper\. Below is a brief overview of what each section contains:

## Appendix BProof Sketch on preserving timescales with SGD

In this section, using mathematical analysis, we provide the intuition why the learning of the multiple\-timescales Successor Features \(SFs\) must be done using Stochastic Gradient Descent \(SGD\) and not the commonly used Adaptive moment estimation \(Adam\)\(Kingma & Ba,[2014](https://arxiv.org/html/2605.26357#bib.bib25)\)\.

For the sake of brevity, we consider a tabular Reinforcement Learning setting where the SFsψ​\(s,a\)\\psi\(s,a\)are not parameterised and depends only on a state\-action pair\(s,a\)\(s,a\):

ψ​\(s,a\)=𝔼π​\[∑t=0∞γt​ϕ​\(st,at\)∣s0=s,a0=a\]\\displaystyle\\psi\(s,a\)=\\mathbb\{E\_\{\\pi\}\}\\bigg\[\\sum\_\{t=0\}^\{\\infty\}\\gamma^\{t\}\\phi\(s\_\{t\},a\_\{t\}\)\\mid s\_\{0\}=s,a\_\{0\}=a\\bigg\]\(15\)
whereγ∈\[0,1\]\\gamma\\in\[0,1\]is the discount factor andϕ\\phiis the basis features\. Without loss of generality, we consider an arbitrary consolidation system withK∈ℤK\\in\\mathbb\{Z\}possible consolidation variables operating atKKpossible timescales, whereK\>1K\>1\. Letuku\_\{k\}be a variable andψuk∈ℝn\\psi\_\{u\_\{k\}\}\\in\\mathbb\{R\}^\{n\}be the SFs operating at timescalek∈\(1,2,…,K\)k\\in\(1,2,\.\.\.,K\)\. The termsgk−1,k/Ck,gk,k\+1/Ckg\_\{k\-1,k\}/\{C\_\{k\}\},g\_\{k,k\+1\}\{/C\_\{k\}\}, whereCk=2k−1C\_\{k\}=2^\{k\-1\}andgk,k\+1∝2−k−2g\_\{k,k\+1\}\\propto 2^\{\-k\-2\}determines the overall timescales of learning the SFsψuk\\psi\_\{u\_\{k\}\}\.

Recall that after applying the Euler’s method to the continuous dynamics \(section[3\.3](https://arxiv.org/html/2605.26357#S3.SS3)\), we get the following update rules for the consolidation variables\(ψu1,ψu2,…,ψuK\)\(\\psi\_\{u\_\{1\}\},\\psi\_\{u\_\{2\}\},\\ldots,\\psi\_\{u\_\{K\}\}\)\. Letηk=Δ​t/Ck\\eta\_\{k\}=\\Delta t/C\_\{k\}and ignoring the first learning phaset\+1/2t\+1/2, at stept\+1t\+1, for the first variableψu1\\psi\_\{u\_\{1\}\}, we get:

ψu1t\+1=ψu1t\+12\+η1​\[g1,2​\(ψu2t−ψ​u1t\+12\)\]\\displaystyle\\psi\_\{u\_\{1\}\}^\{t\+1\}=\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\+\\eta\_\{1\}\[g\_\{1,2\}\\big\(\\psi\_\{u\_\{2\}\}^\{t\}\-\\psi\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\\big\)\]\(16\)For the interior variablesk=2,…,K−1k=2,\\ldots,K\-1:

ψukt\+1=ψukt\+ηk​\[gk−1,k​\(ψuk−1t−ψukt\)\+gk,k\+1​\(ψuk\+1t−ψukt\)\]\\displaystyle\\psi\_\{u\_\{k\}\}^\{t\+1\}=\\psi\_\{u\_\{k\}\}^\{t\}\+\\eta\_\{k\}\\big\[g\_\{k\-1,k\}\(\\psi\_\{u\_\{k\-1\}^\{t\}\}\-\\psi\_\{u\_\{k\}\}^\{t\}\)\+g\_\{k,k\+1\}\(\\psi\_\{u\_\{k\+1\}\}^\{t\}\-\\psi\_\{u\_\{k\}\}^\{t\}\)\\big\]\(17\)For the last variableKK:

ψuKt\+1=ψuKt\+ηK​\[gK−1,K​\(ψuK−1t−ψuKt\)−gK,K\+1​\(ψuKt\)\]\\displaystyle\\psi\_\{u\_\{K\}\}^\{t\+1\}=\\psi\_\{u\_\{K\}\}^\{t\}\+\\eta\_\{K\}\\big\[g\_\{K\-1,K\}\(\\psi\_\{u\_\{K\-1\}\}^\{t\}\-\\psi\_\{u\_\{K\}\}^\{t\}\)\-g\_\{K,K\+1\}\(\\psi\_\{u\_\{K\}\}^\{t\}\)\\big\]\(18\)
Without loss of generality, we consider the case of updatingψu1\\psi\_\{u\_\{1\}\}in Eq\.[16](https://arxiv.org/html/2605.26357#A2.E16), whereu1u\_\{1\}andu2u\_\{2\}represent the first and second consolidation variables\. LetΔ​t=1\\Delta t=1andκ1,2\\kappa\_\{1,2\}be the timescale ratiog1,2C1\\frac\{g\_\{1,2\}\}\{C\_\{1\}\}, the update term forψu1t\+1\\psi\_\{u\_\{1\}\}^\{t\+1\}, corresponding to:

η1​\[g1,2​\(ψu2t−ψu1t\+12\)\]\\displaystyle\\eta\_\{1\}\[g\_\{1,2\}\\big\(\\psi\_\{u\_\{2\}\}^\{t\}\-\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\\big\)\]=g1,2C1​\(ψu2t−ψu1t\+12⏟:=gt\)\\displaystyle=\\frac\{g\_\{1,2\}\}\{C\_\{1\}\}\\big\(\\underbrace\{\\psi\_\{u\_\{2\}\}^\{t\}\-\\psi\_\{u\_\{1\}\}^\{t\+\\frac\{1\}\{2\}\}\}\_\{:=g\_\{t\}\}\\big\)\(19\)=κ1,2⊙gt\\displaystyle=\\kappa\_\{1,2\}\\odot g\_\{t\}\(20\)=gt~\\displaystyle=\\tilde\{g\_\{t\}\}\(21\)In optimization, we usually use a learning rate, such asα∈ℝ\\alpha\\in\\mathbb\{R\}to update the variables\.

As we shall see in the proof sketch below, bothκ1,2∈ℝ\\kappa\_\{1,2\}\\in\\mathbb\{R\}andα∈ℝ\\alpha\\in\\mathbb\{R\}contribute to the effective learning rate\. We will first present our analysis using SGD, followed by Adam\.

### B\.1Stochastic Gradient Descent \(SGD\)

Recall that the update rule for SGD at stepttis defined as:

ψt\+1​\(s,a\)←ψt​\(s,a\)−α⊙gt~\\displaystyle\\psi\_\{t\+1\}\(s,a\)\\leftarrow\\psi\_\{t\}\(s,a\)\-\\alpha\\odot\\tilde\{g\_\{t\}\}\(22\)
###### Proposition B\.1\.

Optimizing Eq\.[21](https://arxiv.org/html/2605.26357#A2.E21)using Stochastic Gradient Descent \(SGD\) ensures that the gradient updates explicitly scales withα\\alpha, thus preserving the relative timescale information\.

Proof\.Without loss of generality, letκ∈\{κ1,2,κ2,3,…,κK,K\+1\}\\kappa\\in\\\{\\kappa\_\{1,2\},\\kappa\_\{2,3\},\.\.\.,\\kappa\_\{K,K\+1\}\\\}:

ψt\+1​\(s,a\)\\displaystyle\\psi\_\{t\+1\}\(s,a\)←ψt​\(s,a\)−α​gt~\\displaystyle\\leftarrow\\psi\_\{t\}\(s,a\)\-\\alpha\\tilde\{g\_\{t\}\}\(23\)=ψt​\(s,a\)−α​\(κ⋅gt\)\\displaystyle=\\psi\_\{t\}\(s,a\)\-\\alpha\(\\kappa\\cdot g\_\{t\}\)Sub​gt~=κ⋅gt​from Eq\.[20](https://arxiv.org/html/2605.26357#A2.E20)\\displaystyle\\text\{Sub \}\\tilde\{g\_\{t\}\}=\\kappa\\cdot g\_\{t\}\\text\{ from Eq\.\}\\ref\{eq:timescale\_ratio\_gradient\}\(24\)=ψt​\(s,a\)−\(α⋅κ⋅gt\)\\displaystyle=\\psi\_\{t\}\(s,a\)\-\(\\alpha\\cdot\\kappa\\cdot g\_\{t\}\)\(25\)It is then trivial to conclude that when the learning rateα=1\\alpha=1, the effective learning rateα⋅κ⋅gt=κ⋅gt\\alpha\\cdot\\kappa\\cdot g\_\{t\}=\\kappa\\cdot g\_\{t\}, thus preserving the relative scale of the updates, even when the timescale ratioκ\\kappadecreases as we move down the chain of dynamic variables due to the fact thatκ1,2\>\>κ2,3\>\>…,κK,K\+1\\kappa\_\{1,2\}\>\>\\kappa\_\{2,3\}\>\>\.\.\.,\\kappa\_\{K,K\+1\}\.□\\square

### B\.2Adaptive moment estimation \(Adam\)

Recall that the update rule for Adam\(Kingma & Ba,[2014](https://arxiv.org/html/2605.26357#bib.bib25)\)at stepttis defined as:

mt←β1⋅mt−1\+\(1−β1\)⋅gt~\\displaystyle m\_\{t\}\\leftarrow\\beta\_\{1\}\\cdot m\_\{t\-1\}\+\(1\-\\beta\_\{1\}\)\\cdot\\tilde\{g\_\{t\}\}First moment\(26\)vt←β2⋅vt−1\+\(1−β2\)⋅gt~2\\displaystyle v\_\{t\}\\leftarrow\\beta\_\{2\}\\cdot v\_\{t\-1\}\+\(1\-\\beta\_\{2\}\)\\cdot\\tilde\{g\_\{t\}\}^\{2\}Second moment\(27\)mt^←mt\(1−β1t\)\\displaystyle\\hat\{m\_\{t\}\}\\leftarrow\\frac\{m\_\{t\}\}\{\(1\-\\beta\_\{1\}^\{t\}\)\}Bias correction for first moment\(28\)vt^←vt\(1−β2t\)\\displaystyle\\hat\{v\_\{t\}\}\\leftarrow\\frac\{v\_\{t\}\}\{\(1\-\\beta\_\{2\}^\{t\}\)\}Bias correction for second moment\(29\)ψt\+1​\(s,a\)←ψt​\(s,a\)−αvt^\+ϵ⋅mt^\\displaystyle\\psi\_\{t\+1\}\(s,a\)\\leftarrow\\psi\_\{t\}\(s,a\)\-\\frac\{\\alpha\}\{\\sqrt\{\\hat\{v\_\{t\}\}\}\+\\epsilon\}\\cdot\\hat\{m\_\{t\}\}\(30\)whereαvt^\+ϵ\\frac\{\\alpha\}\{\\sqrt\{\\hat\{v\_\{t\}\}\}\+\\epsilon\}is the effective learning rate andgt~\\tilde\{g\_\{t\}\}is the update term as defined in Eq\.[21](https://arxiv.org/html/2605.26357#A2.E21)\.

###### Proposition B\.2\.

Optimizing Eq\.[21](https://arxiv.org/html/2605.26357#A2.E21)using Adam results in gradient updates to not preserve the relative timescale information\.

Let’s focus our analysis on the effective learning rateαvt^\+ϵ\\frac\{\\alpha\}\{\\sqrt\{\\hat\{v\_\{t\}\}\}\+\\epsilon\}and once again, without the loss of generality, letκ∈\{κ1,2,κ2,3,…,κK,K\+1\}\\kappa\\in\\\{\\kappa\_\{1,2\},\\kappa\_\{2,3\},\.\.\.,\\kappa\_\{K,K\+1\}\\\}:

αvt^\+ϵ\\displaystyle\\frac\{\\alpha\}\{\\sqrt\{\\hat\{v\_\{t\}\}\}\+\\epsilon\}=αvt\(1−β2t\)\+ϵ\\displaystyle=\\frac\{\\alpha\}\{\\sqrt\{\\frac\{v\_\{t\}\}\{\(1\-\\beta\_\{2\}^\{t\}\)\}\}\+\\epsilon\}Sub Eq\.[29](https://arxiv.org/html/2605.26357#A2.E29)into​vt^\\displaystyle\\text\{Sub Eq\. \\ref\{eq:SecondMomentCorrection\} into \}\\hat\{v\_\{t\}\}=αβ2⋅vt−1\+\(1−β2\)⋅gt~2\(1−β2t\)\+ϵ\\displaystyle=\\frac\{\\alpha\}\{\\sqrt\{\\frac\{\\beta\_\{2\}\\cdot v\_\{t\-1\}\+\(1\-\\beta\_\{2\}\)\\cdot\\tilde\{g\_\{t\}\}^\{2\}\}\{\(1\-\\beta\_\{2\}^\{t\}\)\}\}\+\\epsilon\}Sub Eq\.[27](https://arxiv.org/html/2605.26357#A2.E27)into​vt^\\displaystyle\\text\{Sub Eq\. \\ref\{eq:SecondMoment\} into \}\\hat\{v\_\{t\}\}=αβ2⋅vt−1\+\(1−β2\)⋅\(κ⋅gt\)2\(1−β2t\)\+ϵ\\displaystyle=\\frac\{\\alpha\}\{\\sqrt\{\\frac\{\\beta\_\{2\}\\cdot v\_\{t\-1\}\+\(1\-\\beta\_\{2\}\)\\cdot\(\\kappa\\cdot g\_\{t\}\)^\{2\}\}\{\(1\-\\beta\_\{2\}^\{t\}\)\}\}\+\\epsilon\}Sub​gt~=κ⋅gt​from Eq\.[21](https://arxiv.org/html/2605.26357#A2.E21)\\displaystyle\\text\{Sub \}\\tilde\{g\_\{t\}\}=\\kappa\\cdot g\_\{t\}\\text\{ from Eq\.\}\\ref\{eq:grad\_fast\_beaker\_update\}We can observe that when the learning rateα=1\\alpha=1, the effective learning rate,1vt^\+ϵ\\frac\{1\}\{\\sqrt\{\\hat\{v\_\{t\}\}\}\+\\epsilon\}, increases as the timescale ratio variableκ\\kappadecreases\. This is due to the fact that as we move down the chain of consolidation variables, we will getκ1,2\>\>κ2,3\>\>…,κK,K\+1\\kappa\_\{1,2\}\>\>\\kappa\_\{2,3\}\>\>\.\.\.,\\kappa\_\{K,K\+1\}\. This implies that the timescale will no longer be preserved, as the relative scale of the updates will now become inversely proportional with respect toκ\\kapparather than proportional toκ\\kappa\.□\\square

### B\.3Conclusion

These analyses demonstrate that learning the Successor Features \(SFs\),ψuk\\psi\_\{u\_\{k\}\}, using Stochastic Gradient Descent \(SGD\) rather than Adam is critical for preserving the timescale information intrinsic to these features\. The differential impact of SGD and Adam on the behavior of the updates highlights the importance of choosing an appropriate optimization strategy in RL settings that require maintenance of structured timescale information\.

## Appendix CPseudocode Implementation

In this section, we present the pseudocode of our model \(SF \+ SC\), where we apply the synaptic consolidation mechanism\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)to Successor Features\.

Algorithm 1Learning Successor Features with Synaptic Consolidation1:Determine the number of consolidation variables \(

ψu2,ψu3,…\\psi\_\{u\_\{2\}\},\\psi\_\{u\_\{3\}\},\\ldots\)

2:Initialize reward weight vector

𝒘\\boldsymbol\{w\}
3:Initialize SF

ψθ\\psi\_\{\\theta\}network, SF

ψθ¯\\overline\{\\psi\_\{\\theta\}\}target network

4:Set

ψθu1←ψθ\\psi\_\{\{\\theta\}\_\{u\_\{1\}\}\}\\leftarrow\\psi\_\{\\theta\}
5:Copy

θu1\{\\theta\}\_\{u\_\{1\}\}to the networks of the consolidation variables \(e\.g\.,

ψθu2←θu1,ψθu3←θu1,…\)\\psi\_\{\{\\theta\}\_\{u\_\{2\}\}\}\\leftarrow\\theta\_\{u\_\{1\}\},\\psi\_\{\{\\theta\}\_\{u\_\{3\}\}\}\\leftarrow\\theta\_\{u\_\{1\}\},\\ldots\)
6:for

t:=1t:=1, Tdo

7:Receive observation

StS\_\{t\}from environment

8:

At←ϵA\_\{t\}\\leftarrow\\epsilon\-greedy using

Q\(St,⋅∣𝒘\)Q\(S\_\{t\},\\cdot\\mid\\boldsymbol\{w\}\)
9:Send

AtA\_\{t\}to receive

St\+1S\_\{t\+1\}and

Rt\+1R\_\{t\+1\}from environment

10:

a′∈argmax𝑏​ψθ¯​\(St\+1,b,𝒘\)⊤​𝒘a^\{\\prime\}\\in\\underset\{b\}\{\\mathrm\{argmax\}\}\\;\\overline\{\\psi\_\{\\theta\}\}\(S\_\{t\+1\},b,\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}
11:

y^=Rt\+1\+γ​ψθ¯​\(St\+1,a′,𝒘\)⊤​𝒘\\hat\{y\}=R\_\{t\+1\}\+\\gamma\\overline\{\\psi\_\{\\theta\}\}\(S\_\{t\+1\},a^\{\\prime\},\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}
12:

ϕ←\\phi\\leftarrowL2 normalized output from the encoder of SF

ψ\\psinetwork

13:

l​o​s​sψθ=\(ψθ​\(St,At,𝒘\)⊤​𝒘−y^\)2loss\_\{\\psi\_\{\\theta\}\}=\(\\psi\_\{\\theta\}\(S\_\{t\},A\_\{t\},\\boldsymbol\{w\}\)^\{\\top\}\\boldsymbol\{w\}\-\\hat\{y\}\)^\{2\}
14:

l​o​s​sw=\(ϕ⊤​𝒘−Rt\+1\)2loss\_\{w\}=\(\\phi^\{\\top\}\\boldsymbol\{w\}\-R\_\{t\+1\}\)^\{2\}
15:Gradient descent on

ψθ\\psi\_\{\\theta\}and

𝒘\\boldsymbol\{w\}
16:Set

ψθu1←ψθ\\psi\_\{\{\\theta\}\_\{u\_\{1\}\}\}\\leftarrow\\psi\_\{\\theta\}
17:Update the parameters of the consolidation parameters analytically using Eq\.[12](https://arxiv.org/html/2605.26357#S4.E12), Eq\.[13](https://arxiv.org/html/2605.26357#S4.E13), Eq\.[14](https://arxiv.org/html/2605.26357#S4.E14)\(Stochastic Gradient Descent\)

18:Set

ψθ←ψθu1\\psi\_\{\\theta\}\\leftarrow\\psi\_\{\{\\theta\}\_\{u\_\{1\}\}\}
19:endfor

## Appendix DCode Availability

The repository includes:

- •implementations of all methods evaluated in the MuJoCo experiments;
- •continual non\-stationarity benchmarks for MuJoCo;
- •training and evaluation scripts;
- •hyperparameter configurations used in the paper;
- •instructions for reproducing the reported results\.

The JAX implementation used for the 3D Four Rooms experiments is also publicly available\.666[https://github\.com/raymondchua/multi\-timescale\-successor\-features\-fourrooms](https://github.com/raymondchua/multi-timescale-successor-features-fourrooms)This repository contains the 3D Four Rooms continual non\-stationarity experiments together with the corresponding training, evaluation, and reproducibility code\.

## Appendix EEnvironments

In this section, we present the two environments used throughout this manuscript: the Slippery 3D Four Rooms environment and the MuJoCo control suite, along with the corresponding forms of continuous non\-stationarity introduced in each setting\. In the Slippery 3D Four Rooms environment, slippery probabilities were varied according to a periodic noisy sine function\. In the MuJoCo control suite, we considered a broader range of continuous changes to the embodiment dynamics by perturbing the embodiment mass using periodic noisy sine functions, non\-periodic variants, and Ornstein–Uhlenbeck processes\.

### E\.1Slippery 3D Four Rooms Environment \(Periodic\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x10.png)Figure 10:\(a\)The slippery variant of the 3D Four Rooms environment\. The agent alternates between two tasks: in Task 1, reaching the green box produces \+1 reward and the yellow box \-1, in Task 2, the reward assignment is reversed\. At each step, the agent’s chosen action may be randomly replaced with a probability sampled from the noisy sine function shown in B\. The agent receives only egocentric pixel observations\.\(b\)A noisy sine wave that generates continuously varying slip probabilities, used to stochastically replace the agent’s intended actions\.
### E\.2Continuous Control in MuJoCo \(Periodic\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x11.png)Figure 11:MuJoCo suite with continuous periodic mass changes during training and evaluation for \(a\) Humanoid, \(b\)Walker, \(c\) Quadruped, \(d\) Half\-cheetah\. A periodic noisy sine wave that generates continuously varying mass values, used to stochastically scale the agent’s mass during training and evaluation\.
### E\.3Slippery 3D Four Rooms Environment \(Non\-Periodic\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x12.png)Figure 12:\(a\)Partial view over the first 1 million environment steps and\(b\)complete view over the full 10 million environment training steps of the noisy non\-periodic sine wave used to generate continuously varying slip probabilities that stochastically replace the agent’s intended actions\.
### E\.4Continuous Control in MuJoCo \(Non\-Periodic\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x13.png)Figure 13:MuJoCo suite with continuous non\-periodic mass changes during training and evaluation for \(a\) Humanoid, \(b\)Walker, \(c\) Quadruped, \(d\) Half\-cheetah\. A non\-periodic noisy sine wave that generates continuously varying mass values, used to stochastically scale the agent’s mass during training and evaluation\.
### E\.5Slippery 3D Four Rooms Environment \(Ornstein–Uhlenbeck processes\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x14.png)Figure 14:\(a\)Partial view over the first 1 million environment steps and\(b\)complete view over the full 10 million training steps of the Ornstein–Uhlenbeck \(OU\) process used to generate varying slip probabilities that stochastically replace the agent’s intended actions\.
### E\.6Continuous Control in MuJoCo \(Ornstein–Uhlenbeck processes\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x15.png)Figure 15:MuJoCo suite with continuous mass changes induced using Ornstein–Uhlenbeck\(OU\) processes during training and evaluation for \(a\) Humanoid, \(b\)Walker, \(c\) Quadruped, \(d\) Half\-cheetah\. An Ornstein–Uhlenbeck processes that generate continuously varying mass values, used to stochastically scale the agent’s mass during training and evaluation\.

## Appendix FExperimental details

In this section, we provide more details about the environments used in our experiments\.

### F\.13D Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x16.png)Figure 16:Slippery Four Rooms environmentWe extended this environment used in the Simple Successor Features\(Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\)which was built upon the original 3D Miniworld environment\(Chevalier\-Boisvert et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib12)\)\. In the slippery variant of the Four Rooms environment777[https://github\.com/raymondchua/miniworld\_four\_rooms](https://github.com/raymondchua/miniworld_four_rooms), we mimic wet or icy conditions in all four rooms, rather than just two rooms \(top right and bottom left\)\. The key difference is, unlike in the prior setup which uses a constant pre\-determined slippery probability value, we sample the slippery probabilities using a noisy sine function to ensure continuous changes during training and evaluation\. We kept the same two tasks structure, where the rewards alternate when the task switches\. The agent receives egocentric pixel observations at every step and the actions are moving forward, backwards, turning left and turning right\.

The specific parameters defining the 3D Slippery Four Rooms Environment are detailed in Table[1](https://arxiv.org/html/2605.26357#A6.T1)\.

Table 1:3D Slippery Four Rooms Environment Specific Parameters
### F\.2MuJoCo

In this work, we consider only state observations\. For the embodiments, we chose both Walker, Half\-Cheetah, Quadruped and Humanoid\. We broadly follow the same setup asYarats et al\. \([2021](https://arxiv.org/html/2605.26357#bib.bib53)\)andChua et al\. \([2024](https://arxiv.org/html/2605.26357#bib.bib13)\), and include their models as baselines, which we denote as “TD3” and “SF” respectively\.

The codebase was adapted from the Simple Successor Features repository888[https://github\.com/raymondchua/simple\_successor\_features](https://github.com/raymondchua/simple_successor_features)\. The specific parameters we used in the MuJoCo environment for training broadly follow the ones defined inChua et al\. \([2024](https://arxiv.org/html/2605.26357#bib.bib13)\), with some exceptions, such as the training steps and the learning rate for the reward weight vector\. The specific parameters for MuJoCo are detailed in Table[2](https://arxiv.org/html/2605.26357#A6.T2)\.

Table 2:MuJoCo Environment Specific Parameters

## Appendix GPlasticity\-Stability Analysis in non\-stationary conditions induced by a periodic noisy sinusoidal function

In this section, we present the experimental results for the various MuJoCo embodiments undergoing naturalistic, continuous changes sampled from a periodic noisy sinusoidal function\. Results for the Slippery Four Rooms Environment are presented in Figure[3](https://arxiv.org/html/2605.26357#S3.F3)in the main manuscript\. See Appendix[E](https://arxiv.org/html/2605.26357#A5), specifically Sections[E\.1](https://arxiv.org/html/2605.26357#A5.SS1)and[E\.2](https://arxiv.org/html/2605.26357#A5.SS2), for illustrations of the resulting dynamics\.

### G\.1MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x17.png)Figure 17:Plasticity–stability analysis in the MuJoCo suite under continuous mass changes induced by a periodic noisy sine function during training and evaluation\. We compare a baseline TD3 agent with three variants: \(i\) Continual Backprop \(CBP\), which selectively resets least\-active weights; \(ii\) plasticity injection by resetting the weights in the last layer \(P\-last\); and \(iii\) synaptic consolidation \(SC\)\. Across embodiments, CBP generally struggled to outperform TD3, with complete collapse in Humanoid, while SC improved stability\.

## Appendix HPlasticity\-Stability Analysis in non\-stationary conditions induced by a non\-periodic noisy sinusoidal function

In this section, we present experimental results for the 3D Miniworld environment and various MuJoCo embodiments undergoing naturalistic, continuous changes generated from a non\-periodic noisy sinusoidal function\. See Appendix[E](https://arxiv.org/html/2605.26357#A5), specifically Sections[E\.3](https://arxiv.org/html/2605.26357#A5.SS3)and[E\.4](https://arxiv.org/html/2605.26357#A5.SS4), for illustrations of the resulting dynamics\.

### H\.1Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x18.png)Figure 18:Plasticity–stability analysis in the Slippery Four Rooms environment using non\-periodic noisy sinusoidal function\. The agent undergoes two exposures; after each learning phase, the reward mapping is reversed\.\(a\)Average return per episode\.\(b\)Learning efficiency \(steps to reach a good policy; lower is better\)\. For the plasticity\-injection agent, plasticity was injected once at 10 million environment steps \(end of Exposure 1\)\. In both panels, the agent with synaptic consolidation \(SC\) applied to the SFs \(SF \+ SC, purple\) consistently achieved better learning efficiency when re\-encountering the same set of tasks sequentially in exposure 2\.
### H\.2MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x19.png)Figure 19:Plasticity–stability analysis in the MuJoCo suite under continuous mass changes induced by a non\-periodic noisy sine function during training and evaluation\. We compare a baseline TD3 agent with three variants: \(i\) Continual Backprop \(CBP\), which selectively resets least\-active weights; \(ii\) plasticity injection by resetting the weights in the last layer \(P\-last\); and \(iii\) synaptic consolidation \(SC\)\. Across embodiments, CBP generally struggled to outperform TD3, with complete collapse in Humanoid, while SC improved stability\. Applying SC to SFs leads to improved learning performance\. However, for complex embodiments such as Quadruped and Humanoid, these gains are reduced\. This is unsurprising, as SFs are predictive representations, and the underlying mass dynamics become less predictable in non\-periodic settings\.

## Appendix IPlasticity\-Stability Analysis in non\-stationary conditions induced by Ornstein–Uhlenbeck processes

In this section, we present experimental results for the 3D Miniworld environment and the various MuJoCo embodiments undergoing naturalistic, continuous changes sampled from the Ornstein–Uhlenbeck processes\. See Appendix[E](https://arxiv.org/html/2605.26357#A5), specifically Sections[E\.5](https://arxiv.org/html/2605.26357#A5.SS5)and[E\.6](https://arxiv.org/html/2605.26357#A5.SS6), for illustrations of the resulting dynamics\.

### I\.1Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x20.png)Figure 20:Plasticity–stability analysis in the Slippery Four Rooms environment using OU processes\. The agent undergoes two exposures; after each learning phase, the reward mapping is reversed\.\(a\)Average return per episode\.\(b\)Learning efficiency \(steps to reach a good policy; lower is better\)\. For the plasticity\-injection agent, plasticity was injected once at 10 million environment steps \(end of Exposure 1\)\. In both panels, the agent with synaptic consolidation \(SC\) applied to the SFs \(SF \+ SC, purple\) consistently achieved better learning efficiency when re\-encountering the same set of tasks sequentially in exposure 2\.
### I\.2MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x21.png)Figure 21:Plasticity–stability analysis in the MuJoCo suite under continuous mass changes induced by Ornstein–Uhlenbeck processes during training and evaluation\. We compare a baseline TD3 agent with three variants: \(i\) Continual Backprop \(CBP\), which selectively resets least\-active weights; \(ii\) plasticity injection by resetting the weights in the last layer \(P\-last\); and \(iii\) synaptic consolidation \(SC\)\. Across embodiments, CBP generally struggled to outperform TD3, with complete collapse in Humanoid, while SC improved stability\. Applying SC to SFs \(SF\+SC\) leads to similar performance as applying SC to Q\-values \(TD3\+SC\)\. This is unsurprising, as SFs are predictive representations, and the underlying mass dynamics become less predictable in OU settings\.

## Appendix JSchematic of Synaptic Consolidation for Q\-values vs\. Successor Features

In this section, we present a schematic comparison of applying synaptic consolidation to the parameters of Q\-values versus Successor Features\.

![Refer to caption](https://arxiv.org/html/2605.26357v1/x22.png)Figure 22:Schematic of synaptic consolidation applied to Q\-values and to Successor Features \(SFs\)\.\(a\):\(Kaplanis et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib21)\)showed that adapting the synaptic consolidation mechanism of\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)to Q\-values improves robustness in continual RL\.\(b\):Here, we extend this approach to predictive, generalizable representations using Simple Successor Features\(Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\)\. Consolidated variables \(e\.g\.,Qu2,Qu3,…Q\_\{u\_\{2\}\},Q\_\{u\_\{3\}\},\\ldotsorS​Fu2,S​Fu3,…SF\_\{u\_\{2\}\},SF\_\{u\_\{3\}\},\\ldots\) are computed analytically and therefore lie outside the computational graph used to updateQu1Q\_\{u\_\{1\}\}orS​Fu1SF\_\{u\_\{1\}\}by backpropagation\.
## Appendix KQ\-values vs\. SFs, Elastic Weight Consolidation vs Synaptic Consolidation Comparison

In this section, we compare the effects of applying synaptic consolidation \(SC\)\(Benna & Fusi,[2016](https://arxiv.org/html/2605.26357#bib.bib7)\)and Elastic Weight Consolidation \(EWC\)\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27)\)to either the Q\-value parameters or the Successor Feature \(SF\) parameters\. Across all environments, SC consistently outperforms EWC, regardless of whether the consolidation mechanism is applied to the Q\-values or the SFs\. These results suggest that multi\-timescale consolidation dynamics provide a more effective mechanism for maintaining stability under continuous non\-stationarity than importance\-based regularization alone\.

### K\.1Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x23.png)Figure 23:Comparison of Q\-values and Successor Features \(SFs\), with synaptic consolidation \(SC\) or elastic weight consolidation \(EWC\), in the 3D Slippery Four Rooms environment during training and evaluation\. Applying SC to Q\-values \(green\) and SFs \(purple\) offers higher learning efficiency than their EWC counterparts, requiring fewer steps to learn a good policy\. This demonstrates that SC is more effective than EWC, particularly when applied to SFs, during the agent’s second exposure to the tasks\.
### K\.2MuJoCo suite with continuous mass changes

![Refer to caption](https://arxiv.org/html/2605.26357v1/x24.png)Figure 24:Comparison of Q\-values and Successor Features \(SFs\), with and without synaptic consolidation, on the MuJoCo suite under continuous mass changes during training and evaluation\. Interestingly, unlike Q\-values, applying synaptic consolidation to SFs \(purple\) yields consistently higher learning efficiency\.

## Appendix LQ\-values vs\. SFs, With and Without Synaptic Consolidation Comparison

In this section, we present the results of applying synaptic consolidation to the parameters of Q\-values versus the parameters of Successor Features\.

### L\.1Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x25.png)Figure 25:Comparison of consolidating the parameters of Q\-values and SFs using Synaptic Consolidation \(SC\) using the 3D Slippery Four Rooms environment\.\(left\):Average episode return plot\.\(right\):Number of training steps needed to reach a pre\-determined good policy\. Lesser steps the better\. Applying SC to the SFs \(purple\) yields better learning performance overall\.
### L\.2MuJoCo suite with periodic mass changes

![Refer to caption](https://arxiv.org/html/2605.26357v1/x26.png)Figure 26:Comparison of consolidating the parameters of Q\-values and SFs using Synaptic Consolidation using the MuJoCo suite\. Interestingly, when compared to TD3 \(blue\), SFs \(orange\) learn well in Half\-Cheetah and Walker but not Quadruped and Humanoid\. This is probably due to higher complexity in Quadruped and Humanoid as they have larger state and action spaces\. Overall, applying SC to the SFs \(purple\) yields better learning performance, highlighting their effectiveness when combined together\.
### L\.3MuJoCo suite with non\-periodic mass changes

![Refer to caption](https://arxiv.org/html/2605.26357v1/x27.png)Figure 27:Comparison of consolidating the parameters of Q\-values and SFs using Synaptic Consolidation using the MuJoCo suite when embodiments undergo non\-periodic mass changes\. Interestingly, when compared to TD3 \(blue\), SFs \(orange\) learn well in Half\-Cheetah and Walker but not Quadruped and Humanoid\. This is probably due to higher complexity in Quadruped and Humanoid as they have larger state and action spaces\. Unlike in the periodic mass changes setting, here, applying SC to the SFs \(purple\) only yields better learning performance in simpler embodiments such as Half\-Cheetah and Walker, but struggle in the more complex embodiments such as Quadruped and Humanoid when compared to applying SC to the Q\-values \(green\)\.
### L\.4MuJoCo suite with Ornstein\-Uhlenbeck mass changes

![Refer to caption](https://arxiv.org/html/2605.26357v1/x28.png)Figure 28:Comparison of consolidating the parameters of Q\-values and SFs using Synaptic Consolidation using the MuJoCo suite when embodiments under Ornstein\-Uhlenbeck mass Changes\. In this setting, applying SC to the SFs \(purple\) only yields better learning performance compared to applying SC to Q\-values\.

## Appendix MAnalysis of Fast and Slow Timescale Variables

In this section, we present analyses aimed at understanding the contributions of the different timescale variables\. To do so, we control for the number of consolidation variables\. Increasing the number of consolidation variables allows the Successor Features to be preserved over longer timescales\.

### M\.1Slippery Four Rooms Environment Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x29.png)Figure 29:Analysis of fast and slow timescale variables in the 3D Slippery Four Rooms environment during training and evaluation\. Using synaptic consolidation clearly leads to better learning efficiency, but there is no clear advantage between six and nine consolidation variables\.
### M\.2MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x30.png)Figure 30:Analysis of fast and slow timescale variables on the MuJoCo suite under continuous mass changes during training and evaluation\. Using more consolidation variables \(six, eight or nine\) yields consistently higher learning efficiency, highlighting the importance of slower\-timescale variables\.

## Appendix NArchitecture for recalling consolidated Successor Features using cross\-attention

We adapt the canonical cross\-attention mechanism by letting the reward weight vectorwwserves as the query, while the SF consolidation variables \(excluding the most plasticS​Fu1SF\_\{u\_\{1\}\}\) serve as keys and values\. To construct more discriminative representations, we subtracted each variable from its faster neighbor \(eg\.S​FukSF\_\{u\_\{k\}\}=S​Fuk−S​Fuk−1SF\_\{u\_\{k\}\}\-SF\_\{u\_\{k\-1\}\}\)\. The keys and values were layer\-normalized before projected through learnable weights\(WKeys,WValues\)\(W\_\{\\text\{Keys\}\},W\_\{\\text\{Values\}\}\), while the query vectorwwis projected throughWQueryW\_\{\\text\{Query\}\}\. Attention scores were computed via the query\-keys multiplication, followed by softmax activation, which was then multiplied by the Values to produce a weighted sum, thus integrating information across timescales \(see Appendix[N](https://arxiv.org/html/2605.26357#A14)for more details\)\. Since the deeper SF consolidation variables\(S​Fu2,S​Fu3,…\)\(SF\_\{u\_\{2\}\},SF\_\{u\_\{3\}\},\\ldots\)were computed analytically and not part of the computational graph, we applied a reparameterization trick \(inspired byKingma & Welling \([2013](https://arxiv.org/html/2605.26357#bib.bib26)\)\) to enable joint training of the cross\-attention mechanism with the most plastic variable\(S​Fu1\)\(SF\_\{u\_\{1\}\}\)using the Q\-SF\-TD loss \(Eq\.[6](https://arxiv.org/html/2605.26357#S3.E6)\)\.

![Refer to caption](https://arxiv.org/html/2605.26357v1/x31.png)Figure 31:Using cross\-attention to recall information from the SF consolidation modules\.\(a:A high\-level schematic on how the cross\-attention mechanism is used\.\(b:The computations for the cross\-attention mechanism\. We used the reward weight vectorwwas the query, the SFs consolidation variables except the most plastic one as keys and values \(S​Fu2,S​Fu3,…,S​FuKSF\_\{u\_\{2\}\},SF\_\{u\_\{3\}\},\\ldots,SF\_\{u\_\{K\}\}\)\. Because these SFs consolidation variables are computed analytically, they are not part of the computational graph\. Therefore, we apply a reparameterization trick to add the output of the cross\-attention mechanism to the most plastic SF \(SFu1\)SF\_\{u\_\{1\}\}\)so that the learnable weights \(WQ​u​e​r​y,WK​e​y​s,WV​a​l​u​e​sW\_\{Query\},W\_\{Keys\},W\_\{Values\}\) in the cross\-attention mechanism are learned via the Q\-SF\-TD loss \(Eq\.[6](https://arxiv.org/html/2605.26357#S3.E6)\)\.
## Appendix OCross\-Attention Analysis of Fast and Slow Timescale Variables

In this section, we perform a set of analyses using the cross\-attention architecture \(Figure[31](https://arxiv.org/html/2605.26357#A14.F31)\) to better understand the functional roles of the individual timescale variables\. By allowing interactions across Successor Features consolidated over different timescales, the architecture enables us to examine how rapidly adapting versus slowly varying predictive representations contribute to the stability–plasticity tradeoff under naturalistic non\-stationarity\.

### O\.1Slippery Four Rooms Environment Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x32.png)Figure 32:Analysis of all consolidated variables using Cross\-Attention during training in the 3D Slippery Four Rooms environment\. The cross\-attention probabilities indicate that fast and slow timescale variables were attended to similarly, suggesting nearly equal contribution\. This may be due to the sparse reward structure in the 3D Slippery Four Rooms environment, which affects how discriminate the SFs are given that the SFs are learned via the reward signal using Q\-SF\-TD loss \(Eq\.[6](https://arxiv.org/html/2605.26357#S3.E6)\)
### O\.2MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x33.png)Figure 33:Analysis of all consolidated variables using cross\-attention in the MuJoCo suite under continuous mass changes\. Memory recall was performed solely through the cross\-attention mechanism, rather than by waiting for information to propagate from slower to faster timescale variables\. Unsurprisingly, faster timescale variables were attended to more than slower ones\. Notably, Half\-Cheetah and Walker benefited from memory recall via cross\-attention, whereas Quadruped and Humanoid require more steps to learn\. This may be due to the higher complexity of Quadruped and Humanoid, as their larger state and action spaces reduce the learning efficiency of the cross\-attention mechanism\.![Refer to caption](https://arxiv.org/html/2605.26357v1/x34.png)Figure 34:Learning curves in the MuJoCo suite under continuous mass changes with cross\-attention over consolidated variables\. Faster timescale variables were generally attended to more strongly than slower ones as shown in Figure[33](https://arxiv.org/html/2605.26357#A15.F33)\. Half\-Cheetah and Walker benefited from cross\-attention–based memory recall, whereas Quadruped and Humanoid require more steps to learn, likely due to their higher complexity and larger state–action spaces which reduces the learning efficiency of the cross\-attention mechanism\.

## Appendix PAgents

In this section, we describe our agent as well as the ones we used for comparisons\.

### P\.1Successor Features with Synaptic Consolidation

![Refer to caption](https://arxiv.org/html/2605.26357v1/x35.png)Figure 35:Simple SFs with synaptic consolidation architecture\. Simple SFs were adapted from\(Chua et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib13)\), with TD3\(Lillicrap et al\.,[2015](https://arxiv.org/html/2605.26357#bib.bib30)\)as base model\. The synaptic consolidation variables are updated analytically \(see section[4](https://arxiv.org/html/2605.26357#S4)for more details on the consolidation variables\)\.We swept the task learning rate of the reward weight vector across the values of\{10−5,10−6,…,10−10\}\\\{10^\{\-5\},10^\{\-6\},\\dots,10^\{\-10\}\\\}when optimizing the reward prediction loss \(Eq\.[5](https://arxiv.org/html/2605.26357#S3.E5)\) for the MuJoCo suite\. In the naturalistic, continual non\-stationary setting that we are studying, we find that a lower learning rate for learning the reward weight vectorwwgenerally helps\.

The DQN variant largely follows the same architecture, but without the actor network\. The only major difference is that hidden dimensions are set to 256\. The encoder architecture is the same as DQN encoder\.

Table 3:Simple SF with Synaptic Consolidation HyperparametersTable 4:Taskwwencoding Hyperparameters
### P\.2Elastic Weight Consolidation

For the Elastic Weight Consolidation \(EWC\) agents\(Kirkpatrick et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib27)\), we adapt the online variant\(Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41)\), such that instead the fisher information is computed at everykksteps instead of waiting till the end of the task, thus removing the need for tasks boundaries information\. The EWC loss is defined as:

L​\(θ\)=LT​D​\(θ\)\+∑iλ2​Fi​\(θi−θi∗\)2\\displaystyle L\(\\theta\)=L\_\{TD\}\(\\theta\)\+\\sum\_\{i\}\\frac\{\\lambda\}\{2\}F\_\{i\}\(\\theta\_\{i\}\-\\theta^\{\*\}\_\{i\}\)^\{2\}\(32\)whereiiis the number of parameters,λ∈ℝ\\lambda\\in\\mathbb\{R\}is the regularization factor,LT​DL\_\{TD\}can be either the DQN loss or Q\-ST\-TD loss \(Eq\.[6](https://arxiv.org/html/2605.26357#S3.E6)\), if we are learning SFs\. The fisher information is computed by squaring the gradients of the parameters\. In our experiments, we set the fisher computation interval to every 10k steps, which is the number of steps per episode in MuJoCo\. We also swept the regularization factorλ∈\{12,25,75,125,175\}\\lambda\\in\\\{12,25,75,125,175\\\}, following the same setup as\(Schwarz et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib41)\)\. In our experiments, we findλ=25\\lambda=25works best\.

### P\.3Plasticity Injection Model

For the plasticity injection models, we consider plasticity injection on the last layer \(P\-last\) for both the Slippery Four Rooms environment and the MuJoCo suite\. We follow the same setup as\(Nikishin et al\.,[2023](https://arxiv.org/html/2605.26357#bib.bib35)\), whereby during the plasticity injection stepkk, we retain freeze the parameters of last layer in the artificial neural networkθ\\thetaand introduce a new set of parametersθ′\\theta^\{\{\}^\{\\prime\}\}which is sampled from random initialization\. The set of new of parameters is then further copied, such that we will haveθ′=θ1′=θ2′\\theta^\{\{\}^\{\\prime\}\}=\\theta^\{\{\}^\{\\prime\}\}\_\{1\}=\\theta^\{\{\}^\{\\prime\}\}\_\{2\}and the output of the network is computed using:

hθ​\(x\)\+hθ1′​\(x\)−hθ2′\\displaystyle h\_\{\\theta\}\(x\)\+h\_\{\\theta^\{\{\}^\{\\prime\}\}\_\{1\}\}\(x\)\-h\_\{\\theta^\{\{\}^\{\\prime\}\}\_\{2\}\}\(33\)wherehhis the function of the artificial neural network andxxis the input\. After the plasticity injection stepk\+1k\+1and beyond, onlyθ1′\\theta^\{\{\}^\{\\prime\}\}\_\{1\}is allowed to be updated whileθ′,θ2′\\theta^\{\{\}^\{\\prime\}\},\\theta^\{\{\}^\{\\prime\}\}\_\{2\}are kept frozen, leading \(θ′−θ2′\\theta^\{\{\}^\{\\prime\}\}\-\\theta^\{\{\}^\{\\prime\}\}\_\{2\}\) to be the bias term\.

We experimented with injecting plasticity at 25%, 50% and 75% of the training steps, but observed little effect in our setting\. We also found that this method was ineffective when adapted to TD3 and evaluated in MuJoCo\. Therefore, we included Continual Backprop\(Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)as additional baseline for both the Slippery 3D Four Rooms environment and MuJoCo\.

### P\.4Continual Backprop

The plasticity injection method described above can hurt learning performance when injecting plasticity into the critic network of actor\-critic architecture\. Therefore, we consider another baseline model, known as continual backprop \(CBP\), that is more effective in enhancing plasticity\(Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)\. Rather than re\-initializing parameters at random, CBP selectively injects plasticity into parameters who were least useful for the current task\. This least useful metric is known as the contribution utility, which measures both the hidden unit’s activity and its outgoing connection strength\. For each hidden unitiiin layerllat timett, the contribution utility is defined as:

ul​\[i\]=η×ul​\[i\]\+\(1−η\)×\|hl,i,t\|×∑k=1nl\+2\|wl,i,k,t\|\\displaystyle u\_\{l\}\[i\]=\\eta\\times u\_\{l\}\[i\]\+\(1\-\\eta\)\\times\|h\_\{l,i,t\}\|\\times\\sum\_\{k=1\}^\{n\_\{l\+2\}\}\|w\_\{l,i,k,t\}\|\(34\)wherehl,i,th\_\{l,i,t\}is the output of theiith hidden unit in layerllat timett,wl,i,k,tw\_\{l,i,k,t\}is the weight connecting theiith unit in layerllto thekkth unit in layerl\+1l\+1at timettandnl\+1n\_\{l\+1\}is the number of units in the next layerl\+1l\+1\. This contribution utility can be thought of as a running average of instantaneous contributions, with a decay rateη\\eta\.

At each step, CBP identifies eligible units for re\-initialization based on two criteria, first is the low contribution utility which indicates that the unit has not been useful in recent training phase, and second is the lifespan, which ensures that units are only re\-initialized after allowing to have a sufficient steps of learning\. By periodically resetting under\-utilized units, CBP ensures that the network maintains plasticity throughout learning\. In our experiments, we broadly follow the parameters defined in\(Dohare et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib14)\)\. We swept the replacement rate across the values of\{10−3,10−4,10−5\}\\\{10^\{\-3\},10^\{\-4\},10^\{\-5\}\\\}but observed little effect in our setting, so we kept it at10−410^\{\-4\}\. We also swept the maturity threshold across the values of\{100,1000,10000\}\\\{100,1000,10000\\\}and also observed little effect, so we kept it as 1000\.

In our experiments, unsurprisingly, we found CBP to be much more effective than the plasticity injection model \(P\-last\) introduced in section[P\.3](https://arxiv.org/html/2605.26357#A16.SS3)\. Particularly in Quadruped, Half\-Cheetah and Walker \(Figure[17](https://arxiv.org/html/2605.26357#A7.F17)\), we see that CBP outperforms P\-last\. However, across MuJoCo tasks, CBP generally failed to outperform its base model, TD3, suggesting that in our natural and continually evolving environments, stability rather than plasticity, is the critical bottleneck, since injecting plasticity did not improve performance\.

## Appendix QImplementation Details

For our experimental setup, we used Python 3\(Van Rossum & Drake,[2009](https://arxiv.org/html/2605.26357#bib.bib49)\)as the primary programming language\. The agents and the training framework were developed using Jax\(Bradbury et al\.,[2018](https://arxiv.org/html/2605.26357#bib.bib10); Godwin\* et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib18)\)for the Slippery Four Rooms environment and PyTorch\(Paszke et al\.,[2019](https://arxiv.org/html/2605.26357#bib.bib36)\)for the MuJoCo embodiments\(Todorov et al\.,[2012](https://arxiv.org/html/2605.26357#bib.bib46)\)\. We used the DeepMind Control Suite \(DMC\)\(Tunyasuvunakool et al\.,[2020](https://arxiv.org/html/2605.26357#bib.bib47)\)to manipulate the embodiments\. Flax\(Heek et al\.,[2024](https://arxiv.org/html/2605.26357#bib.bib19)\)was employed for implementing the neural network components\. For data visualization, we used Seaborn\(Waskom,[2021](https://arxiv.org/html/2605.26357#bib.bib50)\), Matplotlib\(Hunter,[2007](https://arxiv.org/html/2605.26357#bib.bib20)\)on Jupyter Notebooks\(Kluyver et al\.,[2016](https://arxiv.org/html/2605.26357#bib.bib28)\)to generate the plots\. The configuration and management of our experiments were performed with Hydra\(Yadan,[2019](https://arxiv.org/html/2605.26357#bib.bib52)\)and Weights & Biases\(Biewald,[2020](https://arxiv.org/html/2605.26357#bib.bib8)\)respectively\. All experiments were conducted using Nvidia A100 GPUs and completed within one to two days max\. The code used in the study will be released in the near future, following an internal review process\.

## Appendix RComputational Complexity

In this section, we report the computational complexity of all methods, measured in frames per second \(FPS\) during training\. Across both the Slippery Four Rooms and the MuJoCo Humanoid experiments, our model \(SF \+ SC\) exhibits the lowest throughput\. This slowdown arises from the consolidation dynamics, which require analytical, sequential updates between consolidation variables, in addition to learning the SFs and the reward weight vector𝒘\.\\boldsymbol\{w\}\.Because these updates are not handled by backpropagation and cannot be parallelised, they introduce a constant\-factor computational overhead inherent to the mechanism\. Moreover, this overhead increases with the number of consolidation variables, where using more variables leads to proportionally lower FPS\.

### R\.1Slippery Four Rooms Environment

![Refer to caption](https://arxiv.org/html/2605.26357v1/x36.png)Figure 36:Comparison of training throughput \(FPS\) for all models in the Slippery Four Rooms environment\. Higher FPS reflects more efficient computation\.![Refer to caption](https://arxiv.org/html/2605.26357v1/x37.png)Figure 37:Comparison of training throughput \(FPS\) for different number of consolidation variables for the SFs within the slippery four rooms environment\. Higher FPS reflects more efficient computation\.
### R\.2MuJoCo \- Humanoid

![Refer to caption](https://arxiv.org/html/2605.26357v1/x38.png)Figure 38:Comparison of training throughput \(FPS\) for all models in the humanoid embodiment within the MuJoCo environment\. Higher FPS reflects more efficient computation\.![Refer to caption](https://arxiv.org/html/2605.26357v1/x39.png)Figure 39:Comparison of training throughput \(FPS\) for different number of consolidation variables for the SFs within the humanoid embodiment within the MuJoCo environment\. Higher FPS reflects more efficient computation\.

## Appendix SComparison with Trust\-Region Methods \(PPO\)

In this section, we include Proximal Policy Optimization \(PPO\)\(Schulman et al\.,[2017](https://arxiv.org/html/2605.26357#bib.bib40)\)as an additional baseline to investigate whether trust\-region optimization alone is sufficient to maintain stability under continuous non\-stationarity\. PPO is particularly relevant in this setting because its clipped objective is designed to stabilize policy updates by constraining large changes in the policy distribution across optimization steps\. Given that PPO relies on parallel sample collection, we also report the number of environment samples used during training to account for differences in data usage across methods\. Despite these stabilization mechanisms, PPO struggles to maintain strong performance under gradual changes in the environment dynamics, suggesting that trust\-region\-based optimization alone may be insufficient for continual learning under persistent non\-stationarity\.

### S\.1MuJoCo suite Results

![Refer to caption](https://arxiv.org/html/2605.26357v1/x40.png)Figure 40:Comparison of PPO \(teal\) with TD3, SF, and their variants with plasticity preservation or stability enhancement mechanisms under continuous mass changes\. \(Left\) Average episode return over training\. \(Middle\) Area under the curve \(AUC\) of the return, summarizing overall performance\. \(Right\) Total number of environment samples used during training\. While PPO leverages parallelized data collection and uses more samples, it achieves lower return and AUC compared to SF \+ SC \(purple\)\. This suggests that trust\-region updates alone are insufficient to maintain stable and efficient learning under continuous non\-stationarity, compared to explicit multi\-timescale consolidation mechanisms\.

## Appendix TQuantifying Continuous Non\-Stationarity

In this section, we present results under varying levels of perturbation in the 3D Four Rooms environment and MuJoCo, quantified across three regimes: mild, moderate, and severe\. For the mild regime, perturbations were capped at 25% of the maximum allowable value\. Here, the maximum corresponds either to a 0\.45 probability that the selected action is replaced by a randomly sampled alternative action in the 3D Four Rooms environment, or to the largest perturbation permitted by the MuJoCo physics engine before the simulation becomes unstable\. For the moderate and severe regimes, perturbations were capped at 50% and 100% of this maximum, respectively\. All perturbations were generated using theperiodiccontinuous\-change setting \(Sections[E\.1](https://arxiv.org/html/2605.26357#A5.SS1)and[E\.2](https://arxiv.org/html/2605.26357#A5.SS2)in Appendix[E](https://arxiv.org/html/2605.26357#A5)\)\.

We observe that, in the 3D Four Rooms environment, applying synaptic consolidation to the Successor Feature \(SF\) parameters consistently yields more robust learning performance across all perturbation regimes\. In contrast, for MuJoCo environments under mild perturbations \(25%\), applying synaptic consolidation to the Q\-value parameters generally results in stronger performance\. However, as the magnitude of perturbation increases in the moderate and severe regimes, consolidating the SF parameters becomes increasingly effective\.

### T\.1Slippery Probabilities Perturbations \(3D Four Rooms\)

![Refer to caption](https://arxiv.org/html/2605.26357v1/x41.png)Figure 41:Quantification of slippery dynamics in the 3D Four Rooms environment\. \(a\) Average episode return, \(b\) minimum number of environment steps required to learn a successful policy, and \(c\) area under the curve \(AUC\) of the episode returns\. We consider three levels of slippery probability variation: mild \(25%\), moderate \(50%\), and severe \(100%\), where the maximum corresponds to a 0\.45 probability that the selected action is replaced by a randomly sampled alternative action\. The results suggest that methods primarily designed to promote plasticity \(CBP, P\-last\) are less effective under continuous non\-stationarity than approaches incorporating synaptic consolidation \(SC\)\. While Successor Features \(SFs\) consistently improve performance across all settings, additional gains are only achieved when SC is applied to SFs\.
### T\.2Mass Perturbations \(MuJoCo\)

#### T\.2\.1Humanoid

![Refer to caption](https://arxiv.org/html/2605.26357v1/x42.png)Figure 42:Quantification of mass changes for the Humanoid embodiment\. We consider three levels of mass dynamics variation: mild \(25%\), moderate \(50%\), and severe \(100%\), corresponding to the maximum change allowed before the physical simulation becomes unstable\. Across these settings, plasticity\-preserving methods \(CBP, P\-last\) are less effective than approaches incorporating synaptic consolidation \(SC\)\. Under moderate—and especially severe—conditions, where the agent experiences substantial changes in dynamics, applying SC to Successor Features \(SF \+ SC\) yields the best performance\. In contrast, under mild changes, the benefits of SC are limited, as the environment remains close to stationary\.
#### T\.2\.2Quadruped

![Refer to caption](https://arxiv.org/html/2605.26357v1/x43.png)Figure 43:Quantification of mass changes for the Quadruped embodiment\. We consider three levels of mass dynamics variation: mild \(25%\), moderate \(50%\), and severe \(100%\), corresponding to the maximum change allowed before the physical simulation becomes unstable\. Across these settings, plasticity\-preserving methods \(CBP, P\-last\) are less effective than approaches incorporating synaptic consolidation \(SC\)\. Under severe conditions, where the agent experiences substantial changes in dynamics, applying SC to Successor Features \(SF \+ SC\) yields the best performance\. In contrast, under mild and moderate changes, applying SC to Q\-values \(TD3 \+ SC\) yields better performance, as the environment remains closer to stationary\.
#### T\.2\.3Half\-Cheetah

![Refer to caption](https://arxiv.org/html/2605.26357v1/x44.png)Figure 44:Quantification of mass changes for the Half\-Cheetah embodiment\. We consider three levels of mass dynamics variation: mild \(25%\), moderate \(50%\), and severe \(100%\), corresponding to the maximum change allowed before the physical simulation becomes unstable\. Across these settings, plasticity\-preserving methods \(CBP, P\-last\) are less effective than approaches incorporating synaptic consolidation \(SC\)\. Under moderate—and especially severe—conditions, where the agent experiences substantial changes in dynamics, applying SC to Successor Features \(SF \+ SC\) yields the best performance\. In contrast, under mild changes, the benefits of SC are limited, as the environment remains close to stationary\.
#### T\.2\.4Walker

![Refer to caption](https://arxiv.org/html/2605.26357v1/x45.png)Figure 45:Quantification of mass changes for the Walker embodiment\. We consider three levels of mass dynamics variation: mild \(25%\), moderate \(50%\), and severe \(100%\), corresponding to the maximum change allowed before the physical simulation becomes unstable\. Across these settings, plasticity\-preserving methods \(CBP, P\-last\) are less effective than approaches incorporating synaptic consolidation \(SC\)\. Under severe conditions, where the agent experiences substantial changes in dynamics, applying SC to Successor Features \(SF \+ SC\) yields the best performance\. In contrast, under mild and moderate changes, applying SC to Q\-values \(TD3 \+ SC\) yields better performance, as the environment remains closer to stationary\.

Similar Articles

On the Stability of Growth in Structural Plasticity

arXiv cs.LG

This academic paper investigates the asymmetry between pruning and growth in structural plasticity for neural networks, showing that newborn units suffer from weaker gradient signals than incumbent units, and proposes interventions to improve integration.