Continual Reinforcement Learning (CRL) agents struggle to adapt to new situations while retaining prior knowledge, leading to a stability-plasticity trade-off. One approach to solving this problem is by using complementary learning systems: one for slow, long-term learning and another for quick, transient adaptation. Anand & Precup (2023) instantiated this idea using permanent and transient value functions, but used the same features in both systems. Intuitively, long-term learning should rely more on parametric representations for broad generalization, while transient learning would be best served by non-parametric representations for situation-specific adaptation. In this paper, we explore this idea both theoretically and empirically. By introducing a novel non-parametric approximation for estimating the transient value function in complex tasks, we demonstrate that our method enables online learning and outperforms competitive baselines on image-based tasks and Craftax-Classic, underscoring the effectiveness of our system-level decomposition.
Figures & tables
Figure 1: Results in 1D chain environment. (a) TD learning using both features results in the lowest RMSVE. (b) The separate variant of PT-TD learning has a lower initial RMSVE following task changes, and its permanent component converges to the correct value after just one task.
Figure 2: Environments used in the experiments.
Figure 3: (a) 2D grid prediction results. PT-TD-Separate variant has the lowest RMSVE in both online and other tasks (shown in solid and dotted lines, respectively). (b) Control results. The PTQ-Separate variant adapts faster when the tasks change.
Figure 4: (a) Transient Approximator. The observation is mapped to value function in four steps: (1) tokenization; (2) hashing; (3) binning; (4) value estimation. (b) Individual Table. Within each table, the value of the query observation is weighted by ρ , and the remaining mass ( 1−ρ ) is distributed among its neighbours to form the estimate.
Figure 5: Image Task Results. Our algorithm adapts rapidly when the task changes, whether task boundaries are available or hidden, while other methods are slow.
Figure 6: Craftax-Classic Results. Our algorithm outperforms other continual learning algorithms, including ablations, in both two-parallel and 1024-parallel environments.
Figure 7: Chain MDP with the two feature sets
Table 8
Figure 8: Value function heatmap for the prediction task
Figure 9: Image Task (500) Results.
Parameter
Value
Hyperparameters
ALG_NAME
PQN
NUM_MINIBATCHES
4
NUM_EPOCHS
1
NORM_INPUT
True
NORM_TYPE
layer_norm
Table 3: Hyperparameters for PQN for Image task.
Parameter
Value
TRANSIENT_TABLE_SIZE
128
NUM_TABLES
2
NUM_HASHES
128
NUM_SLOTS
4
TRANSIENT_LR
0.3
DECAY
0.95
Table 4: PT-Ours HPs.
Parameter
Value
TRANSIENT_LR
0.003
DECAY
1.0
Table 5: PT-NN HPs.
Parameter
Value
TRANSIENT_TABLE_SIZE
512
NUM_TABLES
2
NUM_HASHES
128
TRANSIENT_LR
0.5
DECAY
1.0
Table 6: PT-IHT HPs.
Parameter
Value
TRANSIENT_TABLE_SIZE
128
NUM_TABLES
2
NUM_HASHES
128
NUM_SLOTS
4
TRANSIENT_LR
0.3
DECAY
0.95
Table 7: T-Only HPs.
Figure 10: Hyperparameter tuning results.
Parameter
Value
NUM_TABLES
2
TRANSIENT_TABLE_SIZE
2048
NUM_SLOTS
32
NUM_HASHES
128
CROP_SIZE
7
ρ
0.85
Table 8: PT-Ours HPs for Craftax.
Parameter
Value
NUM_TABLES
16
TRANSIENT_TABLE_SIZE
65536
NUM_SLOTS
32
NUM_HASHES
128
CROP_SIZE
7
TRANSIENT_LR
0.8
Table 9: PT-IHT HPs for Craftax.
Parameter
Value
ALG_NAME
PQN
TOTAL_TIMESTEPS
5×108
TOTAL_TIMESTEPS_DECAY
5×108
NUM_ENVS
1024
NUM_STEPS
32
EPS_START
0.1
Table 10: PQN HPs for Craftax
Figure 11: Parameter Ablation Studies. Rows represent different parameters tested: Minibatch updates (MB), ρ , table configuration, decay parameter, and transient learning rates (v1 and v2). Columns represent standard metrics: Episode Returns, Score, and Episode Lengths.
Figure 12: All achievements in the craftax online experiment.
Figure 13: All achievements in the craftax 250M experiment.
ETH AI Center, Department of Computer Science, ETH Zürich · Mila - Quebec AI Institute, School of Computer Science, McGill University · Mila - Quebec AI Institute +1