FLAT:用于几何精确场景生成的前向潜在三角形泼溅 FLAT: Feedforward Latent Triangle Splatting for Geometrically Accurate Scene Generation
从视频扩散潜空间直接解码三角形,实现单图生成几何精确的3D场景
前置知识
3D Gaussian Splatting (3DGS)
一种使用各向异性3D高斯集合表示场景的显式3D表示方法,每个高斯由位置、旋转、缩放、不透明度和颜色参数化,通过可微渲染实现高质量实时渲染。高斯是体积性的、半透明的团块,适合训练但难以转换为标准图形引擎所需的表面网格。
现有前向场景生成方法如Wonderland和Lyra都基于3DGS,理解其限制(体积性、缺乏明确定义的表面)对理解FLAT为何选择三角形表示至关重要。
视频扩散模型
基于扩散概率模型的视频生成方法,通过逐步去噪从噪声中生成视频帧。大型视频扩散模型在互联网规模数据上学习丰富的先验和隐式3D世界理解,其潜空间包含多视图几何结构编码。常用架构如Wan-2.1结合VAE将视频压缩到潜在空间进行处理。
FLAT的核心创新是首次将视频扩散潜空间直接映射到三角形泼溅,而非传统的图像,理解视频模型如何编码3D信息是理解该方法有效性的基础。
三角形泼溅
一种使用软三角形作为场景表示的可微渲染方法,每个三角形由三个顶点、颜色、平滑度和不透明度定义。通过窗口函数将硬三角形替换为平滑近似,使渲染过程可微,允许通过反向传播优化三角形参数。与高斯相比,三角形是扁平的、明确定义表面的图元。
这是FLAT选择的核心表示,理解其可微渲染原理和相对于高斯的挑战(方向敏感性、梯度稀疏)对理解FLAT的技术贡献至关重要。
可微渲染
一种使渲染过程可求导的技术,允许通过比较渲染图像与目标图像的损失,反向传播梯度来优化场景参数。传统光栅化由于离散性不可微,需要通过连续近似(如窗口函数)实现梯度流动。这是训练3D生成模型的关键技术。
FLAT的训练依赖于可微三角形渲染,理解其梯度流动的挑战以及FLAT如何通过改进窗口函数解决这些问题,对理解方法的技术细节至关重要。
研究动机
现有前向3D场景生成方法如Wonderland、Lyra和Generative Gaussian Splatting都局限于生成3D Gaussian Splatting表示。这些高斯是体积性的、半透明的团块,虽然适合可微渲染训练,但缺乏明确定义的表面,使其无法直接用于大多数图形引擎。图形引擎依赖于不透明的表面表示如三角形或网格。虽然存在从高斯表示提取网格的方法,但这些方法需要复杂的后处理,且在稀疏视图覆盖和小场景尺度下会失败。此外,直接优化不透明的三角形或基于网格的表示由于渲染过程的不可微性而非常困难。更重要的是,训练前向三角形泼溅解码器面临进一步挑战:直接回归顶点容易导致退化解;与体积高斯不同,错误方向的扁平三角形对渲染图像贡献可忽略,导致梯度监督微弱,尤其是在训练早期。这些问题使得稳定的前向非体积图元预测成为一个开放问题。
本文的目标是本文的目标是首次展示可以从压缩的视频扩散潜空间在单次前向传播中直接解码显式的、非体积的表面图元。具体而言,引入FLAT方法,从冻结视频扩散模型的潜空间直接预测半不透明三角形泼溅图元,生成几何精确、物理基础的场景表示。该方法旨在维持与最先进方法相当的视觉质量,同时显著提高几何精度,并提供可选的轻量级细化步骤,将半透明的前向预测转换为完全opaque的三角形,使其直接兼容标准图形管线和游戏引擎。
与已有工作不同的是,本文的独特切入角度在于首次探索从前向角度解码非体积表面图元(三角形)而非体积图元(3D高斯)。这与所有现有的前向场景生成方法形成鲜明对比,后者都局限于3DGS表示。作者认识到这需要仔细选择参数化和可微渲染公式,提出了两个关键技术:射线中心的局部三角形参数化和改进的窗口函数。此外,本文还在相同的训练设置下训练了3DGS和2DGS变体,提供了首次系统性的表示权衡分析,这是现有工作所缺乏的。这种角度不仅解决了表面表示兼容性问题,还首次系统性地比较了不同表示在前向场景生成中的权衡。
核心方法
FLAT的整体思路是复用冻结视频扩散模型的强大生成先验,通过训练一个轻量级场景解码器,将视频潜空间直接映射到显式三角形泼溅表示。给定单张RGB图像和目标相机轨迹,视频生成器输出去噪潜变量,场景解码器融合视频潜变量与相机潜变量,解码三角形泼溅场景表示用于新视角合成。这种设计避免了昂贵的逐场景优化,同时实现了输入视图之外的合理生成和结果场景表示的实时渲染。直觉上,作者认为视频扩散模型在潜空间中隐式编码了多视图几何结构,通过精心设计的参数化和可微渲染,可以直接解码出几何精确的表面图元。
FLAT的核心创新在于首次展示了可以将视频扩散潜空间直接解码为三角形泼溅,并解决了由此带来的两个关键挑战。首先,提出了稳定的扁平图元参数化:每个解码器token预测一个射线中心定义的三角形,使用约束的Cholesky风格形状变换和围绕射线对齐帧的残差旋转,避免了退化三角形和不稳定的世界空间方向。其次,引入了改进的窗口函数,通过乘积形式而非原始三角形泼溅公式中的最大值操作,在可微三角形渲染期间改善图元边界处的梯度流动。这两个关键创新使得稳定的前向非体积图元解码成为可能,与基于3DGS的方法形成本质区别。
方法步骤详情
FLAT方法的完整步骤如下:首先,给定单张RGB图像 $I_0 \in \mathbb{R}^{H \times W \times 3}$ 和目标相机轨迹 $\{P_t\}_{t=1}^T$,其中每个 $P_t = (K_t, R_t, t_t)$ 表示相机内参和外参。第二,通过渲染沿目标相机轨迹构建基于点云的控制视频。第三,控制视频和相机嵌入条件化冻结的视频扩散模型(Uni3C),视频生成器输出去噪潜变量 $z \in \mathbb{R}^{F' \times C' \times H' \times W'}$。第四,相机信息编码为与视频潜变量对齐的每像素射线嵌入。具体地,从像素对齐的Plücker射线嵌入 $r_{pl} = (o \times d, d)$ 开始,其中 $o \in \mathbb{R}^3$ 和 $d \in S^2$ 是射线原点和方向,使用RPPC参数化 $r_{rppc} = (o - (o \cdot d)d, d)$ 更好地暴露射线位置和相对深度。第五,将 $r_{rppc} \in \mathbb{R}^{T \times 6 \times H \times W}$ 分离为参考点和方向分量,分别用预训练VAE编码器编码后融合并通过零初始化块注入场景解码器。第六,场景解码器(修改的Wan-2.1 VAE解码器)将融合特征映射到三角形参数。第七,每个解码器输出token预测单个三角形泼溅的参数:对于局部 $2 \times 2$ 图像区域的锚定射线,网络预测深度值 $D$、三个形状参数、旋转参数、颜色系数、不透明度和平滑度参数 $\sigma$。第八,三角形中心放置在 $r_o + D \cdot r_d$,其面内几何首先在与射线相切的2D坐标系中定义。从正则中心等边三角形开始,用下三角矩阵 $L = \begin{bmatrix} L_{00} & 0 \\ L_{10} & L_{11} \end{bmatrix} \in \mathbb{R}^{2 \times 2}$ 变换,对角项 $L_{00}$ 和 $L_{11}$ 被强制为正,离对角项 $L_{10}$ 控制剪切。第九,将变换后的顶点平移使其质心与沿射线的锚定点重合。第十,使用射线切向帧将局部2D三角形提升到3D。方向参数化为两个残余倾斜角度和一个面内旋转角度。第十一,通过改进的窗口函数实现可微分光栅化。对于投影三角形 $T_m^{2D}$,令 $L_{m,i}(p) = n_{m,i}^\top p + d_{m,i}$ 表示到第 $i$ 条支撑边线的有符号距离,归一化边响应为 $u_{m,i}(p) = -L_{m,i}(p) / \rho_m$,其中 $\rho_m = -\max_i L_{m,i}(s_m)$ 是内切圆半径。应用平移裁剪 $r_{m,i}(p) = \text{clamp}(u_{m,i}(p) + \epsilon, 0, 1)$,最终窗口值为 $I_m(p) = \prod_{i=1}^3 r_{m,i}(p)^{\sigma_m}$。第十二,渲染图像通过按从前到后深度顺序累积所有重叠三角形的贡献获得,遵循标准alpha合成方程。第十三,可选的轻量级细化步骤将预测的三角形汤转换为完全opaque的、游戏引擎就绪的表示,通过50次迭代的不透明度选择阶段、边界顶点合并和修复阶段完成。
技术新颖性
FLAT的技术新颖性体现在多个方面:首次展示了显式的、非体积的表面图元可以从压缩的视频扩散潜空间在单次前向传播中直接解码, formulate了之前未被充分探索的前向扁平图元解码的参数化和训练问题。引入了射线中心的局部三角形参数化,使用约束的Cholesky风格形状变换和围绕射线对齐帧的残差旋转预测,相比直接预测世界空间旋转更数值稳定,避免了早期训练中的不稳定方向、消失渲染支持和模型发散问题。提出了改进的窗口函数,通过乘积形式而非原始三角形泼溅公式中的最大值操作,在可微三角形渲染期间改善梯度流动,这对稳定的前向潜变量模型尤为重要。首次在相同的潜变量解码条件下系统性地比较了3DGS、2DGS和三角形泼溅,表征了渲染质量、几何精度和下游网格兼容性之间的表示权衡。架构上,复用预训练视频VAE的RGB解码器主干而非从头训练小型transformer解码器,简化了具有挑战性问题的优化。提供了首个从前向潜在模型到游戏引擎兼容格式的完整管线,通过轻量级后处理将半透明前向预测转换为完全opaque的三角形。
实验结果
FLAT在前向3D场景生成任务上取得了显著成果,首次实现了从视频扩散潜空间直接解码三角形泼溅。在RealEstate10K数据集上,FLAT三角形模型达到PSNR 21.45、SSIM 0.710、LPIPS 0.245,在DL3DV数据集上达到PSNR 20.04、SSIM 0.627、LPIPS 0.314,几何质量(余弦相似性)在两个数据集上平均达到0.853。相比最先进的3DGS方法如Lyra(RealEstate10K: PSNR 21.79, SSIM 0.752, LPIPS 0.219),FLAT三角形模型在视觉质量上保持竞争力的同时,几何精度显著优于3DGS(余弦相似性0.116)和2DGS(余弦相似性0.587)。实验还训练了FLAT的3DGS和2DGS变体进行公平比较,3DGS变体达到最高视觉保真度(RealEstate10K: PSNR 22.39, SSIM 0.762, LPIPS 0.203),确认了训练管线和设计选择的有效性,而三角形模型在几何精度上占优。在网格转换评估中,FLAT三角形通过简单后处理获得的opaque mesh在RealEstate10K上达到PSNR 21.23、SSIM 0.749、LPIPS 0.388,相比3DGS网格(PSNR 14.18, SSIM 0.619, LPIPS 0.452)提升超过7dB,且顶点数仅0.5M远少于2DGS(5M)和3DGS(4M)。消融研究显示,所有组件的组合对稳定的前向三角形解码至关重要:预测世界空间旋转导致模型发散到完全噪声或空渲染;使用LongLRM Mamba解码器表现不佳,表明其容量不足以解码复杂的非体积图元;改变预测参数化降低训练稳定性;恢复原始窗口公式削弱梯度流动。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 新视角合成(RealEstate10K) | PSNR↑ | 21.45(三角形)/22.39(3DGS) | Lyra 3DGS: 21.79 | 三角形-0.34/3DGS+0.60 |
| 新视角合成(RealEstate10K) | SSIM↑ | 0.710(三角形)/0.762(3DGS) | Lyra 3DGS: 0.752 | 三角形-0.042/3DGS+0.010 |
| 新视角合成(RealEstate10K) | LPIPS↓ | 0.245(三角形)/0.203(3DGS) | Lyra 3DGS: 0.219 | 三角形+0.026/3DGS-0.016 |
| 几何质量(平均余弦相似性) | Cosine Similarity↑ | 0.853(三角形) | 2DGS: 0.587, 3DGS: 0.116 | +0.266 vs 2DGS/+0.737 vs 3DGS |
| 不透明网格转换(RealEstate10K) | PSNR↑ | 21.23 | 2DGS TSDF: 15.89, 3DGS GS2Mesh: 14.18 | +5.34 vs 2DGS/+7.05 vs 3DGS |
| 不透明网格转换(RealEstate10K) | 顶点数 | 0.5M | 2DGS: 5M, 3DGS: 4M | -4.5M vs 2DGS/-3.5M vs 3DGS |
局限与改进
作者承认FLAT在视觉质量上略逊于纯体积表示如3DGS,这是因为高斯的体积特性使其能够通过针状高斯自然处理各种3D结构如半透明雾和薄边缘,并且直接优化像素级指标如PSNR。此外,作者提到训练过程的计算成本较高,需要在8个H100 GPU上进行200,000次迭代,并采用从320p到768p的四阶段渐进训练。代码开源程度未在论文中明确说明。从技术角度观察,FLAT对相机轨迹的质量有一定依赖,作者观察到Uni3C在具有挑战性的轨迹上不能完美匹配输入相机条件,需要使用MapAnything预测重新计算生成视频的相机轨迹。对于稀疏视图覆盖,三角形的几何精度可能受限,虽然作者通过合成数据(25,000张来自S3OD的图像)和真实视频的混合训练缓解了这一问题。射线中心参数化虽然提高了稳定性,但在某些极端几何情况下可能仍产生退化三角形。改进的窗口函数虽然改善了梯度流动,但增加了额外的超参数(如 $\epsilon$ 和 $\sigma_m$)需要调优。
独立分析的弱点
FLAT存在几个可改进的弱点:首先,视觉质量略逊于纯体积表示,尤其在处理半透明物体、薄结构和雾状效果时,这是因为三角形本质上是扁平的、不透明的表面图元,难以模拟体积效应。改进方向可以是混合表示,结合三角形用于主结构和少量高斯用于体积效果。其次,相机轨迹依赖性强,作者需要使用MapAnything重新计算相机轨迹,这增加了管线的复杂性和潜在的误差累积。改进方向可以是在训练中引入更鲁棒的相机预测模块或数据增强来覆盖更多样的轨迹分布。第三,计算成本高,200,000次迭代在8个H100 GPU上的训练成本限制了广泛实验和超参数搜索。改进方向可以是更高效的架构设计、渐进式训练策略或知识蒸馏。第四,对相机质量敏感,从稀疏视图生成的几何精度可能受限。改进方向可以是引入多尺度特征融合或时序一致性约束来增强几何推理。第五,超参数敏感性,虽然比传统网格提取方法低,但窗口函数中的 $\epsilon$ 和 $\sigma_m$ 仍需要调优。改进方向可以是自适应参数化或学习这些超参数。第六,表面提取后的纹理质量可能下降,虽然作者包含修复阶段,但拓扑变化后的视觉保真度恢复可能不完美。改进方向可以是改进纹理传播算法或联合优化几何和外观。
未来方向
作者提出的未来方向包括鼓励进一步在显式几何精确的前向场景生成上的工作,以及可控生成模型与实时3D渲染之间的更紧密集成。基于FLAT的成果,可以延伸的研究方向包括:探索其他非体积图元如四边形、凸包或更高级的图元集合,在几何精度、渲染效率和表示复杂度之间找到更好的权衡。研究自适应表示选择,根据场景局部特性动态选择最适合的图元类型(如平坦区域用三角形,体积区域用高斯)。改进网格转换算法,实现更高质量的拓扑重建和纹理映射,减少视觉保真度损失。扩展到更大规模和更多样化的场景,如室外城市场景、动态场景和交互式环境。集成物理属性如材质、摩擦系数等,使生成的场景不仅几何精确而且物理上可模拟。探索实时编辑能力,如形状修改、纹理替换、场景合成等,增强实用性。研究多模态生成,如从文本、草图、点云等多种输入生成3D场景。改进效率,通过模型压缩、量化或更高效的架构降低推理延迟,使方法能够在移动设备上运行。探索与游戏引擎的深度集成,提供即插即用的资产生成管线。研究长期训练和持续学习,使模型能够适应新的场景分布和用户偏好。
复现评估
FLAT的复现评估涉及多个方面:代码开源程度在论文中未明确说明,项目页面提到 https://flat-splat.github.io,但没有明确说明代码和模型权重是否开源。数据方面,训练使用真实视频(RealEstate10K和DL3DV)和合成数据(25,000张来自S3OD的图像)的混合。真实数据使用RealCam-Vid提供的相机标注,合成数据使用Uni3C模型生成视频并存储轨迹元数据。所有场景使用MapAnything预测度量相机姿势和深度进行尺度统一。伪ground truth表面法线由NormalCrafter预测。评估数据集RealEstate10K和DL3DV是公开的,但相机标注来自RealCam-Vid。算力需求很高,训练在8个H100 GPU上进行200,000次迭代,采用从320p到768p的四阶段渐进训练,这限制了大多数研究机构的复现能力。实现细节使用Uni3C作为视频模型,构建在Wan-2.1之上,生成49到81帧,分辨率432×768。VAE编码器暂时将视频下采样因子 $r_t = 4$,空间下采样 $r_s = 8$。使用AdamW优化器,学习率 $1e^{-4}$。超参数包括损失权重 $\lambda_{rgb} = 1.0$、$\lambda_{perc} = 0.5$、$\lambda_D = 0.01$、$\lambda_N = 0.01$ 和 $\lambda_O = 0.001$,以及不透明度阈值40%。考虑到高算力需求、复杂的训练管线和多数据源混合,复现难度较高,除非作者提供完整的代码、训练脚本和预训练模型。
论文图表
这张图展示了FLAT方法的核心概念:从视频扩散潜空间直接回归软三角形,实现几何精确和高保真的场景生成。图中显示了一个3D场景的可视化,其中三角形表示的3D几何结构清晰地展示了场景的表面细节。图示说明了FLAT如何将单张输入图像通过视频扩散模型转换为三角形泼溅表示,最后生成新视角的渲染结果。视觉上可以看到生成的场景具有清晰的几何结构和高质量的外观。
这张图对理解论文至关重要,因为它直观地展示了FLAT方法的核心创新:从前向角度直接解码三角形而非传统的高斯表示。它帮助读者快速理解方法的工作原理和输出结果,展示了在几何精度和视觉质量之间的平衡。这张图在论文的Introduction部分首先出现,为后续的技术细节提供了直观基础。