4. 世界模型核心架构

世界模型之所以能“在脑内预演未来”,依赖的不是单一网络,而是一条从感知到预测再到重建的完整链路。理解这条链路,是后续做评测指标设计、失效归因和安全验证的前提。本章把世界模型拆成五个可独立审视、又必须协同工作的部件:编码器负责把高维观测压缩成隐空间表示,动力学模型负责在隐空间中推演状态演化,解码器负责把隐状态还原为可观测的预测,序列建模负责维持时间一致性,多模态融合负责把相机、激光雷达、毫米波雷达、IMU、地图等异构输入对齐到同一表征空间。

一、核心原理:五部件分工与数据流

从工程视角看,世界模型可以形式化为一个隐状态序列模型。设 t 时刻的多模态观测为 o_t,编码器 E 输出隐表示 z_t = E(o_t);动力学模型 f 依据历史隐状态与动作 a_t 预测下一隐状态 z_{t+1} = f(z_t, a_t);解码器 D 将隐状态映射回观测空间 ô_{t+1} = D(z_{t+1})。序列建模模块则把单步预测串成轨迹,多模态融合模块在编码前后完成特征对齐。

关键点在于:隐空间不是装饰,而是压缩与抽象的核心。它决定了模型能保留哪些信息、丢弃哪些噪声。隐空间过窄会丢失小目标与细粒度运动,过宽则让动力学学习变得困难、易过拟合。评测时,隐空间维度、连续性、可分性都是可量化对象。

世界模型五部件数据流与隐空间推演 多模态观测 相机图像 激光点云 毫米波 / IMU 地图与定位 编码器 E 特征提取 跨模态对齐 隐空间表示 z 连续 / 离散 维度与容量 语义可分性 时序一致性 动力学模型 f 状态转移 动作条件预测 解码器 D 观测重建 未来帧预测 序列建模:RNN / Transformer / 状态空间模型 时间对齐 · 长时依赖 · 滚动预测 · 误差累积控制 多模态融合 早期 / 中期 / 晚期 评测关注 重建 / 预测 / 一致性

二、方法流程:从观测到预测的六步链路

  1. 多模态预处理与时间同步:将不同频率、不同坐标系的传感器数据对齐到统一时间戳与自车坐标系,处理丢帧与延迟。
  2. 编码与融合:各模态分别编码后,通过注意力、拼接或对比学习融合为统一隐表示。融合时机决定信息保留程度。
  3. 隐空间约束:通过 KL 散度、对比损失、重构损失等约束隐空间结构,使其既紧凑又可预测。
  4. 动力学推演:在隐空间中按动作条件滚动预测,输出未来若干步隐状态序列。
  5. 解码与重建:将隐状态解码为未来图像、BEV 占据栅格、轨迹或语义分割,供下游规划与验证使用。
  6. 误差评估与回灌:对比预测与真实观测,计算重建误差、预测误差与一致性指标,定位失效环节。
工程要点:编码器与解码器通常成对设计,但评测时可以解耦。例如固定编码器只评动力学预测能力,或固定动力学只评解码重建质量,从而把误差归因到具体部件。

三、案例场景:城市路口的多模态预测

以城市无保护左转路口为例。相机提供语义与纹理,激光雷达提供精确几何与距离,毫米波雷达提供速度与穿透能力,IMU 与轮速提供自车运动,高精地图提供车道拓扑。编码器需要把这些异构输入映射到同一隐空间;动力学模型需要预测对向直行车、横穿行人与自车左转动作的交互演化;解码器输出未来 3 秒的 BEV 占据与目标轨迹。

若隐空间只保留外观特征而丢失速度信息,动力学模型会低估对向车接近速度,导致预测轨迹过于乐观。若序列建模窗口过短,模型无法捕捉行人“犹豫—起步”的长时意图,预测会在关键时刻跳变。若多模态融合只在晚期拼接,激光雷达的几何优势可能被图像语义淹没。这些都是评测中必须显式覆盖的场景。

四、误区与复盘:常见架构陷阱

误区 表现 后果 复盘建议
隐空间过大 重构极好但预测发散 动力学难学,长时预测崩溃 加入 KL 或对比约束,做维度消融
融合时机单一 某模态主导,其他被忽略 恶劣天气或遮挡下鲁棒性差 对比早期/中期/晚期融合方案
序列窗口过短 意图预测跳变 交互场景误判 测试不同历史长度与长时依赖
解码目标单一 只评图像重建 忽略几何与运动误差 增加 BEV、轨迹、占据等多任务解码
训练与评测分布不一致 离线指标好,闭环失效 安全验证结论不可信 引入闭环滚动评测与分布偏移测试
安全验证提醒:世界模型的预测误差不会均匀分布,往往集中在交互密集、遮挡严重、长尾行为等高风险场景。评测集必须按场景难度分层,不能只看平均指标。

五、参数与配置示例

以下是一组典型世界模型训练配置片段,用于说明各部件参数如何影响评测重点:

model:
  encoder:
    modality: [camera, lidar, radar, imu, map]
    fusion: mid          # early | mid | late
    latent_dim: 256
  dynamics:
    type: transformer
    history: 8           # 历史帧数
    horizon: 12          # 预测步数
    action_cond: true
  decoder:
    heads: [rgb, bev_occ, traj, seg]
  sequence:
    window: 16
    stride: 2
loss:
  recon: 1.0
  kl: 0.001
  contrastive: 0.1
  traj: 0.5

评测时,latent_dim 决定隐空间容量,history 与 horizon 决定时序难度,fusion 决定多模态信息保留方式,heads 决定可验证的输出维度。把这些参数与指标联动分析,才能判断架构改动是否真正提升安全相关能力。

六、小结:架构即评测对象

编码器、动力学模型、解码器、隐空间、序列建模与多模态融合不是孤立模块,而是一条可拆解、可归因、可验证的链路。做世界模型评测时,应把每个部件都视为潜在失效点:隐空间是否保留关键信息,动力学是否在长时推演中稳定,解码是否覆盖几何与运动,序列建模是否维持一致性,多模态融合是否在退化条件下仍可靠。只有把架构理解透,后续的指标设计、场景构造与安全验证才有坚实依据。