Lift4D:协调单视图3D估计用于野外4D重建 Lift4D: Harmonizing Single-View 3D Estimation for 4D Reconstruction In-the-Wild
从单眼视频重建完整4D动态对象,处理大变形和遮挡
前置知识
3D Gaussian Splatting
一种实时3D场景渲染方法,用高斯椭球体表示3D空间中的点,每个高斯包含位置、旋转、尺度、不透明度和颜色等属性。通过可微的泼溅渲染,可以从任意视角高效渲染高质量图像。相比NeRF,它训练更快、渲染实时,已成为3D重建的主流表示方法。
本文采用可变形3D Gaussian Splatting作为4D表示的基础,理解其表示能力和渲染机制是理解本文方法的基础。
测试时优化
一种推理范式,不需要预训练好端到端模型,而是在推理时针对每个输入样本进行优化。通常包含一个可学习的表示和优化目标,通过梯度下降迭代优化。优点是可以充分利用每个输入的独特信息,缺点是计算成本较高。
Lift4D整个方法基于测试时优化框架,理解这一范式有助于理解为什么本文不依赖大规模4D训练数据。
扩散模型分数蒸馏
一种将预训练扩散模型作为先验用于优化的技术。通过将渲染图像编码到扩散潜空间,采样时间步和噪声,用扩散模型预测去噪后的图像,计算渲染图像和预测图像之间的损失。这相当于从扩散模型中提取知识来指导优化过程。
本文使用视图条件图像扩散先验来补全未观察区域的外观,分数蒸馏是实现这一点的关键技术。
线性混合皮肤
一种计算机动画中常用的骨架驱动变形技术。每个顶点受若干控制节点影响,通过权重函数确定影响程度。顶点的最终位置是控制节点变换后的位置的加权组合。权重通常基于距离衰减,确保变形的平滑性和局部性。
本文用稀疏控制节点和LBS来参数化4D变形,理解这一机制有助于理解如何高效表达复杂非刚性运动。
研究动机
现有4D重建方法面临两大根本局限性。第一类方法直接从视觉输入预测4D表示,受限于4D训练数据的稀缺性。这些方法要么依赖类别特定模板,限制在狭窄的对象域内,要么在缺乏野外泛化所需多样性的合成资产上训练,导致在复杂野外视频上表现不佳。第二类优化方法通过利用更广泛适用的3D先验来规避数据稀缺问题,但难以连接静态先验和动态序列之间的差距。那些利用视频扩散先验的方法在大幅视角变化时性能下降,而仅在初始化时使用图像到3D先验的方法在静态先验和动态序列之间存在域间隙,导致在大变形和遮挡下几何、运动或外观质量下降。野外视频中的对象类别不受约束,可能经历大变形,并遭受遮挡,加上单视点的根本模糊性,使得这些局限性更加严重。
本文的目标是本文的目标是从单眼野外视频中开发一种计算方法来推断通用对象的完整4D重建:给定单个视频,恢复每个动态对象的完整360度几何和外观,以及其在帧间的变形。具体来说,希望重建的4D表示能够处理复杂的野外场景,包括大非刚性变形、严重遮挡,以及从未被相机观察到的区域。不需要依赖多视图数据或类别特定模板,能够从随意的野外视频中可靠地重建动态4D表示。
与已有工作不同的是,本文的独特切入角度在于认识到最先进的单视图3D重建方法(如SAM3D)可以被调整以在优化过程中提供强大的4D先验。关键洞察是虽然天真地独立重建每个视频帧会产生时间上不一致的几何形状,但通过引入因果潜在条件策略可以使这些逐帧3D重建在时间上连贯。更进一步,这些表示仍然是逐帧的,不形成一个在时间上经历变形的连贯3D结构,因此引入了由稀疏控制节点参数化的时变可变形3D表示。这一设计与现有方法形成本质区别:不是依赖4D训练数据,不是仅在初始化时使用先验,也不是在优化过程中使用视频扩散先验,而是通过协调图像到3D重建作为4D推理的先验,实现可泛化的4D重建。
核心方法
Lift4D是一个测试时优化框架,整体思路分为三个阶段。首先,将现有的单视图3D重建模型(SAM3D)调整为产生时间一致的逐帧预测,通过因果潜在条件提供可变形3D高斯泼溅表示的连贯初始化。然后,将逐帧重建蒸馏到一个变形的规范表示中,该表示通过稀疏控制节点和变形MLP参数化,通过渲染监督与输入视频对齐。最后,使用遮挡感知优化方案和视图条件扩散先验来细化外观,处理野外视频中常见的复杂遮挡和未观察区域。整体上,方法通过课程式的测试时优化,将大规模预训练的2D和3D先验分别路由到它们对野外内容可靠的角色中,几何优化阶段利用图像到3D模型的详细几何,外观优化阶段利用扩散先验的合理外观。
核心创新点是因果潜在条件传递和遮挡感知优化的结合。因果潜在条件通过在ODE级别重新使用前一帧的去噪潜变量作为下一帧的噪声先验来强制时间一致性,无需重新训练。具体来说,对于第一帧I0从纯高斯噪声Z00到N(0, I)开始去噪,得到干净的潜变量Z01。对于后续帧i,不是从独立噪声开始,而是在时间步t0到(0, 1]预热ODE,通过混合新鲜噪声与前一帧的干净潜变量:Zit0 = (1-t0)Zi0 + t0 Zi-1,然后从t0积分到1。参数t0权衡时间一致性和逐帧保真度。遮挡感知优化通过深度比较检测被场景内容遮挡的对象区域,结合颜色匹配和扩散先验来补全这些区域,确保4D表示即使在监督不完整的情况下也能保持质量。
方法步骤详情
方法包含三个主要步骤。第一步是因果单视图3D重建:给定视频I = {Ii}i=1到N和对象掩码M = {Mi}i=1到N,使用流匹配图像到3D模型vθ对每个帧进行去噪。通过因果潜变量传播强制时间一致性,每个去噪后的潜变量通过高斯泼溅解码器解码为逐帧高斯Gi和对象到相机变换Ti。第二步是重建引导的可变形3D优化:在规范高斯G★表面初始化Np = 1024个稀疏控制节点{pk}k=1到Np,变形MLPψ预测每个节点的时间变化变换[Rk|tk]k|i到SE(3)。通过线性混合皮肤变形每个规范高斯,目标是最小化3D重建损失Lrec = LCD + Lmv,其中LCD是Chamfer距离对齐位置,Lmv是随机相机视角的多视图渲染损失。第三步是遮挡感知外观重建:冻结变形MLP,添加更密集的控制节点来细化外观。通过深度比较Miocc = Dscene < Diπc来land Mi检测遮挡区域,使用颜色匹配生成完成参考图像Iifull = Miocc到odot Ĩiπc + Mi到odot Ii,结合渲染损失Lrender和分数蒸馏损失LSDS优化外观。
技术新颖性
技术新颖性体现在多个方面。首先,因果潜在条件传递是首个在推理时(无需重新训练)使逐帧3D重建时间一致的方法,通过在ODE级别耦合相邻潜变量实现,与需要重新训练的时序模型形成本质区别。其次,可变形4D表示将几何和外观优化解耦,先通过逐帧重建蒸馏变形场,再冻结变形细化外观,避免了已有方法中几何和外观相互干扰的问题。再次,遮挡感知渲染监督是首个在野外视频中显式建模场景-对象遮挡的方案,通过深度比较和颜色匹配生成干净的参考图像,而不是天真地对原始视频进行光度监督。最后,方法采用课程式优化策略,将几何和外观先验分别路由到它们最可靠的阶段,与同时使用所有先验的方法相比,提高了优化稳定性和最终质量。
实验结果
Lift4D在多个数据集上展现出优异的4D重建性能。在野外Pexels数据集上,CLIP图像得分达到0.780,CLIP文本得分达到0.286,端点误差(EPE)仅为0.072,显著优于所有基线方法。具体来说,CLIP图像得分比最佳基线STAG4D的0.757提升了3.0%,比L4GM的0.756提升了3.2%;CLIP文本得分比STAG4D的0.264提升了8.3%;EPE比最佳基线PAD3R的0.119降低了39.5%。在DAVIS数据集上,CLIP图像得分达到0.715,CLIP文本得分达到0.292,EPE为0.161,CLIP图像得分比最佳基线V2M4的0.637提升了12.2%,CLIP文本得分比L4GM的0.235提升了24.3%。在合成Consistent4D数据集上,LPIPS为0.116,FVD为592.44,CLIP得分为0.950,LPIPS比最佳基线DM4D的0.128降低了9.4%,FVD比最佳基线DM4D的688.84降低了14.0%,CLIP得分比DM4D的0.936提升了1.5%。消融实验显示,去除因果潜变量传播导致LPIPS从0.116增加到0.120,FVD从592.44增加到627.90,CLIP得分从0.950下降到0.945;去除运动正则化Lreg导致LPIPS增加到0.122,FVD增加到794.82,CLIP得分下降到0.943;去除图像先验LSDS导致LPIPS增加到0.170,FVD增加到1242.32,CLIP得分下降到0.848,证明图像先验对补全未观察区域至关重要。单个包含32帧的对象视频在一张H200卡上约30分钟完成重建,优化使用10,000次3D优化迭代和总共20,000次迭代,AdamW优化器。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 野外4D重建 | CLIP图像得分 | 0.780 | STAG4D 0.757 | +3.0% |
| 野外4D重建 | CLIP文本得分 | 0.286 | STAG4D 0.264 | +8.3% |
| 野外4D重建 | 端点误差(EPE) | 0.072 | PAD3R 0.119 | -39.5% |
| 合成4D重建 | LPIPS | 0.116 | DM4D 0.128 | -9.4% |
| 合成4D重建 | FVD | 592.44 | DM4D 688.84 | -14.0% |
| 合成4D重建 | CLIP得分 | 0.950 | DM4D 0.936 | +1.5% |
局限与改进
作者承认了几个局限性。首先,方法依赖初始3D重建的质量,当SAM3D产生糟糕的几何或布局时,这些错误会传播到规范表示或遮挡感知帧重建。这通常发生在高帧率和薄对象的视频中,由于逐帧变换中的跳变,优化难以解释。其次,条件时间步t0需要平衡,高t0可能抑制合理的变形,低t0可能无法防止几何闪烁,默认使用t0 = 0.2,但某些序列可能需要调优,刚性对象序列可能受益于更高的t0。此外,观察到方法在处理更复杂场景交互时可能受限,例如人类抓取等涉及复杂交互的场景。最后,方法依赖于SAM3D和Depth Anything 3等预训练模型的质量,这些模型在极端场景下的失败会影响最终重建结果。
独立分析的弱点
方法存在几个潜在的弱点。第一,级联架构导致误差累积,SAM3D初始预测的错误会无监督地传播到后续优化阶段。具体来说,当SAM3D产生错误的布局或几何时,变形MLP可能学习到错误的运动场,导致最终4D表示扭曲。改进方向可以是引入一个自监督的对齐步骤,在优化过程中迭代修正初始3D重建,或者添加一个几何一致性损失来检测和纠正严重错误。第二,时间一致性参数t0是固定的,无法适应不同视频的动态特性。对于变形较大的对象(如跳舞的人),可能需要较低的t0以允许更多灵活性,而对于刚性对象(如移动的汽车),可能需要较高的t0以增强一致性。改进方向可以是自适应选择t0,根据估计的运动幅度或相机运动自动调整。第三,方法假设对象在视频中的可见部分足够支持重建,但对于极端遮挡(如对象大部分时间被遮挡),仅靠扩散先验可能无法恢复准确的几何。改进方向可以是利用视频时序信息进行传播,或者引入对象类别先验来约束补全。第四,优化时间较长(30分钟),对于实时应用不够高效。改进方向可以是学习一个初始化网络,预测规范高斯和变形的初始估计,减少优化迭代次数。
未来方向
作者提出改进底层架构的几何估计骨干是一个有前景的方向,进一步增强Lift4D对更复杂场景(如人类抓取)的泛化能力。基于论文成果,可以延伸多个未来研究方向。首先,可以探索更丰富的变形表示,如使用图神经网络或Transformer来建模控制节点之间的关系,以捕捉更复杂的非刚性运动模式。其次,可以扩展到多对象场景,处理对象间的交互和遮挡,这需要引入对象间关系建模和联合优化策略。第三,可以研究更有效的先验路由策略,自动判断何时使用几何先验、何时使用外观先验、何时同时使用,减少手动设计优化课程。第四,可以探索交互式4D重建,允许用户通过稀疏标注(如关键点、轮廓)来指导优化,提高重建的准确性。第五,可以将方法与实时渲染技术结合,通过渐进式优化或部分优化实现实时预览,提高用户体验。第六,可以研究4D重建的下游应用,如4D编辑、4D动画合成、4D场景理解等,充分发挥完整4D表示的价值。
复现评估
论文提供了项目网页,但未在论文中明确说明代码和数据的开源情况。实验设置描述较为详细,包括超参数选择(t0 = 0.2,Np = 1024个控制节点,每个高斯k=4个最近控制节点,扩散时间步从[0.2, 0.5]均匀采样,指导尺度设置为3.0)。评估数据集包括合成Consistent4D数据集(7个视频,每个32帧)、野外Pexels数据集(10个视频)和DAVIS数据集(8个视频),所有视频长度在77到100帧之间。计算需求为单个包含32帧的对象视频在一张H200卡上约30分钟完成重建,优化使用10,000次3D优化迭代和总共20,000次迭代。评估指标包括LPIPS、FVD、CLIP得分和EPE,其中LPIPS衡量感知相似性,FVD衡量视频真实性和时序一致性,CLIP得分衡量语义相似性,EPE衡量3D运动准确性。复现需要访问SAM3D、Depth Anything 3、Stable Zero123等预训练模型,这些模型大多公开可用。总体而言,基于提供的详细信息,复现难度中等,主要挑战在于计算资源需求和多个预训练模型的集成。
论文图表
这个表格展示了消融实验的结果,评估了不同3D或图像先验的影响。表格包含四个设置:完整Lift4D、去除因果潜变量传播(Batch-wise {Gi})、去除运动正则化Lreg、去除图像先验LSDS。完整Lift4D表现最佳(LPIPS 0.116, FVD 592.44, CLIP 0.950)。去除因果潜变量传播导致性能下降(LPIPS 0.120, FVD 627.90, CLIP 0.945),去除Lreg导致更大下降(LPIPS 0.122, FVD 794.82, CLIP 0.943),去除LSDS导致最严重的下降(LPIPS 0.170, FVD 1242.32, CLIP 0.848)。
这个表格对理解论文的设计选择和每个组件的贡献至关重要。它通过消融实验定量展示了因果潜变量传播、运动正则化和图像先验的重要性。特别是去除LSDS导致的性能下降最大(FVD从592.44增加到1242.32,CLIP得分从0.950下降到0.848),证明了图像先验对补全未观察区域的关键作用。这些结果直接支持了论文的核心设计决策。
这张图展示了在合成Consistent4D数据集上重建4D对象的定性结果。图中包含多个合成对象的输入视频和Lift4D与多个基线(L4GM, STAG4D, PAD3R, DreamMesh4D, V2M4)的重建结果对比。Lift4D能够鲁棒地重建丰富和完整的4D对象几何和纹理,对于更简单的合成案例也是如此。相比之下,基线方法恢复更简单的几何和纹理或显示错误的变形。图中展示了多种合成对象,证明方法在理想条件下的有效性。
这张图对理解论文在合成数据上的表现至关重要。它展示了Lift4D在有真实标注的数据集上的定性表现,与Table 2的定量结果相互支持。图中证明了方法即使在相对简单的合成案例上也能优于现有方法,这表明方法的优势不仅限于复杂的野外场景,而是具有普遍适用性。这也支持了论文中声称的方法鲁棒性。