Extend3D:城镇规模3D场景生成 Extend3D: Town-Scale 3D Generation
无需训练的单图像到城镇规模3D场景生成方法,通过扩展潜在空间和重叠块式流实现高保真度。
前置知识
潜在流模型
一种生成模型,通过学习从噪声分布到目标分布的向量场来生成样本。在3D生成中,它通常在潜在空间(如体素化稀疏结构或基于集合的潜在表示)中运行。模型通过求解常微分方程(ODE)沿着学习的向量场逐步去噪,从随机噪声生成高质量的3D表示。与扩散模型不同,流模型具有确定性,可以直接生成而不需要迭代采样步骤。
Extend3D基于预训练的3D对象生成模型Trellis,它使用潜在流模型架构。理解这种架构对于理解Extend3D如何扩展潜在空间以及为什么需要进行重叠块式处理至关重要。
稀疏结构和结构化潜在
稀疏结构是Trellis中的第一阶段输出,表示为占用体素坐标的集合{pi}⊂[M]³,定义了3D空间中哪些位置存在物体。结构化潜在(SLAT)是第二阶段输出,为稀疏结构中的每个坐标分配高维特征向量,形成一个基于集合的潜在表示。SLAT可以解码为3D高斯、辐射场或网格等多种3D表示形式。
Extend3D需要在扩展的潜在空间中同时处理稀疏结构和SLAT两个阶段,理解这两个概念的区别和联系对于理解方法的两阶段生成流程和不同的优化策略至关重要。
SDEdit
一种图像编辑方法,通过对原始图像的潜在表示添加一定程度的噪声,然后从这个部分噪声的潜在开始去噪过程,生成保留原始图像结构但满足新条件的编辑结果。公式为:Z_{t_start} = (1-t_start)·Z₀^{(g)} + t_start·noise。它通过控制噪声程度平衡原图信息保持和生成模型创造性。
Extend3D采用SDEdit作为关键组件来初始化和完善场景结构,特别是引入了under-noising变体(t_start > t_noise),使模型将遮挡区域视为额外噪声并自然填充,这是解决单目深度估计不完整问题的核心创新。
单目深度估计
从单张2D图像推断每个像素的深度信息,生成深度图或点云。MoGe-2是本文使用的方法,它可以提供具有度量尺度的几何信息。由于视角限制,单目深度估计无法推断遮挡区域(如建筑物背面、室内被遮挡的角落),导致生成的点云存在空洞。
单目深度估计为Extend3D提供了关键的几何先验,但其固有局限性(无法处理遮挡区域)正是Extend3D需要通过under-noising SDEdit来解决的问题。理解这个先验的作用和限制对于理解方法的动机和贡献至关重要。
研究动机
现有3D生成方法在处理大规模场景时面临严重限制。大多数当前3D数据集(如Objaverse、3D-FUTURE)主要由以对象为中心的数据组成,缺乏多个对象复杂排列和背景的场景案例。因此,以数据为中心的方法无法生成大型通用场景。现有的潜在生成模型(如Trellis、Hunyuan3D)使用固定大小的潜在空间表示3D数据,限制了生成结果的细节水平。随着3D场景规模增大,输出会因潜在维度有限而变得模糊,类似于低分辨率图像。此外,直接将高分辨率图像生成方法(如MultiDiffusion、DemoFusion)应用于扩展3D潜在生成时失败,因为模型的图像条件、3D和对象中心性的独特动力学导致地面消失或补丁相关性差等问题。现有的训练free 3D场景生成方法(如SynCity、3DTown)要么产生可见的接缝,要么只能处理受限的输入类型,要么忽略某些输入信息(如旋转)。
本文的目标是本文的目标是开发一个无需训练的管道,能够从单个场景图像生成高保真度、可扩展的大规模3D场景。具体而言,希望解决固定潜在空间的限制,使模型能够生成比训练时使用的对象尺度更大的场景;克服对象中心模型的局限性(如地面消失、无法生成子场景、对象随机旋转),确保生成的场景保持空间一致性和几何完整性;实现图像条件的严格空间对齐,使生成的3D场景忠实于输入图像的结构和外观;避免以往outpainting方法中的补丁不一致和接缝问题,生成连续、自然的3D场景。
与已有工作不同的是,本文的独特切入角度是将训练free高分辨率图像生成中的扩展潜在和重叠块式思想,与对象中心3D生成模型结合,同时引入3D感知优化目标来解决对象中心模型与场景动力学的不匹配。与以往顺序生成3D子场景块的方法不同,Extend3D通过重叠块同时生成,使相邻块能够相互影响,增加准确重建3D表示的可能性。更重要的是,本文发现了将3D结构的不完整性在3D细化过程中视为噪声可以通过under-noising实现3D补全的概念,这是一个新颖的见解。此外,与仅仅提供指导的方法不同,Extend3D优化整个场景的中间潜在,使用专门为3D场景生成设计的损失函数,假设先验具有3D结构和纹理的真实知识,这在之前的训练free 3D场景生成工作中未被探索。
核心方法
Extend3D的整体思路是扩展预训练3D对象生成模型的潜在空间,使其能够表示更大规模的3D场景,然后通过重叠块式流处理和先验优化来生成高质量的3D场景。直觉上,这就像将一个只能生成单个对象的3D模型的能力放大,让它能够生成包含多个对象的完整场景。技术路线包括三个核心组件:扩展潜在空间,在x和y方向扩展3D潜在,使其能够表示比原始模型更大的场景;重叠块式流,将扩展的潜在空间划分为重叠的块,每个块由预训练模型独立处理但通过重叠区域相互耦合;先验优化,使用单目深度估计的点云和输入图像作为先验来初始化和优化潜在,确保空间对齐和几何一致性。方法分为两个阶段:稀疏结构生成和结构化潜在生成。在稀疏结构生成中,使用点云先验和迭代under-noising SDEdit来初始化和完善结构;在结构化潜在生成中,使用图像先验和3D感知优化来细化纹理和外观。
核心创新点有三个:潜在空间扩展与重叠块式流的结合,通过扩展潜在空间(a倍和b倍)并使用重叠块式流(滑动窗口),使对象中心模型能够生成场景规模的3D内容。重叠区域允许相邻块相互影响,防止接缝和对象重复;Under-noising SDEdit,不同于标准SDEdit(t_start = t_noise),Extend3D使用t_start > t_noise,使模型将遮挡区域视为额外噪声并在去噪过程中自然填充,这是解决单目深度估计不完整问题的关键;3D感知优化目标,引入专门为3D场景设计的损失函数,在稀疏结构阶段使用占用损失防止初始化的体素消失,在结构化潜在阶段使用渲染损失细化纹理。这些优化确保去噪轨迹保持与子场景动力学一致,防止对象中心模型将场景扭曲为对象中心结构。
方法步骤详情
Extend3D的完整步骤如下:输入处理阶段,给定单张场景图像I,使用MoGe-2单目深度估计器提取点云P和相机参数;稀疏结构生成阶段,将点云P体素化为初始占用网格O₀∈R^{aM×bM×M},使用编码器E将O₀编码为潜在Z₀^{(g)},应用under-noising公式Z_{t_start} = (1-t_noise)·Z₀^{(g)} + t_noise·noise,其中t_start > t_noise,使用重叠块式流(滑动窗口大小N³或M³,步长N/d或M/d)对扩展的稀疏结构潜在Z_t^{SS}进行去噪,每个块与对应的图像块条件ψ_{i,j}(I)配对,在每个时间步使用Adam优化器优化向量场v̂_t,目标是最小化稀疏结构损失L_SS,迭代执行SDEdit:O_{n+1} = SDEdit(O_n),通常n_iter=5次;结构化潜在生成阶段,从稀疏结构{p_i}初始化SLAT潜在Z₁,使用相同但更细的重叠块式流(d=4)对扩展的SLAT潜在Z_t^{SLAT}进行去噪,在每个时间步使用Adam优化器优化向量场v̂_t,目标是最小化结构化潜在损失L_SLAT = LPIPS(Î, I) - SSIM(Î, I),其中Î = Render(D_GS(Z_t^{SLAT} - t·v̂_t), P);解码阶段,使用解码器D_GS和D_mesh将最终的SLAT Z₀解码为3D高斯和网格表示。
技术新颖性
Extend3D的技术新颖性体现在四个方面:跨模态方法迁移,首次将训练free高分辨率图像生成中的扩展潜在和重叠块式思想成功迁移到3D领域,针对3D生成的独特挑战(对象中心性、3D几何约束)进行了专门设计;Under-noising概念,提出了under-noising这一新颖概念,通过使去噪程度高于加噪程度(t_start > t_noise),让模型将不完整结构视为噪声并在去噪过程中自然补全,这与传统的outpainting或inpainting方法有本质区别;双阶段3D感知优化,设计了针对稀疏结构和结构化潜在的不同优化目标,占用损失确保几何结构保持,渲染损失细化外观,这种分离式优化比统一优化更有效;迭代完善框架,提出了迭代SDEdit框架,每次迭代都基于上一次的结果进一步完善场景,逐步填充遮挡区域并细化细节,这种迭代思想在3D生成中是新颖的。
实验结果
通过人类偏好研究和定量实验,Extend3D在几何、保真度、外观和完整性四个方面都显著优于先前方法。人类偏好研究(10名参与者,14张图像)显示,Extend3D在几何方面对Hunyuan3D-2.1的胜率为73.6%,对EvoScene为87.1%;在保真度方面对Hunyuan3D-2.1为75.7%,对EvoScene为87.9%;在外观方面对Hunyuan3D-2.1为75.0%,对EvoScene为87.1%;在完整性方面对Hunyuan3D-2.1为75.0%,对EvoScene为87.1%。定量评估(100张图像)显示,Extend3D在LPIPS(0.240,越低越好)、SSIM(0.611,越高越好)、PSNR(20.4,越高越好)上都达到最优,显著优于Trellis(0.650/0.239/10.0)、Hunyuan3D(0.683/0.255/10.4)和EvoScene(0.482/0.310/13.2)。几何评估(45个UrbanScene3D场景)显示,Extend3D在Chamfer Distance(0.0086)上优于EvoScene(0.0188)和3DTown(0.0192),在F-score(0.05阈值)上达到0.694,优于EvoScene(0.498)。与SynCity的比较显示,Extend3D在CLIP分数(0.276 vs 0.251)、HPSv3(3.519 vs 3.254)和Intra-LPIPS(0.571 vs 0.631,越低越好)上都更优,表明更好的文本兼容性、质量和补丁一致性。消融实验表明,增加分割因子d(从2到4)显著改善LPIPS(0.251到0.240)、SSIM(0.598到0.611)、PSNR(19.8到20.4);初始化将CD从0.0348降到0.0083,F-score从0.261提升到0.693;稀疏结构优化将CD进一步降到0.0078,F-score提升到0.708;under-noising(t_noise=0.6, t_start=0.8)在所有指标上都优于标准SDEdit(t_noise=t_start=0.6)和over-noising(t_noise=0.8, t_start=0.6)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 人类偏好-几何 | 胜率 | 87.1 (vs EvoScene) | EvoScene | +75.1个百分点 |
| 外观保真度-LPIPS | LPIPS | 0.240 | Trellis | -63.1% |
| 外观保真度-SSIM | SSIM | 0.611 | Trellis | +155.6% |
| 外观保真度-PSNR | PSNR | 20.4 | Trellis | +104.0% |
| 几何精度-Chamfer Distance | CD | 0.0086 | EvoScene | -54.3% |
| 几何精度-F-score | F-score | 0.694 | EvoScene | +39.4% |
| 文本兼容性-CLIP | CLIP | 0.276 | SynCity | +10.0% |
| 补丁一致性-Intra-LPIPS | Intra-LPIPS | 0.571 | SynCity | -9.5% |
局限与改进
作者承认的局限性包括:遮挡区域补全有时不完整,例如在补充材料A.7中展示的房间案例,某些遮挡区域仍然存在空洞,没有完全填充;SLAT优化需要大量内存,特别是对于大型场景,峰值GPU内存使用可达28GB(a=b=2, d=4, n_iter=5),其中大部分用于SLAT优化过程,对于a=b=6的扩展,不使用SLAT优化时峰值内存为61GB;对街道级别图像的性能有限,由于x和y坐标尺度的显著不匹配(由灭点引起),框架在街道级别图像上表现有限。作者自己的观察包括:几何与纹理的权衡,SLAT优化虽然增强了3D场景的纹理,但有时会降低几何质量(F-score从0.708降到0.694),表明几何和纹理之间存在权衡;迭代次数的权衡,增加SDEdit迭代次数(n_iter)会改善场景完整性,但可能降低几何细节(n_iter=2时F-score=0.688,n_iter=3时降到0.677);计算复杂度,方法的计算复杂度为O(abd²n_iter),更大的扩展因子a、b、分割因子d和迭代次数n_iter都会增加推理时间,存在输出质量与推理时间的权衡。
独立分析的弱点
独立分析的弱点包括:依赖单目深度估计的质量,MoGe-2的误差直接影响初始化质量,虽然under-noising可以部分缓解,但深度估计的系统性偏差仍可能传播到最终结果,改进方向是集成多视图几何约束或使用更鲁棒的深度估计方法,如结合立体匹配或激光雷达数据(如果可用);对极端视角的敏感性,当输入图像包含极端透视(如低角度拍摄)时,扩展潜在空间的xy均匀扩展与实际投影几何不匹配,导致失真,改进方向是引入非均匀的潜在扩展策略,根据透视畸变自适应调整扩展因子;材质和光照的一致性,虽然方法确保了几何连续性,但不同块之间可能存在光照和材质的不一致,改进方向是引入全局光照一致性约束或使用可微分渲染来优化跨块的光照参数;缺乏语义理解,方法没有显式建模场景语义,可能导致语义不一致(如室内外边界模糊),改进方向是引入语义分割先验或使用多模态模型来确保语义一致性。
未来方向
作者提出的未来研究方向:支持更广泛的图像类型,解决街道级别图像中的尺度不匹配问题,实现从更广泛视角和场景类型的3D生成;优化SLAT的内存效率,减少SLAT优化过程的内存占用,使方法能够在更有限的GPU资源上运行更大的场景。基于成果可延伸的未来方向:多模态条件生成,扩展方法以支持文本、草图、布局等多种输入条件的组合,提供更灵活的控制方式;动态场景生成,扩展到包含动态元素(如行人、车辆)的4D场景生成;编辑和交互,开发基于Extend3D的场景编辑工具,允许用户修改特定区域或添加/删除对象;层次化场景表示,引入层次化的潜在表示,支持从粗到细的场景生成和编辑,类似于LT3SD但保持训练free的优点;真实场景重建,将Extend3D与神经辐射场或3D高斯泼溅结合,实现从真实视频序列的高质量场景重建和生成。
复现评估
复现评估:Extend3D的代码和模型权重尚未开源(截至论文发表),这限制了完全复现的能力。然而,方法的各个组件都有公开实现:Trellis(结构化3D潜在)、MoGe-2(单目深度估计)、SDEdit(扩散编辑)。数据方面:论文使用了多种数据源进行评估,包括UrbanScene3D(45个合成场景,带ground truth网格)、ChatGPT生成的图像、Flux.1生成的图像、Google Earth图像和CarlaSC数据集。这些数据集都是公开可访问的。算力要求:论文使用NVIDIA RTX Pro 6000 GPU,峰值内存使用28GB(a=b=2, d=4, n_iter=5,包含SLAT优化)。不使用SLAT优化时,内存需求降到14GB。推理时间为14.1分钟(与SynCity的52.0分钟和EvoScene的35.3分钟相比)。对于a=b=6的扩展,不使用SLAT优化时峰值内存为61GB。复现难度:中等。主要挑战包括实现重叠块式流的精确映射(特别是反向映射(ϕ_{i,j})⁻¹和(ϕ_{i,j}^{SLAT})⁻¹)、调整under-noising的超参数(t_noise和t_start)以及实现Adam优化器的向量场优化。论文在补充材料中提供了详细的算法伪代码(算法1、2、3)和严格的数学定义(Sec. A.5),这有助于复现。
论文图表
这个表格展示了不同SDEdit迭代次数n_iter对几何结果的影响。n_iter=0(仅深度估计)时CD=0.0079, F-score=0.647。n_iter=1时改善到CD=0.0075, F-score=0.681。n_iter=2时轻微改善(CD=0.0077, F-score=0.688)。n_iter=3时退化(CD=0.0082, F-score=0.677)。
这个表格对理解论文至关重要,因为它揭示了迭代SDEdit的权衡:增加迭代次数可以改善场景完整性,但过多迭代可能降低几何细节。这解释了为什么论文选择n_iter=5作为平衡点。