将几何基础模型复用于多视角扩散 Repurposing Geometric Foundation Models for Multi-view Diffusion
用几何基础模型的特征空间替代VAE作为多视角扩散的潜空间,实现更快更好的新视角合成
前置知识
新视角合成 (Novel View Synthesis, NVS)
给定一个场景的若干已知视角图像及其相机位姿,预测从新相机角度观察该场景时应该看到的图像。这个任务要求模型理解场景的三维结构,对被遮挡区域进行合理的几何补全,并保证生成图像与已知视角在空间结构上保持一致。经典方法如NeRF需要密集多视角输入和逐场景优化,而基于扩散模型的现代方法尝试从稀疏输入(如1-4张图)直接生成新视角。
这是本文要解决的核心任务。理解NVS的难点——跨视角几何一致性——才能理解为什么作者要提出用几何特征空间替代传统VAE潜空间。
潜空间扩散模型 (Latent Diffusion Model, LDM)
将扩散过程从原始像素空间转移到一个压缩的潜空间中进行。具体做法是先用一个编码器(通常是VAE)将图像压缩为低维特征表示,然后在该特征空间中训练扩散模型去噪,最后用解码器将生成的特征还原为图像。这样做的好处是大幅降低计算成本(潜空间分辨率远低于像素空间),同时保留图像的主要语义信息。Stable Diffusion就是LDM的典型应用。
本文的核心论点就是:LDM中潜空间的选择对生成质量至关重要,而现有NVS方法使用的VAE潜空间并非最优选择。
几何基础模型 (Geometric Foundation Model)
一类以纯前馈方式从图像预测三维几何信息的大规模预训练模型,典型代表包括Depth Anything 3 (DA3)和VGGT。这些模型通常基于Vision Transformer架构,通过3D注意力机制联合处理多视角输入,能够同时预测深度图、相机参数等几何信息。其内部特征在多个中间层(通常L=4层)提取,浅层特征保留更多光度细节(颜色、纹理),深层特征编码更抽象的几何结构。已有研究表明这些模型的内部特征天然编码了强跨视角几何对应关系。
本文的核心创新就是将几何基础模型的特征空间'挪用'为扩散模型的潜空间,因此理解这些模型的工作原理和特征层次结构是理解本文方法的基础。
Flow Matching
一种训练扩散模型的目标函数,与传统的去噪分数匹配不同,flow matching直接学习从噪声分布到数据分布的速度场(velocity field)。具体来说,它定义一条从噪声 $z_0$ 到数据 $z_1$ 的线性插值路径 $z_t = (1-t)z_0 + t z_1$,然后训练网络预测该路径上每一点的速度 $u_t$。相比DDPM等方法,flow matching训练更稳定,且推理时可以使用更高效的ODE求解器。
本文的多视角扩散模型采用flow matching目标进行训练,理解这一训练范式有助于把握模型的训练细节和推理流程。
Plücker坐标 (Plücker Coordinates)
一种用于参数化三维空间中直线的六维表示方法。对于图像中的每个像素,从其对应的相机光心出发、穿过该像素的射线可以用一个6D向量表示:前3维是射线方向,后3维是射线方向与光心位置的叉积。Plücker坐标将相机的内外参数信息编码为逐像素的几何嵌入,使得模型能够精确知道每个像素对应的三维射线方向。
本文使用Plücker射线嵌入作为相机条件注入扩散模型,确保生成的图像严格遵循目标视角的几何约束。这是实现精确视角控制的关键技术细节。
研究动机
在新视角合成(NVS)任务中,现有基于扩散模型的方法面临一个根本性的架构矛盾:它们使用的潜空间(通常是2D VAE的潜空间)本身是为单张图像压缩设计的,缺乏跨视角的几何结构信息。这意味着扩散模型需要从零开始学习跨视角的几何对应关系,把'理解三维空间结构'这个本应由潜空间承担的负担完全推给了网络本身。具体而言,当场景中存在大面积遮挡、视角变化剧烈或需要对不可见区域进行合理补全时,VAE潜空间中的特征是视角独立的(view-independent),不同视角的特征之间没有内在的几何关联,导致模型即使学习了大量数据,生成的图像在像素层面可能看起来逼真,但在三维空间中存在几何不一致——比如同一个物体在不同视角下形状扭曲、深度不连贯、或出现重复结构。此外,为了弥补潜空间缺乏几何先验的缺陷,现有方法往往需要依赖外部深度估计器进行warping作为条件输入(如MVGenMaster),这增加了推理时的复杂度和对深度估计精度的依赖,一旦深度估计失败就会产生明显的伪影。
本文的目标是本文的目标是设计一个对NVS任务而言更合适的潜空间——一个本身就编码了丰富跨视角几何结构的潜空间。具体来说,作者希望找到一种表征,使得:(1)扩散模型在其中训练时收敛更快(因为不需要从头学习几何对应);(2)生成的新视角在二维图像质量(PSNR、SSIM、LPIPS)和三维一致性(相机位姿误差、重投影误差)上都优于VAE和DINO潜空间的基线;(3)不需要依赖外部几何条件(如深度图warping)就能实现高质量的多视角生成;(4)生成的潜特征能直接解码为几何信息(深度图、相机位姿),实现'一次生成,同时获得外观和几何'。
与已有工作不同的是,本文抓住了一个被已有工作忽视的关键洞察:几何基础模型(如DA3、VGGT)的特征空间天然具备跨视角几何对应关系,这正是NVS任务所需要的。已有的RAE等工作已经证明,在语义丰富的特征空间(如DINO)中做扩散比在VAE空间中更快更好,但这些工作都聚焦于单张图像生成,从未探索过几何感知的特征空间是否能为多视角生成带来更大收益。本文的独特视角在于:与其在VAE空间中通过额外模块注入几何信息(如depth warping、3D attention),不如直接选择一个'自带几何'的潜空间,让几何结构成为表征的固有属性而非外部监督信号。这一思路不仅简化了模型架构(无需外部深度估计器),还从根本上改变了扩散模型的学习任务——从'学习几何'变为'利用已有的几何先验进行生成'。
核心方法
本文提出Geometric Latent Diffusion (GLD)框架,核心思路可以用一个类比理解:如果把扩散模型比作一个画家,那么传统方法是在一张白纸(VAE潜空间)上作画,画家需要自己理解物体的三维结构才能画出正确的视角;而GLD则给画家提供了一张已经画好透视网格的纸(几何基础模型的特征空间),画家可以直接在网格基础上填色,不需要从零理解透视关系。技术路线上,GLD采用三阶段流水线:首先,训练一个RGB解码器验证DA3特征空间确实能高保真地还原图像(确认'这张纸够好');然后,通过系统实验确定最优的'边界层'——即扩散模型需要显式生成到哪一层特征,更深层的特征可以通过冻结的DA3编码器从边界层特征传播得到(避免生成所有4层特征的巨大计算开销);最后,采用级联方案生成浅层特征以保证跨层对齐。整个框架以Depth Anything 3 (DA3)为主干网络,操作在其4层中间特征上,每个级别独立训练一个扩散模型。
GLD最核心的创新在于发现并验证了:几何基础模型的中间特征空间可以作为多视角扩散的'即插即用'潜空间,且在该空间中训练的扩散模型比在VAE或DINO空间中训练的模型收敛快4.4倍,同时在二维和三维指标上都更优。这一发现的关键在于几何基础模型的特征天然编码了跨视角的几何对应——论文通过PCK(Percentage of Correct Keypoints)指标量化证明,DA3的Level 1和Level 2特征在跨视角匹配上的表现(PCK 35.98和40.70)甚至超过了专门为语义特征设计的DINOv2(PCK 31.64)。更进一步,作者发现并非所有特征层都同样适合作为扩散潜空间:通过从几何对应(PCK)和光度重建(PSNR)两个维度分析4层特征,发现Level 1是'甜蜜点'——它同时具备足够强的几何对应(PCK 35.98,接近Level 2的40.70)和足够丰富的光度信息(重建PSNR 25.36),而Level 0虽然光度信息最丰富但几何对应很弱(PCK 22.25),Level 2和Level 3虽然几何对应强但丢失了大量细节。这个发现为潜空间设计提供了明确的指导原则。
方法步骤详情
GLD的完整流程分为以下步骤:第一步,特征提取与归一化。将输入的多视角图像(V=8个视角,其中N个为源视角,M个为目标视角)送入冻结的DA3-Base编码器,提取4层中间特征 $\{F_l\}_{l=0}^{3}$,每层维度为 $V \times T \times 1536$(T为token序列长度)。在训练扩散模型前,对每层特征按通道计算训练集上的均值和方差进行标准化。第二步,RGB解码器训练。训练一个ViT-based解码器 $D_{rgb}$ 从多层特征还原RGB图像,训练时采用层级dropout策略随机屏蔽某些层以增强鲁棒性。解码器在Re10K和DL3DV上训练,使用L1+LPIPS+GAN组合损失。第三步,边界层确定。独立训练4个扩散模型 $\{M_l\}_{l=0}^{3}$,每个负责生成特定层的特征。通过实验发现边界 $k=1$ 效果最优。第四步,级联特征生成。以Level 1为边界,先用 $M_1$ 生成Level 1特征,再用级联模型 $M_{1 \to 0}$ 以Level 1特征为条件生成Level 0特征。$M_{1 \to 0}$ 在训练时以带噪声的Ground Truth Level 1特征作为条件,以增强对推理时噪声输入的鲁棒性。第五步,特征传播与解码。将生成的Level 1特征 $\tilde{F}_1$ 送入冻结的DA3编码器的第7-11个block,确定性地推导出Level 2和Level 3特征。最后,RGB解码器 $D_{rgb}$ 接收完整的4层特征 $\{\tilde{F}_0, \tilde{F}_1, \tilde{F}_2, \tilde{F}_3\}$ 输出目标视角的RGB图像,同时DA3原始解码器 $D_{geo}$ 可直接输出深度图和射线图。
技术新颖性
GLD的技术新颖性体现在三个层面。首先,在概念层面,它是首个将几何基础模型的特征空间作为多视角扩散潜空间的工作。此前RAE和SVG虽然探索了语义编码器(如DINO、MAE)的特征空间用于扩散,但这些工作都局限于单张图像生成,从未考虑过几何感知的表征。本文将'任务特定潜空间设计'这一理念从2D生成推广到了3D感知的多视角生成。其次,在方法层面,边层级联策略(cascaded feature generation)是一个巧妙的设计:通过让浅层特征的生成条件化于已生成的深层特征,保证了特征层级间的对齐,这比独立生成每一层特征再拼接更有效(实验验证PSNR从18.81提升到19.00)。第三,在工程层面,'冻结编码器+传播深层特征'的设计使得生成4层特征的实际计算量远小于显式生成所有层——Level 2和Level 3的传播仅需0.15秒,而生成Level 1需要37.8秒,这种不对称利用避免了全层生成的计算瓶颈。
实验结果
GLD在多个维度上展示了其优越性。在2D图像质量方面,GLD在Re10K数据集上达到PSNR 16.362、SSIM 0.630、LPIPS 0.431,在DL3DV上达到PSNR 15.499、SSIM 0.468、LPIPS 0.438,全面超越VAE基线(Re10K: PSNR 15.656, SSIM 0.606)和DINO基线(Re10K: PSNR 15.638, SSIM 0.601),甚至超越了利用大规模文本到图像预训练的最先进方法如MVGenMaster(Re10K: PSNR 15.226)和Matrix3D(Re10K: PSNR 14.490)。在3D一致性方面,GLD的优势更加显著:在Re10K上ATE为0.211(VAE基线0.278,降低24%),RPEr为7.07(VAE基线8.68,降低19%),RPEt为0.444(VAE基线0.552,降低20%)。在DL3DV上优势更大,ATE为0.209(VAE基线0.589,降低65%),RPEr为5.75(VAE基线15.00,降低62%)。训练效率方面,GLD的收敛速度是VAE潜空间的4.4倍、DINO潜空间的4.4倍。在零样本泛化到Mip-NeRF 360(物体中心场景,训练时未见过此类数据)时,GLD仍取得PSNR 14.542和ATE 0.589的最优成绩,远超VAE(ATE 1.221)和DINO(ATE 0.949)。深度评估方面,在ETH3D数据集上,GLD的深度预测AbsRel为0.160(Matrix3D为0.197),$\delta < 1$准确率为0.800(Matrix3D为0.731)。消融实验还发现,随着源视角数量减少(从N=4到N=1),GLD相对基线的3D指标优势进一步扩大,说明几何先验在视觉线索稀缺时尤为关键。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 新视角合成 (Re10K) | PSNR ↑ | 16.362 | VAE: 15.656, MVGenMaster: 15.226 | +0.71 (vs VAE), +1.14 (vs MVGenMaster) |
| 新视角合成 (Re10K) | SSIM ↑ | 0.630 | VAE: 0.606, MVGenMaster: 0.588 | +0.024 (vs VAE), +0.042 (vs MVGenMaster) |
| 新视角合成 (Re10K) | LPIPS ↓ | 0.431 | VAE: 0.456, MVGenMaster: 0.456 | -0.025 (vs VAE) |
| 新视角合成 (Re10K) | ATE ↓ | 0.211 | VAE: 0.278, MVGenMaster: 0.282 | -0.067 (vs VAE, 24% down) |
| 新视角合成 (Re10K) | RPEr ↓ | 7.07 | VAE: 8.68, DINO: 15.59 | -1.61 (vs VAE, 19% down) |
| 新视角合成 (DL3DV) | PSNR ↑ | 15.499 | VAE: 14.725, MVGenMaster: 14.565 | +0.77 (vs VAE) |
| 新视角合成 (DL3DV) | ATE ↓ | 0.209 | VAE: 0.589, MVGenMaster: 0.281 | -0.380 (vs VAE, 65% down) |
| 新视角合成 (DL3DV) | RPEr ↓ | 5.75 | VAE: 15.00, MVGenMaster: 6.93 | -9.25 (vs VAE, 62% down) |
| 零样本泛化 (Mip-NeRF 360) | PSNR ↑ | 14.542 | VAE: 13.942, MVGenMaster: 14.170 | +0.60 (vs VAE) |
| 零样本泛化 (Mip-NeRF 360) | ATE ↓ | 0.589 | VAE: 1.221, DINO: 0.949 | -0.632 (vs VAE, 52% down) |
| 深度估计 (ETH3D) | AbsRel ↓ | 0.160 | Matrix3D: 0.197 | -0.037 (19% down) |
| 深度估计 (ETH3D) | δ < 1 ↑ | 0.800 | Matrix3D: 0.731 | +0.069 |
局限与改进
GLD存在几个明显的局限性。首先,推理速度是主要瓶颈:整个生成流程需要66.8秒(单场景),其中Level 1采样37.8秒、级联Level 0采样28.4秒,而VAE基线仅需28.0秒,DINO基线35.2秒。这是因为GLD需要两次串行的扩散采样过程,且DA3特征的token数量较多。其次,模型在极端遮挡或视角覆盖极稀疏的场景下可能产生幻觉内容或伪影,这是扩散模型固有的问题。第三,虽然GLD在零样本泛化上表现不错(Mip-NeRF 360),但在物体中心场景上的3D指标(ATE 0.589)仍略逊于经过大规模T2I预训练并使用物体中心数据微调的MVGenMaster(ATE 0.320)和Matrix3D(ATE 0.548)。第四,GLD的性能高度依赖于几何基础模型(DA3)本身的质量——如果基础模型在某些场景上表现不佳(如极端光照变化、大时间间隔输入),GLD也会受到影响。最后,当前实验仅在室内/室外场景级数据上训练,尚未覆盖更多样化的场景类型(如户外大场景、高度动态场景等)。
独立分析的弱点
GLD有几个值得关注的弱点。第一,推理效率:66.8秒的推理时间使得GLD难以应用于实时或近实时场景。改进方向包括:(1)使用一致性模型(Consistency Model)或蒸馏技术将多步采样压缩为单步或少步生成;(2)探索更高效的潜空间分辨率,当前DA3-Base的patch size为1导致token数量较多,可以考虑使用更小的ViT变体或对特征做空间下采样后再做扩散。第二,潜空间选择依赖人工经验:当前通过穷举实验确定Level 1为最优边界层,缺乏理论指导。改进方向是设计自动化的目标函数来评估各层特征的'扩散友好度',结合几何对应分数和光度信息量的帕累托最优解。第三,级联生成引入了串行依赖:Level 0必须等Level 1生成完毕才能开始,限制了并行化。可以探索将两层的生成统一到一个模型中,通过多头输出或分层去噪策略同时生成多层特征。第四,RGB解码器的训练是独立于扩散模型的,两者之间没有联合优化。端到端的微调虽然计算量大,但可能进一步提升生成质量。
未来方向
基于GLD的成果,有几个值得探索的方向。首先,将GLD框架扩展到视频生成:几何基础模型的跨帧一致性特征天然适合时序连贯的视频新视角合成,可以将时间维度纳入扩散过程。其次,探索更强大的几何基础模型作为潜空间:随着DA3和VGGT等模型的快速迭代,更新的主干网络可能提供更丰富的几何特征,进一步提升GLD的性能。第三,将GLD与文本条件结合:当前GLD仅支持图像条件输入,引入文本描述可以实现文本引导的新视角合成或场景编辑。第四,自适应边界层选择:不同场景复杂度可能需要不同的边界层深度,训练一个轻量级路由器根据输入场景自动选择最优边界层可以进一步提升效率和质量。第五,利用GLD生成的高质量几何信息做下游任务,如3D重建、场景理解、机器人导航等,将生成模型与感知任务打通。
复现评估
GLD的复现条件相对友好。代码方面,论文提供了项目主页链接(https://cvlab-kaist.github.io/GLD),但未明确说明是否已开源完整代码和预训练权重。主干网络DA3-Base是公开可用的,RGB解码器和扩散模型的架构细节在论文中有详细描述(附录A.2),包括层数、隐藏维度、注意力头数等完整超参数。数据方面,训练使用了4个公开数据集(Re10K、DL3DV、HyperSim、TartanAir),评估使用了3个公开数据集(Re10K、DL3DV、Mip-NeRF 360、ETH3D),均为学术界常用基准。算力需求方面,训练使用8张NVIDIA B200 GPU训练175k步,这对大多数实验室而言是一个可承受的规模。RGB解码器训练约需170k步。总体而言,如果代码开源,复现难度中等——主要挑战在于数据预处理(多视角采样、相机位姿归一化)和扩散模型训练的超参数调优。
论文图表