酷应用

加速RL探索效率，CMU、谷歌、斯坦福提出以弱监督学习解纠缠表征

百家作者：机器之心 2020-05-13 13:43:08

选自arXiv

作者：Lisa Lee等

机器之心编译

机器之心编辑部

巨大的探索空间阻碍了强化学习（RL）的发挥，这篇论文通过弱监督学习从广泛的目标空间中分离出有语义意义的表征空间，从而增强 RL 的学习速度与泛化性能。

通用型智能体必须通过与真实环境交互来高效学习各种任务。典型的方法是人为地定义一组奖励函数，令智能体仅学习那些由奖励函数诱导出的任务。然而，定义与调整这些奖励函数需要耗费大量精力，并且使用者需要针对其关心的任务设置特定奖励函数，这为他们增添了额外的负担。而设计一个既能提供充足学习信号又能在算法收敛时诱导正确行为的奖励函数，是非常有挑战性的。

最近，来自 CMU、谷歌大脑和斯坦福大学的研究者发布一项研究，展示了如何通过弱监督以最小负担为智能体提供有用信息，以及如何利用这些监督帮助智能体在环境中学习。研究者探索了一种在目标趋向强化学习（goal-conditioned RL）设置中使用弱监督的方法。

研究者提出的弱监督智能体不需要通过探索和学习来达到每个目标状态，而是只需学习沿着有意义的变化轴达到相应状态，无需关注与解决人类指定任务无关的状态维度。重点是，研究者提出通过弱监督来执行此类约束，而不是列举目标或任务及其对应奖励。

这项工作的主要贡献是弱监督控制（weakly-supervised control，WSC），这是一个将弱监督引入 RL 的简单框架。该方法学习一个有语义意义的表征空间，智能体可以使用该表征空间生成自己的目标，获取距离函数，并执行定向探索。

WSC 包含两个阶段：首先基于弱标注离线数据学习状态的解纠缠表征，然后使用解纠缠表征约束 RL 智能体的探索空间。

实验结果表明，学习解纠缠表征能够加快强化学习在多种操作任务上的学习速度，并提高其泛化能力。此外，研究者还发现 WSC 能够产生可解释的潜在策略（latent policy），其潜在目标直接与环境的可控特征保持一致。

论文链接：https://arxiv.org/abs/2004.02860

了解 WSC 之前你需要先知道这些

目标趋向强化学习

研究者团队通过元组 (S, A, P, H, G) 定义有限时域下目标趋向的马尔可夫决策过程，其中 S 是观测空间，A 是动作空间，P (s′ | s, a) 表示一个未知动态函数，H 表示最大时间长度，G ⊆ S 表示目标空间。

在目标趋向 RL 中，研究者通过优化预期累积奖励

来训练策略 π_θ (a_t | s_t, g)，从而在目标空间中达到目标 g〜G，其中 R_g(s) 是由目标 g ∈ G 和观测值 s ∈ S 之间的距离度量定义的奖励函数。

在低维度任务中，我们可以简单地将奖励视为状态空间中的负 ℓ_2 距离。然而，在高维度空间（如图像）中定义距离度量更具挑战性。先前关于视觉目标趋向（visual goal-conditioned）的 RL 工作训练了一个额外的状态表征模型，例如变分自编码器（VAE encoder）e^{VAE}：S→Z^{VAE}。这些方法基于编码状态和目标训练一个策略，并使用潜在空间中的 ℓ_2 距离来定义奖励函数：

弱监督解纠缠表征

该研究提出的方法在 RL 环境中使用弱监督解纠缠表征学习。解纠缠表征学习旨在学习数据的可解释表征，表征的每一个维度度量一个独特的变化因子（factor of variation），这些因子是数据生成的基础（示例参见图 2）。

图 2：基于视觉的机械臂操作环境示意图。

该研究使用一种叫作 rank pairing 的弱监督形式，其中数据集 D := {(s_1, s_2, y)} 由观测值 {s_1 , s_2 } 与弱二值化标签 y ∈ {0, 1}^K 组成，y_k = 1(f_k(s_1) < f_k(s_2)) 表示观测值 s_1 的第 k 个因子的值是否小于 s_2 的相应因子值。

使用这些数据，Shu et al. (2019) 提出的弱监督方法通过优化以下损失函数训练出编码器 e : S → Z、生成器 G : Z → S 和判别器 D：

弱监督强化学习问题

问题描述

给定弱标注数据集 D: = {((s_1, s_2, y)}，它由观测值 {s_1, s_2} 和弱二值化标签 y∈{0,1}^K 组成，其中 y_k = 1(f_k(s_1)<f_k(s2)) 表示观察值 s_1 的第 k 个因子值是否小于 s_2 的相应因子值。

除了这些标签外，使用者还可以指定索引子集 I ⊆ [K]，用来表示哪些因子（f_1，...，f_K）∈ F 与解决某一类任务有关。在训练期间，智能体可以与环境交互，但除了 D 中的弱标签外不接受任何监督（即没有奖励）。

在测试阶段，采样未知的目标因子 f_I^∗ ∈ F_I，则智能体接收到目标观测（如目标图像），其因子等于 f_I^*。智能体的目标是学习 latent-conditioned RL 策略，以最小化目标距离

新方法：弱监督控制（WSC）

该研究提出的弱监督 RL 训练框架 WSC 包含两个阶段：首先基于弱标注离线数据学习状态的解纠缠表征，然后使用解纠缠表征来约束 RL 智能体的探索空间。

图 3：WSC 框架示意图。

从观测中学习解纠缠表征

研究者提出的解纠缠表征学习基于上文介绍的 Shu et al. (2019) 方法构建。当然，理论上也可以使用其他类似的方法。该方法通过优化公式 1 中的损失函数来训练编码器、生成器和判别器。在训练完解纠缠模型后，研究者丢弃了生成器与判别器，仅用编码器来定义目标空间，并计算状态之间的距离。

结构化目标生成与距离函数

该研究提出的新方法将目标空间定义为学得的解纠缠潜在空间 Z_I，限制在索引 I 下。其目标采样分布的定义如下：

其中，Z^{min}_I 与 Z^{max}_I 分别表示对潜在值在元素层面上逐个取最小或最大。

在每一次迭代中，该方法从 p(Z_I) 中直接采样潜在目标 z_g，或从 replay buffer 中采样图像观测，并将其编码为解纠缠模型 z_g = e_I (s_g )。然后，执行该策略得到轨迹 (s_1, a_1, ..., s_T)，从而尝试该目标。在基于 replay buffer 采样 transition (s_t, a_t, s_t+1, z_g) 时，研究者使用 hindsight re-labeling 和修改后的目标来提供额外的训练信号。也就是说，研究者有时会使用修改后的目标 z′_g 重新标注 transition (s_t, a_t, s_t+1, z′_g)。

该方法将奖励函数定义为解纠缠潜在空间中的负 ℓ_2 距离：

整个 WSC 框架的伪代码如下所示：

实验

在图 4 中，研究者团队在 Sawyer 环境中的视觉目标趋向任务（参见图 2）上，对比了其提出的新方法和先前的 SOTA 目标趋向 RL 方法。

图 4：不同方法在视觉目标趋向任务上的性能随训练步的变化情况。弱监督控制（WSC）的学习速度超过之前的 SOTA 目标趋向 RL 方法（HER、RIG、SkewFit），尤其是在环境复杂性提高的情况下。因此，我们可以看到，在（学得的）语义解纠缠潜在空间中进行定向探索和目标采样比在 VAE 潜在空间中进行纯粹无监督的探索更加有效。

在图 5 中，研究者评估了针对视觉目标趋向任务的训练策略，并比较了每个时间步上的潜在目标距离与真实目标距离。