让行为基础模型在线探索并实时适配新任务,无需预设奖励数据。
Exploration and Online Transfer with Behavioral Foundation Models

- 用行为基础模型生成探索策略,实现在线零样本迁移。
- 在线学习中通过置信上界思想优化探索,收敛到最优策略。
- 适合需要实时反馈、无法提前获取奖励的任务场景。
零样本强化学习旨在训练一个代理,在仅使用无奖励轨迹的情况下,能为任意奖励函数生成最优策略,且无需在迁移时额外学习。这类模型因其任务泛化能力,被称为「行为基础模型」(BFMs)。然而现有框架假设迁移阶段可提前获得状态-奖励对数据集,以便选择最佳策略;但在实际中若奖励为黑盒(如用户直接反馈),则无法预先生成此类数据,必须通过环境交互来观测奖励。这与传统强化学习中通过试错探索发现奖励的设定不一致。本文提出解决这一在线迁移问题的新方法,核心思想是利用行为基础模型自身生成探索策略。我们将该在线学习问题建模为类似多臂赌博机的探索-利用问题:每步由算法推荐一个策略,行为基础模型在环境中执行,获得奖励与新状态,重复此过程直至收敛至最优策略。在常见线性奖励近似背景下,我们基于上置信界思想推导出一种形式,通过最小化不确定性矩阵的特征值实现有效探索。我们在简单环境中进行了定性和定量评估,验证了该方法的有效性。
原文摘要 · Abstract (English)
Zero-shot Transfer in Reinforcement Learning (RL) aims to train an agent that can generate optimal policies for any reward function, without additional learning at transfer time, while training only on reward-free trajectories. For their generality over tasks, such models are sometimes called ``Behavioral Foundation Models'' (BFMs). While they have shown strong performances and improvements in recent years, the current framework and algorithms still assume that, during the transfer phase, the agent is informed offline about the reward (the task to solve) through a dataset of state-reward pairs, which it uses to pick the best policy to deploy. However, in practice if the reward is a black-box (e.g. direct user feedback), it is not possible to generate such a dataset: it is necessary to observe the reward through interactions with the environment. In other words, the current framework of offline transfer is not aligned with the traditional RL setting of online learning through trial-and-error, which requires exploration in order to find rewards. This paper proposes to tackle this new online transfer in zero-shot RL, with the key insight that the BFM itself can be used to generate exploration policies. We show that it is possible to frame this online learning problem in terms of a bandit-like exploration-exploitation problem. More precisely, at each step the bandit algorithm recommends a policy, the BFM executes it in the environment, which yields a reward and a new state; we repeat the process until we converge to the optimal policy. In the popular context of linear reward approximation, we derive a formulation inspired by Upper Confidence Bound and show that exploration can be achieved through the minimization of the eigenvalues of an uncertainty matrix. We evaluate qualitatively and quantitatively our framework on a simple environment to validate the concept of our method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。