基于真实世界的城市场景世界模拟模型:首尔世界模型 Grounding World Simulation Models in a Real-World Metropolis
首尔世界模型:用街景图像检索增强,实现真实城市级别的世界模拟视频生成
前置知识
世界模型(World Model)
世界模型是一种学习环境内部表示并预测其未来状态的模型。在视频生成领域,世界模型将生成模型作为动力学模型,根据过去观测和动作来预测未来观测,模拟环境如何随时间演化。给定起始图像、动作输入和文本提示,世界模型可以生成动态和交互式的环境,包括物体运动、天气变化和物理交互。当前的世界模型通常在想象的环境中操作,而本文将其扩展到真实物理世界。
本文提出的首尔世界模型(SWM)正是在世界模型的基础上,将其应用场景从想象环境扩展到真实城市。理解世界模型的基本概念和工作原理是理解本文贡献的前提。
检索增强生成(Retrieval-Augmented Generation, RAG)
检索增强生成是一种在生成过程中引入外部知识的技术。在本文中,RAG被用于视频生成:给定地理位置坐标和相机轨迹,系统从地理索引的街景数据库中检索附近的街景图像,然后将这些图像作为条件来指导视频生成。这种设计使得生成的视频能够锚定到真实世界的几何结构和外观,而不是完全由模型想象。
RAG是SWM的核心机制,使得世界模拟模型能够“接地”到真实世界。理解RAG如何工作以及它在视频生成中的创新应用是理解本文的关键。
自回归视频生成(Autoregressive Video Generation)
自回归视频生成是一种逐块(chunk-by-chunk)生成视频的方法。模型首先生成一个块(包含若干帧),然后使用前一个块的输出作为历史条件来生成下一个块。这种方法允许生成任意长度的视频,但存在误差累积问题:每一步的误差会累积,导致长期生成质量下降。现有的解决方案包括注意力机制(attention sink),但本文提出了更先进的虚拟前瞻槽(Virtual Lookahead Sink)。
SWM采用自回归方式生成长视频(可达数百米),误差累积是长期生成的核心挑战。理解自回归生成的工作原理和挑战是理解本文技术贡献的基础。
扩散变换器(Diffusion Transformer, DiT)
扩散变换器是一种结合扩散模型和Transformer架构的生成模型。它将UNet骨干替换为Transformer,实现更好的可扩展性和质量。SWM基于Cosmos-Predict2.5-2B预训练模型进行微调,这是一个2B参数的DiT,包含28个块、16个注意力头和2048的隐藏维度。它在16通道的潜空间中操作,通过3D VAE(4倍时间压缩和8倍空间压缩)从像素空间压缩而来。
SWM的技术架构基于DiT,理解其基本结构有助于理解模型如何处理视频生成任务,包括潜空间操作、VAE压缩和注意力机制。
教师强迫与自强迫(Teacher Forcing vs Self-Forcing)
教师强迫(TF)在训练时使用真实的历史帧作为条件,而在推理时使用模型自己的输出。自强迫(SF)则在训练时就使用模型自己生成的历史,使得训练和推理条件一致。SWM同时评估了这两种配置:TF模式使用T=77帧、H=5历史帧、K=5个参考图像;SF模式使用较短的块(12帧)、H=3历史帧、K=1个参考图像,在单H100 GPU上实现15.2 fps。
这两种训练策略的选择影响生成质量和推理速度。理解它们的区别有助于理解SWM的实验设计和性能权衡。
3D VAE与潜空间压缩
3D变分自编码器(3D VAE)是一种将高维视频数据压缩到低维潜空间的模型。在本文中,3D VAE实现4倍时间压缩(每4帧压缩为1个潜表示)和8倍空间压缩。这种压缩对于处理长视频至关重要,但也带来了一个技术挑战:孤立的关键帧无法形成有效的4帧组,因此本文提出了间歇性冻帧策略(intermittent freeze-frame strategy)来解决这个问题。
理解3D VAE的压缩机制是理解SWM视图插值流水线的基础,特别是理解为什么需要间歇性冻帧策略而不是简单的通道拼接。
研究动机
当前的视频世界模型(如Aether、DeepVerse、Yume1.5、HY-World1.5、FantasyWorld、Lingbot等)在生成视觉上看似合理但实际上是人工虚构的环境。这些模型给定一个起始图像后,所有超出该图像范围的内容——例如未见过的街道几何结构、远处的建筑物——都由模型想象生成。虽然大规模3D重建系统(如NeRF)可以建模真实城市,但它们本质上是静态的,缺乏生成模拟能力。在自动驾驶场景测试、城市规划可视化和基于位置的探索等应用中,需要能够在真实城市环境中进行动态模拟的世界模型,但目前还没有任何世界模拟模型能够锚定到特定的真实世界位置。
本文的目标是本文的目标是实现真实世界接地(real-world grounded)的视频世界模拟:开发一个能够生成空间保真、时间一致、长距离(可达数百米)视频的模型,这些视频基于真实城市的街景图像进行条件生成。具体来说,首尔世界模型(SWM)需要:(1)在给定地理位置坐标、相机动作和文本提示时,检索附近的街景图像并作为条件;(2)解决街景图像与目标场景之间的时序不对齐问题;(3)克服街景数据稀疏性和轨迹多样性不足的问题;(4)在自回归长距离生成中防止误差累积。
与已有工作不同的是,本文的独特切入角度在于将世界模拟模型从想象环境扩展到真实物理世界。虽然视频生成和世界模型领域已经取得了显著进展,但所有现有工作都完全在想象的或合成的环境中操作。SWM的创新在于:(1)利用广泛可用的街景图像作为位置特定的视觉参考,通过检索增强生成将视频生成锚定到真实世界;(2)提出跨时序配对策略,训练模型区分持久结构和瞬态内容;(3)开发视图插值流水线,从稀疏的街景图像合成时间连续的训练视频;(4)提出虚拟前瞻槽机制,通过在生成块中动态更新注意力锚点来解决长距离误差累积问题。
核心方法
SWM的整体思路是通过检索增强条件生成来实现真实世界接地的视频世界模拟。给定用户指定的起始位置、相机运动和文本提示,SWM从地理索引的街景数据库中检索附近的图像,然后将这些图像作为几何和语义条件来指导视频生成。技术路线包括三个关键组件:(1)数据构建——从真实街景和合成城市数据构建训练对,通过跨时序配对解决时序不对齐,通过视图插值解决数据稀疏性;(2)模型架构——基于预训练的DiT进行微调,设计几何参照和语义参照两条条件注入路径;(3)生成策略——提出虚拟前瞻槽机制,在每个生成块中动态更新注意力锚点,防止长距离误差累积。整个系统在440K首尔街景图像、12.7K CARLA合成视频和Waymo驾驶数据上训练,使用AdamW优化器,学习率4.8e-5,在24块H100 GPU上训练10K迭代。
SWM的核心创新点在于三个技术突破:(1)跨时序配对(Cross-temporal Pairing)——在训练时,参考图像和目标序列必须来自不同的时间戳,这迫使模型学习区分持久的场景结构和瞬态的动态内容(如车辆、行人),而不是简单地复制参考图像中的所有内容。这与现有方法的本质区别是:现有方法要么使用同时间的参考(导致模型学习虚假的时序相关性),要么没有专门处理时序不对齐问题。(2)虚拟前瞻槽(Virtual Lookahead Sink)——与传统的注意力槽(使用第一帧作为固定全局上下文)不同,VL Sink在每个生成块中检索未来位置的街景图像,作为虚拟目的地来重新锚定生成。这解决了传统静态锚点随着相机远离起始位置而变得越来越无关的问题。(3)间歇性冻帧策略——解决了3D VAE时间压缩与关键帧条件注入的不匹配问题,确保每个关键帧形成完整的4帧组。
方法步骤详情
SWM的方法分为以下几个步骤:(1)数据构建阶段:收集1.2M首尔全景图像,通过跨时序配对选择参考-目标对,使用视图插值流水线从稀疏关键帧合成连续视频,用Qwen2.5-VL-72B生成文本描述,用Depth Anything V3估计深度图和相机位姿。同时从CARLA模拟器生成12.7K合成视频,覆盖行人、车辆和自由相机轨迹。(2)检索阶段:对于每个目标块,给定目标相机轨迹 $C^{(i)}$,通过最近邻搜索找到候选街景位置,再通过深度重投影过滤保留重投影像素覆盖率超过阈值的参考,得到最多K个针孔参考 $X^{(i)}_{ref}$ 及其相机位姿 $C^{(i)}_{ref}$ 和深度估计 $D^{(i)}_{ref}$。(3)条件注入阶段:几何参照——对每个目标帧,将空间最近的参考通过深度前向投影重投影到目标视角,得到扭曲视频 $X^{(i)}_{warp}$,经3D VAE编码后通道拼接到噪声潜表示;语义参照——将原始参考图像编码为潜表示,沿时间轴拼接到目标潜表示,允许每个目标潜表示关注所有K个参考。(4)虚拟前瞻槽:给定目标轨迹终点,检索最近的街景图像作为虚拟未来目的地,编码为单个潜表示 $z^{(i)}_{VL}$,分配到超出当前生成窗口的RoPE时间位置,时间偏移为 $\Delta_{VL}$。(5)生成阶段:模型自回归地生成视频,每个块条件化于文本提示 $P^{(i)}$、目标相机轨迹 $C^{(i)}$、噪声潜表示 $Z^{(i)}$、历史潜表示 $Z^{(i)}_{hist}$、几何参照 $X^{(i)}_{warp}$、语义参照 $X^{(i)}_{ref}$ 和虚拟前瞻槽 $z^{(i)}_{VL}$。
技术新颖性
SWM的技术新颖性体现在多个层面:(1)任务定义层面——首次提出真实世界接地的世界模拟任务,将世界模型从想象环境扩展到真实城市,这是一个全新的研究方向,现有工作都没有直接探索过。(2)数据构建层面——跨时序配对策略是解决街景图像与生成视频时序不对齐的关键创新。与现有方法(使用同时间参考或忽略时序问题)不同,跨时序配对从训练数据层面强制模型学习区分持久结构和瞬态内容。视图插值流水线中的间歇性冻帧策略解决了3D VAE时间压缩与关键帧条件注入的技术不匹配,这在现有工作中没有被讨论过。(3)模型架构层面——几何参照和语义参照的双路径条件注入设计是新的,将空间布局信息和外观细节信息分别处理,实现互补的条件效果。(4)生成策略层面——虚拟前瞻槽是针对检索增强长距离生成的新机制,与传统注意力槽(使用第一帧作为固定锚点)有本质区别:VL Sink动态更新,使用未来位置的参考,且在训练时通过随机时间偏移让模型学习锚点接近度的影响。(5)评估层面——构建了两个新的城市级评估基准(Busan-City-Bench和Ann-Arbor-City-Bench),并引入了包含跨城市泛化评估的实验设计,这些都是新的评估维度。
实验结果
SWM在两个城市级基准测试中全面超越了现有方法。在Busan-City-Bench上,SWM (TF) 取得FID 28.43(对比次优的HY-World1.5的49.63,提升42.7%)、FVD 301.76(对比544.04,提升44.5%)、mPSNR 14.56(对比11.87,提升22.7%)、mLPIPS 0.392(对比0.588,降低33.3%)。在Ann-Arbor-City-Bench上,SWM (TF) 取得FID 56.61(对比67.02,提升15.5%)、mPSNR 15.18(对比14.26,提升6.5%)、mLPIPS 0.481(对比0.575,降低16.3%)。相机跟随精度方面,SWM在Busan上取得RotErr 0.020和TransErr 0.015,显著优于所有基线(最佳基线RotErr为0.030,TransErr为0.073)。消融实验证明了每个组件的重要性:移除跨时序配对导致Busan FID从28.43恶化到44.74(提升57.4%),证明这是最关键的组件;移除合成数据导致FID轻微改善但相机跟随精度和3D一致性显著下降;移除几何参照导致FID从28.43恶化到33.01;移除语义参照导致FID从28.43恶化到30.27。注意力槽比较实验显示:无槽时FID为33.06,第一帧槽为32.71,第一位置槽为32.41,而VL Sink取得最佳FID 28.43。长期生成实验(1,460帧)进一步验证了VL Sink的优势:FID 25.13(对比无槽的37.37,提升32.8%),证明了其在公里级生成中的稳定性。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 城市级视频世界模拟(Busan-City-Bench) | FID | 28.43 | 49.63 (HY-World1.5) | 42.7%降低(数值越低越好) |
| 城市级视频世界模拟(Busan-City-Bench) | FVD | 301.76 | 544.04 (HY-World1.5) | 44.5%降低(数值越低越好) |
| 城市级视频世界模拟(Busan-City-Bench) | mPSNR (masked) | 14.56 | 11.87 (FantasyWorld) | 22.7%提升(数值越高越好) |
| 城市级视频世界模拟(Busan-City-Bench) | mLPIPS (masked) | 0.392 | 0.588 (HY-World1.5) | 33.3%降低(数值越低越好) |
| 城市级视频世界模拟(Ann-Arbor-City-Bench) | FID | 56.61 | 67.02 (HY-World1.5) | 15.5%降低(数值越低越好) |
| 城市级视频世界模拟(Ann-Arbor-City-Bench) | mPSNR (masked) | 15.18 | 14.26 (HY-World1.5) | 6.5%提升(数值越高越好) |
| 城市级视频世界模拟(Ann-Arbor-City-Bench) | mLPIPS (masked) | 0.481 | 0.575 (HY-World1.5) | 16.3%降低(数值越低越好) |
| 视图插值(Waymo测试集) | PSNR | 25.03 (间歇性冻帧) | 22.52 (通道拼接) | 11.1%提升 |
| 视图插值(Waymo测试集) | LPIPS | 0.162 (间歇性冻帧) | 0.245 (通道拼接) | 33.9%降低 |
| 长期生成(1,460帧扩展基准) | FID | 25.13 (VL Sink) | 37.37 (无注意力槽) | 32.8%降低 |
局限与改进
SWM存在以下几个主要局限性:(1)生成质量依赖训练数据质量——由于城市范围的视频数据不易获得,SWM使用视图插值流水线从稀疏的街景图像序列合成时间连续的视频作为训练监督,但这些插值视频的质量低于真实捕获的视频。随着真实视频数据的可用性增加,引入真实视频数据可以进一步提升生成质量。(2)街景数据捕获模式的限制——街景图像通常在相等距离间隔(而非相等时间间隔)捕获。当捕获车辆减速或停车时,连续街景帧可能跨越较大的时间间隙。尽管作者基于捕获时间元数据进行过滤,但噪声元数据导致一些时间不一致的序列通过过滤器。当这些序列被转换为插值训练视频时,动态物体(如车辆)可能在帧之间突然出现或消失,这种伪影传播到训练模型中,导致偶尔生成车辆突然出现或消失的情况。(3)地理覆盖范围限制——SWM仅在首尔进行训练,尽管在釜山和安娜堡的评估展示了跨城市泛化能力,但模型在这些城市的表现仍弱于首尔。此外,当前模型仅支持已有街景数据覆盖的区域,对于没有街景数据的区域无法生成。(4)计算资源需求——模型需要在24块H100 GPU上训练,自强迫模式在单H100 GPU上推理速度为15.2 fps,对于实时应用可能仍然较慢。(5)文本风格变化的局限性——作者观察到,涉及场景内事件(如洪水、火灾、怪物出现)的文本提示泛化良好,但涉及全局风格变化(如昼夜转换、天气变化)的提示在街景数据微调后往往不能被忠实地跟随。
独立分析的弱点
基于独立分析,SWM存在以下几个值得关注的弱点:(1)视图插值质量瓶颈——虽然间歇性冻帧策略解决了3D VAE压缩的关键帧对齐问题,但插值视频的质量仍然低于真实视频。特别是在动态物体密集的场景中,插值可能导致物体运动不自然。改进方向包括:开发更先进的视频插值模型,或结合光流引导的插值策略来改善动态区域的过渡。(2)跨时序配对的数据效率——跨时序配对要求参考图像和目标序列来自不同时间戳,这可能减少可用的训练数据对数量。改进方向包括:开发更精细的时序过滤策略,或设计半监督学习方法来利用同时间戳的数据。(3)参考数量与质量的权衡——消融实验显示K=1时FID最佳但mPSNR下降,说明参考数量和质量之间存在权衡。改进方向包括:设计自适应参考选择策略,根据场景复杂度动态调整参考数量;或开发参考质量评估机制,优先选择高质量的参考图像。(4)相机轨迹多样性——尽管合成数据覆盖了行人、车辆和自由相机轨迹,但真实街景数据主要是驾驶视角。改进方向包括:收集更多样化的街景数据(如行人视角、无人机视角),或开发更好的域适应技术来弥合不同视角之间的差距。(5)文本控制精度——作者承认涉及全局风格变化的文本提示(如昼夜转换)在微调后效果下降。改进方向包括:设计分层文本控制机制,分别控制局部场景变化和全局风格变化;或采用更精细的微调策略来保持预训练模型的风格控制能力。
未来方向
作者提出的未来研究方向和基于本文成果可延伸的方向包括:(1)扩展到更多城市——当前SWM仅在首尔训练,但展示了跨城市泛化能力。未来工作可以构建覆盖全球主要城市的世界模型,或开发零样本/少样本的城市适应方法。(2)引入更多模态——当前SWM使用文本提示和相机轨迹作为条件,未来可以引入更多模态如3D场景图、语义分割图、深度图等,提供更丰富的场景控制。(3)实时交互式生成——当前自强迫模式的速度为15.2 fps,未来可以通过模型压缩、量化或蒸馏技术提高推理速度,实现真正的实时交互式城市探索。(4)结合大语言模型——将SWM与LLM结合,实现更智能的场景理解和生成控制,例如根据用户的自然语言描述自动规划相机轨迹和生成场景。(5)多智能体模拟——扩展SWM支持多智能体交互模拟,如模拟多个自动驾驶车辆在城市中的交互,或模拟行人群体的行为。(6)物理一致性增强——引入物理引擎或物理约束,确保生成的视频遵守物理规律,如车辆运动的物理约束、行人步态的生物力学约束等。(7)长期记忆与导航——结合空间记忆机制,实现长距离导航任务,如从A点到B点的自主探索和视频生成。(8)应用开发——基于SWM开发实际应用,如城市规划可视化工具、自动驾驶场景生成平台、沉浸式城市探索体验等。
复现评估
从复现角度来看,SWM的复现面临以下挑战:(1)开源情况——论文提供了项目页面(https://seoul-world-model.github.io),但未明确说明代码和模型权重是否开源。数据集方面,真实街景数据来自NAVER Map,需要相应的数据访问权限;合成数据使用CARLA模拟器生成,可以公开获取;Waymo数据集是公开可用的。(2)数据准备——SWM的训练数据构建流程复杂,包括1.2M全景图像的收集和处理、跨时序配对的选择、视图插值视频的合成、文本描述的生成(使用Qwen2.5-VL-72B)、深度和相机位姿估计(使用Depth Anything V3)等多个步骤。这些步骤需要大量的计算资源和数据处理能力。(3)算力需求——模型在24块H100 GPU上训练10K迭代,这是非常高的算力需求。对于大多数研究机构来说,这可能是复现的主要障碍。(4)环境依赖——SWM依赖多个预训练模型(Cosmos-Predict2.5-2B、Depth Anything V3、Qwen2.5-VL-72B),这些模型的版本和配置可能影响复现结果。(5)评估基准——作者构建了两个新的评估基准(Busan-City-Bench和Ann-Arbor-City-Bench),但这些基准的构建细节和数据是否公开尚不明确。总体而言,SWM的复现难度较高,主要障碍在于算力需求和数据准备的复杂性。如果作者能够开源代码、模型权重和评估基准,将大大降低复现难度。
论文图表
该图展示了SWM的核心能力:在首尔地图上放置一条相机轨迹,生成对应位置的连续动态视频。左侧是首尔城市地图,标示了轨迹起点(0 km)和终点(1.8 km和1.2 km);右侧是生成的视频帧,展示了真实的城市环境。特别值得注意的是,用户可以通过文本提示来改变场景,如“点燃汽车”和“让哥斯拉出现在摩天大楼之间”,而模型仍然保持对真实城市布局的接地。
这张图是论文的核心展示,直观地传达了SWM的主要贡献:在真实城市中进行长距离视频生成,并支持文本提示控制场景变化。它展示了SWM与现有想象环境世界模型的本质区别。
该图对比了两种注意力槽设计:(a) 传统注意力槽——使用第一帧作为固定全局上下文,随着相机远离起始位置,锚点的指导作用减弱。(b) 虚拟前瞻槽(本文方法)——在每个生成块中动态检索未来位置的街景图像作为虚拟目的地,持续重新锚定生成。这种方法使得锚点始终与当前生成区域相关,解决了传统静态锚点的局限性。
这张图展示了SWM的关键创新之一——虚拟前瞻槽机制。理解传统注意力槽的局限性和VL Sink的解决方案是理解本文长期生成能力的基础。
该图展示了不同注意力槽策略在长距离生成中的性能变化:使用200帧滑动窗口计算FID,比较无槽、第一帧槽、第一位置槽和VL Sink。VL Sink在整个生成过程中保持最低的FID和最慢的性能下降,证明了其在长距离生成中的优势。
这张图量化地展示了不同注意力槽策略的长期性能,是验证VL Sink有效性的关键证据。滑动窗口FID是衡量长距离生成质量的重要指标。
该图展示了SWM的典型失败模式:车辆偶尔会突然出现或消失。这是由于训练数据中存在时间不一致的街景序列(元数据噪声导致),这些序列被转换为训练视频时产生的伪影传播到模型中。
这张图诚实地展示了SWM的局限性,帮助读者理解模型的失败模式和潜在改进方向。