← 返回 2026-04-09

一帧值一令牌:基于增量令牌的高效生成式世界建模 A Frame is Worth One Token: Efficient Generative World Modeling with Delta Tokens

Tommie Kerssies, Gabriele Berton, Ju He, Qihang Yu, Wufei Ma, Daan de Geus, Gijs Dubbelman, Liang-Chieh Chen 📅 2026-04-06 👍 12 2026-07-13 08:36
世界模型 生成式模型 自动驾驶 自监督学习 视觉基础模型 视频预测 语义预测

单令牌编码帧间差分,35×参数、2000×FLOPs缩减的生成式世界模型。

前置知识

世界模型 (World Model)

世界模型是预测环境未来状态的生成式模型,给定过去观察(可能含动作)即可推断未来可能的状态分布。它在自动驾驶、机器人、游戏AI中扮演「决策模拟器」的角色,使智能体能「想象」未来并据此规划安全动作。区分判别式与生成式两类:判别式只能给出唯一确定性预测,生成式能输出多种可能未来。

本文核心是构建一个生成式世界模型。理解「为何需要生成式」(未来不确定,需推理多种可能)以及「为何现有生成式昂贵」(像素级建模+多步去噪)是抓住本文动机的关键。

视觉基础模型 (Vision Foundation Model, VFM)

VFM指在大规模数据上自监督或弱监督预训练的通用视觉模型(如DINOv3、CLIP),其特征空间富含语义结构。不同于为像素重建优化的潜空间,VFM特征天然适合下游理解任务。本文使用DINOv3 ViT-B作为冻结特征提取器,patch大小16×16,输出14×14=196个patch令牌(256×256输入)。

本文所有创新都建立在「VFM特征空间比像素空间更适合决策」这一前提上。读懂本文需要理解为何在VFM特征空间预测比在像素空间重建更高效。

最佳多选 (Best-of-Many, BoM) 训练目标

BoM是一种简单生成式训练策略:从高斯分布采样K个噪声查询,每个查询生成一个未来预测,只对「与真值最接近的那个预测」计算损失并反向传播。这鼓励模型把不同随机输入映射到不同合理未来,从而在单次前向传播中实现多样化采样,无需迭代去噪。

BoM是本文把判别式模型变为生成式模型的关键机制。需理解其与传统扩散模型的根本区别:BoM无显式分布目标,靠「赢者通吃」机制隐式学习多样本映射。

Vision Transformer (ViT)

ViT将图像划分为固定大小的patch(如16×16),把每个patch线性映射为令牌后送入标准Transformer自注意力层。本文的DeltaTok编码器和解码器都是ViT-B配置(12层、12头、768维),改造自HuggingFace DINOv3实现,但去掉了patch embedding层(因输入已是VFM令牌)。

理解ViT的patch令牌概念(每帧对应H×W个空间令牌)才能体会本文「用单个令牌替代H×W个令牌」的压缩比含义(如512×512输入下1024×压缩)。

帧间差分编码 (Inter-frame Delta Coding)

经典视频压缩(如H.264/AVC)的核心技术:连续帧大部分内容静止,只需编码「差异」而非全帧。本文的DeltaTok将这一思想迁移到VFM特征空间:编码器输入xt-1和xt,输出一个「增量令牌」zt,解码器用xt-1+zt重建xt。当场景近静态时,zt接近零向量。

DeltaTok的命名和核心思路都来自这一传统。但需区分:传统差分是逐像素运动向量,DeltaTok是单个语义令牌向量,可处理遮挡和新物体。

研究动机

现有生成式世界模型在自动驾驶等需要「推理多种可能未来」的场景中面临三个核心效率瓶颈。第一,它们的表示空间是为像素重建优化的潜空间(如Cosmos用7B扩散解码器从512×1024像素生成),建模大量与决策无关的视觉细节(如背景树叶渲染);第二,离散自回归模型和连续扩散模型都需多次前向传播才能生成单个未来假设(扩散需数十次去噪迭代);第三,它们未利用连续帧之间的时空冗余——背景静态区域在前后帧完全一致,却被重复编码。论文Table 3清晰展示这一代价:Cosmos-12B生成单个未来需6.4×10^7 GFLOPs,是判别式DINO-world(5.8×10^3 GFLOPs)的11000倍,但预测质量却低于更小的DINO-world(Cityscapes mid-horizon mIoU 49.1/53.3 vs 49.8)。更糟的是判别式模型在多模态未来下会「坍缩到条件均值」(Walker et al. 2016的经典结论),无法表达行人可能左转或右转等离散事件,因此无法支撑安全决策。

本文的目标是本文的具体目标是构建一个既「能高效生成多种合理未来」又「在参数和计算量上比现有生成式世界模型小数量级」的通用生成式世界模型。具体可量化的目标包括:1)参数量比Cosmos-12B少35×以上;2)FLOPs比Cosmos-12B少2000×以上(对应Figure 2的柱状对比:31 TFLOPs vs 60,000/64,000 TFLOPs);3)在dense forecasting benchmark上,best-of-20预测质量在VSPW、Cityscapes、KITTI三大数据集六个指标上全部超过Cosmos;4)mean预测质量回退到判别式基线水平(避免退化为噪声)。

与已有工作不同的是,本文的独特切入角度在于「VFM特征空间中的帧间差分压缩」。与已有工作相比有三个本质差异:1)相比在像素潜空间做生成的Cosmos、GAIA-1等工作,本文直接在冻结DINOv3特征空间预测,绕开像素重建;2)相比同样在VFM空间工作的DINO-world、DINO-Foresight等判别式模型,本文通过BoM目标扩展为生成式,单次前向可出多样本;3)核心创新是DeltaTok——传统视频tokenizer(如RefTok)训练重建像素且保持空间结构,本文反其道而行:扔掉空间维度,把「如何把xt-1变成xt」这个变化量压缩成单个一维令牌,让视频从3D时空表示退化为1D时序序列,把BoM的K次多样性搜索直接放在令牌空间内执行(无需解码完整特征图)。这一思路借鉴了H.264的运动补偿思想,但用单令牌替代逐像素运动向量,自然处理遮挡和场景切换。

核心方法

DeltaWorld的整体思路是「用最少的令牌表达最大的未来信息」,分三层递进实现。直觉上,你可以把世界想象成「大部分时间不变,偶尔变化」的视频流,与其预测下一帧的完整画面(256个空间令牌),不如只预测「和上一帧相比发生了什么变化」(1个增量令牌)。技术上由三个组件协作:1)冻结的DINOv3 ViT-B作为特征提取器,把每帧编码为14×14×768的空间令牌网格;2)单独训练的DeltaTok tokenizer,把(xt-1, xt)压缩成单个delta令牌zt,解码时从xt-1恢复xt;3)预测器f,以前t个delta令牌为上下文,结合K个随机查询q^k,输出下一个delta令牌。BoM训练时只监督「最接近真值」的那条轨迹,让不同查询自然对应不同未来;推理时一次前向即得K个不同delta令牌序列,再分别解码回特征图。输入序列前置一个黑帧让z1自然编码第一帧的绝对特征。这种「在令牌空间做生成、只在需要时解码回空间」的设计把预测器开销压到总FLOPs的0.5%(Table B:预测器0.26 GFLOPs vs 解码器46.12 GFLOPs),且预测器序列长度与H×W无关。

DeltaWorld和已有生成式世界模型有四个本质区别。第一,表示层面:Cosmos等用7B扩散解码器在像素潜空间生成,DeltaWorld在冻结VFM特征空间预测,像素解码可后处理也可省略;第二,生成机制:扩散模型需数十次迭代去噪生成单样本,DeltaWorld用BoM+随机查询实现单次前向多样本;第三,token化粒度:传统视频tokenizer保持H×W个空间令牌(如16×16=256个)逐patch表示一帧,DeltaTok把整个时空体压缩成1D时序序列,每帧仅1个令牌(在512×512下等价1024×压缩);第四,监督信号:判别式世界模型监督唯一确定预测,BoM+delta机制把「无变化」作为天然先验(预测零向量即等于「复制上一帧」),使模型只需学习「哪里变了」,大幅降低学习难度。这种「变化量先验」正是论文Table 2中step(3) mean mIoU恢复到判别式基线水平(Cityscapes 45.5 vs 45.4)的根本原因。

方法步骤详情

实现分四步。第一步VFM特征提取:用冻结DINOv3 ViT-B把t帧视频编码为X1:t=(x1,...,xt),其中xi∈R^(H×W×D)为patch令牌网格,D=768。第二步DeltaTok tokenizer训练:编码器g输入(xt-1,xt,zinit)输出单令牌zt∈R^D,解码器h从(xt-1,zt)重建xt̂,两者是ViT-B Transformer,用MSE损失L_tok=||xt-xt̂||^2训练50K迭代。第三步预测器训练:f用ViT-B配置,对每个查询q^k(从N(0, 0.022I)采样)输出ẑt+1=f(q^k, Z1:t, T1:t, τt+1)。BoM损失L_BoM=Σ(h,w) ℓ(x_{t+1,h,w}, x̂^k*_{t+1,h,w}),k*=argmin_k Σ(h,w)ℓ(...),仅最优预测反向传播。K=256(主实验),训练300K迭代,batch=1024,序列长度8帧,∆τ∈[1/25,1/3]秒。第四步推理与解码:生成20条delta序列,每条用不同查询自回归展开3步,用DeltaTok解码器从xt恢复xt+1特征图,最后用DINOv3特征上训练的线性分割/深度头输出。关键工程:3D RoPE简化为1D旋转(前60维),block-causal mask简化为标准因果mask,每帧仅1令牌无需空间注意力。

技术新颖性

技术新颖性体现在三方面。第一,delta token化是首个把帧间差分思想与VFM特征空间结合并极端化(单令牌/帧)的工作。传统motion-residual框架(如光流RAFT、Video RefTok)保留空间维度输出H×W个运动向量,DeltaTok扔掉空间维度输出单个语义令牌,因此能优雅处理遮挡/新物体(warp-based方法在这些场景失效),且可通过预测零向量退化为「绝对编码」(传统差分无此能力)。第二,BoM与单令牌空间的耦合是优雅的:传统BoM用于H×W空间特征图时,K次采样意味着K×H×W个输出,计算昂贵;DeltaTok让BoM在单令牌空间执行,K次采样只多K×1个输出,预测器开销从97%降到0.5%。第三,「无变化=零令牌」的天然先验是论文发现的隐式正则化:Table 2中frame compression(step 2)因单令牌容量不足而预测质量下降,delta compression(step 3)却因「只需编码变化」反而恢复并超越基线,这意味着delta先验有效抵消了单令牌容量瓶颈。第四,跨架构迁移性:在DINO-Foresight(PCA特征+factorized space-time attention)中替换为delta token,token数减少2048×但性能持平(Table D),证明delta token是一般性方法而非过拟合某个架构。

Outline of DeltaWorld.
Figure 1: Outline of DeltaWorld.
Overview of DeltaTok.
Figure 3: Overview of DeltaTok.
Overview of DeltaWorld.
Figure 4: Overview of DeltaWorld.

实验结果

Table 3在512×512下,DeltaWorld全面领先Cosmos。VSPW短程mIoU 55.4(best)/53.7(mean) vs Cosmos-12B 51.7/50.7;VSPW中程50.1/46.7 vs 47.7/45.5;Cityscapes短程65.8/63.9 vs 55.3/56.0(+10.5);Cityscapes中程55.4/51.3 vs 53.3/51.2;KITTI短程RMSE 3.00/3.17 vs 3.72/3.71;KITTI中程3.88/4.17 vs 4.01/4.14。效率上,DeltaWorld-0.3B需3.1×10^4 GFLOPs(20样本),比Cosmos-12B(6.4×10^7)少2065×,参数少40×。Table 2消融揭示各组件贡献:step(0)判别式基线44.8/45.4;step(1)加BoM后best升到47.0/46.8但mean崩到39.4/31.1;step(2)加frame compression让时间缩到0.2×,mean恢复40.3/35.5仍低于基线;step(3)再加delta compression,best达46.8/48.7,mean恢复到44.4/45.5几乎追平基线,证明delta先验是修复mean崩塌的关键。

Evaluation datasets.
Table 1: Evaluation datasets.
Towards an efficient generative world model.
Table 2: Towards an efficient generative world model.
Dense forecasting.
Table 3: Dense forecasting.
Performance comparison.
Figure 2: Performance comparison.
Best-of-Many sample scaling.
Figure 5: Best-of-Many sample scaling.
Diverse sampled futures.
Figure 6: Diverse sampled futures.
查看结构化数据
任务指标本文基线提升
VSPW语义分割 (短程~0.2s) mIoU ↑ (best-of-20 / mean) 55.4 / 53.7 Cosmos-12B: 51.7/50.7;DINO-world: 54.0 (single) best +3.7,mean +3.0 (vs Cosmos-12B)
VSPW语义分割 (中程~0.6s, 3步自回归) mIoU ↑ (best-of-20 / mean) 50.1 / 46.7 Cosmos-12B: 47.7/45.5;DINO-world: 47.9 (single) best +2.4,mean +1.2 (vs Cosmos-12B)
Cityscapes语义分割 (短程~0.2s) mIoU ↑ (best-of-20 / mean) 65.8 / 63.9 Cosmos-12B: 55.3/56.0;DINO-world: 62.0 (single) best +10.5,mean +7.9 (vs Cosmos-12B)
Cityscapes语义分割 (中程~0.6s) mIoU ↑ (best-of-20 / mean) 55.4 / 51.3 Cosmos-12B: 53.3/51.2;DINO-world: 49.8 (single) best +2.1,mean +0.1 (vs Cosmos-12B)
KITTI单目深度估计 (短程~0.2s) RMSE ↓ (best-of-20 / mean) 3.00 / 3.17 Cosmos-12B: 3.72/3.71;DINO-world: 3.16 (single) best -0.72,mean -0.54 (vs Cosmos-12B)
KITTI单目深度估计 (中程~0.6s) RMSE ↓ (best-of-20 / mean) 3.88 / 4.17 Cosmos-12B: 4.01/4.14;DINO-world: 4.07 (single) best -0.13,mean +0.03 (基本持平)
计算效率 GFLOPs ↓ (生成20样本) / 参数量 3.1×10^4 GFLOPs / 0.3B参数 Cosmos-12B: 6.4×10^7 GFLOPs / 12B参数;Cosmos-4B: 6.0×10^7 / 4B参数 FLOPs 2065×更少,参数 40×更少

局限与改进

作者在Appendix D承认两大局限。第一,分布建模无理论保证:BoM无显式去噪目标对应数据分布,样本覆盖度受训练K限制,查询空间利用无机制保证,近似真实后验无理论保证。作者观察到查询可被解读为「隐式动作条件」但缺证明。第二,误差累积:训练时tokenizer并行解码(基于真实xt-1),推理时串行展开,每步xt+1̂作为下一步条件输入,tokenizer重建误差跨步累积;自回归rollout中预测器误差同样放大。作者提议用tokenizer在训练时对自己重建帧算delta来缓解。其他隐含局限:1)DINOv3 ViT-B特征空间设上限,Cityscapes mid-horizon mean 51.3仍比Present上界低19.2点;2)4帧+3步评估可能低估真实部署误差;3)K=256在大batch下需仔细调参;4)评估仅限三种密集预测任务,未在闭环规划中验证。

独立分析的弱点

独立分析有五点。第一,VFM特征依赖:所有性能建立在DINOv3 ViT-B足够丰富这一假设上,Table D迁移实验未探索其他VFM(CLIP、SigLIP、DINOv3 ViT-L等)的扩展规律,对需细粒度空间信息的下游任务(光流、检测框)单令牌可能容量不足。第二,K选择敏感:Figure 5显示K=1→1024 best单调上升(47.0→50.7)但作者未给出选择指南,实际部署需场景化调优。第三,两阶段训练非最优:tokenizer用MSE独立训练缺乏预测任务梯度,可能保留对预测无用的重建信息,可参考Diffusion Transformers with RAE思路端到端联合训练。第四,评估协议边界:best-of-20是「20次尝试选最优」的乐观估计,应补充FID/FVD等无条件样本质量指标。第五,BoM「赢家通吃」梯度稀疏:K=1024仅1个查询收到梯度,可能限制大规模K下的训练稳定性,需调研重要性加权或Gumbel-Softmax等替代。

未来方向

作者在结论中明确指出未来工作方向:「scaling predictor size, context length, and rollout depth」。基于本文成果可延伸的具体方向包括:第一,闭环规划验证:在自动驾驶模拟器中接入DeltaWorld作为世界模型,验证多样本未来是否能提升规划安全性(碰撞率、轨迹合理性),这才是生成式世界模型的根本价值。第二,tokenizer自重建训练以缓解误差累积,让tokenizer在训练时模拟推理时的串行解码。第三,扩散+delta混合:保留BoM的高效性,但用小规模扩散头在delta token空间做精修,可能获得分布建模的理论保证和更好的样本多样性。第四,长上下文+长程预测:本文只评测4帧上下文+3步rollout,未来可探索16+帧+10+步,并研究「增量先验+长程记忆」如何结合(如偶尔插入「重置令牌」绝对编码关键帧)。第五,多模态条件生成:把查询从纯高斯噪声扩展为可解释的「意图向量」(如转向角、目标点),实现action-conditional world model。第六,跨VFM的delta token研究:系统比较DINOv2、CLIP、SigLIP等不同VFM下delta token的压缩比-质量权衡。第七,把delta token用于视频理解而非生成(如动作识别、视频问答),验证其作为通用视频表示的潜力。

复现评估

可复现性整体良好但有缺口。代码与权重已开源:deltatok.github.io;主架构基于公开DINOv3 ViT-B和HuggingFace Transformers,tokenizer是标准ViT-B连续自编码器,BoM目标代码简单透明。训练数据用约4M视频样本(11秒@16FPS),DINO-world官方数据未公开,论文用相同数据训练自己的DINO-world†作基线。算力方面,Table 2显示训练用8×H200 GPU节点+torch.compile+BF16混合精度;tokenizer 50K迭代约数小时;主实验300K迭代×512×512需数天到一周。关键超参(K=256, batch=1024, lr=1e-4 with 5K warmup, ∆τ~U[1/25,1/3]秒, 序列长度8)已详列。复现难度中等:熟悉PyTorch的研究者约2-3周可跑通主表,但完整对比Cosmos需访问其官方代码(受限padding+9帧固定上下文)和大量GPU资源。