提出新方法让强化学习模型持续吸收旧知识,避免遗忘且提升新任务表现。
Continual Knowledge Adaptation for Reinforcement Learning
- 用任务专属知识向量池动态调用历史经验,防止遗忘
- 在三个基准上提升整体性能4.20%,前向迁移效率提高8.02%
- 自适应合并相似知识,降低内存占用,适合长期学习场景
强化学习使智能体通过与环境交互学习最优行为。然而真实环境通常是非平稳的,要求智能体持续适应新任务和变化条件。尽管持续强化学习能跨任务学习,现有方法常面临灾难性遗忘和知识利用效率低的问题。为此,我们提出持续知识适配强化学习(CKA-RL),实现历史知识的积累与高效利用。具体而言,引入持续知识适配策略,维护任务特定的知识向量池,并动态使用历史知识以适应新任务。该过程通过保留并调整关键模型参数,缓解灾难性遗忘,促进任务间有效知识迁移。此外,提出自适应知识融合机制,合并相似知识向量,解决可扩展性问题,在降低内存开销的同时确保核心知识保留。在三个基准上的实验表明,所提 CKA-RL 显著优于现有先进方法,整体性能提升 4.20%,前向迁移性能提升 8.02%。源代码已开源:https://github.com/Fhujinwu/CKA-RL。
原文摘要 · Abstract (English)
Reinforcement Learning enables agents to learn optimal behaviors through interactions with environments. However, real-world environments are typically non-stationary, requiring agents to continuously adapt to new tasks and changing conditions. Although Continual Reinforcement Learning facilitates learning across multiple tasks, existing methods often suffer from catastrophic forgetting and inefficient knowledge utilization. To address these challenges, we propose Continual Knowledge Adaptation for Reinforcement Learning (CKA-RL), which enables the accumulation and effective utilization of historical knowledge. Specifically, we introduce a Continual Knowledge Adaptation strategy, which involves maintaining a task-specific knowledge vector pool and dynamically using historical knowledge to adapt the agent to new tasks. This process mitigates catastrophic forgetting and enables efficient knowledge transfer across tasks by preserving and adapting critical model parameters. Additionally, we propose an Adaptive Knowledge Merging mechanism that combines similar knowledge vectors to address scalability challenges, reducing memory requirements while ensuring the retention of essential knowledge. Experiments on three benchmarks demonstrate that the proposed CKA-RL outperforms state-of-the-art methods, achieving an improvement of 4.20% in overall performance and 8.02% in forward transfer. The source code is available at https://github.com/Fhujinwu/CKA-RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。