AnchorWorld:基于视图演化定制的具身第一人称世界模拟 AnchorWorld: Embodied Egocentric World Simulation with View-based Evolution Customization
通过混合视图动作控制和锚点视图定制实现可交互、可演化的第一人称世界模拟
前置知识
具身AI(Embodied AI)
具身AI是指智能体通过身体与物理环境进行交互的学习范式,强调感知、行动与环境反馈的闭环。与传统纯视觉任务不同,具身AI需要智能体在3D空间中主动探索,其行为会影响环境状态,形成相互依赖的动态系统。典型应用包括机器人导航、虚拟现实中的用户行为模拟等。
本文的核心场景是第一人称视角下的交互式世界模拟,需要理解人体动作如何影响视觉观察,这正是具身AI的核心问题。读懂本文需要理解为什么传统非具身方法无法满足第一人称交互的需求。
SMPL-X人体模型
SMPL-X是一个参数化3D人体模型,用theta参数表示人体姿态,用beta参数表示体型。模型包含22个主要关节,每个关节状态由3D位置和3D轴角旋转向量构成,可表示为M属于实数的f乘k乘6维度,其中f是帧数,k是关节数。该模型能生成完整的3D人体网格,广泛应用于姿态估计和动作捕捉。
本文使用SMPL-X模型表示人体动作条件,理解它的参数化表示对于理解方法如何编码和注入动作信号至关重要。论文中提到的22个主要身体关节排除手部姿态,正是基于当前第一人称数据中手部估计不可靠的考虑。
流匹配(Flow Matching)
流匹配是一种新型的生成模型训练范式,通过学习从噪声分布到目标分布的向量场来生成样本。给定噪声样本x0和目标样本x1,流匹配最小化期望vt与条件向量场ut的平方误差,其中xt是x0和x1的线性插值。相比扩散模型,流匹配训练更稳定且采样更快。
本文采用Wan2.2 TI2V 5B作为基础模型,这是一个基于流匹配的DiT视频生成模型。理解流匹配有助于理解为什么选择这个基础模型,以及如何在其上添加条件控制。
旋转位置编码(3D RoPE)
旋转位置编码是一种基于复数旋转的位置表示方法,最初用于NLP领域。3D RoPE将其扩展到3D空间,通过在不同轴向上应用旋转来编码3D位置信息。给定查询q和键k,RoPE通过将它们乘以与位置p相关的角度的指数形式来计算旋转后的表示。这种方法能自然地处理相对位置关系。
本文使用3D RoPE来区分不同的锚点视图,在位置嵌入空间中为它们分配不同的帧轴位置。理解3D RoPE有助于理解模型如何区分来自不同3D位置的锚点视图,避免它们相互混淆。
研究动机
现有交互式世界模型在实用性场景所需的多样化控制能力方面存在两个根本性局限。首先,许多方法依赖键盘输入、相机轨迹或文本提示等简化控制信号,虽然方便导航但不能反映人类从第一人称视角如何自然行动。例如,在虚拟现实和具身AI应用中,用户的头部运动决定相机朝向,身体运动驱动导航,协调动作塑造与附近物体的交互方式,这些细粒度的交互行为是键盘或文本无法精确表达的。其次,对于世界本身的定义,现有方法通常通过初始帧、全局提示或历史上下文隐式确定环境,因此新观察到的区域约束较弱。这使得难以指定特定3D位置应该存在什么内容,或者局部场景状态应该如何随时间演化。例如,用户想要在房间角落放置一把椅子并让它在5分钟后自动移动到窗边,现有方法无法实现这种精确的局部状态控制。
本文的目标是本文提出AnchorWorld框架,旨在构建一个可定制世界的具身第一人称视角世界模拟系统,实现两种互补的控制形式:通过人体3D运动实现第一人称导航和交互的精确控制,以及通过姿态关联的锚点视图实现明确的局部世界状态定制和演化。具体而言,系统需要能够:理解用户的全身运动包括根轨迹、肢体动作和头部运动如何影响第一人称视觉观察,支持用户在选定的3D位置指定局部状态,在视角变化时保持这些局部状态,根据文本描述引导局部场景的动态演化,即使在初始视野之外的区域也能正确推理和演化场景状态。
与已有工作不同的是,本文的独特切入角度是同时解决动作控制和世界定制两个问题,而不是像现有工作那样只关注其中一个方面。PlayerOne关注全身运动控制但缺乏场景信息,CaM关注视角一致性但没有动作输入,而本文将两者统一在同一个框架中。另一个关键创新是引入第三人称视角视频作为辅助训练数据,利用它们提供的完整人体运动和交互上下文来弥补第一人称视频中身体部分通常位于视野之外导致监督稀疏的问题。这种混合视图训练策略让模型能够从外部视角学习到完整的人体与场景交互先验,然后在第一人称视角中应用这些知识。
核心方法
AnchorWorld的整体思路是构建一个统一的条件视频生成框架,能够同时响应人体动作输入和世界状态定制指令。从直觉上理解,这就像导演一部第一人称视角的电影,导演既可以通过告诉演员如何移动来控制镜头视角,也可以在场景的不同位置布置道具并规定它们如何变化。技术路线上,本文基于Wan2.2 TI2V 5B流匹配DiT模型,在其基础上添加了两个关键的条件注入模块:空间姿态注意力机制用于处理人体动作条件,以及锚点视图上下文注入机制用于处理世界定制信号。模型采用渐进式多阶段训练策略,依次学习第三人称动作控制、第一人称动作控制、静态场景一致性和动态场景演化,每个阶段都建立在前一个阶段的稳定基础上。
核心创新点包括三个方面:混合视图人体动作控制通过投影方式表示动作条件,将全身运动序列与相机轨迹结合,使模型能够在任意视点下将3D人体运动投影到2D视觉观察。这种设计允许先在第三人称视频上预训练获得投影知识和人场景交互先验,然后在第一人称视频中适配。姿态关联的锚点视图定制中每个锚点视图包含RGB图像指定局部视觉外观、6自由度相机姿态提供空间定位、以及演化提示描述动态变化,三者共同提供完整的局部世界先验。渐进式多阶段训练策略中Stage I在200K内部真实单人动作视频和101K UE合成视频上学习第三人称动作控制,Stage II在100K第一人称样本上学习第一人称动作控制,Stage III在25K大视角变化样本上学习静态锚点视图一致性,Stage IV在25K静态样本加10K动态样本上学习文本驱动的局部状态演化。
方法步骤详情
方法步骤的完整描述如下:首先,输入处理阶段接收人体运动序列和相机姿态序列,运动编码器将运动投影到潜在嵌入,相机编码器将相机姿态编码,同时应用时间下采样以匹配VAE编码视频潜在的时间分辨率。其次,空间姿态注意力步骤将视频token与人体运动token和相机姿态token沿空间维度拼接,通过空间自注意力块处理,然后使用截断操作丢弃辅助姿态token,只保留更新后的视频特征。第三,锚点视图上下文注入将锚点视图图像编码为潜在token,与视频潜在token沿帧维度拼接,使用3D RoPE为不同锚点视图分配不同的帧轴位置以区分它们。第四,视图姿态注入将相机姿态编码为嵌入,空间广播到匹配潜在分辨率,在自注意力层之前加到视觉token上。最后,文本驱动的锚点视图演化通过交叉注意力注入演化描述,为保持局部性使用注意力掩码,对于文本提示,其键只对生成的视频token和对应的锚点视图token可见,其他位置设为负无穷。
技术新颖性
技术新颖性主要体现在四个方面:投影式动作控制表示与PlayerOne的部分解耦运动注入不同,本文通过结合全身运动和相机轨迹实现投影式控制,让模型能够在任意视点下理解3D运动到2D观察的映射,这是第一次将这种投影机制应用于具身第一人称视频生成。异构视图联合训练通过统一的投影式动作条件框架支持第三人称和第一人称视频的联合训练,利用第三人称视频的完整运动监督来补偿第一人称视频的稀疏监督,这种混合视图训练策略在之前的具身视频生成工作中未被探索。锚点视图的三元组表示与现有方法只用图像或只用姿态不同,本文将RGB外观、3D姿态和演化提示组合成统一的锚点视图表示,这是第一个支持可演化局部世界定制的具身世界模型。掩码交叉注意力实现局部控制通过限制文本提示只影响特定视频token和对应锚点视图token,实现了锚点特定的文本控制,这种精细化的注意力掩码设计使得多个锚点视图可以独立演化而不相互干扰。
实验结果
实验结果表明AnchorWorld在场景一致性、相机准确性和文本对齐方面均显著优于基线方法。在第一人称静态场景测试中,本文方法的匹配像素数量达到4493.4K,显著高于PlayerOne的3961.6K和CaM-Ego的4379.4K;CLIP-V语义相似度为0.885,高于PlayerOne的0.845和CaM-Ego的0.872;PSNR为16.06,SSIM为0.578,均优于所有基线。相机轨迹误差方面,ATE为0.112,优于PlayerOne的0.131和CaM-Ego的0.125;RTE为0.029,优于PlayerOne的0.037和CaM-Ego的0.032;RRE为3.145,优于PlayerOne的3.741和CaM-Ego的3.207。在UE CineScene测试中,本文方法在CLIP-V(0.851 vs PlayerOne的0.787)和RRE(1.656 vs PlayerOne的2.438)上表现突出,证明在虚幻引擎合成场景中的强泛化能力。最令人印象深刻的是在第一人称动态场景中,本文方法的文本对齐得分达到0.717,远高于PlayerOne-Scene的0.449和CaM-Ego的0.385,这表明演化提示控制的有效性。VBench视频质量指标显示本文方法在主体一致性(0.8167)、背景一致性(0.9041)、运动平滑度(0.9832)等方面均达到或超过基线水平。消融研究表明,移除Stage I第三人称视频训练会显著恶化相机控制精度(ATE从0.112增加到0.125),移除锚点视图姿态会降低场景一致性(匹配像素从4493.4K下降到4401.7K),验证了每个关键设计的必要性。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 第一人称静态场景-场景一致性 | 匹配像素(K) | 4493.4 | PlayerOne: 3961.6, CaM-Ego: 4379.4 | 相比PlayerOne提升13.4%,相比CaM-Ego提升2.6% |
| 第一人称静态场景-语义相似度 | CLIP-V | 0.885 | PlayerOne: 0.845, CaM-Ego: 0.872 | 相比PlayerOne提升4.7%,相比CaM-Ego提升1.5% |
| 第一人称静态场景-相机精度 | ATE | 0.112 | PlayerOne: 0.131, CaM-Ego: 0.125 | 相比PlayerOne降低14.5%,相比CaM-Ego降低10.4% |
| 第一人称动态场景-文本对齐 | VideoAlign-TA | 0.717 | PlayerOne-Scene: 0.449, CaM-Ego: 0.385 | 相比PlayerOne-Scene提升59.7%,相比CaM-Ego提升86.2% |
| UE CineScene-场景一致性 | CLIP-V | 0.851 | PlayerOne: 0.787 | 提升8.1% |
| UE CineScene-相机精度 | RRE | 1.656 | PlayerOne: 2.438 | 降低32.1% |
局限与改进
作者承认的局限性包括三个方面:长期探索挑战,本文主要关注短视频片段场景,但实现更长时间范围的世界探索和交互对于未来进步至关重要。具身智能体在第一人称设置中可能持续与环境交互并探索更大规模的场景,在此过程中模型必须实时更新由自身动作引起的环境状态变化。开放世界泛化,本文的训练数据主要聚焦于受限的场景集合,未来收集开放世界数据以构建更广泛的环境并支持更长时间范围的世界探索将是重要方向。多样化动态场景,由于当前第一人称训练数据的限制,这些数据通常为同一动态人类活动提供多个视角,本文的实证实现对所有锚点视图先验使用全局一致的演化描述,主要关注与人相关的活动而非多样化的动态场景。从自身观察来看,还存在以下局限:细粒度纹理细节不一致,当场景的局部区域包含复杂结构和丰富纹理细节时,本文方法可能产生不一致的细粒度细节结果。这在很大程度上受到基础模型能力的限制,具体来说,实验中使用的Wan TI2V 2.2 5B的VAE空间下采样因子为16,导致潜在空间维度的高压缩比从而丢失细细节信息。模糊生成伪影,第一人称训练数据包含大量具有快速视角变化的视频,这经常导致帧中出现运动模糊,因此生成的结果也可能表现出类似的模糊伪影。此外,由于基础模型的限制和第一人称数据中常见的快速运动,生成的手部可能遭受视觉质量下降。
独立分析的弱点
从独立分析的角度,可以识别出以下具体弱点和改进方向:手部姿态缺失问题,由于当前第一人称数据中手部估计不可靠,本文排除手部姿态,这限制了精细交互操作的模拟。改进方向可以是开发更鲁棒的第一人称手部姿态估计方法,或者设计能够从不完整姿态估计中学习的模型架构,例如通过部分观测推断完整姿态。锚点视图中的玩家污染,训练数据中锚点视图可能包含第一人称玩家,虽然模型通过第一人称视频监督和姿态信息学会了忽略玩家,但这仍然是一种次优设置。改进方向可以是在训练时应用更精细的玩家掩码和修复技术,或者设计能够显式建模锚点视图应该独立于玩家这一约束的损失函数。全局演化描述的限制,当前实现对所有锚点视图使用相同的演化描述,这限制了场景中不同区域独立演化的能力。改进方向可以是设计锚点特定的演化描述机制,同时保持它们之间的空间一致性,例如通过演化描述之间的空间传播或约束。颜色不一致性,Ego-Exo4D数据集中第三人称和第一人称视频之间存在明显的颜色差异,虽然模型能够利用锚点视图的有价值场景信息同时保持与初始第一人称帧一致的色调,但这仍然是一个挑战。改进方向可以是在训练时引入颜色一致性损失,或者使用风格迁移技术来对齐不同视角的色调。
未来方向
作者提出的未来工作方向包括:实时自回归交互,扩展框架以支持实时自回归交互,这对于第一人称设置中的长时间范围探索至关重要。具身智能体持续与环境交互并探索更大规模场景时,模型必须实时更新由自身动作引起的环境状态变化,这需要在模型内更强地强调长期记忆机制。开放世界数据收集,收集开放世界数据以构建更广泛的环境并支持更长时间范围的世界探索。这将包括更多样化的场景类型、更复杂的环境结构和更丰富的人类环境交互模式。多样化动态场景建模,扩展框架以支持更多样化的场景和锚点特定的动态控制,同时结合世界的自然动态演化,从而构建更真实和时间丰富的世界。这将超越当前主要关注的人相关活动,涵盖环境自身的动态变化、天气变化、光照变化等。基于本文成果可以延伸的未来方向包括:多智能体交互,当前工作主要关注单个人类用户的交互,未来可以扩展到多个智能体在同一环境中的协调交互,这将需要建模智能体之间的相互影响和协作行为。物理一致的模拟,当前方法主要关注视觉一致性,未来可以引入物理引擎或物理约束来确保生成的世界不仅在视觉上合理,而且在物理上是一致的,例如物体碰撞、重力、摩擦等物理规律的遵循。跨模态世界定制,除了当前使用的文本演化提示,可以探索其他模态如语音、手势、草图作为世界定制的输入,使交互更加自然和多样化。
复现评估
从复现评估的角度分析:本文使用了多种数据集,包括内部策划的200K真实单人动作视频、101K来自MultiCamVideo的UE合成视频、100K第一人称样本来自Ego-Exo4D和LEMMA,以及25K静态样本和10K动态样本。这些数据集中,Ego-Exo4D和LEMMA是公开可用的,MultiCamVideo代码不可用,内部策划数据未公开。人体运动和锚点视图姿态估计使用GVHMR,这是一个公开工具,但需要一定的GPU资源运行。训练在16块NVIDIA 80G GPU上进行,总计算量约1406 GPU小时,这是一个相当大的计算需求,对于没有大规模计算资源的实验室来说复现成本很高。基础模型Wan2.2 TI2V 5B是公开可用的,但需要相应的许可证。本文没有明确提供代码或模型权重,考虑到训练的计算成本和数据集的可获得性,完整复现的难度较高。不过,如果作者发布代码和预训练模型,基于这些权重进行推理和小规模微调应该是可行的。评估指标如GIM-based Mat. Pix.、CLIP-V、PSNR、SSIM、LPIPS、ATE、RTE、RRE、VideoAlign-TA和VBench都是标准或公开可用的工具,便于复现实验结果。总体而言,完整复现训练过程难度较大,但复现推理和部分实验如使用预训练模型在不同测试集上评估是中等难度。
论文图表
这个图展示了两种类型的失败案例。左侧展示了当场景局部区域包含复杂结构和丰富纹理细节时,方法可能产生不一致的细粒度细节结果,可以看到墙壁表面的细节在不同帧之间不一致。右侧展示了由于第一人称训练数据包含快速视角变化导致的模糊帧,生成的结果也表现出类似的模糊伪影,特别是在手部区域。
这个图对理解方法的局限性非常重要,它诚实地展示了方法在特定情况下的失败模式。读者可以从图中看到两种不同的失败类型:细粒度纹理不一致和模糊生成伪影,这些都与基础模型能力和训练数据特性有关。这个图也间接指出了未来改进的方向,如使用更强大的基础模型或改进数据预处理。