← 返回 2026-03-20

生成模型懂空间:释放隐式三维先验用于场景理解 Generation Models Know Space: Unleashing Implicit 3D Priors for Scene Understanding

Xianjin Wu, Dingkang Liang, Tianrui Feng, Kui Xia, Yumeng Zhang, Xiaofan Li, Xiao Tan, Xiang Bai 📅 2026-03-19 👍 95 2026-07-13 08:36
3D场景理解 多模态大语言模型 扩散模型 空间推理 视频生成模型

从视频生成模型提取3D先验增强MLLM空间理解

前置知识

Multimodal Large Language Model (MLLM)

多模态大语言模型是将视觉编码器(如SigLIP)与大语言模型结合的架构,能够同时处理图像/视频和文本输入。其工作流程是:首先用视觉编码器提取图像特征,通过投影层映射到LLM的嵌入空间,然后与文本token一起输入LLM进行推理。标准训练目标是最大化响应token序列的似然:$\mathcal{L}_{\mathrm{CE}}(\Theta) = -\sum_{i=1}^{L} \log p_{\Theta}(y_i \mid y_{<i}, \mathbf{x}, \mathbf{v})$。然而这种监督信号是稀疏离散的,空间误差(如预测'左'vs'右')仅被视为普通的token不匹配,缺乏几何度量约束。

本文的核心目标是改进MLLM的空间理解能力,理解MLLM的工作原理和局限性是理解本文动机的基础。

Video Diffusion Model (视频扩散模型)

视频扩散模型是一类基于扩散过程的视频生成模型,如Wan2.1、Sora等。现代视频生成器采用Diffusion Transformer (DiT)架构,通过Flow Matching训练:给定干净潜在视频 $\mathbf{z}_0$,采样高斯噪声 $\boldsymbol{\epsilon} \sim \mathcal{N}(0, I)$ 和时间 $t \sim \mathcal{U}(0,1)$,训练流网络 $v_\psi(\cdot)$ 回归目标速度:$\mathcal{L}_{\mathrm{FM}}(\psi) = \mathbb{E}_{\mathbf{z}_0, \boldsymbol{\epsilon}, t} \|\mathbf{u}_t - v_\psi(\mathbf{z}_t, t, \mathbf{c})\|^2$。关键洞察是:为了生成时间连贯的视频,这些模型必须隐式学习3D结构先验和物理规律。

本文的核心创新是将视频生成模型作为'潜在世界模拟器'来提取3D先验,理解其工作原理是理解方法的关键。

DiT (Diffusion Transformer)

Diffusion Transformer是将Transformer架构应用于扩散模型的架构,替代了传统的UNet结构。DiT利用全局注意力机制捕获整体上下文,在视频生成中展现出优异的多视角一致性(大于96%对应分数)。相比UNet架构的局部归纳偏置和有限感受野,DiT的全局注意力能更好地保持长程几何依赖关系。实验表明,基于DiT的模型(如Wan2.1)在3D理解任务上显著优于基于UNet的模型。

论文通过实验发现DiT架构比UNet架构提供更好的3D先验,这是选择生成模型的重要依据。

Multi-view Correspondence Score (多视角对应分数)

这是论文引入的量化指标,用于评估不同特征骨干的几何完整性。给定一个从V个视角观察的3D场景,使用真值相机外参和深度将各视角的编码器特征 $F_v$ 投影到共享的全局体素网格中。对于在两个不同视角 $v_i$ 和 $v_j$ 中观察到的特定体素 $m$,提取对应特征向量 $\mathbf{h}_{m,v_i}$ 和 $\mathbf{h}_{m,v_j}$,一致性分数定义为余弦相似度:$S_{\mathrm{voxel}}^{(m)} = \frac{\mathbf{h}_{m,v_i}^\top \mathbf{h}_{m,v_j}}{\|\mathbf{h}_{m,v_i}\| \|\mathbf{h}_{m,v_j}\|}$。更高分数表示模型隐式地对齐了同一3D结构的不同视角。

这个指标是论文验证生成模型具有3D先验的关键证据,也是选择DiT架构的实验依据。

Adaptive Gated Fusion (自适应门控融合)

这是论文提出的核心融合机制,用于弥合生成特征和语义特征之间的分布差异。对于第 $i$ 个空间token $\mathbf{F}_i$,计算标量门控 $g_i \in [0,1]$:$g_i = \sigma(\mathbf{W}_g^\top \mathrm{Concat}[\mathrm{LN}(\mathbf{F}_{\mathrm{gen},i}), \mathrm{LN}(\mathbf{F}_{\mathrm{sem},i})] + b_g)$,其中 $\sigma$ 是sigmoid函数,LN是层归一化。最终融合表示为凸组合:$\mathbf{F}_i^{\mathrm{fused}} = (1-g_i) \cdot \mathbf{F}_{\mathrm{gen},i} + g_i \cdot \mathbf{F}_{\mathrm{sem},i}$。这个门控充当语义-几何仲裁器,对识别任务优先语义先验,对空间推理任务动态转向生成世界知识。

这是将生成先验与语义特征整合的关键技术创新,直接决定了方法的有效性。

研究动机

多模态大语言模型(MLLM)在语义理解方面表现出色,但存在严重的'空间盲视'问题,在细粒度几何推理和物理动态理解方面表现不佳。标准的交叉熵损失在词汇空间中计算,将空间误差(如预测'左'vs'右')视为普通的token不匹配,缺乏几何度量约束,导致判别编码器往往关注语义存在而非精确的空间结构。现有解决方案主要依赖两种范式:第一种直接利用显式3D模态(如点云或深度),但受限于高质量3D数据的稀缺性;第二种采用几何脚手架方法,通过额外的重建或蒸馏将2D特征提升到3D空间,但需要复杂的多阶段训练流程或特定任务的几何标注(如深度、相机位姿)。这些方法在数据稀缺性和泛化能力方面面临根本性挑战。

本文的目标是本文的目标是探索一种全新范式:利用大规模视频生成模型学到的隐式空间先验来增强MLLM的3D理解能力。具体而言,论文希望验证现代视频生成器在合成时间连贯视频的过程中,是否隐式地学习了鲁棒的3D结构先验和物理规律,并将这些先验提取出来用于下游3D视觉理解任务,从而在不需要显式3D监督的情况下实现优越的性能。

与已有工作不同的是,与以往方法的根本区别在于,本文不再依赖显式3D模态或复杂几何监督,而是发现了一个被忽视但日益有前景的范式:现代视频生成模型(如Wan2.1)在大规模视频数据上训练,其训练目标隐式地奖励与3D几何和物理动态一致的表示。论文的关键洞察是:视频扩散模型在不同帧之间展现出惊人的多视角一致性,模型捕获了物体在不同视角下的结构完整性,这意味着其内部存在鲁棒的3D几何表示。通过将这些模型重新定位为'潜在世界模拟器'(Latent World Simulator),可以提取隐式几何表示作为补充特征流,与语义编码器集成以改进判别式3D理解。

核心方法

VEGA-3D的整体思路是将视频生成模型重新定位为'潜在世界模拟器',提取其隐式学习的3D空间先验,并通过自适应门控融合机制与语义特征整合。直觉上,视频生成模型为了产生时间连贯的视频,必须在内部建立对3D结构和物理规律的理解——遮挡需要保持物体身份一致性,相机运动揭示深度相关的表观运动,交互必须遵循一致的动力学。这些约束促使潜在表示编码几何一致的结构和运动,从而在没有显式3D监督的情况下产生强的学习3D先验。技术路线是双分支视觉编码机制:一个分支使用判别编码器(如SigLIP)提取高级语义特征,另一个分支使用冻结的视频生成模型(如Wan2.1-T2V 1.3B)通过噪声注入激活其3D感知能力,然后通过token级自适应门控融合将两个异构特征流整合。

核心创新点有两个层面。第一,发现并验证了视频生成模型的隐式3D先验:论文引入多视角对应分数指标,通过ScanNet数据集定量验证了不同特征骨干的几何完整性。实验发现DiT架构(如Wan2.1)的全局注意力机制能够捕获整体上下文,实现大于96%的对应分数,显著优于基于UNet的模型(如SVD、Stable Diffusion)。更重要的是,论文揭示了多视角一致性与下游3D理解性能之间的强正相关关系。第二,提出噪声注入激活策略:静态潜在表示 $\mathbf{z}_0$ 不足以完全激活生成模型的推理能力,扩散模型主要在对损坏信号进行活跃去噪期间强制结构连贯性。因此,论文沿Flow Matching加噪路径扰动干净潜在:$\mathbf{z}_k = (1-t_k)\mathbf{z}_0 + t_k \boldsymbol{\epsilon}$,使用空文本提示确保激活的特征仅依赖视觉信号和模型学到的物理规律,最小化语义幻觉。这与以往方法的本质区别在于:不需要显式3D监督或复杂渲染流程,而是释放生成模型中已有的潜在物理先验。

方法步骤详情

方法分为三个逻辑阶段。第一阶段是3D感知分析(3D Awareness Analysis):在ScanNet测试集上,使用真值相机外参和深度将编码器特征投影到共享体素网格,计算多视角对应分数,验证生成模型的几何能力,并发现对应分数与下游性能的正相关关系。第二阶段是潜在世界模拟(Latent World Simulation):给定输入视频序列 $\mathbf{V} \in \mathbb{R}^{T \times H \times W \times 3}$,首先通过VAE映射到低维潜在空间 $\mathbf{z}_0 = \mathcal{E}(\mathbf{V})$;然后选择离散时间步索引 $k$(默认 $k=300$,即 $t_k=0.3$),采样噪声 $\boldsymbol{\epsilon} \sim \mathcal{N}(0, I)$,构造 $\mathbf{z}_k = (1-t_k)\mathbf{z}_0 + t_k \boldsymbol{\epsilon}$;将 $\mathbf{z}_k$ 输入骨干网络 $\Phi(\cdot)$,使用空文本提示 $c_{\text{text}} = ""$,从特定中间DiT层 $l$(默认第20层)提取特征 $\mathbf{f}_{\mathrm{gen}} = \Phi^{(l)}(\mathbf{z}_k, k; c_{\text{text}} = "")$;经过自适应平均池化匹配语义token化,得到生成表示 $\mathbf{f}_{\mathrm{gen}} \in \mathbb{R}^{T \times N \times D_{\mathrm{gen}}}$。第三阶段是弥合生成-语义鸿沟(Bridging the Generative and Semantic Gap):通过独立的MLP投影器 $P_{\mathrm{gen}}$ 和 $P_{\mathrm{sem}}$ 将两个流投影到LLM的隐藏维度 $D_{\mathrm{llm}}$;然后应用自适应门控融合机制,对每个空间token计算门控值 $g_i$,生成最终融合表示 $\mathbf{F}_i^{\mathrm{fused}} = (1-g_i) \cdot \mathbf{F}_{\mathrm{gen},i} + g_i \cdot \mathbf{F}_{\mathrm{sem},i}$。

技术新颖性

技术新颖性体现在多个方面。首先,这是首次系统性地探索将视频生成模型的隐式3D先验用于判别式3D理解任务的框架,开创了一个全新的研究范式。论文不仅提出了方法,还通过多视角对应分数指标提供了定量证据,建立了生成模型几何能力的评估标准。其次,噪声注入激活策略具有独创性:论文发现静态潜在表示无法充分利用生成模型的推理能力,必须通过沿Flow Matching路径的噪声扰动来激活其去噪过程中的结构理解能力。这一发现对理解扩散模型的内部表示具有重要启示。第三,token级自适应门控融合机制设计精巧:它不是简单的特征平均或拼接,而是为每个token位置学习独立的门控权重,充当语义-几何仲裁器,能够根据任务需求动态调整两种先验的贡献比例。第四,框架的即插即用特性使其能够无缝集成到不同的基线架构(Video-3D LLM、Qwen2.5VL-7B、OpenVLA-OFT),展示了广泛的适用性。

隐式3D先验的可视化
Fig. 2: 隐式3D先验的可视化
VEGA-3D框架概览
Fig. 4: VEGA-3D框架概览
自适应门控融合
Fig. 5: 自适应门控融合
噪声注入和DiT深度的消融研究
Fig. 6: 噪声注入和DiT深度的消融研究

实验结果

论文在三个代表性评估轴上进行了广泛实验。在3D场景理解方面,VEGA-3D在Video-3D LLM基线上持续改进,特别是在定位中心任务上表现突出:ScanRefer Acc@0.5从51.7提升到56.2(+4.5),SQA3D EM从58.6提升到61.3(+2.7),ScanQA CIDEr从102.1提升到106.3(+4.2)。这些增益表明隐式3D结构感知作为鲁棒空间锚点,有效减少了标准MLLM的空间模糊性。在空间推理方面,VEGA-3D在VSI-Bench上实现了50.5%的平均准确率,在路由规划(Route Planning)子任务上达到60.5%(比基线提升2.7%)。在机器人操作方面,VEGA-3D在LIBERO基准上实现97.3%的平均成功率,超越了高度饱和的OpenVLA-OFT基线(97.0%),特别是在长期任务上从94.4%提升到95.2%。消融研究揭示了几个关键发现:(1) 生成特征单独使用会导致性能大幅下降(Scan2Cap CIDEr从83.8降至25.2),确认生成先验是补充而非替代语义表示;(2) 性能在中间噪声水平($t_k=0.3$)和中间DiT层(第20层)达到峰值,表明最丰富的空间线索来自中间表示和中等去噪阶段;(3) 多视角一致性与下游性能存在强正相关关系,验证了对应分数作为3D能力预测指标的有效性。

3D场景理解基准的性能对比
Table 1: 3D场景理解基准的性能对比
VSI-Bench空间推理基准的对比
Table 2: VSI-Bench空间推理基准的对比
LIBERO机器人操作基准的对比
Table 3: LIBERO机器人操作基准的对比
不同判别和生成基础模型的实验
Table 4: 不同判别和生成基础模型的实验
不同特征融合模块的消融研究
Table 5: 不同特征融合模块的消融研究
特征分析
Fig. 3: 特征分析
推理开销分析
Fig. 7: 推理开销分析
查看结构化数据
任务指标本文基线提升
3D场景理解 - ScanRefer Acc@0.25 / Acc@0.5 63.2 / 56.2 58.1 / 51.7 (Video-3D LLM) +5.1 / +4.5
3D场景理解 - Multi3DRefer F1@0.25 / F1@0.5 60.8 / 55.1 58.0 / 52.7 (Video-3D LLM) +2.8 / +2.4
3D场景理解 - Scan2Cap CIDEr@0.5 / BLEU-4@0.5 83.2 / 42.2 83.8 / 41.3 (Video-3D LLM) -0.6 / +0.9
3D场景理解 - ScanQA CIDEr / EM 106.3 / 30.4 102.1 / 30.1 (Video-3D LLM) +4.2 / +0.3
3D场景理解 - SQA3D EM 61.3 58.6 (Video-3D LLM) +2.7
空间推理 - VSI-Bench Average Accuracy 50.5 48.9 (Qwen2.5VL-7B finetuned) +1.6
机器人操作 - LIBERO Average Success Rate 97.3% 97.0% (OpenVLA-OFT) +0.3%

局限与改进

论文承认的主要局限性是推理开销增加:集成视频扩散骨干增加了推理成本,包括延迟、内存和计算量的增加。具体数据显示,在832x480分辨率下,缓存后的推理延迟为70.35ms,而完整推理延迟为92.08ms;内存占用从28GB增加到29GB;计算量从约200 tflops增加到约325 tflops。此外,从个人观察来看:(1) 论文仅在ScanNet室内场景上进行了详细验证,对于室外场景、动态场景或大规模场景的泛化能力未得到验证;(2) 生成模型的选择对性能有显著影响,但论文未深入分析不同生成模型的适用场景和选择标准;(3) Scan2Cap的CIDEr指标出现了轻微下降(-0.6),暗示强调结构线索可能会削弱细粒度词汇细节,语义-几何权衡机制需要进一步优化;(4) 论文使用了8张H100 GPU进行实验,对于资源有限的研究者来说复现门槛较高。

独立分析的弱点

论文存在几个值得深入分析的弱点。首先,推理效率问题:虽然论文提出了缓存机制(每场景缓存一次特征,所有问题复用),但对于实时应用(如机器人导航)来说,额外的生成模型前向传播仍然不可忽视。改进方向包括:将生成先验蒸馏到轻量级编码器中,或开发更高效的特征提取策略,如仅在关键帧上运行生成模型。其次,生成模型的噪声水平和层选择需要手动调优:论文默认使用 $k=300$ 和第20层,但这些超参数可能因场景和任务而异。未来可以探索自适应选择策略,根据输入场景的复杂度动态调整。第三,对于长文本或复杂空间关系的处理能力有待验证:论文主要在单物体定位和简单空间关系上验证,对于涉及多个物体间复杂空间推理的场景(如'在A和B之间但在C左侧')的表现未知。第四,语义-几何权衡机制可以进一步优化:当前的门控融合是全局学习的,对于不同任务可能需要不同的权衡策略,可以探索任务感知的门控机制。

未来方向

作者提出的主要未来方向包括:(1) 将这些先验蒸馏到轻量级编码器中,以降低推理开销;(2) 将框架扩展到更动态的场景理解任务。基于已有成果,还可以延伸以下方向:(3) 探索更多类型的生成模型作为先验来源,如3D生成模型(NeRF、Gaussian Splatting)或音频-视觉生成模型;(4) 将方法应用于更广泛的具身AI任务,如导航、操作规划等;(5) 研究生成先验的可解释性,理解模型在去噪过程中学到的具体3D知识;(6) 开发在线自适应机制,根据任务难度动态决定是否激活生成分支;(7) 探索生成先验在少样本或零样本3D理解中的应用。

复现评估

论文的复现条件相对友好。代码已在GitHub公开(https://github.com/H-EmbodVis/VEGA-3D),使用的主要模型(Wan2.1-T2V 1.3B、SigLIP)都是公开可用的。数据方面,ScanNet、ScanRefer、Multi3DRefer、ScanQA、SQA3D等基准数据集都是标准的公开数据集。然而,复现的主要挑战在于算力需求:论文使用8张H100 GPU进行所有实验,这对于大多数研究机构来说是较高的门槛。训练配置(batch size 128、学习率 $1 \times 10^{-5}$ 等)也暗示了较大的内存需求。此外,论文中的关键超参数(如 $k=300$、第20层、体素大小0.1)都是通过实验确定的,新场景可能需要重新调优。总体而言,对于拥有充足算力的研究者,复现应该是可行的,但对于资源有限的团队,可能需要考虑更轻量化的实现方案。