1B 世界模型跑赢闭源旗舰 21 个百分点：迭代深度成为新的 scaling 轴

译序：为什么 LoopWM 值得完整读

“Current world models face a fundamental tension: faithful long-horizon simulation demands deep computation, but deeper models are expensive to deploy and prone to compounding errors.”

—— Looped World Models, arXiv 2606.18208, 2026-06-16 提交

做世界模型（World Model, WM）的人都被两件事反复折磨过：

1. 长时序会崩——单步预测再准，rollout 几十步之后误差就滚雪球。论文开篇引的 Xiao 2020 和 Talvitie 2017 研究的正是这个问题，十年了没有干净解法。
2. 加深就变贵——加几层 transformer 单步确实更准，但参数量、推理延迟、显存占用同步上涨。自动驾驶、机器人这类实时场景扛不住。

Looped World Models（LoopWM）给的答案是把这两件事解耦：不加深层数，把同一组参数迭代更多次。参数量不动，计算深度靠循环次数堆出来。

实验结果相当激进：约 1B 参数的模型，在 ScienceWorld 文本环境的 14 个子任务上，EM 平均超过 claude-opus-4-6-max 21.2 个百分点，其中 Lifespan 子任务从 0% 拉到 100%。论文摘要把这概括为"最高 100× 的参数效率"。

读完全文，我的判断是：这篇论文真正立住的东西不是"1B 打败百倍大的闭源旗舰"，而是验证了一条新的 scaling 轴——iterative latent depth（迭代深度），与模型规模、数据规模正交。它的对比基线全部是 LLM，没有和 DreamerV3、IRIS 这些经典世界模型正面交手，所以"世界模型 SOTA"还谈不上；但作为 looped 架构在世界模型上的第一次系统落地，方法和实验链条是完整的。

目录

• 一、论文速览
• 二、问题：长时序模拟的两难
• 三、方法：把深度换成迭代次数
• 四、实验一：ScienceWorld
• 五、实验二：AlfWorld
• 六、局限与未解问题
• 七、实践判断：什么时候值得试
• 八、常见问题
• 九、关键判断
• 十、参考链接

一、论文速览

一个先要交代的背景：这篇论文的评估协议是把世界建模表述成文本预测任务——给模型连续 5 个动作，让它预测最终状态，按 EM / Token F1 / BLEU-4 / Entity 四个指标打分。这个设定让闭源 LLM 可以直接当基线跑，但也意味着所有结论都在"文本状态预测"这个框架里，跟 RL 社区常跑的 Atari、Crafter 不是一回事。这一点后面第六节还会展开。

二、问题：长时序模拟的两难

2.1 世界模型与两个失败模式

世界模型学习环境演化规律。最简形式写作：

$$h_{k+1} = f_\theta(h_k, a_k, o_k)$$

它是 sample-efficient RL 的基石：PlaNet（Hafner 2019）证明 agent 可以纯从像素学 latent dynamics 并在线规划，确立了 RSSM 这个基础架构；DreamerV3（Hafner 2025）用一套超参在 150 多个任务上达到人类水平，是这条线的顶点。之后 transformer 介入：IRIS 用自回归 transformer 处理离散 latent token，TransDreamer 引入 Transformer State-Space Model，Δ-IRIS 改进 tokenization，DIAMOND 用扩散模型生成画面，EMERALD 在 Crafter 上拿到 SOTA；更大尺度上 Sora 和 Genie 展示了视频生成模型可以当通用世界模拟器。

论文把长时序模拟的困难归到两个根因：

1. 误差累积：每步预测误差独立存在，K 步 rollout 后轨迹质量快速退化（引 Xiao 2020、Talvitie 2017、Luo 2022）。
2. 加深不经济：靠加深层数对抗退化，参数量和推理成本成比例上涨，实时部署变得昂贵。

常规思路是"单步更强 + 步数更少"。论文反着走：单步内用同一组参数多迭代几次，用循环次数换深度。

2.2 looped transformer 不是新概念，但世界模型是空白

把同一组 transformer block 反复应用到同一个 latent 上，可以追溯到 Universal Transformer（Dehghani 2019）——权重跨深度共享，配一个源自 Adaptive Computation Time（Graves 2016）的自适应停机机制。早期理论工作证明 looped 模型能以常数参数量模拟梯度下降、牛顿法、动态规划（Giannou 2023），还能用不到 10% 的参数达到与标准 transformer 可比的 in-context learning 性能（Yang 2023）。

近两年这条线在语言模型上快速推进：Ouro（Zhu 2025）做到约 2-3× 参数效率并引入熵正则早退；Geiping 2025 的 recurrent-depth 模型靠增加推理时循环次数扩展测试时算力；Parcae（Prairie 2026）把 looped 前向重建成残差流上的非线性动力系统，用负对角参数化约束状态转移矩阵的谱范数，专治 looped 模型的训练不稳定。

论文对这条线的总结是一句话：以上工作全部只在语言建模里开发和评估，looped world model 完全无人做过。LoopWM 补的就是这个空白。

三、方法：把深度换成迭代次数

3.1 为什么 looped 结构对世界模型格外合适

论文的关键观察：环境动力学本身近似一个迭代过程——状态 $s_t$ 通过（近似）平稳的物理规律演化到 $s_{t+1}$。而 looped transformer 的计算图恰好同构：

$$h_{t+1} = \bar{A} h_t + \bar{B} e + \bar{\mathcal{R}}(h_t, e)$$

$\bar{A}$ 管状态保留，$\bar{B}$ 管输入注入，$\bar{\mathcal{R}}$ 是 transformer 非线性。同一个共享 block 反复迭代，结构上对应物理规律的反复应用。

需要注意的是，论文自己把这个对应关系说得很克制：“conceptual rather than exact”——内循环并不直接代表物理时间，只是在迭代精化一个 latent 转移估计。

直觉上：模拟一个 5 步的物理过程，可以用 5 层 transformer 一次过，也可以用 1 层跑 5 次。后者参数只有 1/5，代价是单步容量变小——所以每个 block 要足够通用，能处理不同复杂度的转移。

3.2 四模块架构

动力学核心内部是三段式：Prelude（$L_\mathcal{P}$ 层，非共享，把上一 latent、观测嵌入、动作嵌入拼成条件信号）→ Recurrent Block（$L_\mathcal{R}$ 层，参数共享，迭代 $T$ 次，$h^{(0)} \sim \mathcal{N}(0, \sigma^2 I)$ 初始化）→ Coda（$L_\mathcal{C}$ 层，非共享，输出终态）。论文没有披露这些层数的具体取值。

整个 forward 是嵌套双循环：

• 外层：环境步 $k = 0, \ldots, K-1$，每步注入一个动作；
• 内层：latent refinement $t = 0, \ldots, T-1$，共享 block 反复精化 $h$。

有效深度是 $K \times T$ 次共享参数前向，解码器只跑一次（延迟解码时）。另外论文提到，如果做 Dreamer 式的 imagination 训练，编码器整个被旁路：$h_{k+1} = \mathcal{L}_\theta(h_k, \mathbf{0}, u_k)$，只靠动作自回归滚动。

3.3 让迭代不崩的三道保险

第一道：谱约束状态保留。 反复迭代同一组参数，数值误差会累积放大。LoopWM 沿用 Parcae 的做法，把状态保留项用连续时间矩阵参数化：

$$A = \mathrm{diag}(-\exp(\mathbf{a})), \quad \bar{A} = \exp(\Delta \cdot A)$$

$\mathbf{a}$ 和 $\Delta$ 均可学习，离散化用零阶保持。由于 $A$ 的对角严格为负，$\exp(\cdot)$ 把每个元素映到 $(0,1)$，$\bar{A}$ 的谱半径严格小于 1——收缩映射，无论内循环迭代多少次，hidden state 都有界。论文强调这个约束是构造性成立的：全程不需要梯度裁剪、事后归一化或敏感的超参调优。

第二道：Poisson 变深训练 + 截断 BPTT。 训练时内循环次数不固定，从 Poisson 分布采样：

$$T \sim \mathrm{Poisson}(\mu_{\mathrm{rec}})$$

$\mu_{\mathrm{rec}}$ 是可学习标量。一个容易忽略的实现细节：$T$ 在每个 micro-batch 内逐 sequence 独立采样，而不是像 Geiping 2025 那样按整个 micro-batch 采样——论文明确说这个改动降低了训练目标方差，“经验上消除了大部分 loss spike”，而 loss spike 正是 looped 模型训练最痛的问题。反向传播在循环方向上截断到 $\mu_{\mathrm{bwd}} = \lceil \mu_{\mathrm{rec}} / 2 \rceil$ 步以控制显存。

第三道：熵正则。 推理启用早退时（见 3.4），训练损失加一项 $-\alpha,\mathbb{E}[\sum_t H(g^{(t)})]$，其中 $H$ 是 exit 概率的二元熵，$\alpha$ 是正则系数。作用是防止 exit gate 塌缩到平凡解——要么第一次迭代就退出，要么永不退出。

世界模型损失本身是三项加权和：观测重建 + 奖励预测 + 终止预测，权重 $\lambda_r$、$\lambda_c$ 平衡。

3.4 推理：exit gate 与测试时算力扩展

推理时不需要每个转移都跑同样的迭代次数。LoopWM 用一个轻量 exit gate——单层 MLP 加 sigmoid：

$$g^{(t)} = \sigma(\mathbf{w}_g^\top h^{(t)} + b_g)$$

每次迭代算一次 gate 值，超过阈值 $\tau$ 就停，用当前的 $h^{(t)}$ 作为终态。阈值 $\tau$ 论文没有给具体数值，属于需要经验调的量。

收益的量级论文给了一笔账：对比一个 100 层的固定深度基线，简单的自由飞行段单次循环（比如 4 层）就收敛，该步推理 FLOPs 降约 25×；一条长 rollout 里简单转移占多数的话，聚合 FLOPs 降幅可达两个数量级。注意这是一笔分析账（论文标注的是 “Consider a 100-layer fixed-depth baseline” 的思想实验），论文没有放早退机制在 ScienceWorld 上的实测对比表。

另外一个设计：推理时最大循环次数 $T_{\max}$ 可以超过训练均值 $\mu_{\mathrm{rec}}$——多给迭代次数，预测继续被精化。这就是"测试时算力扩展"，与语言模型里 test-time compute scaling 的思路同源。

3.5 延迟解码：只在终点解一次

最后一个设计：把解码从每个 transition 推迟到 terminal step。

常规做法每个环境步都解码出预测——如果只关心长时序规划（plan K 步看终态），中间每步解码都是白花的开销。LoopWM 的替代方案是 K 步无解码 latent rollout，最后只解一次。训练目标相应调整为两部分：

1. 终端预测损失：解码 $h_K$，对齐最终观测、奖励、终止标志；
2. latent trajectory regularizer：没有中间监督时，中间 latent 可能漂进"谱稳定但语义无意义"的区域，所以用冻结编码器对中间真实观测编码（stop-gradient），做 latent 一致性约束。

直接训大 K 不稳定（梯度要穿过 $K \times T$ 次共享参数前向），论文用课程学习：K 从 1 开始（等价于逐步解码），按 $\min(K_{\max}, 1 + \lfloor \mathrm{step}/\Delta \rfloor)$ 的日程渐进拉长。

延迟解码还带来两种推理模式：

• Planning mode：给定候选动作序列，做无解码 rollout，只评估终态 $h_K$。解码器调用从 K 次降到 1 次，配合内循环早退，长时序规划的总 FLOPs 可降两个数量级；
• Monitoring mode：安全攸关场景需要中途看状态时，用一个轻量投影头 $g_\omega$ 在任意步产出低维状态摘要，不必调用完整解码器。

论文把这个设计概括为职责分离：“latent dynamics reasoning（内外双循环）+ observation grounding（单次 terminal decode）"。对比表里它同时对照了 Dreamer（每步都跑奖励和价值头）、MuZero（固定深度动力学函数）和 ETD（looped 但不支持动作条件转移）。

四、实验一：ScienceWorld

4.1 评测协议

ScienceWorld（Wang 2022）是基于文本的科学实验环境，agent 用自然语言操控烧杯、加热、做化学反应、读电路。论文从中取了 14 个子任务：Boil、Chemistry、Conductivity、Find、Freeze、Genetics、Grow、Incline、LifeStages、Lifespan、Melt、Power、StateChange、Thermometer。

协议：给模型连续 5 个动作，预测最终状态。四个指标——EM（Exact Match）、Token F1、BLEU-4、Entity。这是世界建模任务，不是 decision-making 任务：模型不选动作，只预测状态。

闭源 LLM 怎么当世界模型基线？因为任务被表述为文本预测，LLM 可以直接 prompt 出答案。claude-opus-4-6-max（Anthropic，2026）是主基线。

4.2 主结果

论文 Table 2（LoopWM 与 claude-opus-4-6-max，14 个子任务）：

EM 平均差 21.2 个百分点。逐任务看，EM 上 LoopWM 11 胜 3 平 0 负——Chemistry、LifeStages、Thermometer 三个子任务与 Claude 持平，其余全部领先。

最有戏剧性的是 Lifespan：Claude 的 EM 是 0%（Token F1 倒有 61.4%），LoopWM 四个指标全部 100%。论文原文的说法是 “In the most extreme cases, it improves the scores on Lifespan from 0% to 100%"。

参数量的对比论文原话是：约 1B 的 LoopWM 比 claude-opus-4-6-max 这类闭源 API 模型小 100 倍以上。摘要里"最高 100× 参数效率"的口径则是相对"传统方法”（conventional approaches）说的——两个口径不完全等同，引用时需要留意。

4.3 另外两个基线

Table 3 补了两个更小的基线，结论一致：

论文的解释是这两个模型"比其他基线更小”，分数低合乎预期。值得一提的是 Entity 指标：qwen-3.5-flash 平均 63.0%、gemini 73.8%，并没有崩——小模型在实体抽取上的表现远好于它们的 EM。

4.4 这些数字说明什么，不能说明什么

这个 benchmark 段需要回答三个问题：

1. 测的是什么：固定 5 步动作序列下的文本状态预测。它考察的是"环境建模的忠实度"，不是策略性能——没有任何 RL 交互、探索或规划在跑。
2. 数字反映系统的哪部分：+21.2 pp 主要来自 looped 架构在多步科学实验这类结构化转移上的拟合能力，延迟解码的贡献另行分解（见下）。Lifespan 那种单任务 0→100 的跃迁，更可能是 Claude 在这类不常见任务格式上的特定失效，而非 LoopWM 的普遍优势——论文自己也只用"most extreme cases"来定位它。
3. 不能推出什么：LoopWM 没有和 IRIS、DIAMOND、EMERALD、DreamerV3 中的任何一个直接对比，也没有在任何标准 RL 环境上跑分。“1B 打败闭源旗舰"只在"文本世界建模 + 这三个 LLM 基线"的范围里成立，不能外推为"1B 世界模型全面超越大模型”。

延迟解码的单独分析在论文 4.3 节：Table 5 给全部任务的平均相对提升（相对 gemini-3-flash-preview-thinking，公式 $(\text{Ours}-\text{Baseline})/\text{Baseline} \times 100%$），EM 依次为 Step1 +73.2%、Step2 +54.5%、Step3 +103.6%、Step4 +82.9%、Step5 +113.8%；Table 6 起逐子任务展开（vs gemini 覆盖 Boil 到 Power 共 10 个子任务，另有一套 vs qwen 的表）。单任务单步上确实常见三位数相对提升——Boil 的 Step4 EM 相对提升 +700.9%——但根因是基线在这些点上的绝对分数极低，相对百分比被放大。读这批表时盯绝对分比盯相对百分比安全。

五、实验二：AlfWorld

5.1 四模型完整排名

AlfWorld（Côté 2018）是基于 TextWorld 的家庭任务环境，子任务按动作类型分 clean、cool、heat、look、pick 五类，协议与 ScienceWorld 相同（连 5 动作预测终态）。

论文 Table 4 的 Overall 排名（四个模型）：

LoopWM 拿下 BLEU 第一（生成文本最贴近 ground truth），不过领先 Gemini 只有 0.6 个百分点；EM 第二，距 Claude 1.4 个百分点；Token F1 也是第二，反超 Claude 7.8 个百分点，但输给 Gemini。论文原文的概括是 “it gives the best result on the BLEU metrics among four models, and ranks in second place on EM and Token F1”。

5.2 论文自己的误差分析

AlfWorld 上 LoopWM 没有像 ScienceWorld 那样领先，论文给的解释不是架构瓶颈，而是一条具体的误差分析：逐动作类别检查后发现模型的 Entity 分数偏低，且这个模式在大多数动作类别上都成立，“future optimization can focus on the entity scores to further enhance the model”。顺带一提，Entity 指标上两个小基线（Qwen、Gemini）反而占据前两名，四个模型的强弱格局在这个指标上整个倒转。

论文没有把 AlfWorld 的相对弱势归因到长程依赖或迭代次数上限——解读时不要替论文补这个论证。

六、局限与未解问题

对比基线全是 LLM。 这是全文最大的实验留白。论文标题意义上的"世界模型"社区（Dreamer 系、IRIS 系、扩散系）一个都没进场，Atari、DMControl、Crafter 上没有数字。LoopWM 与这些家族的定位差异，论文在 Broader Impacts 里承认"更明确的定位分析会让贡献更容易解读"，留给未来。

关键设计没有消融。 谱约束、Poisson 逐 sequence 采样、熵正则、exit gate——论文用分析和引言里的动机支撑它们，但没有"去掉某一件性能掉多少"的消融表。唯一做了系统性实验分析的是延迟解码（Table 5-44 那一大批）。“四件套缺一不可"是论文的设计主张，实证支撑目前只覆盖其中一件。

视觉环境只有可行性声明。 Broader Impacts 的原话值得整段读：

“We have also verified in continuous visual environments that optimization is feasible and that the training loss is consistently reducible, which supports the practicality of the proposed architecture beyond the environments highlighted in this paper. The main limitation at this stage is therefore not a lack of empirical support, but that the manuscript does not yet fully expose the breadth of validation already completed.”

也就是说，作者声称连续视觉环境上的验证已经做了一部分，只是这篇稿件"有意选择性地披露”（intentionally selective in disclosure scope）。SOTA 数字要等后续工作，而且这篇论文的披露策略本身就是解读时需要留意的信号。

scaling law 没做完。 结论一节的原话：“the present paper stops short of providing a more complete scaling law characterization across broader task and compute ranges”。Step 1 到 Step 5 的实验已经显示迭代深度像一条 scaling 维度那样起作用，但跨任务、跨算力档位的完整曲线还没有。论文还顺带说了一句工程经验：训练受益于课程式的策略，逐步解锁架构能力——他们把这看作"让新架构在大规模下可靠可训的实操配方"，而不是方法缺陷。

没有代码。 截至本文写作（2026-07），论文与 arXiv 页面均未给出开源仓库，复现只能靠自己按公式实现。

七、实践判断：什么时候值得试

值得优先尝试的场景：

1. 文本/离散状态的环境模拟：ScienceWorld 式的多步结构化转移是 LoopWM 目前唯一被验证过的主场；
2. 参数预算紧张的长时序 rollout：长 rollout 要把动力学模型跑成百上千次，参数量小的模型在每一步都省，收益是复利的；
3. 算力异质的状态分布：简单转移占多数、偶尔来一个复杂事件的场景，早退机制的按需分配正好用上。

先不要用的场景：

1. 需要跟现有 WM 正面比较的场合：没有任何与 Dreamer 系/transformer 系/扩散系的对比数据，选型时拿不出横向证据；
2. 长程文本一致性优先的任务：AlfWorld 的 EM/Entity 表现说明 1B 模型在实体级精度上有明显短板（论文自己的误差分析指向这里）；
3. 超短 horizon：迭代精化还没展开就到终点了，浅层 transformer 一次过更划算；
4. 零训练数据的场景：共享算子的学习需要充分的任务多样性，纯 zero-shot 不适用。

如果你想自己复现，论文给出的可依赖要素按重要性排：谱约束参数化（没有它，反复迭代会让数值误差无界累积，hidden state 失去保证——这是 Parcae 和本文共同的动机）；Poisson 变深训练且必须逐 sequence 采样（论文明确对比了按 micro-batch 采样的高方差）；截断 BPTT 的步长取 $\lceil \mu_{\mathrm{rec}}/2 \rceil$。exit gate 的阈值 $\tau$ 和熵正则系数 $\alpha$ 论文都没给数值，属于自己调的部分。另外别忽略延迟解码的训练配方——终端损失加 latent 一致性正则，K 用课程从 1 涨上去。

八、常见问题

Q1：LoopWM 能直接用于在线 RL 训练吗？

架构上预留了这条路：论文 3.1 明确写了 Dreamer 式 imagination 训练的用法（编码器旁路，纯动作自回归滚动 latent）。但论文的所有实验都是离线世界建模评估，没有跑过任何 RL 交互。要在线用，参考 DreamerV3 的 Actor-Critic 框架把 LoopWM 当 dynamics model 接进去，效果未知。

Q2：1B 打百倍大的闭源模型，是不是过拟合？

没有证据支持过拟合的判断，但要小心另一个方向的误读。14 个子任务 11 胜 3 平、换一个环境（AlfWorld）就只是第二梯队——这个模式更像"架构与任务结构的匹配"，而不是全面碾压。论文也没有做 train/test 划分的细节披露来直接回应这个问题。

Q3：代码开源了吗？

没有。论文和 arXiv 页面都没有仓库链接（截至 2026-07）。想自己实现，关键点是谱约束参数化、逐 sequence 的 Poisson 采样、截断 BPTT，这三处缺了任何一处都容易复现失败——按论文的叙述，它们分别对应稳定性、方差和显存三个坑。

Q4：能处理图像/视频吗？

架构上没有障碍：观测编码器本来就写成"卷积或 ViT"。论文在 Broader Impacts 里说连续视觉环境上的优化可行性已验证，但没放数字。真正的门槛在解码端——图像/视频的观测重建比文本重得多，延迟解码恰好把这部分开销推迟并削减了，这可能是这个架构在视觉场景的真正卖点，但目前只是推测。

Q5：与 Sora / Genie 这类生成式世界模型相比呢？

目标函数不同。Sora/Genie 是生成式的，主打高质量画面，评估看视觉保真度；LoopWM 是预测式的，主打状态预测精度（EM/F1/BLEU），给 agent 规划用。论文在结论里把 LoopWM 与 RSSM 系、自回归视频 token 系、扩散系并列为一族中"一个 distinct point"，明确说定位分析留待未来。

Q6：训练要什么硬件？

论文通篇没有披露训练硬件、时长和数据规模，任何具体配置建议都是猜。能确定的只有参数量级（约 1B）和任务形态（文本环境），这两点意味着它不属于"只有大厂能训"的那类工作。

九、关键判断

1. 迭代深度立住了一条新 scaling 轴——与模型规模、数据规模正交，这在世界模型上是第一次被系统验证；Step 1-5 的相对提升曲线支持迭代深度作为一条 scaling 维度成立，完整 scaling law 还没做。
2. “1B 打败闭源旗舰"要读全限定语——成立的范围是文本世界建模、14 个 ScienceWorld 子任务、EM 口径、对比对象是三个 LLM。换环境、换指标（Entity）、换基线家族，优势都会缩水甚至反转。
3. 稳定性设计是方法的核心资产——谱约束构造性保证 $\rho(\bar{A}) < 1$，配合逐 sequence Poisson 采样消 loss spike、截断 BPTT 控显存，这套组合拳让 looped 架构在世界模型上可训。没有消融，但动机链条完整。
4. 延迟解码是实验上被验证最充分的一件——Table 5-44 的全部分解实验都在讲它；Planning mode 下解码次数从 K 降到 1，与早退叠加可降两个数量级 FLOPs。
5. 下一篇该看什么——视觉环境数字（作者称已有部分验证）、与 Dreamer 系的正面对比、跨算力的 scaling 曲线、开源代码。这四件事任何一件落地，这条线都值得重估。

十、参考链接

• 论文：Looped World Models（arXiv 2606.18208），2026-06-16 提交，CC BY 4.0
• 机构：FaceMind Research Asia；Leading Contributors：Hongyuan Adam Lu、Z.L. Victor Wei
• 代码/数据：论文未给出开源仓库

引用文献（论文 Related Work 选录）

• Looped/自适应计算：Universal Transformer（Dehghani 2019）、Adaptive Computation Time（Graves 2016）、Parcae: Scaling Laws For Stable Looped Language Models（Prairie 2026）、Ouro（Zhu 2025）、LoopFormer（Jeddi 2026）、Hyperloop Transformers（Zeitoun 2026）
• 世界模型基础：PlaNet（Hafner 2019）、Dreamer 系列至 DreamerV3（Hafner 2025）
• Transformer 系 WM：IRIS（Micheli 2023）、TransDreamer（Chen 2022）、Δ-IRIS（Micheli 2024）、DIAMOND（Alonso 2024）、EMERALD（Burchi & Timofte 2025）
• 生成式 WM：Sora（OpenAI 2024）、Genie（Bruce 2024；Google DeepMind 2025）
• 评测基准：ScienceWorld（Wang 2022）、AlfWorld（Côté 2018）
• 误差累积：Talvitie 2017、Xiao 2020、Luo 2022

附录：核心流程示意

按论文 3.2 节的三段式结构整理（符号与论文一致；各超参数值论文未披露，此处只示结构）：

# 单个环境步的 forward（论文式 (3)）
e = encoder(o_k)                      # 卷积/ViT
u = action_embedder(a_k)
cond = prelude([h_prev, e, u])        # L_P 层，非共享
h = normal_init(sigma)                # 序列起始 h(0) ~ N(0, sigma^2 I)；时序 rollout 承接上一步终态
for t in range(T):                    # T ~ Poisson(mu_rec)，逐 sequence 采样
    if inference and exit_gate(h) > tau:   # 单层 MLP + sigmoid
        break
    h = exp(delta * A_bar) @ h + B_bar @ cond + recurrent_block(h, cond)
    # A = diag(-exp(a))，谱半径 rho(A_bar) < 1 构造性成立
obs_hat, r_hat, c_hat = coda_and_heads(h)   # L_C 层 + 预测头

# 延迟解码：K 步 rollout 只在最后解一次
for k in range(K):                    # 训练时 K 按课程从 1 渐增
    h = env_step(h, a[k])             # 即上面的单步 forward，不解码
prediction = decoder(h)               # 唯一一次 terminal decode

反向传播在循环方向截断到 $\lceil \mu_{\mathrm{rec}}/2 \rceil$ 步；启用早退时训练损失加熵正则项防止 gate 塌缩。