精读笔记

Problem Setting

论文标题：SWAP: Symmetric Equivariant World-Model for Agile Robot Parkour（arXiv preprint / 2026）。这篇论文不是在解决一般意义上的四足 parkour，也不是单纯把 Dreamer/RSSM 搬到机器人上；它解决的是极限 parkour 中 latent representation 与控制策略的几何结构不一致问题。gap leaping 和 box climbing 的困难在于可行解空间极窄，成功依赖接触时序、双侧力分配和地形 affordance 的精确耦合。已有无约束 world model 会把左高右低和右高左低、左腿支撑和右腿支撑当成独立模式，从而增加样本需求，并在高难度任务中放大探索失败。关键矛盾是：任务物理上高度对称，但学习系统的函数类并不尊重这种对称；当动作空间接近动力学边界时，这种 mismatch 不只是低效，而会直接改变 policy 收敛到的行为类型。

Motivation

已有路线的问题在于 symmetry 通常只作为 data augmentation、loss penalty 或 policy-level trick 出现，而不是作为 world model 的 latent geometry。对于短历史 proprioceptive estimator，这种缺口不明显，因为模型只需要补状态；但对于 latent world model，它还要承担未来接触与几何变化的预测，latent coordinate 是否对齐物理对称性会影响 downstream policy 的可搜索性。作者真正抓住的是：world model 的冗余编码会把一个本来等价的镜像 MDP 切成两个经验分布，导致有限样本下的 prediction 和 value learning 都不稳定。缺的不是更大的网络，而是一个使左右镜像经验共享梯度、共享 latent dynamics、共享 policy structure 的硬归纳偏置。

Core Idea

核心思想是把四足 locomotion 建模为 Symmetric MDP，并将 Z2 左右反射约束硬编码进从 perception 到 latent dynamics 再到 actor-critic 的整条链路。镜像 observation 不允许映射到任意 latent，而必须映射到成对交换后的 latent；镜像 latent rollout 不允许漂移到另一套坐标，而必须保持等变；actor 在镜像状态下必须输出镜像动作，critic 对镜像状态必须给出相同 value。这样做改变的是建模方式：不是让网络从数据中“发现”左右对称，而是把左右对称变成函数空间本身的合法性条件。

这和 prior 的本质区别在于，它不是在输入层扩数据，也不是在 loss 上惩罚不对称，而是让不满足 symmetry 的函数根本不可表示。理论上，这对应 SMDP 中存在 equivariant optimal policy 和 invariant value 的结构；直觉上，它把左右两侧经验折叠到同一个 latent manifold 上，使 unilateral terrain 上学到的几何-接触模式可以无损迁移到 mirrored terrain。它更 scalable 的原因不是 compute 更多，而是有效 hypothesis space 更小，且 latent 表示与 policy 输出的对称坐标系一致。

Method

方法应理解为三个机制，而不是模块堆叠。第一，equivariant world model 解决的是 latent dynamics 的坐标任意性：RSSM 的 stochastic state 和 deterministic state 被组织成左右成对的表示，encoder、recurrent transition、prior/posterior 和 decoder 都保持反射等变，使视觉深度、本体状态和辅助地形重建共享同一个对称 latent space。核心变化是 world model 不再学习两套镜像动力学，而是在结构上复用它们。

第二，equivariant actor + invariant critic 解决的是 policy 可能破坏 latent symmetry 的问题。仅有对称 world model 并不能保证控制输出对称，尤其在多接触 box climbing 中，policy 很容易利用某个单侧偏置获得短期 reward。actor 的硬等变约束把镜像状态的动作选择绑定起来，critic 的不变性减少镜像状态 value mismatch，从而降低 asymmetric local optimum 的吸引力。

第三，heightmap reconstruction、AMP、curriculum 和低频 world model / 高频控制是让系统可训练、可部署的工程支架。heightmap auxiliary target 强迫 latent 捕获地形几何；AMP 约束运动风格；curriculum 把极限任务逐步推到可达边界。这些组件重要，但从论文的科学 claim 看，它们主要保证训练稳定和覆盖难度，不是最核心的新机制。

Key Insight / Why It Works

最重要的 insight 是：在极限 locomotion 中，symmetry 不是审美约束，而是 optimization prior。接近物理边界时，成功轨迹稀疏，policy 很容易被早期探索中的随机单侧成功带偏，形成不可扩展的 asymmetric strategy。SWAP 通过硬等变把这些单侧偶然性压回对称函数类，使左右接触、左右视觉几何、左右动作输出在训练梯度上绑定。它真正有效的部分大概率是 representation alignment + reduced hypothesis space，而不是 world model 本身的“想象能力”。

最可能的核心贡献是把 symmetry 放进 latent world model，而不只是 policy。mirrored terrain reconstruction error 的对比说明，无约束 RSSM 即便在训练方向上重建很好，也没有学到可镜像复用的 latent geometry；这会导致 OOD mirror 时 perception-to-state 失配。只加 equivariant world model 已经能显著改善 gap leaping，说明很多收益来自更好的 terrain-contact representation。

policy 端约束在 box climbing 中更关键，因为 box climbing 是多接触闭链式协调问题，value landscape 中存在大量“撞墙但不摔”“单腿扒边”等局部策略。full SWAP 优于 w/o Eq-Policy，说明 actor symmetry 在抑制这些局部解上有实质作用。SymLoss 改善有限也符合预期：软约束在高奖励梯度和接触非光滑性面前很容易被破坏，不能保证函数类级别的 equivariance。

需要保留怀疑：论文中的 record-breaking 真机结果不应全部归因于 symmetry。Apollo 机器人更大更重，训练使用 6000 env、GPU rendering、curriculum、AMP、terrain diversity 和大量工程 randomization；这些 scaling / engineering 可能贡献很大。户外 zero-shot 展示更像系统能力展示，缺乏对照，无法证明是 symmetry 带来的 robust perception under water/darkness/grass。所谓 world model 的 long-horizon proactive prediction 在文中也没有被充分隔离验证，可能更多是结构化状态估计与低频 terrain memory，而不是显式 planning。

Relation To Prior Work

最接近的谱系有三条：Dreamer/RSSM/DayDreamer/WMP 的 latent world model，legged parkour 的 model-free/distillation/curriculum 系统，以及 equivariant RL / symmetry-aware locomotion。SWAP 的新意不是发明 equivariance，也不是发明 RSSM，而是把 hard equivariance 贯穿到 RSSM latent dynamics 与 actor-critic 的接口处。相对于 WMP，它增加的是物理对称坐标系；相对于 symmetry policy work，它把 symmetry 前移到 perception 和 temporal prediction；相对于 data augmentation / SymLoss，它把对称性从训练偏好变成架构约束。

看似新的部分中，SMDP 推导、actor equivariance、critic invariance 都是已有思想的自然应用；真正新增的信息是：在视觉 parkour world model 中，latent dynamics 的对称结构会显著影响极限接触策略的可学性。它属于“结构化 world model for embodied control”的演化，而不是纯 RL 算法创新。论文最有价值的定位是证明：对于 embodied RL，world model 的 latent geometry 不能只追求 reconstruction/prediction loss，还要和系统的物理群结构及 policy action space 对齐。

Dataset / Evaluation

评估覆盖了仿真 ablation、mirrored terrain OOD、极限 gap/box、真机室内记录和户外 zero-shot 展示。最能支持核心 claim 的是 unilateral-to-mirrored terrain transfer，因为这个实验直接测试硬等变是否带来结构泛化，并且重建误差分析对 latent symmetry 给出了相对直接的证据。极限 parkour ablation 也有价值，尤其区分了 full SWAP、只去 policy equivariance、全去 equivariance、SymLoss，能说明 hard architecture 比 soft penalty 更有效。

但 evaluation 也有明显边界。mirrored OOD 只验证了一个预定义群上的泛化，不等价于一般几何泛化；训练 terrain 和测试 terrain 的生成机制可能仍高度重叠，只是方向镜像。真机和户外结果强，但不是严格 benchmark：没有同平台同训练预算的 baseline 真机对照，也没有失败统计、随机种子、环境覆盖量或部署选择标准。表格中的 record-breaking 绝对距离还受到机器人尺寸、质量、执行器功率和平台差异影响，归一化后优势没有绝对数值那么清晰。因此实验支持“symmetry 是有效结构先验”，但不足以单独证明“world-model reasoning 推动了物理边界”。

Limitation

方法成立依赖一个强前提：任务、机器人、观测、执行器和 reward 在左右反射下近似等价。一旦出现单侧载荷、腿部磨损、传感器偏心、地面摩擦左右不均、任务本身要求非对称动作，hard equivariance 可能从先验变成偏置错误。论文没有讨论如何在 symmetry broken 的情况下自适应放松约束。

泛化上限也受限。SWAP 对 mirror OOD 很强，但这是由手工指定 group transformation 直接覆盖的 OOD 类型；对 stepping stones、稀疏 foothold、绕障、动态障碍、非对称 affordance 的泛化没有系统验证。作者自己也承认未充分评估离散 foothold 场景，这很关键，因为这类任务常常需要打破双侧同步或选择单侧策略。

增益归因不完全清晰。训练包含 curriculum、AMP、辅助 heightmap reconstruction、大规模并行仿真、视觉 rendering 和工程 randomization；这些因素可能与 equivariance 强耦合。尤其真机 outdoor robustness 可能主要来自 data coverage、domain randomization 和平台控制裕度，而不是 symmetry。world model 的作用也可能更接近 latent state estimator / memory reuse，而非真正的 long-horizon planning；文中未充分说明 imagination horizon、policy 是否使用 rollout planning、以及去掉 world model 但保留对称 policy 的对照。

另一个隐含成本是 transformation table 的人工工程化。每个 observation/action/heightmap 分量都要手工定义符号翻转和排列，扩展到不同机器人形态、多传感器布局或更复杂群结构时会变得脆弱。它不是自动发现结构，而是把专家知识硬编码进网络。

Takeaway

• 第一，极限 embodied control 中，latent structure 比单纯模型容量更重要；world model 如果没有和物理 symmetry 对齐，会把可共享经验拆散，进而污染 policy optimization。
• 第二，hard equivariance 比 augmentation / symmetry loss 更适合高动态接触任务，因为接触任务的 reward landscape 非光滑、局部解多，软约束很容易被局部高回报行为突破。
• 第三，未来值得做的不是继续堆更大的 world model，而是研究可放松、可学习、可诊断的 symmetry：在严格对称时共享，在 symmetry broken 时允许偏离。
• 第四，这篇论文可迁移的 insight 是“把任务群结构贯穿 perception-latent-control 接口”，这比只在 policy 输入输出上做 equivariance 更有普适价值，尤其适用于双臂操作、类人运动、多足机器人和任何存在可交换子结构的 embodied system。

一句话总结

SWAP 是一篇把左右反射等变从 policy trick 提升为 world-model-to-control 全链路结构先验的四足 parkour 工作，真正贡献在于证明对称对齐的 latent dynamics 能显著改善极限接触任务中的表示效率、镜像泛化和策略优化。