解决跨空间持续离线强化学习难题,实现多环境任务无缝衔接。
Solving Continual Offline RL through Selective Weights Activation on Aligned Spaces
- 通过向量量化对齐不同任务的状态与动作空间,构建统一训练基础。
- 利用稀疏掩码选择性激活权重,在15个任务上超越16个基线模型。
- 适用于真实世界多环境学习,突破传统方法仅限同空间的限制。
持续离线强化学习(CORL)在基于扩散模型的终身学习系统中展现出强大能力,能建模轨迹的联合分布。然而,多数研究仅关注观察与动作空间相同的有限持续任务设置,偏离了在多种环境中训练智能体的实际需求。为此,我们提出向量量化持续扩散器(VQ-CD),打破不同任务间空间差异的壁垒。该方法包含两个互补部分:首先,通过向量量化对齐各任务不同的状态与动作空间,实现统一空间下的持续训练;其次,设计一个附带逆动态模型的统一扩散模型,根据任务相关的稀疏掩码选择性激活不同权重,以掌握所有任务。我们在15个持续学习任务上进行了广泛实验,涵盖常规设置(相同空间)与通用设置(不同空间)。相比16个基线方法,本方法达到最先进性能。
原文摘要 · Abstract (English)
Continual offline reinforcement learning (CORL) has shown impressive ability in diffusion-based lifelong learning systems by modeling the joint distributions of trajectories. However, most research only focuses on limited continual task settings where the tasks have the same observation and action space, which deviates from the realistic demands of training agents in various environments. In view of this, we propose Vector-Quantized Continual Diffuser, named VQ-CD, to break the barrier of different spaces between various tasks. Specifically, our method contains two complementary sections, where the quantization spaces alignment provides a unified basis for the selective weights activation. In the quantized spaces alignment, we leverage vector quantization to align the different state and action spaces of various tasks, facilitating continual training in the same space. Then, we propose to leverage a unified diffusion model attached by the inverse dynamic model to master all tasks by selectively activating different weights according to the task-related sparse masks. Finally, we conduct extensive experiments on 15 continual learning (CL) tasks, including conventional CL task settings (identical state and action spaces) and general CL task settings (various state and action spaces). Compared with 16 baselines, our method reaches the SOTA performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。