Policy Plasticity Matters in Offline-to-Online Reinforcement Learning: Refitting Offline Policies for Online Adaptation
Organizations: Zhejiang University · Carnegie Mellon University · Nanyang Technological University · Zhejiang University & Zhejiang Lab
Abstract
Offline-to-Online Reinforcement Learning (O2O RL) has emerged as a practical paradigm that pre-trains the policy using static offline datasets and subsequently adapts the policy through online interactions. Existing O2O methods primarily address the transition through value calibration, while generally treating the offline-trained policy as a given initialization. We instead study O2O adaptation from the perspective of network plasticity, asking whether the offline-trained policy remains sufficiently adaptable for online learning. Controlled experiments show that prolonged optimization on static offline data progressively reduces network plasticity even after offline performance has largely saturated, and that lower plasticity is associated with weaker subsequent online improvement. Motivated by these observations, we propose REstoring plasticity via Fresh Initialization and policy Transfer (REFIT), a lightweight model-level method for the O2O transition. Before online fine-tuning, REFIT distills the offline policy into a freshly initialized student while temporarily freezing a random subset of student units, transferring the learned offline behavior to a more plastic policy initialization. Extensive experiments on D4RL and OGBench demonstrate that REFIT consistently achieves higher aggregate performance than existing O2O plug-in methods across both Cal-QL and IQL backbones, while plasticity diagnostics and ablations provide further evidence of restored network plasticity.
Figures & tables
| Environment | IQL ( Kostrikov et al., 2021 ) | Cal-QL ( Nakamoto et al., 2023 ) | ||||
|---|---|---|---|---|---|---|
| Baseline | OPT | Ours (REFIT) | Baseline | PARS | Ours (REFIT) | |
| antmaze-m-p | 76.2 1.5 | 88.0 3.0 | 86.6 4.8 | 92.2 2.1 | 92.4 3.0 | 95.4 1.7 |
| antmaze-m-d | 71.6 2.7 | 88.8 1.2 | 89.2 2.7 | 93.2 0.8 | 89.6 2.9 | 96.2 0.8 |
| antmaze-l-p | 46.0 2.8 | 61.6 5.1 | 63.8 8.3 | 70.6 7.5 | 78.6 3.6 | 78.2 5.0 |
| antmaze-l-d | 50.6 5.7 | 63.6 3.6 | 65.2 4.0 | 53.6 17.4 | 79.2 12.4 | 80.2 5.0 |
| AntMaze total | 244.4 | 302.0 | 304.8 | 309.6 | 339.8 | 350.0 |
| Method | hc-m | hc-r | wal-m-r | hop-m-r | ant-m-p | ant-m-d | Score |
|---|---|---|---|---|---|---|---|
| Baseline | 48.1 0.1 | 13.8 3.9 | 74.8 2.8 | 77.5 18.1 | 76.2 1.5 | 71.6 2.7 | 362.0 (-23.3%) |
| Reset | 47.3 0.4 | 38.6 2.4 | 84.0 10.6 | 86.5 29.9 | 71.2 6.8 | 74.0 5.5 | 401.6 (-14.9%) |
| Reset + Distillation | 49.8 0.2 | 45.7 1.9 | 89.3 11.2 | 97.9 6.2 | 83.5 1.8 | 79.2 23.6 | 445.4 (-5.6%) |
| REFIT | 50.0 0.3 | 46.3 1.4 | 96.9 2.2 | 102.7 0.4 | 86.6 4.8 | 89.2 2.7 | 471.7 |
| Env | 0.0 | 0.25 | 0.5 | 0.75 | 1.0 |
|---|---|---|---|---|---|
| halfcheetah-m | 49.8 0.2 | 50.0 0.2 | 50.0 0.3 | 50.1 0.4 | 47.3 0.4 |
| halfcheetah-r | 45.7 1.9 | 46.2 1.3 | 46.3 1.4 | 46.5 1.8 | 38.6 2.4 |
| antmaze-m-p | 83.5 1.8 | 86.0 3.7 | 86.6 4.8 | 85.8 4.1 | 71.2 6.8 |
| antmaze-m-d | 79.2 23.6 | 86.4 3.0 | 89.2 2.7 | 84.4 1.5 | 74.0 5.5 |
| Total | 258.2 | 268.6 | 272.1 | 266.8 | 231.1 |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Suite | Token | Meaning | Example |
|---|---|---|---|
| Environment | hc | halfcheetah | hc-m |
| hop | hopper | hop-m-r | |
| wal | walker2d | wal-r | |
| ant | antmaze | ant-l-d | |
| MuJoCo Postfix | r | random | hc-r |
| m | medium | hop-m |
| Method | hc-m | hc-r | wal-m-r | hop-m-r | ant-m-p | ant-m-d | Total |
|---|---|---|---|---|---|---|---|
| IQL | 48.1 0.1 | 13.8 3.9 | 74.8 2.8 | 77.5 18.1 | 76.2 1.5 | 71.6 2.7 | 362.0 |
| IQL+CPR | 51.7 0.5 | 26.0 4.0 | 86.0 11.7 | 101.8 3.9 | 66.0 3.3 | 71.0 6.8 | 402.5 |
| IQL+REFIT | 50.0 0.3 | 46.3 1.4 | 96.9 2.2 | 102.7 0.4 | 86.6 4.8 | 89.2 2.7 | 471.7 |
| Cal-QL | 65.3 3.0 | 33.4 3.8 | 86.7 8.7 | 68.9 7.8 | 92.2 2.1 | 93.2 0.8 | 439.7 |
| Cal-QL+CPR | 43.3 0.6 | 2.2 0.1 | 17.6 2.8 | 30.3 7.9 | 27.0 10.3 | 26.0 13.9 | 146.3 |
| Cal-QL+REFIT | 51.1 0.9 | 39.2 3.5 | 97.9 5.8 | 87.6 7.5 | 95.4 1.7 | 96.2 0.8 | 467.4 |
| Environment | MSE | ||
|---|---|---|---|
| antmaze-m-d | 86.9 6.7 | 87.0 5.2 | 89.2 2.7 |
| antmaze-m-p | 87.8 2.7 | 84.8 5.2 | 86.6 4.8 |
| halfcheetah-m | 49.9 0.2 | 50.0 0.6 | 50.0 0.3 |
| halfcheetah-r | 46.3 1.5 | 47.5 1.7 | 46.3 1.4 |
| Total Score | 270.9 2.9 | 269.2 2.9 | 272.1 2.8 |
| Method | antmaze-m-d | antmaze-m-p | halfcheetah-m | halfcheetah-r | Total Score |
|---|---|---|---|---|---|
| Default | 89.2 2.7 | 86.6 4.8 | 50.0 0.3 | 46.3 1.4 | 272.1 2.8 |
| Copy Std | 86.4 4.5 | 84.3 4.0 | 50.1 0.3 | 46.9 2.9 | 267.7 2.3 |
Offline-To-Online RL
Offline-to-online reinforcement learning adapts policies learned from fixed datasets through further interaction with the environment.
See all 69 papers