arXiv:2504.07896cs.LGcs.AI2025-04被引 24

让预训练智能体在几轮互动中快速提升零样本表现,避免性能下降。

Fast Adaptation with Behavioral Foundation Models

  • 在预训练任务嵌入空间中搜索,快速优化零样本策略。
  • 仅需数十回合即实现10%-40%性能提升,超越现有基线。
  • 适合需要高效适应新任务的强化学习场景。

无监督零样本强化学习已成为预训练行为基础模型(BFM)的强大范式,使智能体能以零样本方式通过奖励函数解决多种下游任务,无需测试时学习或规划。这依赖于同时学习自监督任务嵌入与近最优行为,并通过推理过程直接检索给定奖励函数对应的潜在任务嵌入和策略。尽管效果显著,零样本策略常因无监督训练、嵌入误差及推理过程而次优。本文提出快速适应策略,在少量在线交互中改进BFM的零样本性能,且不造成性能下降。我们发现现有BFM已掌握比其推理过程识别出的更优策略,因此适合快速适应。为此,我们设计了基于演员-评论家和仅演员的快速适应方法,探索预训练BFM的低维任务嵌入空间,迅速提升任意下游任务的表现。该方法缓解了微调预训练强化学习模型常见的初始“遗忘”阶段。我们在多个导航与运动领域对四种前沿零样本强化学习方法进行了评估,结果表明,仅需几十个环境交互回合,性能提升达10%-40%,优于现有基线。

原文摘要 · Abstract (English)

Unsupervised zero-shot reinforcement learning (RL) has emerged as a powerful paradigm for pretraining behavioral foundation models (BFMs), enabling agents to solve a wide range of downstream tasks specified via reward functions in a zero-shot fashion, i.e., without additional test-time learning or planning. This is achieved by learning self-supervised task embeddings alongside corresponding near-optimal behaviors and incorporating an inference procedure to directly retrieve the latent task embedding and associated policy for any given reward function. Despite promising results, zero-shot policies are often suboptimal due to errors induced by the unsupervised training process, the embedding, and the inference procedure. In this paper, we focus on devising fast adaptation strategies to improve the zero-shot performance of BFMs in a few steps of online interaction with the environment while avoiding any performance drop during the adaptation process. Notably, we demonstrate that existing BFMs learn a set of skills containing more performant policies than those identified by their inference procedure, making them well-suited for fast adaptation. Motivated by this observation, we propose both actor-critic and actor-only fast adaptation strategies that search in the low-dimensional task-embedding space of the pre-trained BFM to rapidly improve the performance of its zero-shot policies on any downstream task. Notably, our approach mitigates the initial "unlearning" phase commonly observed when fine-tuning pre-trained RL models. We evaluate our fast adaptation strategies on top of four state-of-the-art zero-shot RL methods in multiple navigation and locomotion domains. Our results show that they achieve 10-40% improvement over their zero-shot performance in a few tens of episodes, outperforming existing baselines.

强化学习快速适应基础模型零样本

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