JEPA4Japan · 教程

第 26 章——线性探针:潜在状态编码了哪些物理变量?

1,420字 4分钟阅读 #LeWorldModel#世界模型#JEPA

用合适的基线探测位置、速度、角度和目标距离,同时区分可解码性、模型使用、相关性与因果性。

课程进度 课程大纲 已发布 48/48 课

第 0 部分——阅读指南:我们究竟要学什么?

  1. 01 第 0 章——开始之前 已发布

第 1 部分——世界模型:智能体内部的沙盒

  1. 02 第 1 章——智能体为什么需要“想象未来” 已发布
  2. 03 第 2 章——为什么不直接预测下一张图像? 已发布
  3. 04 第 3 章——JEPA 思想:预测意义,而不是图像的复制品 已发布
  4. 05 第 4 章——用一张图看懂 LeWM 已发布

第 2 部分——把图像变成状态:LeWM 架构

  1. 06 第 5 章——轨迹数据:对模型而言,世界不是图像集合 已发布
  2. 07 第 6 章——视觉编码器:为每一帧签发“状态护照” 已发布
  3. 08 第 7 章——动力学预测器:在头脑中推动时间前进 已发布
  4. 09 第 8 章——一次完整的前向传播:跟随一个批次从头走到尾 已发布

第 3 部分——防止模型作弊:预测损失与 SIGReg

  1. 10 第 9 章——最危险的捷径:表示崩塌 已发布
  2. 11 第 10 章——预测损失:模型怎样学习下一步 已发布
  3. 12 第 11 章——SIGReg 的直觉:让表示空间“呼吸” 已发布
  4. 13 第 12 章——只保留最少却足够的数学 已发布
  5. 14 第 13 章——原始 LeWM 的端到端训练机制 已发布
  6. 15 第 14 章——训练一个不会立即崩塌的模型 已发布

第 4 部分——让模型行动起来:在潜在空间中规划

  1. 16 第 15 章——目标条件规划:从“我在哪里?”到“我要去哪里?” 已发布
  2. 17 第 16 章——潜在欧氏距离:方便,但未必可靠 已发布
  3. 18 第 17 章——CEM:用淘汰赛搜索动作 已发布
  4. 19 第 18 章——MPC:不要一次信任模型太久 已发布
  5. 20 第 19 章——长视界滚动:小误差怎样滚成大失败 已发布
  6. 21 第 20 章——从零实现一个最小 LeWM 规划器 已发布

第 5 部分——工程复现:从论文到可运行系统

  1. 22 第 21 章——官方代码仓库与实验环境 已发布
  2. 23 第 22 章——第一个实验:TwoRoom 冒烟测试 已发布
  3. 24 第 23 章——第二个实验:复现 PushT 已发布
  4. 25 第 24 章——怎样公平评估世界模型 已发布
  5. 26 第 25 章——失败诊断手册 已发布

第 6 部分——LeWM 究竟学到了什么?

  1. 27 第 26 章——线性探针:潜在状态编码了哪些物理变量? 当前课程
  2. 28 第 27 章——为潜在空间做一次“健康检查” 已发布
  3. 29 第 28 章——预期违背:模型会对“不可能事件”感到惊讶吗? 已发布
  4. 30 第 29 章——怎样严谨讨论“理解世界” 已发布

第 7 部分——为什么“预测准确”仍可能“规划很差”

  1. 31 第 30 章——训练目标与规划目标之间的差距 已发布
  2. 32 第 31 章——全局不崩塌不保证保留任务相关动力学 已发布
  3. 33 第 32 章——各向同性高斯先验何时过强? 已发布
  4. 34 第 33 章——长视界规划:预测得更远,还是规划得更聪明? 已发布
  5. 35 第 34 章——从位置距离到任务进度 已发布
  6. 36 第 35 章——多任务学习、真实机器人与视觉干扰 已发布
  7. 37 第 36 章——理论边界:何时能辨识真实状态? 已发布

第 8 部分——从复现者到研究者

  1. 38 第 37 章——设计一项可信的 LeWM 改进实验 已发布
  2. 39 第 38 章——十二个可执行的研究项目 已发布
  3. 40 第 39 章——LeWM 研究中的开放问题 已发布

附录

  1. 41 附录 A——最低限度的必需数学工具箱 已发布
  2. 42 附录 B——PyTorch 实现速查 已发布
  3. 43 附录 C——完整张量形状表 已发布
  4. 44 附录 D——实验配置卡 已发布
  5. 45 附录 E——论文时间线与证据等级 已发布
  6. 46 附录 F——术语表 已发布
  7. 47 附录 G——复现检查清单 已发布
  8. 48 附录 H——专家审阅检查清单 已发布

先看全景

  1. 冻结保持一个明确命名的表征不变。
  2. 划分回合不让相邻帧泄漏。
  3. 拟合读出器先用线性直尺,再用受控 MLP。
  4. 只说“可恢复”是否被使用需要另一项测试。
探针是一台外挂扫描仪。它能读取信号,却不负责驱动引擎。

一个可训练探针接到冻结的 LeWM 嵌入上,而世界模型本身保持不变。

一个小故事

机场扫描仪能看见行李箱中的钥匙。这说明扫描仪可以从当前视角恢复钥匙信息,却不能证明旅客知道钥匙在那里,也不能证明旅客会使用它。

探针的工作方式相同。冻结训练后的表征,收集其潜在向量,再训练一个小型读出器,把这些向量映射到模拟器标签。成功读出说明:对于这个表征、读出器、数据划分和分布,该信息在统计上可以访问。它不能证明预测器或规划器会使用该信息。

真正规则

明确指出连接位置:编码器输出、投影后的状态、预测器输入和预测器输出是不同对象。冻结检查点、评估模式、裁剪、缩放、归一化、池化、投影器和历史构建方式。如果探针梯度更新了编码器,实验就已经变成监督微调。

应在提取嵌入或拟合目标归一化之前,划分完整回合。在训练回合上训练,在验证回合上选择超参数,最后只在从未触碰的测试回合上评估一次。随机帧划分把相邻帧放在两侧时,它们是近似重复样本,并不代表独立泛化。

LeWM v3 报告了下列探针目标:

  • TwoRoom:二维智能体位置;
  • PushT:智能体位置、物块位置、物块角度;
  • OGBench-Cube:关节位置/速度、末端执行器位置/yaw、夹爪状态、物块位置/quaternion/yaw。

目标距离是教程扩展,不是 v3 探针目标。论文为线性和非线性探针报告 MSE 与 Pearson 相关系数。存在尺度偏差时,相关系数仍可能很高;平均 MSE 则可能掩盖某个糟糕维度或罕见接触切片。

线性探针测试简单的仿射可访问性。MLP 测试更丰富假设类中的可访问性,也可能利用捷径。二者都应报告容量、正则化、优化和调参预算。

可能骗过我们的把戏

设想每个 TwoRoom 回合都在大致相同的时间穿过门口,并且角落中有一个微弱标记会随帧索引变亮。随机帧划分会让 MLP——甚至线性探针——通过时间预测位置。

现在改为按回合划分、随机改变策略速度,并移除标记。如果性能下降,原结果仍然只意味着这一点:在时间与位置耦合的场景中,位置可以恢复。

使用下列对照:常量均值预测器、简单像素或特权可观测量基线、未训练编码器、打乱标签,以及训练与测试曲线。TwoRoom 按开阔地面、墙壁和门口切片;PushT 则在标签可靠时,按自由移动、接近、首次接触、滑动和旋转切片。

若要测试信息是否被使用,应在探测之后进行干预。扰动与某变量相关的子空间,并在相同回合和候选动作上,与等大小的随机方向、保范数噪声和不干预设置比较。行为改变仍可能来自纠缠,或落入分布外潜在状态;因此即便如此,也只能支持“在该项干预下与行为有关”,不能自动建立因果性。

实验凭据

后续结果让这条边界变得具体:

  • Objective Bottleneck 报告,在一个 TwoRoom 后续系统中,终端潜在 L2 与真实终点距离之间 Pearson (r=0.426),而 ridge 位置探针在留出数据上达到 (R^2=0.9922)。
  • SCALE 报告,一项状态回归对照在部分任务上能够达到或超过完整嵌入的可解码性,却不能达到同样的距离对齐;与之相比,SCALE 在 15 个任务—求解器平均值中改善 15 个,该对照改善 13 个。
  • Decision-Metric Alignment 报告,四种未崩塌 PushT 变体之间的探针 (R^2) 差异小于 0.03,而 Plan–Real Spearman 从 +0.280 提高到约 +0.410–+0.420,在线成功率范围为 49.3%–92.7%。每项配置只有一次训练运行;三个随机种子测量的是评估变异。

这些是彼此独立、由作者报告的协议。它们说明可解码性可以与面向规划器的几何分离;却不能确定某个通用机制。原始 v3 的探针实现细节不足以精确重建每一项报告中的探针。

快速检查

  1. 为了让探针描述原始表征,什么必须保持冻结?
  2. 相比线性探针成功,MLP 成功多说明了什么,又不能说明什么?
  3. 哪种干预能把主张从“可恢复”推进到“与行为有关”?