SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2604.25898 · 2026

TSN-Affinity: Similarity-Driven Parameter Reuse for Continual Offline Reinforcement Learning

Kamil Faber, Dominik Żurek, Roberto Corizzo, Marcin Pietron, Paweł Gajewski

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 7 functions out of this paper's own repositories and ran 6 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
anonymized-for-submission123/tsn-affinity canonical 6 of 7
FunctionStatusWhere it lives
build_atari_task_specs Ran anonymized-for-submission123/tsn-affinity/bin/clb-build.py
pointer only (licence: NONE) · get_code("e05055097833c360")
compute_matrices Ran anonymized-for-submission123/tsn-affinity/analyze_runs.py
pointer only (licence: NONE) · get_code("bec8fd98a85466fa")
load_results Ran anonymized-for-submission123/tsn-affinity/analyze_runs.py
pointer only (licence: NONE) · get_code("c8f60efab3e8b421")
obs_diff_stats Ran anonymized-for-submission123/tsn-affinity/bin/clb-run-dt-atari-single.py
pointer only (licence: NONE) · get_code("3664869b0da5a7c2")
pick_target_return Ran anonymized-for-submission123/tsn-affinity/bin/clb-run-atari-dt.py
pointer only (licence: NONE) · get_code("bccf997621a1224e")
replay_actions Ran anonymized-for-submission123/tsn-affinity/bin/clb-run-dt-atari-single.py
pointer only (licence: NONE) · get_code("7046184b8fe25d67")
infer_action_map_small_discrete Not yet run anonymized-for-submission123/tsn-affinity/bin/clb-run-dt-atari-single.py
pointer only (licence: NONE) · get_code("6e66225402cff576")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Continual offline reinforcement learning (CORL) aims to learn a sequence of tasks from datasets collected over time while preserving performance on previously learned tasks. This setting corresponds to domains where new tasks arise over time, but adapting the model in live environment interactions is expensive, risky, or impossible. However, CORL inherits the dual difficulty of offline reinforcement learning and adapting while preventing catastrophic forgetting. Replay-based continual learning approaches remain a strong baseline but incur memory overhead and suffer from a distribution mismatch between replayed samples and newly learned policies. At the same time, architectural continual learning methods have shown strong potential in supervised learning but remain underexplored in CORL. In this work, we propose TSN-Affinity, a novel CORL method based on TinySubNetworks and Decision Transformer. The method enables task-specific parameterization and controlled knowledge sharing through a RL-aware reuse strategy that routes tasks according to action compatibility and latent similarity. We evaluate the approach on benchmarks based on Atari games and simulations of manipulation tasks with the Franka Emika Panda robotic arm, covering both discrete and continuous control. Results show strong retention from sparse SubNetworks, with routing further improving multitask performance. Our findings suggest that similarity-guided architectural reuse is a strong and viable alternative to replay-based strategies in a CORL setting. Our code is available at: https://github.com/anonymized-for-submission123/tsn-affinity.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2604.25898")
get_code_for_paper("2604.25898")
have("2604.25898")

Connect an agent — have() is free.