JEPA4Japan · 教程

第 9 章——最危险的捷径:表示崩塌

1,200字 3分钟阅读 #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. 完美作弊预测差距变成零
模型可以通过抹掉整个世界,赢下预测游戏。

健康的点云、单点崩塌和狭窄的维度崩塌具有不同几何形态。

一个小故事

设想一家护照签发处,给每位旅客都盖上同一张卡片。它的记录一致得惊人:每一本“下一张护照”都与每次预测完全相符。但这些护照无法区分任何人。

这就是完全表征崩塌(complete representation collapse)。维度崩塌(dimensional collapse)更加隐蔽:护照之间虽有变化,但几乎所有信息都挤在高维潜在空间里的一条细路上。较低的内在维数也可能完全合理,因此“狭窄”是一个需要结合数据和任务解读的警告,而不是自动判决。

真正规则

这个漏洞只需四行就能写完:

encode(any_image) -> the_same_vector
predict(any_vector, any_action) -> the_same_vector
prediction_gap -> zero
state distinctions -> gone

它之所以成立,是因为下一状态目标也由同一个可训练编码器产生,而不是一份固定的物理答案。因此,只采用预测损失的训练允许常量解;这不代表每次低损失运行都会走到这个解。

SIGReg 加入了一个群体分布形状目标。单个重复点不可能匹配非退化的等方标准高斯,因此完全采用常量的捷径会与理想联合目标发生冲突。这里的结论刻意保持狭窄。一个分布广阔的点云仍可能编码背景颜色、忽略动作,或把状态排列成误导规划的几何结构。

崩塌也不等同于过拟合。一个模型可能在训练集和验证集上同时崩塌;也可能保留丰富的训练样本差异,却严重泛化不良。因此既要测量表征结构,也要测量留出数据上的行为。

下列筛查手段应配合使用:

  • 每坐标方差,用来发现明显收缩;
  • 协方差谱,用来发现经过旋转的狭窄子空间;
  • 有效秩,用来比较匹配设置下的不同运行;
  • 成对距离分布和重复率;
  • 能回溯到原始观测的最近邻面板;
  • 用于检验控制利用情况的动作交换敏感性。

它们都不是证明。类似单位阵的协方差只描述二阶结构;满秩不能证明动力学正确;看起来漂亮的邻居可能只是遵循颜色,而不是可达性。

容易骗过我们的把戏

分别使用和不使用 SIGReg,训练两组其余条件匹配的小型 TwoRoom 模型。在每个检查点上,始终使用同一批探针数据,记录预测损失、原始 SIGReg、方差、协方差谱、有效秩、成对距离、最近邻和动作交换结果。

假设并不是“没有 SIGReg 的运行一定会崩塌”,而是:移除群体分布项会重新引入一个预测损失本身无法排除的平凡最优解。如果点云仍然分布广泛,说明避免了完全崩塌,但仍然必须进行动作和滚动展开测试。如果一批多样化输入变成狭窄的潜在点云,这条证据会比仅在一个狭小房间角落得到的秩数值更有力。

证据凭据

LeWorldModel v3 指出了常量捷径并使用 SIGReg。LeJEPA v3 解释了完全崩塌与维度崩塌,并以隐藏的 X 形示例说明,简单的坐标轴检查为何会漏掉依赖关系。冻结的 train.py 分别记录预测损失、SIGReg 和总损失。

单随机种子的独立 TwoRoom 报告测得位置探针 Pearson 相关系数为 0.9988,接近论文中的 0.996。这说明在该表征中,位置可由线性方法读出;却不能证明是 SIGReg 导致了这一结果、预测器确实使用了位置,或表征学到了物理规律。

快速检查

  1. 为什么常量编码能得到完美的下一潜在状态一致性?
  2. 维度崩塌与过拟合有何不同?
  3. 为什么必须让多项潜在空间诊断相互印证,才能判定崩塌?