Type 2 · Promising method Requested 证据 B 收录

Hierarchical World Models with Implicit Dynamics

Gaoyue Zhou, Yvonne Wu, Zichen Jeff Cui et al. Mido Assran, Lerrel Pinto, Yann LeCun

全部作者(7)

Gaoyue Zhou, Yvonne Wu, Zichen Jeff Cui, Nicolas Ballas, Mido Assran, Lerrel Pinto, Yann LeCun

发布 · 收录于 2026-11-03

让扩散模型只负责“Δ 步之内能到哪里”,把它的输入噪声当作高层动作来搜索子目标,再用低层世界模型求出具体动作,就能在不需要奖励和价值函数的情况下做长时程规划。

任务
目标条件的长时程规划:给起点和目标状态,求动作序列
环境
Wall、PointMaze(4 种尺寸)、PushT(推 T 形块),另有视觉版 PushT
数据
离线、无奖励标注的随机轨迹
硬件
仅仿真,无真机实验
主要结果
PushT 随机目标成功率 0.77,单层世界模型 0.21

证据等级 B · 仿真充分 (A 真机 + 多基准 + 消融 / B 仿真充分 / C 单一基准或无代码)

动机

上方是高层:从当前状态出发,每一跳不超过 Δ 步,连成一串子目标通向终点;下方是低层:在相邻子目标之间求出具体动作。
原文图 · P.2 上方是高层:从当前状态出发,每一跳不超过 Δ 步,连成一串子目标通向终点;下方是低层:在相邻子目标之间求出具体动作。 图片来源:Zhou et al., Hierarchical World Models with Implicit Dynamics (arXiv:2610.42146) · 查看原文

世界模型(world model) 是一个学出来的“环境模拟器”:给当前状态和动作,预测下一步状态。用它做规划,就是在脑子里试各种动作,挑最能到达目标的那串。

问题在于任务一长就不行: - 搜索空间随步数指数增长; - 每一步的预测误差会层层累积(→ P.1)。

已有的分层方法要么依赖奖励、价值函数或预训练技能来判断子目标能不能到(→ P.2),要么学一套“宏动作”。作者指出,强力的优化器会钻学到的模型的空子,找到现实里不存在的“虫洞”式转移(→ P.2)。所以难点是:高层动作空间既要压缩轨迹,又要“守规矩”,不让规划器搜出假路径。

方法

  1. 1编码把当前观测和目标编码成潜状态
  2. 2撒噪声采样一串高斯噪声,当作高层动作
  3. 3想子目标扩散模型把噪声变成Δ步内可达的状态
  4. 4搜索CEM调噪声,让最后子目标靠近目标
  5. 5求动作低层世界模型+CEM找到达子目标的动作
  6. 6执行重规划执行完一轮后从新状态重新规划

比喻:像在陌生城市走路去远处。高层只问“从这里走 5 分钟能到哪些地方”,每掷一次骰子(一个噪声向量)就得到一个真实可达的落脚点;把几次骰子连起来,看最后离目的地多近。低层再回答“具体怎么走到下一个落脚点”。因为每个骰子结果都落在真实可达的地方,搜骰子比搜一套学出来的宏动作更安全。

1. 高层:隐式动力学模型(implicit dynamics)(→ P.4-5)。这是一个条件扩散模型(DDPM,一种从噪声一步步去噪生成样本的模型),输入当前状态,生成 Δ 步之后的一个可能状态。推理时用 DDIM(一种确定性的去噪采样方式),于是“当前状态 + 噪声”唯一对应一个子目标。噪声空间是标准高斯分布,平滑、各向同性,作者把它当作高层动作空间。

2. 低层:确定性世界模型(→ P.5)。沿用 DINO-WM 的 transformer 预测器:给过去几步状态和动作,预测下一步。规划时把它当“逆动力学求解器”:用 CEM(交叉熵方法,一种反复采样、保留最好一批再采样的优化法)找一串动作,使预测终点最接近子目标。

3. 噪声正则(→ P.6)。规划代价 = 最后一个子目标到目标的距离 + 一项惩罚:噪声分量绝对值超过阈值 τ 就罚,防止规划器跑到高斯分布的稀有区域,生成不真实的子目标。

灰色上半部分是高层世界模型,红圈里的噪声 z 是被优化的高层动作;蓝色下半部分是低层世界模型,橙圈里的动作 a 也在测试时优化,两处虚线框分别是两层的规划代价。
原文图 · P.4 灰色上半部分是高层世界模型,红圈里的噪声 z 是被优化的高层动作;蓝色下半部分是低层世界模型,橙圈里的动作 a 也在测试时优化,两处虚线框分别是两层的规划代价。 图片来源:Zhou et al., Hierarchical World Models with Implicit Dynamics (arXiv:2610.42146) · 查看原文
左列是输入噪声网格(颜色标位置),中列是本文把它映射成的可达状态,颜色过渡平滑且留在起点附近;右列是普通大跳帧世界模型,相邻输入会跳到很远、形成不连续的区域。
原文图 · P.26 左列是输入噪声网格(颜色标位置),中列是本文把它映射成的可达状态,颜色过渡平滑且留在起点附近;右列是普通大跳帧世界模型,相邻输入会跳到很远、形成不连续的区域。 图片来源:Zhou et al., Hierarchical World Models with Implicit Dynamics (arXiv:2610.42146) · 查看原文

实验

PushT 随机目标的成功率(状态输入) SR · 越高越好 ↑
PushT 随机目标的成功率(状态输入)(SR,越高越好 ↑):本文 ID-WM(本文) 0.77;SL-WM 0.21;GCBC 0.21;HIQL 0.02;H-Diffuser 0.01;Diffuser 0 本文 ID-WM · 本文 0.77 SL-WM 0.21 GCBC 0.21 HIQL 0.02 H-Diffuser 0.01 Diffuser 0 00.40.8

本文方法 对比方法 · 数据来源:Table 1, P.8

视觉 PushT,目标在 75 步之外的成功率 SR · 越高越好 ↑
视觉 PushT,目标在 75 步之外的成功率(SR,越高越好 ↑):本文 ID-WM(本文) 0.5;HWM 0.14;DINO-WM 0 本文 ID-WM · 本文 0.5 HWM 0.14 DINO-WM 0 00.250.5

本文方法 对比方法 · 数据来源:Table 3, P.10

对照:给单层世界模型同样或更多算力(PushT 随机目标) SR · 越高越好 ↑
对照:给单层世界模型同样或更多算力(PushT 随机目标)(SR,越高越好 ↑):本文 ID-WM(本文) 0.77;SL-WM(耗时对齐) 0.28;SL-WM(CEM 预算对齐) 0.21 本文 ID-WM · 本文 0.77 SL-WM(耗时对齐) 0.28 SL-WM(CEM 预算对齐) 0.21 00.40.8

本文方法 对比方法 · 数据来源:Table 8, P.21

指标:成功率 SR(越高越好);PushT 另报 T 块覆盖率 Cov(越高越好);附录还有到达步数(越少越好)和推理时间(越短越好)。每个设置随机采 50 组起点-目标,报 3 个种子的均值(→ P.7-8)。所有实验固定 Δ=5(含跳帧即 25 个环境步)、每次 5 个子目标,不按任务调参(→ P.7)。

对比对象:模仿学习 GCBC、离线强化学习 HIQL、扩散规划 Diffuser 和 H-Diffuser、单层世界模型 MPC(SL-WM),以及同期的分层世界模型 HWM。

主要结果: - 表 1(→ P.8):本文在每一行都最好或并列最好。差距最大的是 PushT 随机目标,本文 SR 0.77,其余方法都不超过 0.21。GCBC 在固定目标上也能到 1.00,说明真正的难点是分布外目标。 - 和 HWM 比(→ P.9 表 2):PushT 四个设置本文都明显更高(例如随机目标 0.77 对 0.20)。 - 扩展性(→ P.9 图 7):迷宫越大,基线掉得越多,本文下降最少;PushT 目标距离从 25 步加到 150 步,本文接近满分,单层世界模型快速掉下去。 - 视觉 PushT(→ P.10 表 3):目标 25 步远时 HWM 更好(0.96 对 0.90);50、75 步和随机目标时本文最好(如 75 步 0.50 对 0.14)。

站得住的结论:附录把单层和本文的 CEM 预算设成一样,又额外给单层三倍重规划来对齐耗时,单层只从 0.21 升到 0.28(→ P.21 表 8),说明收益来自分层。消融显示 Δ=1 时成功率都不超过 0.10,Δ=5、5 个子目标时 0.96(→ P.20 表 4);噪声正则和更多 CEM 样本都有帮助(→ P.20 图 10)。

代价:到达一个目标的推理时间本文 53.08 s,单层世界模型 16.93 s,HIQL 5.61 s,Diffuser 24 min(→ P.21 表 6,单张 L40S)。路径效率上,迷宫里 HIQL 平均 34.1 步,本文 163.9 步,因为本文代价只看终点距离、不优化路径长短(→ P.20-21 表 5)。

左图横轴是迷宫从小到大,看各方法成功率下降多少,棕线是本文;右图横轴是 PushT 目标距离,蓝线本文基本持平,橙线单层世界模型快速下降。
原文图 · P.9 左图横轴是迷宫从小到大,看各方法成功率下降多少,棕线是本文;右图横轴是 PushT 目标距离,蓝线本文基本持平,橙线单层世界模型快速下降。 图片来源:Zhou et al., Hierarchical World Models with Implicit Dynamics (arXiv:2610.42146) · 查看原文
每行从左到右是生成的子目标图像,最右灰底是目标;上行本文的 T 块一步步转向目标姿态,下行 AVDC 的 T 块几乎不动。
原文图 · P.10 每行从左到右是生成的子目标图像,最右灰底是目标;上行本文的 T 块一步步转向目标姿态,下行 AVDC 的 T 块几乎不动。 图片来源:Zhou et al., Hierarchical World Models with Implicit Dynamics (arXiv:2610.42146) · 查看原文

要点 Take-home

  1. 把扩散模型的初始噪声当高层动作,规划器在平滑的高斯空间里搜子目标。
  2. 随机目标推T:成功率0.77,单层世界模型0.21,给三倍算力也仅0.28。
  3. 代价是推理慢:每个目标约53秒,是单层世界模型的约3倍。
  4. 目前只在二维仿真和64×64图像上验证,还没有真机。

Claude 读完全文后的提炼。

更多:先读哪 · 值得看的理由 · 质疑

先读哪

先看第 2 页图 1 和第 8 页表 1,再看第 21 页表 8

值得看的理由

  1. 把扩散模型的初始噪声当作高层动作,不靠价值函数或预训练技能就能做分层世界模型规划 → P.5 第 3.1.1 节
  2. 算力对齐对照说明增益来自分层:单层世界模型给三倍重规划仍只有 0.28,本文 0.77 → P.21 表 8

质疑

  • 全部实验是二维仿真小环境,视觉实验仅 64×64 PushT,且 HWM 基线是作者自行复现(官方未开源该版本) → P.18 第 6.4 (g) 节

审稿人笔记

表 1 中 PushT-Aug 随机目标成功率为 1.00,而表 4 同一默认配置为 0.96,两者可能来自不同次运行,引用时注意。