面向可交互世界的精准编辑与灵活参考图引用 Precise Editing and Flexible Referencing for Interactable Worlds
首个支持流式编辑指令与参考图随时注入的视频世界模型 EditWorld
前置知识
视频世界模型 (Video World Model)
一类以视频生成为核心的可交互世界模拟器。模型以一张初始图像作为世界先验,随后根据用户的流式输入(键盘/鼠标动作、相机位姿、文本指令)逐块自回归地生成后续视频,让用户能在一个由生成模型想象出来的世界中持续探索。代表工作包括 YUME、Matrix-Game、HY-World 1.5、LingBot-World、Genie 3 等,近两年演进极快。
本文的 EditWorld 就是一个视频世界模型,它把这类模型的能力从导航探索扩展到精准修改世界内容。不理解这个载体,就无法理解论文要解决的问题位置。
流匹配与速度预测 (Flow Matching)
扩散模型的一种训练范式:对干净数据 $x$ 采样时间步 $\tau \sim U(0,1)$ 和高斯噪声 $\epsilon$,构造噪声样本 $x^\tau = (1-\tau)x + \tau\epsilon$,训练网络预测速度场 $v = \epsilon - x$,损失形如 $\|v_\theta - (\epsilon - x)\|_2^2$。推理时从纯噪声出发沿速度场迭代去噪即得样本。
EditWorld 的自回归与双向训练目标(公式 5-8)完全建立在流匹配之上,理解 $\tau$、$x^\tau$ 和速度目标才能读懂它的损失设计与蒸馏流程。
分块因果注意力 (Chunk-Causal Attention)
自回归视频生成不以单帧而以块(chunk,本文每块 $L$ 帧)为单位推进。生成第 $t$ 块时,干净块可以看到全部此前的干净块,而当前正在去噪的噪声块只能看到之前的干净块和它自己(论文公式 2)。这样既保持时间因果性、训练时可并行、推理时可流式,又避免未来信息泄漏。
这是 EditWorld 实现边生成边编辑的架构基础,其 Gated Causal Attention 就是在这个注意力模式上扩展出来的。
RoPE 旋转位置编码
把相对位置信息注入注意力机制的方法:对 query 和 key 按各自位置的角度做旋转变换,使注意力打分自然依赖二者的相对位置。视频模型中通常时间维与空间维各有独立的 RoPE 坐标系,是当前主流视频 DiT 的标配组件。
本文给参考图 token 分配负的时间 RoPE 坐标 $m(i+1)$,把它们推到视频时间轴之外,抑制模型直接复制粘贴参考图。不懂 RoPE 就无法理解这一设计的动机。
KV 缓存与稀疏上下文
自回归推理时,历史 token 的 key/value 会被缓存以避免重复计算,但缓存随视频长度线性增长。稀疏注意力只保留一部分历史 KV(如 sink 锚点块、最近若干块、按相关性检索出的历史块),让上下文预算固定,从而支持任意长的流式生成。
EditWorld 的 Sparse Context 机制(sink + 最近两块 + Top-$k$ 检索)正是为约束 KV 预算而设计,且训练时需要配套的随机稀疏采样来匹配推理分布。
Teacher Forcing 与误差累积
训练自回归模型时用真实数据而非模型自身的生成结果作为历史上下文,训练稳定但推理时会遇到训练时从未见过的受污染历史,误差逐步放大(exposure bias)。Self-resampling、Self-Forcing 等技术通过在训练中逐渐掺入模型自生成的历史来缓解此问题。
EditWorld 采用退火式 self-resampling 换入自生成历史,是它能在 240-480 帧、多轮编辑序列下保持稳定一致的关键训练技巧。
研究动机
现有视频世界模型的能力重心几乎全部压在导航上:YUME、Matrix-Game、HY-World 1.5、LingBot-World 等都聚焦于让用户用键盘鼠标或相机轨迹在生成的世界里漫游,对世界内容本身的控制非常有限。近期的尝试开始引入文本驱动的事件生成——YUME 1.5 支持文本控制的世界事件,HY-WorldPlay 1.5 支持跨场景的可提示事件,LingBot-World 2.0 扩展了文本事件种类,DreamX-World 通过事件指令微调实现可组合事件控制——但它们本质上都是触发或生成新事件,而不是精确修改世界里已经存在的内容,例如把街边某辆车换成马车、移走一座桥、给整个场景做风格化这类增、删、替换、风格化级别的精准编辑它们做不到。另一条线是参考图条件:XGEN-JING 和 ABot-World 支持用初始参考图保持身份信息,但参考图只能在开始时给定一次,用户无法在交互过程中随时把新参考图的内容注入已生成的世界,也无法精确控制参考内容出现的时机。此外,现有评测基准几乎都只测导航能力和视频保真度,流式编辑能力没有任何系统性的度量标准。
本文的目标是论文目标是把视频世界模型从只能探索推进到可精准修改。具体而言,EditWorld 要让用户在自回归生成世界的过程中做到三件事:第一,通过流式的编辑指令持续修改世界内容,覆盖全局编辑(天气、季节、时间、光照、色调、艺术风格的整体改变)和局部编辑(元素的添加、删除、替换以及颜色、材质、形状、状态等局部属性修改),且编辑要精确命中目标、不破坏无关内容;第二,支持灵活的参考图引用——参考图可以在交互的任意阶段注入而不仅限于初始化,从而实现把参考图里的这辆车加到人行道上这类带时机控制的多轮操作;第三,在获得编辑能力的同时保持长时程生成的稳定性与世界一致性,通用世界建模能力(导航、画质、物理合理性)不退化,能在 WBench 上与商业世界模型同台竞争。
与已有工作不同的是,本文的独特切入角度是把编辑内化为世界生成的第一公民,而不是生成之后再做视频编辑的后处理。技术上它提出了三个关键观察与设计:其一,编辑指令应当按时间门控生效——数据管线为每个视频块标注 before/during/after 三态编辑状态,编辑指令只在 during 状态注入交叉注意力,避免编辑提前泄漏或滞后;其二,参考图的可见性应由文本上下文门控——视频块只有在自己的提示词(含前两块的提示)提到参考内容时才被允许注意该参考图,把何时注入参考这一对齐问题显式写进注意力掩码,这是已有工作完全没有做的;其三,给参考图 token 赋负时间 RoPE 偏移并采用单向注意力,把参考图定位为条件信号而非可复制的像素来源。评测侧则补上了空白:现有 benchmark 只测导航,作者构建了 WBench-Editing(约 150 例、每例 240-480 帧、1-3 条流式编辑指令)来系统评测流式世界编辑。
核心方法
直觉上,EditWorld 想让世界模型在边走边看的同时听懂修改指令并随手改世界。技术上,它把世界生成形式化为因果视频生成:设视频块序列 $V=\{x_1,\dots,x_T\}$,每块 $x_t\in\mathbb{R}^{L\times H\times W\times C}$ 含 $L$ 帧,用户输入序列 $C=\{c_1,\dots,c_T\}$ 包含文本提示、动作/相机位姿和参考图,生成按 $p_\theta(x_{1:T}\mid c_{1:T})=\prod_{t=1}^{T}p_\theta(x_t\mid x_{<t},c_{\leq t})$ 逐块分解。模型底座是双向视频世界模型 LingBot-World-Base,先用 teacher forcing 适配为自回归生成,再叠加三个模块:Gated Causal Attention(分块因果自注意力 + 时间门控的编辑指令交叉注意力 + 单向门控的参考图注意力)、Sparse Context(固定预算的历史上下文:sink 块 + 最近两块 + Top-$k$ 检索块),以及一套训练课程(自回归+双向联合目标、退火 self-resampling、PCM+SGF 两阶段 4-NFE 蒸馏)。数据侧构建专门管线,从导航数据集(Sekai、OmniWorld、SpatialVID)和编辑数据集(Ditto-1M、OpenVE-3M)合成带编辑标注与参考图的长视频,为精细修改和参考引导提供监督。
核心创新是 Gated Causal Attention,它回答了编辑指令何时生效和参考图何时可见这两个此前无人解决的问题。对文本:每个块 $x_t$ 的交叉注意力始终看到场景描述 $p_{\text{scene}}$,而编辑指令 $p_{\text{edit}}$ 只在该块被标注为 during 状态时激活;同时为避免提示切换造成画面突变,每个块还会注意前两块的提示 $p_{t-1}$ 和 $p_{t-2}$,形成短期文本上下文。对参考图:参考图经 VAE 编码后与视频潜变量在时间维拼接,但注意力是单向门控的——参考图 token 只看自己那张图内部(不被视频 token 污染),视频块 $x_t$ 只有当其可见提示($p_t,p_{t-1},p_{t-2}$)的语义提到该参考时才被允许注意它;且第 $i$ 张参考图的全部 token 被赋负时间 RoPE 坐标 $m(i+1)$,把参考推离视频时间轴,抑制直接复制粘贴。另一创新是 Sparse Context:生成 $x_t$ 时始终保留 sink 块 $x_0$(保存初始世界状态)和最近两块,再从剩余历史中用预 RoPE query/key 平均池化后的余弦相似度 $s_{t,j}=\langle\bar{Q}_t,\bar{K}_j\rangle/(\|\bar{Q}_t\|\|\bar{K}_j\|)$ 检索 Top-$k$ 块,使上下文预算恒定:$\mathcal{A}_t=\mathcal{S}_t\cup\mathrm{TopK}_{j\in\mathcal{H}_t}(s_{t,j},k)\cup\mathcal{R}_t$。
方法步骤详情
整体流程分数据、架构、训练三步。第一步数据合成:全局编辑管线从约 200 个主属性加 100 个辅助属性的编辑词表中,用 Qwen3.6-27B 按视频内容采样属性组合并实例化为与场景一致的编辑提示,用 Qwen-Image-Edit 生成锚点帧 $F_a$ 的编辑版本;保留 $F_a$ 之前的原始片段为未编辑前缀,用 Wan2.2-FLF2V-A14B-Control 以过渡起点帧 $F_t$ 为首帧条件、编辑锚帧为末帧条件、原始视频对应区间的深度图为结构控制信号合成平滑过渡段,再用 Wan2.2-I2V-A14B-Control 以编辑锚帧为初始帧、深度序列为控制延拓后续内容,最后三段拼接成完整视频。局部编辑管线则从 Ditto-1M/OpenVE-3M 经两阶段 VLM 过滤(先判断差异是否属于局部编辑,再评语义一致性与画质)选出源-编辑视频对,选源帧 $F_s$ 与目标帧 $F_t$,用同一深度控制 FLF2V 模型合成过渡段后拼接。第二步标注:为每段视频生成场景描述(静态不变内容)、精修后的编辑指令(动态变化)、逐 chunk 的 before/during/after 编辑状态,并用 ViPE 重估全部样本的相机内外参;参考图子集用 Grounded-SAM 分割目标对象、Qwen-Image-Edit 修补瑕疵,同时改写指令剔除参考图已表达的信息以防语义泄漏,迫使模型从参考图中提取内容。第三步训练:在 LingBot-World-Base 上以 teacher forcing 起步,联合优化自回归损失 $\mathcal{L}_{\mathrm{AR}}=\mathbb{E}\|v_\theta(x_i^\tau,\tau\mid x_{<i},c_{\leq i})-(\epsilon_i-x_i)\|_2^2$ 与双向损失 $\mathcal{L}_{\mathrm{BI}}$,总目标 $\mathcal{L}=\mathcal{L}_{\mathrm{AR}}+\lambda_{\mathrm{BI}}\mathcal{L}_{\mathrm{BI}}$;随后以退火调度逐步把真实历史替换为模型自生成历史(self-resampling);最后两阶段蒸馏——先用 PCM 在每块 4 次函数评估(NFE)下做轨迹蒸馏,再用 Self Gradient Forcing(SGF)做分布蒸馏,历史 KV 重算梯度但不回传整条 rollout。
技术新颖性
技术新颖性体现在四个层面。其一,任务定义新:此前世界模型只做触发新事件(YUME 1.5、DreamX-World)或初始参考条件(XGEN-JING、ABot-World),本文首次把增/删/替换/风格化的精准编辑和任意时刻的参考图注入纳入自回归世界生成的闭环。其二,条件门控机制新:编辑指令按数据驱动的 during 状态门控激活、参考图可见性由文本上下文门控,这套把时机控制写进注意力掩码的设计是本文独有的;负 RoPE 边距分离参考 token 与视频 token 也是对 RoPE 的巧妙用法——正向用位置对齐语义、负向用偏移抑制复制。其三,长时程机制新:sink + recent + Top-$k$ 检索的 Sparse Context 与训练时的随机稀疏采样相配合,保证推理时固定预算上下文与训练分布匹配,比 AlayaWorld 的 3D 重投影缓存、WorldKV 的视角检索更轻量,不依赖显式 3D 表示。其四,训练配方新:作者发现纯因果目标会让模型偏向视频续写而弱化条件响应,故用共享参数的 AR+BI 双注意力模式联合训练(消融显示 Detail Accuracy 从 34.9 提升到 53.2),再加退火 self-resampling 与 PCM+SGF 蒸馏构成完整课程。此外 WBench-Editing 本身也是对整个领域评测生态的贡献。
实验结果
核心结果集中在自建的 WBench-Editing(约 150 例,每例 15-30 秒即 240-480 帧,1-3 条流式编辑指令,评测分 Editing/Navigation/Quality/Setting/Consistency/Physical 六类,编辑指标由 VLM 问答评估)。公平起见所有对比模型都使用文本指令和自回归变体。EditWorld 取得 Overall 73.8,比第二名 YUME 1.5(67.0)高 6.8 分;核心 Editing 指标达 80.0,比第二名(YUME 1.5 与 LingBot-World 的 54.8)高 25.2 分,优势悬殊;Physical 67.9 也是全场最佳,说明编辑后的世界保持物理合理;Navigation 74.3、Quality 74.4、Setting 60.9、Consistency 83.8 均属第一梯队,证明编辑能力没有以牺牲世界建模为代价。参考图子集上(Table 2),EditWorld Overall 74.0、Editing 74.6,显著超过 XGEN-JING(BI) 的 68.7/63.6、SolarWM 的 64.6/19.8 和 ABot-World 的 57.3/25.0,且是唯一支持在生成中途注入参考图的模型——定性对比显示 XGEN-JING 因参考只在初始化给定,无法按多轮顺序先加手推车再加黄色气球,两者一起出现了。通用能力方面(Table 3,WBench),双向 SFT 模型得 81.2 排第四,把底座 LingBot-World(base-camera) 从 78.5 提升上来,其中 Setting 从 72.6 涨到 84.6;AR SFT 模型 77.7 仍有竞争力;蒸馏后的 4-step AR 模型 79.4,其中 Setting 85.9、Consistency 89.2,比 AR SFT 平均提升 1.7 分、Setting 提升 8.2 分。消融(Table 4)显示 AR+BI 联合训练把 Editing Overall 从 69.7 提到 80.0,Editing Presence 82.6→93.0、Semantic Alignment 79.0→89.2、Editing Completion 73.5→86.2、Detail Accuracy 34.9→53.2,而 Edit Cleanliness 基本不变(78.3→78.1),说明双向监督主要强化对编辑指令的细粒度遵循。图 4 的定性消融表明单向参考注意力比双向更好地保留参考细节,短期文本上下文能消除提示切换时的场景跳变。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| WBench-Editing 流式世界编辑(文本指令、自回归变体) | Overall | 73.8 | YUME 1.5:67.0(第二名) | +6.8 |
| WBench-Editing 编辑能力 | Editing | 80.0 | YUME 1.5 / LingBot-World:54.8(第二名) | +25.2 |
| WBench-Editing 物理合理性 | Physical | 67.9(全场最佳) | SolarWM(H3):67.7(次优) | +0.2 |
| WBench-Editing 参考图子集 | Overall / Editing | 74.0 / 74.6 | XGEN-JING(BI):68.7 / 63.6 | +5.3 / +11.0 |
| WBench 通用世界建模(双向 SFT) | Average(排名第 4) | 81.2 | LingBot-World (base-camera):78.5 | +2.7(Setting 72.6→84.6) |
| WBench 通用世界建模(4-step AR 蒸馏) | Average / Setting | 79.4 / 85.9 | EditWorld (AR, SFT):77.7 / 77.7 | +1.7 / +8.2 |
| 消融:AR-only vs AR+BI 联合训练 | Editing Overall / Detail Accuracy | 80.0 / 53.2(AR+BI) | 仅 AR:69.7 / 34.9 | +10.3 / +18.3 |
局限与改进
作者承认的局限可以从数字反推:其一,WBench-Editing 上 Setting 只有 60.9,明显低于 SolarWM 的 67.1 和 Zing-0.5 的 67.7,说明对场景级设置/风格化转换的遵循仍不完美;Consistency 83.8 也低于 AlayaWorld 的 93.6,多轮编辑后的世界一致性还有差距;Quality 74.4 不及 YUME 1.5 的 76.1 和 LingBot-World 的 75.8。其二,WBench 上即使双向 SFT 版本也只排第四(81.2,落后于 Alaya-EVOKE-Turbo 82.0、HiDream-O1-World 80.9、Zing-0.5 等 81.0 档),AR SFT 版本 77.7 属于第一梯队但非顶尖。其三,公平对比时所有模型都用文本指令,参考图能力只在与三个模型的小子集上比较。我自己的观察还有几点:评测完全依赖 VLM 问答、规模仅约 150 例且由作者自建,可能存在偏向自家模型的系统性偏置;数据合成的监督上限受制于 Wan2.2、Qwen-Image-Edit 等生成模型的成图质量,所谓编辑真值本身就是模型生成的;编辑状态(before/during/after)由 VLM 自动分配,标注噪声会直接传导到门控机制;参考图可见性依赖提示词语义匹配,若用户指令含糊(如把那个东西加进来),门控可能失效;论文未报告推理延迟、显存占用和实际帧率,实时程度未知;Sparse Context 的 Top-k 检索基于外观相似度,回访早期区域时若外观已因编辑而改变,可能检索不到原始状态,长程物体恒常性存疑。
独立分析的弱点
独立分析有五个弱点。第一,参考门控对指令措辞敏感:参考图可见性由 $p_t,p_{t-1},p_{t-2}$ 是否语义提到参考决定,但这实际是学出来的软对齐,遇到把刚才那张图里的东西加进来这类指代模糊的指令时可能误开门或漏开门,改进方向是引入显式的参考选择模块,或让用户直接指定参考生效的时间区间。第二,编辑监督是合成的:训练真值由 Wan2.2 FLF2V/I2V 生成,过渡段的物理合理性与身份保持继承自这些模型的缺陷,模型可能学到编辑等于渐变变形的伪影模式,改进方向是引入真实拍摄的前后对比视频或游戏引擎渲染的精确编辑数据。第三,VLM 标注链路长且脆弱:场景描述、指令改写、状态划分全部依赖 VLM,任一环出错(如把静态内容写进编辑指令、语义泄漏未剔除干净)都会污染门控信号,可用人工抽检加一致性校验形成闭环。第四,缺乏持久世界记忆:sink 块只有 1 个、检索基于外观相似度,跨大范围回访或多次编辑叠加时早期被编辑内容的状态难以精确恢复,可结合 3D 重投影缓存或显式世界状态表。第五,缺少效率与交互性证据:没有报告端到端延迟、单块生成耗时与用户研究,4-NFE 蒸馏后的真实体验无法量化,应补充 FPS/显存曲线与真人盲测。
未来方向
作者的展望是让用户能持续、灵活地精修、重塑和扩展世界。基于本文成果可自然延伸的方向包括:第一,编辑的持久性与可撤销性——为编辑过的对象维护显式状态存储,支持撤销上一步编辑和跨回访的状态恢复,可与 AlayaWorld 的 3D 重投影缓存思路结合;第二,非文本粒度的编辑指定——点击式掩码、拖拽式空间指定与参考图三者融合,绕开纯文本门控的歧义;第三,物理一致编辑——当前 Physical 67.9 已是最佳但仍有空间,可引入物理仿真先验(流体、刚体)作为过渡段合成或损失约束;第四,多模态交互——语音指令、草图输入、多参考图组合注入与参考间冲突消解;第五,智能体驱动的世界搭建——把 EditWorld 作为具身智能体或游戏引擎的可编辑沙盒后端,由 LLM 规划多轮编辑序列并自动验证结果;第六,评测扩展——WBench-Editing 目前约 150 例、纯 VLM 评分,可扩大规模、加入人类偏好评测以及光流一致性、重投影误差等物理合理性定量指标;第七,更长时程——把 Sparse Context 与分层记忆结合,向分钟级多轮编辑推进,对标 SANA-WM 的分钟级生成与 HY-World 1.5 的长时程几何一致性。
复现评估
复现难度中高。有利条件:论文给出了 GitHub 仓库(github.com/leoisufa/EditWorld,完整权重与训练脚本的开源程度需以仓库实际内容为准);底座 LingBot-World-Base 及所有数据源(Sekai、OmniWorld、SpatialVID、Ditto-1M、OpenVE-3M)均为公开资源;管线依赖的工具全部开源或可获取——Qwen3.6-27B、Qwen-Image-Edit、Wan2.2-FLF2V/I2V-A14B-Control(VideoX-Fun)、ViPE、Grounded-SAM、PCM、SGF。不利条件:完整复现需要跑通多阶段流水线——数据合成(大规模视频生成本身就需可观 GPU 时长)、AR+BI 联合 SFT、退火 self-resampling、两阶段蒸馏(PCM 加 SGF);论文未披露模型参数量、训练卡时、批量大小、$\lambda_{\mathrm{BI}}$、检索数量 $k$、负 RoPE 边距 $m$、块长 $L$ 等关键超参,训练算力估计在数十张 A100/H100 级别、周期以周计;评测依赖 WBench 框架与 VLM 评分,对齐评测器版本才能复现分数。对普通研究者的建议路线:先复用其开源数据合成管线的思路在小模型上验证门控机制的有效性,或等待官方放出权重与配置后做推理级复现,再考虑训练级复现。
论文图表
31 个模型版本在原始 WBench(Quality/Setting/Interaction/Consistency/Physical)上的榜单:EditWorld (BI, SFT) 81.2 排第四(Quality 74.8、Setting 84.6、Interaction 80.3、Consistency 87.8、Physical 67.7),相比底座 LingBot-World(base-camera) 的 78.5 有提升;EditWorld (AR, SFT) 77.7;EditWorld (AR, 4-step) 79.4(Setting 85.9、Consistency 89.2)。榜首为 Alaya-EVOKE-Turbo 82.0。
回应关键质疑——加入编辑能力是否损害通用世界建模:答案是没有,SFT 后反而超越底座并进入榜单前四,蒸馏版还大幅提升 Setting,说明编辑训练与通用能力兼容。