【世界模型】World Observer:面向持久世界建模的联合主体-观察者生成器

图 1:概念示意图。World Observer 与actor 协同生成全景observer。(a) 解耦观察者可实现对主体不可见区域的控制。(b) 可灵活部署多个观察者以追踪被遮挡区域。

目录

标题:《WORLD OBSERVER: JOINT ACTOR-OBSERVER GENERATION FOR PERSISTENT WORLD MODELING》

来源:KAIST AI

链接:https://cvlab-kaist.github.io/world-observer

摘要

  本文提出了一种名为 World Observer(世界观察者) 的架构,旨在解决传统视频世界模型中"物体离开视野后失去状态记录并在重新进入视野时无法保持一致性"的问题。该方法通过将"观察"与"行动"解耦,联合生成智能体视角的参与者(Actor)视频流和一个或多个监控周边环境的全景观察者(Observer)视频流,从而在全局视角下持续维护并演化物体的状态。

一. 核心架构与联合建模

  World Observer 在 3D VAE 的潜在空间中,使用单个预训练的视频扩散 Transformer (DiT) 联合生成 Actor 和 Observer 两个视频流。

  • 输入与生成机制 :Actor 流接收相机轨迹 C a \mathbf{C}^a Ca、提示词 P a \mathbf{P}^a Pa 和噪声潜变量 Z a \mathbf{Z}^a Za 以生成 Z ^ a \hat{\mathbf{Z}}^a Z^a 。Observer 流也同理接收 C o \mathbf{C}^o Co、 P o \mathbf{P}^o Po 和 Z o \mathbf{Z}^o Zo 。 使用单个预训练的DiT(具体为 Cosmos-Predict2.5)来共同处理这两个流视频 :以包含 T T T 帧的块自回归生成,并以历史潜变量 Z hist a \mathbf{Z}^a_{\text{hist}} Zhista 和 Z hist o \mathbf{Z}^o_{\text{hist}} Zhisto 作为条件。
  • 视角-时间对齐 (View-time alignment):将 Actor 和 Observer 的潜变量与历史潜变量拼接为单一序列,并在 RoPE 时间轴上为匹配的潜变量分配相同位置,使自注意力机制能够在相同时步跨流交换信息。
  • 解耦提示 (Decoupled actor-observer prompting ):两者使用独立的提示词进行条件化, Observer 提示词 P o \mathbf{P}^o Po 负责控制和描述视野外的全局事件演化,相当于模型的记忆与控制中心 。

图 4:模型概览。(a)World Observer 利用共享的 DiT 同步生成主体流与全景观察者流。(b)观察者汇点将高分辨率初始全景图转换为四个透视参考图,以支持细粒度外观建模。

二.训练目标 (Training Objective)

  模型使用流匹配 (flow matching) 进行训练。对于每一个流 s ∈ { a , o } s \in \{a, o\} s∈{a,o},采样高斯噪声 ϵ ∼ N ( 0 , I ) \epsilon \sim \mathcal{N}(0, I) ϵ∼N(0,I) 和共享时步 k ∈ 0 , 1 k \in 0, 1 k∈0,1,构造插值潜变量:

Z ~ k s = ( 1 − k ) Z s + k ϵ \tilde{\mathbf{Z}}^s_k = (1 - k)\mathbf{Z}^s + k\epsilon Z~ks=(1−k)Zs+kϵ

整体优化目标为:

L = E ∑ s ∈ { a , o } ∥ v θ s ( Z \~ seq , k , P a , P o , k ) − ( ϵ − Z s ) ∥ 2 2 \mathcal{L} = \mathbb{E} \left \\sum_{s \\in \\{a, o\\}} \\left\\Vert{} \\mathbf{v}_\\theta\^s(\\tilde{\\mathbf{Z}}_{\\text{seq}, k}, \\mathbf{P}\^a, \\mathbf{P}\^o, k) - (\\epsilon - \\mathbf{Z}\^s) \\right\\Vert{}_2\^2 \\right L=E s∈{a,o}∑ vθs(Z~seq,k,Pa,Po,k)−(ϵ−Zs) 22

其中 Z ~ seq , k \tilde{\mathbf{Z}}_{\text{seq}, k} Z~seq,k 表示包含插值目标潜变量和干净历史潜变量的联合序列。

三. 全景观察者建模 (Panoramic Observer Modeling)

  为了确保 Actor 和 Observer 之间的几何一致性和视觉质量,系统引入了以下机制:

  • 全景变换 (Panoramic warping) :利用初始全景图 x 0 o x^o_0 x0o 和深度 D 0 D_0 D0,将共享源扭曲到各个流的目标位姿 c 0 → t o c^o_{0 \to t} c0→to。扭曲公式为:
    x warp , t s = Render ( Unproj ( x 0 o , D 0 o ) , c 0 → t o ) x^s_{\text{warp}, t} = \text{Render}(\text{Unproj}(x^o_0, D^o_0), c^o_{0 \to t}) xwarp,ts=Render(Unproj(x0o,D0o),c0→to)
  • 解耦分辨率 (Decoupled resolution):因为 Observer 仅需维持粗略的空间运动状态而不需要精细展示,其生成分辨率 R o R^o Ro 被设定为低于 Actor 的分辨率 R a R^a Ra。
  • 观察者池 (Observer Sink ):为了弥补全景图几何畸变带来的精细外观丢失,系统从高分辨率初始全景图裁剪出4个视角(水平偏角相差90°),编码为潜变量 z os v z^v_{\text{os}} zosv ,并附加到生成序列中 ,以便在物体重新进入视野时提供高分辨率参考。更新操作如下:
    Z seq ← Z seq ; { z os v } v , P seq ← P seq ; { H + L + Δ os + v δ os } v \mathbf{Z}{\text{seq}} \leftarrow \\mathbf{Z}_{\\text{seq}}; \\{z\^v_{\\text{os}}\\}_v, \quad \mathbf{P}{\text{seq}} \leftarrow \\mathbf{P}_{\\text{seq}}; \\{H + L + \\Delta_{\\text{os}} + v\\delta_{\\text{os}}\\}_v Zseq←Zseq;{zosv}v,Pseq←Pseq;{H+L+Δos+vδos}v

图 5:自注意力机制的可视化。针对重新进入物体的查询向量会关注其处于"出视野状态"的观察者区域;而在非联合模型(基线模型)中,该物体在经历出视野间隔后未能重新进入。

四. 灵活的观察者部署 (Flexible Observer Placement)

  对于复杂场景或互相遮挡的区域,单一 Observer 可能不足,系统可扩展为 N > 1 N > 1 N>1 的多观察者模式 。此时,所有 N N N 个 Observer 流被拼接到单一序列中由同一个 DiT 联合生成。系统会动态选择与当前 Actor 位置最近的初始 Observer 全景图作为视角扭曲和构建 Observer Sink 的几何源,以确保稳定的相机控制和连贯的全局状态维护。

  

训练与推理流程

  训练阶段 (Training):

  • 数据输入:使用 77 帧的视频块进行训练。
  • 特征提取:将成对的 Actor 视角视频和 Observer 全景视频编码为潜变量。
  • 联合去噪:将两者的潜变量拼接,输入到微调的 Cosmos-Predict2.5 DiT 模型中。模型通过流匹配(Flow Matching)目标函数,学习如何根据各自的文本提示词和相机轨迹条件去噪生成这两个流。
  • 多阶段微调:先训练单观察者模型 1.2 万步,然后在此检查点基础上,使用合成数据额外训练 6000 步得到多观察者模型。

  推理阶段 (Inference):

  • 自回归生成:以块(Chunk)为单位自回归生成长视频。当前块生成时,会将上一个块的尾部潜变量作为历史条件输入。
  • 状态演化:Actor 的视野随着提示词和相机轨迹移动,离开视野的物体在 Actor 画面中消失,但在 Observer 流中根据自身的动态逻辑继续演化。
  • 重返视野:当 Actor 的相机再次对准该区域时,DiT 通过自注意力机制从 Observer 当前的时步中读取物体的最新位置和状态,并结合 Observer Sink 提供的高清细节,在 Actor 的高分辨率画面中重新渲染该物体。

  评估指标 :文章引入了全新的世界空间指标,通过将物体提升到 3D 空间计算真实位移。其中 OOV-F 衡量物体能否正确地离开和重新进入视野,而 OOV-D (分为对比真值的 OOV-D gt \text{OOV-D}{\text{gt}} OOV-Dgt 和对比自身轨迹的 OOV-D self \text{OOV-D}{\text{self}} OOV-Dself) 则用于测量离开视野期间运动状态的准确性。

  

实验

实施细节

  模型与训练设置 :使用 8 张 NVIDIA H100 GPU,以 4.8e-5 的学习率(AdamW 优化器)对 Cosmos-Predict2.5 进行微调。训练数据包含从 2.3 万个真实视频采样的 13.8 万个片段,以及从 1.2 万个合成视频采样的 7.2 万个片段,视频块长度为 77 帧。

  单/多观察者训练 :单观察者模型以批次大小 16 训练了 1.2 万次迭代。多观察者模型从单观察者检查点启动,使用 2 个观察者在合成数据上以批次大小 8 额外训练了 6000 次迭代。

  分辨率设置 :Actor 渲染分辨率为 1280×704,Observer 分辨率为 640×320,Observer Sink 的偏移量设为 50。

  评估基准 :构建了包含真实全景视频和合成 CARLA 视频的保留测试集,每种各 100 个序列(共 200 个),长度为 77 帧。评估在单个视频块内进行,以排除内存检索的干扰,直接测试模型维持视野外动态的能力。测试序列均包含相机的来回旋转以及经历视野外运动的物体。

  基线对比与常规指标:对比了通用视频世界模型、全景世界模型和视野外方法(全景模型的输出被裁剪为透视视图以公平对比)。常规指标包括用于视觉/时间质量的 FID、FVD 和 VBench 图像质量,用于相机跟随精度的旋转误差(RotErr)和位移误差(TransErr),以及用于静态区域 3D 贴合度的掩码 PSNR 和 LPIPS。

图 3:数据集概览。上方为真实场景示例,下方为合成场景示例,分别采用移动配置与双固定观测者配置。彩色方框标示跨视角对应关系,最右侧示意图则展示相机配置与运动轨迹。

实验结果 (Results)

  生成能力:World Observer 展现了强大的状态记忆(离开视野的区域保持一致的演化)、状态可控性(观察者提示词可驱动视野外事件)、长视角自回归一致性,以及多观察者对遮挡区域的有效追踪。

图 6:定性结果。World Observer 支持以下四种模式:(a) 状态记忆,(b) 观察者引导控制,© 长时程生成,以及 (d) 多观察者模式。插图展示了观察者的视角。

图 7:与其他模型的定性对比 。红色方框分别标记目标物体在退出前及重新进入后的状态。现有模型往往会出现目标物体丢失、冻结或重建结果不一致的情况,而 World Observer 模型则能更好地保持目标物体的身份标识及其在视线外时段内的状态演化。

  对比优势:与其他模型相比,World Observer 在视觉/时间保真度、相机跟随精度和 3D 贴合度上表现优异,同时取得了最强的视野外动态(OOV)评分。它保持了较高的 OOV-F 和 OOV-D,意味着重新进入视野的物体既不会冻结,也不会变成"冒名顶替者"。增加观察者数量可通过覆盖更多区域进一步提升 OOV-F,同时保持 OOV-D 的竞争力。

相关推荐
杀生丸学AI2 天前
【世界模型】WorldCrafter:基于隐式三维感知记忆的视频世界模型
人工智能·三维重建·扩散模型·视频生成·世界模型·4dgs·动态重建
杀生丸学AI3 天前
【三维重建】GaussianGPT:迈向自回归三维高斯场景生成
深度学习·3d·transformer·三维重建·自回归·高斯泼贱
惊鸿一博5 天前
3DGS_可视化工具SIBR_gaussianViewer_app_使用说明/快捷键
人工智能·3dgs
I Am a robert girl5 天前
HelixWorld 源码剖析:当世界模型第一次“开口说话”
python·多模态·扩散模型·世界模型·自回归·空间音频·流式推理
RockHopper20259 天前
AI时代模型生命周期与软件资产重构:概要版
人工智能·系统架构·软件工程·ai编程·世界模型
I Am a robert girl10 天前
从零读懂世界模型的持续学习:一份组合式基准的源码级拆解
学习·具身智能·持续学习·灾难性遗忘·机器人学习·世界模型·组合式基准
RockHopper202511 天前
面向工业现实的原生数字化工程框架概要说明
人工智能·智能体·世界模型·工业数字化
天涯明月199313 天前
世界模型:原理、范式与工程实践
大数据·人工智能·大模型·具身智能·世界模型