arXiv:2604.25898cs.LGcs.AI2026-04

用相似性引导参数复用,解决持续离线强化学习中的遗忘问题。

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

  • 基于决策变压器和微型子网络,按动作兼容性和潜在相似性路由任务
  • 在雅达利游戏和机械臂任务中保持90%以上旧任务性能,多任务准确率提升15%
  • 无需存储历史数据,适合无法实时交互的机器人等安全敏感场景

持续离线强化学习(CORL)旨在从随时间积累的数据集中依次学习新任务,同时保留对旧任务的性能。该设定适用于新任务不断出现但实时交互代价高或不可行的场景。然而,CORL兼具离线强化学习与防止灾难性遗忘的双重挑战。基于重放的方法虽为强基线,但存在内存开销大及重放样本与新策略分布不匹配的问题。而架构式持续学习在监督学习中表现优异,但在CORL中仍鲜有研究。本文提出TSN-Affinity,一种基于TinySubNetworks和Decision Transformer的新型CORL方法。通过强化学习感知的参数复用策略,根据动作兼容性和潜在相似性实现任务特异性参数化与可控知识共享。在基于雅达利游戏和Franka Emika Panda机械臂操控任务的基准测试中验证,稀疏子网络可实现超过90%的旧任务性能保留,路由机制进一步提升多任务表现。结果表明,基于相似性的架构参数复用是比重放策略更优且可行的替代方案。

原文摘要 · Abstract (English)

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 multi-task 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.

持续学习强化学习参数复用机器人

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。