ReWorld: An Interactive World Model with Long-Horizon Memory

TL;DR

ReWorld通过分离控制与记忆训练,采用混合注意力窗口和边缘缓存,实现长时记忆与实时控制的平衡。

cs.AI 🔴 高级 2026-08-25 80 次浏览
Zhifei Chen Luozhou Wang Guibao Shen Dongyu Yan Shuai Yang Tianshuo Xu Yihua Du Wei Wang Tianyi Gui Lianghua Huang Yingcong Chen
交互式世界模型 长时记忆 注意力机制 边缘缓存 多源数据

核心发现

方法论

ReWorld采用分离训练策略,将控制和记忆在不同窗口下训练,利用混合每头注意力窗口(局部与全局)和随机头路由实现能力解耦。模型在训练时引入chunk-drop增强鲁棒性,推理时结合pose索引的边缘存储(landmark bank)和有限KV缓存,实现长时间跨度的场景回忆。通过分布匹配蒸馏(distribution-matching distillation)和LoRA适配器,模型在保持高保真度的同时实现实时流媒体生成。多源数据管线将合成、真实与游戏画面统一尺度,确保动作一致性。模型在六个交互世界基准中表现出最佳控制精度(旋转误差11.95°)和视频质量,尤其在长滚动回放中仍能准确重现起始视角。

关键结果

  • 在控制保真度方面,ReWorld实现了11.95°的旋转误差和最高的相机运动一致性,超越六个最新交互模型;在长达64秒的回放中,固定12块缓存依然能重建起始视角,显示其长时记忆能力。模型在多源数据融合下,能在不同风格(写实、游戏、风格化)中流畅生成704×1280视频,且推理速度满足实时需求。
  • 通过边缘存储和chunk-drop训练,模型在有限KV预算下保持高质量回放,验证了边缘存储策略的有效性。随机头路由确保训练中不同能力不绑定到特定头,增强模型泛化能力。蒸馏技术使模型在四步采样下实现多模态高保真输出,兼顾控制与长时记忆。
  • 在控制、记忆和视频质量三轴评估中,ReWorld均优于对比模型,特别是在长距离场景回忆和动作跟随方面表现突出。模型在处理复杂路径和反复访问场景时,展示了强大的空间记忆和动作响应能力。

研究意义

该研究突破了交互式世界模型在长时记忆与实时控制间的矛盾,提出了分离训练与边缘存储结合的创新框架。模型不仅提升了虚拟环境中的场景重访与连续性,还为未来自主导航、虚拟现实等应用提供了基础技术支持。其多源数据融合和高效推理机制,为复杂场景下的实时交互带来新可能,推动虚拟环境智能化发展。

技术贡献

ReWorld的核心技术在于通过混合注意力窗口和随机头路由实现能力解耦,解决控制与记忆的冲突问题。引入pose索引的边缘存储机制,有效支持长时场景回忆。chunk-drop训练增强模型对稀疏历史的鲁棒性,结合分布匹配蒸馏和LoRA适配器,实现高效的多模态流媒体生成。模型在不增加参数的情况下,兼顾高保真度与实时性,提供了创新的架构设计与训练策略。

新颖性

本研究首次在交互式世界模型中引入分离训练策略,利用混合每头注意力窗口和随机头路由实现能力解耦,突破了传统模型在长时记忆与控制响应上的限制。结合pose索引的边缘存储机制,显著提升了长距离场景回忆能力。蒸馏技术的创新应用,使模型在保持高质量的同时实现实时推理,展现出极强的工程创新性。

局限性

  • 模型对极端复杂场景或动态变化剧烈的环境仍存在记忆失效或控制偏差的问题,尤其在超长时间跨度的连续回放中可能出现信息遗失。
  • 边缘存储策略依赖于场景的空间冗余度,对于高度稀疏或非重复场景,效果可能受限。
  • 训练过程中对多源数据的统一尺度调整和pose校准存在一定误差,影响模型的泛化能力和精确性。

未来方向

未来将探索更高效的边缘存储策略,提升模型在极端环境下的鲁棒性。结合强化学习优化路径规划和场景记忆,增强自主导航能力。同时,计划扩展多模态输入(如声音、触觉)以丰富交互体验,推动虚拟环境的智能化发展。

AI 总览摘要

ReWorld提出了一种创新的交互式世界模型,旨在解决长时记忆与实时控制之间的根本矛盾。传统模型在追求短期响应和长远记忆时,常常陷入能力绑定或性能退化的困境。本文通过将控制和记忆的训练分离,采用混合每头注意力窗口和随机路由机制,有效实现能力解耦,确保模型在推理时既能快速响应当前动作,又能长时间回忆场景信息。

模型引入pose索引的边缘存储(landmark bank),结合有限的KV缓存,支持在长时间跨度内准确重建场景。训练中采用chunk-drop策略,使模型适应稀疏历史,提升鲁棒性。通过多源数据管线,将合成、真实与游戏画面统一尺度,确保动作一致性。蒸馏技术则使模型在四步采样下实现高保真、多模态流媒体生成,满足实时交互需求。

在六个交互基准测试中,ReWorld在控制精度和视频质量方面均优于现有最先进模型,特别是在长回放和场景重访任务中表现出色。这一技术突破为虚拟环境中的自主导航、虚拟现实等应用提供了坚实基础,推动智能交互系统向更高水平发展。未来,模型将继续优化边缘存储策略,增强在复杂环境中的表现,并扩展多模态交互能力,开启虚拟世界的无限可能。

深度分析

研究背景

虚拟环境中的交互式世界模型经历了从简单场景模拟到复杂长时记忆的演变。早期方法如WorldGAN和VideoGPT主要关注场景生成和短期响应,但缺乏长时记忆能力。近年来,基于Transformer的模型如VQ-VAE和Video Diffusion在生成质量上取得突破,但在长距离场景回忆和实时控制方面仍受限。多源数据融合、边缘存储和能力解耦成为研究热点,推动模型在虚拟现实、自主导航等领域的应用。尽管如此,长时记忆与实时响应的平衡仍是核心难题。

核心问题

现有交互式世界模型在实现长时间跨度的场景回忆与快速响应控制之间存在矛盾。短窗口训练难以支持长时记忆,长时记忆机制在推理时受限于有限缓存,导致场景重访失真。此外,模型在多源、多风格数据融合时存在尺度不一致和 pose 校准误差,影响整体性能。如何在保证实时性同时实现长时记忆,是当前技术的瓶颈。

核心创新

ReWorld的主要创新包括:1)分离控制与记忆训练,利用混合注意力窗口实现能力解耦;2)引入pose索引的边缘存储(landmark bank),支持长时场景回忆;3)chunk-drop训练增强模型对稀疏历史的鲁棒性;4)多源数据管线确保动作尺度一致;5)蒸馏技术实现四步采样,兼顾高保真与实时性。这些创新共同推动模型在长时记忆与控制响应上的突破。

方法详解

  • �� 训练阶段:将模型的注意力头分为局部(短窗口)和全局(长窗口)两组,通过随机头路由在每次训练中切换角色,确保每个头同时学习两种能力。
  • �� 控制机制:引入pose索引的注意力(MRoPE),根据相机姿态检索缓存内容,增强场景重访能力。
  • �� 动作注入:将用户指令的姿态信息直接融入模型,确保动作响应的准确性。
  • �� 缓存策略:在推理时,将有限的KV缓存与边缘存储(landmark bank)结合,存储关键场景片段,并通过姿态相似性检索。
  • �� Chunk-drop训练:随机遮挡部分历史信息,训练模型在稀疏历史下保持性能。
  • �� 蒸馏:采用distribution-matching蒸馏,将多步生成压缩为四步,确保实时推理。
  • �� 多源数据:统一合成、真实和游戏画面,调整尺度,确保动作一致性。

实验设计

模型在六个交互基准中测试,包括控制精度、长时记忆和视频质量。使用合成、真实和游戏数据,评估旋转误差、场景重访准确率和视频帧一致性。对比多种缓存策略和训练方法,验证边缘存储和chunk-drop的有效性。超参数如12块缓存、6个landmark和64秒回放时间,均经过调优以获得最佳性能。

结果分析

ReWorld在控制精度方面实现11.95°旋转误差,超越对比模型;在长回放任务中,固定12块缓存仍能重建起始视角,显示其长时记忆能力。模型在多源数据融合下,生成704×1280高质量视频,且推理速度满足实时需求。边缘存储和chunk-drop策略有效缓解了缓存限制带来的性能下降,验证了系统设计的合理性。

应用场景

该模型适用于虚拟现实、虚拟导览、游戏开发和自主导航等场景。只需提供动作指令和少量存储空间,即可实现长时场景重访和高质量实时生成。未来可结合强化学习优化路径规划,增强自主决策能力,推动虚拟环境的智能化。

局限与展望

模型在极端复杂或动态变化环境中仍存在记忆失效问题,长时间跨度的连续回放可能出现信息遗失。边缘存储策略对场景的空间冗余度敏感,稀疏或非重复场景效果有限。训练时尺度调整和pose校准误差影响模型泛化,未来需优化数据预处理和存储机制。

通俗解读 非专业人士也能看懂

想象你在玩一个非常复杂的拼图游戏,你需要记住每一块拼图的位置,还要根据指令快速拼出新的图案。传统的拼图游戏只能记住最近几块拼图,远处的拼图容易忘记。而ReWorld就像是给你一套特殊的记忆系统,既能快速找到最近的拼图,又能记住远处的重要拼图。它用一种聪明的方法,把重要的拼图存放在一个专门的“记忆仓库”里,随时可以找到。这样,无论你玩多长时间,都能记得之前拼过的内容,也能快速响应新指令,就像在一个虚拟世界中自由探索一样。这种技术让虚拟环境变得更真实、更连续,也更智能。

简单解释 像给14岁少年讲一样

想象你在玩一个超级酷的电子游戏,你可以控制一个角色在一个虚拟世界里走来走去。以前的游戏只能记住你刚刚走过的路,但如果你想回到很久以前走过的地方,就会忘记细节。ReWorld就像给游戏装上了一个神奇的记忆助手,它可以记住很多很远的地方,还能帮你快速找到那些记忆。它用一种特别的方法,把重要的场景存放在一个“记忆仓库”里,每次你想回到以前的地方,只要用你的动作指令,模型就能准确地带你回去。这样,你的游戏体验就会变得更流畅、更真实,就像在一个真正的世界里一样。

原文摘要

An interactive world model must follow the user's actions, remember the places it has shown, and stream in real time. The tension is structural: control wants a short horizon, memory wants an unbounded one. ReWorld separates the two during training and bounds them at inference. Mixed per-head attention windows confine most heads to the recent past while a small set of global heads attends over the entire history, and random head routing keeps either capability from binding to particular heads; random chunk dropping makes sparse histories in-distribution. At inference the whole past lives under a fixed budget: a bounded KV cache backed by a pose-indexed landmark bank, from which the model retrieves the landmarks nearest the current pose. A metric-scale-aligned data engine places eight sources -- Unreal-rendered fly-throughs, game roaming, and real-world footage -- on one physical action scale, so the same key press moves the camera the same distance in every source, and palindrome trajectories supply the revisit evidence that memory training needs. Distribution-matching distillation confined to a LoRA adapter then compresses sampling to four steps: one backbone serves both a high-fidelity multi-step mode and a real-time interactive one, streaming 704x1280 video across photorealistic, game-style, and stylized worlds. Under a three-axis protocol covering action following, long-horizon recall, and video quality, against six recent interactive world models it attains the best control fidelity ($11.95^\circ$ rotation error and the best camera-motion consistency) and the best generation quality; and on minute-long out-and-back rollouts ($64$\,s, $384$ latents), its fixed 12-chunk cache still regenerates the starting view -- at rollout lengths where a sliding window has long evicted the evidence and full-KV attention runs out of memory.

cs.AI