文献题目：Prompting Decision Transformer for Few-Shot Policy Generalization

摘要

人类可以利用先前的经验并从少量演示中学习新任务。与旨在通过更好的算法设计实现快速适应的离线元强化学习相比，我们研究了架构归纳偏差对少样本学习能力的影响。我们提出了一种基于提示的决策转换器 (Prompt-DT)，它利用 Transformer 架构的顺序建模能力和提示框架来实现离线 RL 中的小样本自适应。我们设计了轨迹提示，其中包含小样本演示的片段，并对特定于任务的信息进行编码以指导策略生成。我们在五个 MuJoCo 控制基准测试中的实验表明，Prompt-DT 是一个强大的少样本学习器，无需对未见目标任务进行任何额外微调。 Prompt-DT 的性能大大优于其变体和强大的元离线 RL 基线，其轨迹提示仅包含几个时间步长。 Prompt-DT 对提示长度的变化也很稳健，并且可以推广到分布外 (OOD) 环境。项目页面:
https://mxu34.github.io/PromptDT/。

引言

离线强化学习(离线RL) (Levine et al.，2020)旨在从一组不访问环境的行为策略收集的轨迹中学习最优策略。这种数据驱动的方法在许多情况下是必不可少的，在这些情况下，在线交互可能是昂贵的(如机器人或教育代理)和危险的(如自动驾驶或医疗保健)。最近的一些研究表明，这种方法在支持数据驱动的游戏环境政策学习(Chen等人，2021年)、机器人操作行为(Ebert等人，2018年;Kalashnikov等人，2018年)和机器人导航技能(Kahn等人，2021年)。
然而，我们发现离线RL的内在困难之一是无法推广到看不见的任务。虽然代理可能能够从训练任务中获得良好的状态覆盖，但由于分布的转移，它仍然很难在测试任务中找到一个好的策略。因此，Mitchell等人(2021)最近的工作考虑了旨在解决离线RL中的泛化问题的离线元RL设置。提出了使用优势加权回归(Peng et al.， 2019)作为子例程RL算法的Meta-Actor批评家优势加权(Meta-Actor Critic with Advantage weighted, MACAW)算法，并以元学习目标优化agent的适应能力(Finn et al.， 2017a)。
虽然元学习方法通过算法学习的视角解决了这个问题，但我们的目的是在这项工作中调查架构归纳偏差的力量。众所周知，Transformer (Vaswani et al.， 2017)模型在对大规模数据集进行预训练时，能够进行少镜头或零镜头学习。此外，自然语言处理(NLP)最近的工作(Liu et al.， 2021;Brown et al.， 2020)认为，==基于提示的框架是适应新任务的有效范式，如翻译和问题回答。在基于提示的框架中，提示包含关于任务的有价值的信息，并作为输入的前缀预先挂起。因此，它将少镜头或零镜头的泛化问题转化为条件序列生成，这是这些大型Transformer模型的优势所在。==最近，Chen等人(2021)表明，除了自然语言之外，Transformer体系结构还可以对轨迹数据具有强大的序列建模能力，在离线RL上实现了最先进的结果。在这项工作中，我们的目标是解决这样一个问题:我们能否利用NLP中的基于提示的框架，并将其适应于离线RL的环境，以实现对不可见任务的少镜头泛化?
==值得注意的是，将基于提示的方法应用于RL问题并非易事。==在自然语言处理(NLP)中，语言模型(lm)在大量的原始文本(包括互联网上几乎所有的信息)上进行预先训练。此外，大多数NLP任务可以重写为标准的空白填充格式作为提示。这些提示作为查询从预先训练的lm中提取正确的信息。在RL中，由于不同任务之间的内在差异，预训练的模型是否有足够的知识来解决不可预见的任务是值得怀疑的。我们建议使用几发子弹的演示作为提示，我们称之为轨迹提示。我们没有进行无监督的语言模型预训练，而是集中于有监督地训练一个代理，该代理可以模仿这些演示来生成新策略，而无需进行微调。在我们基于提示的离线RL框架中，代理首先使用从同一领域/环境中的不同任务收集的离线轨迹进行训练。对于每个任务，代理学习预测目标轨迹，同时对从同一任务中采样的轨迹提示进行条件反射。在评估期间，给代理一个新任务和一些新轨迹(总步长小于或等于15)来构建提示。无需进行额外的调整，代理应该利用这些轨迹中显示的任务信息，并为新任务生成策略。该框架功能强大且具有吸引力，原因有很多:它允许代理利用从不同任务收集的离线轨迹，代理可以执行少镜头学习，在不更新代理的情况下适应新场景。
我们将这种方法称为基于提示的决策转换器(prompt - dt)，它利用Transformer体系结构的顺序建模能力和提示框架来实现离线RL中的少镜头适应。我们的贡献如下。
1.我们提出了Prompt-DT，这是一个基于transformer的模型，通过由少量轨迹构建的短轨迹提示来学习适应看不见的任务。
2.我们在五个MuJoCo控制基准测试中的实验表明，Prompt-DT是一个强大的少弹学习器，无需对目标任务进行任何额外的微调，大大超过了强元离线RL基准。
3.我们的分析表明了基于提示的框架的必要性，以及提示长度的鲁棒性和提示质量的敏感性。
4.提示dt可以推广到非分布任务，而之前的方法都失败了。

预赛

在线和离线元强化学习

强化学习 (RL) 问题是一个顺序决策问题，其中学习代理与环境交互并优化其控制策略以获得最优值。动态环境中的每个顺序决策任务通常建模为马尔可夫决策过程 (MDP)（Sutton & Barto，2018），由元组 $M = (S, A, P, R, µ)$ 表示。 $S$ 和 $A$ 是状态空间和动作空间。 $P : S \times A \times S \to R$ 是转移模型，其中 $P (s, a, s^{'})$ 是在状态 $s$ 采取行动 $a$ 到达状态 $s^{'}$ 的概率。 $R : S \to R$ 是奖励函数。 $µ$ 是初始状态分布。在每一步，RL 智能体通过基于当前状态 $s$ 采取行动 $a$ 与环境交互，观察奖励 $r$ 和从环境中产生的下一个状态 $s^{'}$ 。顺序决策任务的目标是找到一个策略 $π : S \times A \to R$ 来优化预期的累积奖励
通常，RL 是在线执行的，其中代理迭代地采取行动并从环境中接收反馈。然而，这可能并不总是可行的，因为 RL 算法由于样本效率普遍较低而可能需要大量训练数据。这使得在线环境中的培训非常耗时。在一些现实世界的安全关键环境中，在训练阶段在线部署代理可能会导致灾难性的失败。我们考虑离线 RL 设置（Levine 等人，2020），其目的是从使用（可能未知的）行为策略预先收集的数据中学习策略。在此设置中，代理可以访问包含一组轨迹的数据集 $D$ 。轨迹 ${s_0, a_0, r_0, s_1, a_1, r_1, . . . , s_T , a_T , r_T \}$ 使用环境中的行为策略进行采样。 agent 需要在不与环境本身交互的情况下，仅使用数据集 $D$ 找到最优策略。
最初提出在线和离线 RL 设置是为了在一项任务中找到最优策略。如果设计的学习代理能够在学习了一些任务后用少量新收集的数据适应类似任务，则可以进一步提高 RL 的效率，这主要是在元 RL 下开发的（Finn 等人，2017a） . 最近，meta-RL 已扩展到离线设置，旨在通过预先收集的数据快速适应新任务。在 Mitchell 等人 (2021) 提出的离线元强化学习设置中，代理被赋予一组任务 $T$ ，其中任务 $T_i ∈ T$ 定义为 $M_i, π_i)$ ，包含 MDP $M_i$ 和行为策略 $π_i$ . 对于每个任务 $T_i$ ，代理都提供了一个数据集 $D_i$ ，其中包含使用 $π_i$ 采样的轨迹。该代理使用表示为 $T^{train}$ 的训练任务子集进行训练，并期望在与 $T^{train}$ 不相交的一组测试任务 $T^{test}$ 中找到最优策略。

决策Transformer

已在 NLP（De vlin 等人，2018 年）和计算机视觉（Carion 等人，2020 年）中广泛研究的 Transformer 表现出优于基于 RNN 的架构。它最近被用于解决 RL 问题，因为它在对长序列数据建模时具有效率和可扩展性。离线强化学习的决策转换器 (Chen et al., 2021) 将学习策略视为顺序建模问题。它建议使用在不同时间步长 $t$ 收集的状态 $s_t$ 、动作 $a_t$ 和 reward-to-go $\hat r_t$ 元组对轨迹进行建模。 reward-to go 是从当前时间步到 episode 结束的累积奖励。这种新颖的表示不包括单步奖励 $r_t$ ，而是有助于指导行动选择以优化回报。在时间步 $t$ ，Decision Transformer 将自回归轨迹序列 $τ$ 作为输入，其中包含最近的 $K$ 步历史记录。
当使用离线收集的数据进行训练时，

。在测试过程中，
其中 G*是一集的目标总回报。每个轨迹 $τ$ 对应于标准 Transformer 模型中的 $3 K$ 个令牌。为了对序列时间步信息进行编码，Decision Transformer 将相同的时间步嵌入连接到 $s_t$ 、 $a_t$ 和 $\hat r_t$ 的嵌入。对应于状态标记的每个头都被训练为通过在连续动作空间时最小化均方损失来预测动作。

基于提示的决策转换器

本节介绍基于提示的决策转换器 (Prompt-DT)，这是一种用于小样本策略泛化的新型转换器架构，如图 1 所示。

在这里插入图片描述

图 1. 用于小样本策略泛化的 Prompt-DT。左边显示了每个任务 $T_i ∈ T^{train} ∪ T^{test}$ 的少量样本演示数据集 $P_i$ 。轨迹提示定义为从存储在 $P_i$ 中的各种事件中采样的长度为 $K$ 的轨迹序列。在预训练和小样本评估中，Prompt-DT 将轨迹提示增强和最近的 $K$ 步历史作为输入，并自回归输出与输入序列中每个状态对应的动作。

4.1 问题表述

在使用离线数据对一组任务进行训练后，我们将离线少样本 RL 问题形式化为针对新任务的少样本策略泛化问题。训练集 $T^{train}$ 中的每个任务 $T_i$ 都与数据集 $D_i$ 相关联，其中包含使用未知行为策略 $π_i$ 预先收集的轨迹。与使用特定于任务的离线数据或在线交互更新模型权重的离线元 RL 相比，我们希望在没有微调或梯度更新的情况下实现泛化，从而保持高效率并避免由于参数移动而导致的灾难性遗忘。
为了在 RL 的上下文中实现少样本学习，我们假设存在一个数据集 $P_i$ ，其中包含每个任务 $T_i ∈ T^{train} ∪ T ^{test}$ 的少样本演示。对于训练任务 $T_i ∈ T^{train}$ ，我们让 $P_i$ 是离线数据集 $D_i$ 的一个子集。测试任务 $T_i ∈ T$ 测试的 $P_i$ 可以通过人类实验者或行为策略获得。在这项工作中，我们希望设计一种架构，可以直接提取存储在演示数据集 $P_i$ 中的独特任务特定信息，并利用这些信息来指导策略生成。

4.2 及时表示

包含任务特定说明的文本提示使大型 Transformer 模型能够在不更改 NLP 任务中的模型参数的情况下生成答案。在 RL 的背景下，最近引入了可以作为提示的文本描述来解决 Atari 视频游戏和多模式家庭任务（Shridhar 等人，2020）。此类文本描述通常需要预定义的语言模板，并且可能需要大量人力来进行注释。
相反，我们将 RL 的轨迹提示定义为由几个轨迹段组成的序列。每个轨迹段包含多个状态 $s *$ 、动作 $a *$ 和 reward-to go $\hat r*$ 元组 $\hat r*)$ ，遵循 Decision Transformer 中的轨迹表示。每个带有上标的元素都与轨迹提示相关联。由于每个顺序决策任务 $T_i$ 都可以建模为 MDP $M_i$ ，因此轨迹提示可以存储部分到完整的信息，以通过隐式捕获转换模型和奖励函数来指定任务。通过直接从少镜头演示数据集 $P_i$ 中采样轨迹段，与文本提示相比，轨迹提示相对容易获得。形式上，我们将任务 $T_i$ 的轨迹提示 $τ_i$ 定义为
$K *$ 是提示中存储的环境步骤数。请注意，我们选择的 $K *$ 比任务的范围短得多。因此，轨迹提示仅包含帮助识别任务所需的信息，但没有足够的信息供智能体模仿。

4.3 提示 DT 体系结构

我们的 Prompt-DT 架构建立在 Decision Transformer (Chen et al., 2021) 的基础上，并通过提示增强序列建模问题的视角解决了离线小样本 RL 问题。建议的轨迹提示允许对 Decision Transformer 进行最小的架构更改以进行泛化。对于每个任务 $T_i$ ，Prompt-DT 将 $τ$ 输入作为输入，其中包含从 $P_i$ 获得的 $K *$ 步轨迹提示和从 $D_i$ 采样的最近 $K$ 步历史记录。形式上 $τ^{input} = (τ*_i, τ_i)$ 。由于每个时间步的数据对是一个三元组 $\hat r)$ ，因此输入序列对应于 Transformer 中的 $3 (K * + K)$ 个标记。 Prompt-DT 自回归地输出与输入序列中的状态标记相对应的头部 $K * + K$ 动作。
在实现中，我们利用随机轨迹提示 $τ*_i$ 来提高训练稳定性并避免算法 2 中的过度拟合。任务 $T_i$ 的随机轨迹提示 $τ*_i$ 由长度为 $H$ 且 $K * = J H$ 的 $J$ 个轨迹段组成。
遵循 Decision Transformer 的结构，我们利用具有线性层的 GPT 模型来获取标记嵌入，并将相同的位置嵌入添加到对应于相同时间步长的标记中。

4.4 算法

我们在算法 1 和算法 3 中总结了 Prompt-DT 的训练和测试算法。
在训练期间，Prompt-DT 最大限度地减少预测动作与提示和最近历史数据中的动作之间的误差。通过学习存储在轨迹提示中的目标动作，Prompt-DT 被激励提取存储在轨迹提示中的特定于任务的信息，并将其与最近的历史记录相结合以预测未来的动作。在连续设置中，Prompt-DT 通过梯度下降最小化均方损失。在每个训练步骤中，我们对包含每个训练任务的提示历史对的批次 $B$ 进行采样。我们的批次 $B$ 不是使用从一个训练任务中采样的数据批次迭代地进行梯度更新，而是通过聚合 $T^{train}$ 中所有任务的梯度估计来帮助稳定训练。
在测试时，我们假设离线预训练的 Prompt DT 可以针对 $T^{test}$ 中的每个评估任务与模拟器进行交互。这种在线评估假设类似于训练有素的 RL 代理的实际部署。我们基于演示集 $P_i$ 采样随机轨迹提示，用于类似于训练过程的任务 $T_i ∈ T^{test}$ 。 Prompt DT 然后生成将提示和最近的上下文作为输入的操作。在每一集的开始，我们用全零初始化最近的历史 $τ$ 用于 Prompt-DT 预测，并用流式收集的数据更新它。

实验

我们进行实验来测试所提出的 Prompt-DT 的小样本泛化能力，并将度量作为情节累积奖励。我们的目标是根据经验回答以下问题：1）Prompt-DT 能否实现少镜头策略泛化？ 2）提示的数量和质量如何影响小样本泛化能力？ 3）Prompt-DT 是否泛化到分布外的任务？

## 5.1 环境和数据集

我们评估了如下描述的五个元强化学习控制任务（Finn 等人，2017a；Rothfuss 等人，2018；Mitchell 等人，2021；Yu 等人，2020a；Todorov 等人，2012） .
Cheetah-dir。 Cheetah-dir 中有两个任务，目标方向分别为向前和向后。猎豹智能体沿着目标方向获得高速奖励。训练集和测试集是相等的，都包含两个任务。
Cheetah-vel。 Cheetah-vel 中有 40 个目标速度不同的任务。目标速度从区间 [0,3] 中统一采样。代理因目标速度的 l2 错误而受到惩罚。我们拿出 5 个任务来构建测试集，并用剩下的 35 个任务进行训练。
Ant-dir。 Ant-dir 中有 50 个不同目标方向的任务在二维空间中统一采样。 8 关节蚂蚁沿目标方向获得高速奖励。我们抽取 5 个任务进行测试，其余的用于训练。
Dial。在 Dial 中，任务是控制一个 6-DOF Jaco
机器人到达位于数字键盘中的目标号码。一共有10个数字，对应10个任务。我们训练 6 个任务并测试 4 个任务。 Dial 比 Meta-World reach-v2 更复杂，因为它直接控制 6 个关节。
Meta-World reach-v2。在 Meta-World reach-v2 中，任务是控制 Sawyer 机器人的末端执行器到达 3D 空间中的目标位置。代理直接控制末端执行器的 XYZ 位置。每个任务都有不同的目标位置。我们训练 15 个任务并测试 5 个任务。
Cheetah-dir、Cheetah-vel 和 Ant-dir 中的实验严格遵循 Mitchell et al(2021) 中的数据集和设置。所有三个任务中的代理都受到大控制信号的惩罚。 Cheetah-dir 和 Ant-dir 的数据集包含用于使用 Soft Actor Critic 训练 RL 代理的完整回放缓冲区（Haarnoja 等人，2018）。 Cheetah vel 的数据集包含使用 TD3 (Fujimoto et al., 2018) 训练的回放缓冲区，以提高训练稳定性。在 Meta-World reach-v2 (Yu et al., 2020a) 和 Dial (Shiarlis et al., 2018) 中，我们使用两种环境中提供的脚本专家策略收集专家轨迹。