跳转到内容

目标条件 RL 与 HER

目标条件策略 π(a∣s,g)\pi(a|s,g) 用一个网络解决一族任务;事后经验回放(HER)把未达成目标的轨迹重新标注为达成了实际到达的状态,把稀疏奖励变稠密。本章讲 HER 的实现与目标采样策略,以及目标条件价值函数与距离学习的联系。

目标条件策略:一个网络,一族任务

Section titled “目标条件策略:一个网络,一族任务”

目标条件策略 π(a∣s,g)\pi(a \mid s, g) 把目标 gg 作为输入的一部分,用一个网络解决一族任务("到达状态 gg"的所有实例)。好处:

  • 任务间的结构共享:到达 A 与到达 B 的技能(移动、避障)是同一套,学一个比学 NN 个省;
  • 零样本泛化到新目标(同族内);
  • 与分层 RL(技能 = 隐式目标,→ 选项框架与技能发现)、语言条件任务(目标 = 语言指令)天然衔接。

训练的困难是稀疏奖励:奖励只在 sT=gs_T = g 时给(到达才有),探索的期望时间随状态空间指数增长(→ 探索与利用)。

事后经验回放(Hindsight Experience Replay, Andrychowicz et al., 2017)的关键洞察:一条未达成目标的轨迹,对"它实际到达的目标"来说就是成功。机械臂想抓 A 却碰到了 B——这条轨迹没教"怎么抓 A",但完美地教了"怎么碰 B"。

实现:把轨迹重标注(relabel)——对轨迹上实际到达的状态 sTs_T,把它作为事后目标 g′=sTg' = s_T 存入回放缓冲:

(s0,a0,…,sT; g)⟹(s0,a0,…,sT; g′=sT, r=1).(s_0, a_0, \dots, s_T;\ g) \quad \Longrightarrow \quad (s_0, a_0, \dots, s_T;\ g' = s_T,\ r = 1).

稀疏奖励被机械地变稠密:每条轨迹(无论成败)都贡献成功的经验。训练循环与普通 off-policy RL 相同(DDPG/SAC + 回放),只是每个转移带着(可能重标注的)目标。

为什么有效:目标条件结构使"成功经验"可以跨目标复用——学到"从这个状态这样动能到达那",对任何目标都对。HER 是稀疏奖励机械臂任务的标准解法。

重标注用什么目标,是 HER 的核心超参:

  • future(最常用):从轨迹之后的状态中采样 kk 个作为事后目标——"顺路"的目标,最相关;
  • final:用轨迹的终态——简单但只教"到达终点";
  • episode:同回合的任意状态;random:其他回合的状态——增加多样性但相关性低。

kk(每条轨迹重标注几个目标)通常 4–8。与课程式出题(→ 目标导向的探索)的关系:HER 的"事后目标"天然是当前能力边缘上的目标(刚到达的),等于免费的自动课程。

目标表示决定泛化:

  • 状态目标(向量):最简单,泛化限于同状态空间;
  • 图像目标:需要表示学习(在潜空间比较),泛化到视觉相似的目标;
  • 语言目标:接语言模型,泛化到语义空间(→ 机器人)。

理论联系:目标条件价值函数 V(s,g)V(s, g) 与状态间的(学习出的)距离有对应——V(s,g)V(s, g) 的负值近似一种可学习的度量(时间距离),这是 goal-conditioned RL 与距离学习/表度学习(metric learning)交叉的来源。

  • HER 是稀疏奖励机械臂任务的标准解法
  • 目标表示(状态/图像/语言)决定泛化
  • 与第六卷目标导向探索衔接