跳转至

GenRecon: Bridging Generative Priors for Multi-View 3D Scene Reconstruction

会议: NeurIPS 2026
arXiv: 2605.23888
OpenReview: https://openreview.net/forum?id=eUqP4liWTd
论文: https://arxiv.org/abs/2605.23888
领域: 重建与渲染
关键词: 3D 场景重建、生成先验、多视角三维生成、流匹配、PBR 网格重建

一句话总结

将多视角三维场景重建构建为基于重叠空间分块的条件 3D 生成任务,通过基于投影的空域锚定与置换不变融合机制将 Trellis.2 物体级生成先验提升至场景级,直接输出带 PBR 材质的高保真连续网格。

研究背景与动机

从多视角 RGB 图像重建高保真三维场景是计算机视觉与图形学的核心课题,对于 AR/VR、具身机器人、物理仿真以及数字内容创作具有不可替代的价值。在游戏引擎导入和沉浸式渲染等下游创作场景中,三维重建不仅要求恢复粗糙的几何轮廓,更亟需具有高几何完整度、光滑连贯表面以及支持任意重光照与材质编辑的 PBR(基于物理的渲染)三角网格资产。然而,从稀疏多视角图像反求三维空间本质上是一个严重欠约束的逆问题;真实拍摄中不可避免地存在大面积弱纹理区域、重复结构、大视差遮挡以及非朗伯体反射,传统的跨视角光度一致性约束在这些区域极易失效。

现有的三维重建路线在面对内容创作的高质量需求时普遍陷入两难。经典的逐场景优化方法(如 COLMAP、NeuS、MonoSDF 以及 3D/2D Gaussian Splatting)完全受制于多视角观测的几何交会,在弱纹理与视线遮挡区域频繁产生破洞、悬浮噪点或过度平滑的畸变;新兴的前馈式几何回归模型(如 DUSt3R、Depth Anything 3、FineRecon)虽具备前馈先验,但作为确定性回归网络,其无法为未观测或严重遮挡区域合理补全几何结构,且输出大多局限于非结构化的稠密深度图、点云或离散高斯基元,难以直接导入标准光栅化管线。与此同时,以 Trellis.2 为代表的三维生成模型在物体级资产合成上展现出惊人的几何逼真度与完整 PBR 材质生成能力,但其原始设计严重依赖单张无位姿图像输入,缺少绝对空间坐标系锚定与多视角几何控制,无法直接拓展至大尺度场景的联合重建。

将传统重建严格的几何对齐能力与生成模型强大的结构补全先验有机融合,成为打破这一瓶颈的关键切入点。核心 idea:将多视角场景重建重构为在重叠空间分块上的条件 3D 流匹配生成过程,设计基于相机投影与置换不变聚合的 3D 条件注入通路将物体级生成先验引入绝对场景坐标系,并在全局流匹配轨迹中协同去噪生成无缝拼接的场景级 PBR 连续网格。

方法详解

整体框架

GenRecon 的核心理念是将场景级重建解耦为一组在空间中相互重叠的局部 3D 分块(chunks)的联合条件生成。系统输入为稀疏且无序的 \(N\) 张 RGB 图像 \(\{I_n\}_{n=1}^N\)。首先运行运动恢复结构(COLMAP)估计相机内参 \(\{K_n\}\)、外参 \(\{T_n\}\) 以及稀疏场景点云 \(P\);通过对点云执行统计与半径滤波,提取场景的物理三维边界包围盒 \(\mathcal{B}\)。随后,系统将 \(\mathcal{B}\) 划分为一组边长为 \(L\) 的固定立方体分块集合 \(\mathcal{C} = \{c_1, \dots, c_K\}\),相邻分块间保持至少 \(m=0.25\) 的空间重叠边距。

在生成模型层面,GenRecon 继承了先进的物体级生成模型 Trellis.2,该模型基于流匹配(Flow Matching)架构,分阶段生成粗糙占用网格(分辨率 16)以及高精度的稀疏几何与 PBR 纹理隐变量(分辨率 32)。为了使单视角无位姿模型能够服从真实多视角图像的相机位姿引导,GenRecon 构建了一条全局 3D 条件通路:使用预训练 DINOv3 提取各视角 2D 特征,通过针孔投影变换将多视角特征抬升至全局三维体素空间,并执行置换不变的多视角特征统计融合,构建全局条件网格 \(G\)。在推理去噪阶段,所有分块在单次流匹配轨迹中依据各自裁剪的局部条件切片 \(G^{(c)}\) 协同去噪,通过在重叠区域加权平均与边界敏感更新消除接缝,最终经由 Trellis.2 解码器输出完整的场景级 PBR 三角网格 \(\mathcal{M}\)。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    A["多视角 RGB 图像 + SfM 相机位姿与稀疏点云"] --> B["场景外包围盒估计与重叠分块划分"]
    B --> C["基于投影的空域锚定多视角特征提升<br/>DINOv3 2D 特征逆投影至局部 3D 体素网格"]
    C --> D["置换不变的跨视角残差融合机制<br/>计算 cross-view 均值/方差与自适应加权聚合"]
    D --> E["全局 3D 条件裁剪与注入 DiT 块"]
    E --> F["重叠分块联合流匹配与边界敏感同步<br/>MultiDiffusion 协同去噪与边界过滤聚合"]
    F --> G["Trellis.2 解码输出场景级 PBR 连续网格 ℳ"]

关键设计

1. 基于投影的空域锚定多视角特征提升:将非位姿物体级先验锚定至绝对场景坐标系

Trellis.2 等现代 3D 生成模型在预训练时仅接受单张无位姿图像,生成的资产尺度、中心位置与朝向完全取决于训练集的隐式分布;直接将其用于场景重建会导致各分块尺度错乱、朝向漂移,根本无法与真实相机视角对齐。针对该痛点,GenRecon 摒弃了传统的跨注意力图像嵌入输入方式,设计了显式绑定的三维投影条件通路。

具体而言,对于每个局部空间分块 \(c\) 及其可见视角集合 \(\mathcal{V}_c\),分块定义在其规范坐标系立方体 \(V_c = [0, L]^3\) 中,世界位移为 \(t_c \in \mathbb{R}^3\)。网络首先利用预训练 DINOv3 提取各视角的稠密 2D 特征图 \(F_n\)。对于分块体素网格中的任意物理坐标 \(x \in V_c\),利用相机几何将其透视投影至第 \(n\) 个相机的图像平面: $\(\pi_n(x) = K_n T_n^{-1}(x + t_c)\)$ 进而双线性插值检索对应的 2D 视觉特征,形成单视角 3D 特征体素网格 \(G_n^{(c)}(x) = F_n(\pi_n(x))\)。这一投影操作将 2D 图像线索精确锚定到了确定的三维几何位置上,使生成模型直接感知绝对物理空间坐标与相机透视约束,摆脱了以往模型需要隐式揣测尺度的缺陷。在 DiT 的 30 个去噪 Transformer 块中,条件特征经由零初始化投影层以残差形式注入,确保模型在训练初期完好保留 Trellis.2 既有的结构与材质先验。

2. 置换不变的跨视角残差融合机制:解耦视角顺序并自适应加权多视角观测

在无约束的多视角场景采集中,不同空间分块的可见视角数量极不均匀,且输入视角缺乏自然的先后顺序。若使用依赖序列顺序的自注意力或固定通道拼接,网络容易过拟合到特定的视角排列,且无法灵活适应视角增减与视角遮挡。

GenRecon 借鉴 IBRNet 的统计聚合思想,提出了一种置换不变的体素级特征自适应融合网络。对于某一空间体素 \(x\),设其聚合了来自所有可见视角的特征集合 \(\{f_i\}_{i=1}^N\)。算法首先计算跨视角的全局均值 \(\mu\) 与方差 \(\sigma^2\): $\(\mu = \frac{1}{N}\sum_{i=1}^N f_i, \quad \sigma^2 = \frac{1}{N}\sum_{i=1}^N f_i^2 - \mu^2\)$ 全局统计量反映了该体素在不同视角下的一致性与方差(方差较小通常意味着无遮挡且投影准确)。随后,两组参数共享输入的紧凑 MLP(三层特征网络 \(\mathrm{MLP}_{\text{feat}}\) 与两层权重网络 \(\mathrm{MLP}_{\text{weight}}\))分别提取局部特征残差与视角贡献打分: $\(f'_i = \mathrm{MLP}_{\text{feat}}([f_i, \mu, \sigma^2]), \quad w_i = \mathrm{MLP}_{\text{weight}}([f_i, \mu, \sigma^2])\)$ 最终体素输出特征为均值特征叠加上 softmax 归一化加权的视角残差: $\(f_{\text{out}} = \mu + \sum_{i=1}^N \alpha_i f'_i, \quad \text{其中 } \alpha_i = \mathrm{softmax}_i(w_i)\)$ \(\mathrm{MLP}_{\text{feat}}\) 的最后一层同样零初始化,使得训练起点严格等价于视角均值池化。该机制既严格保证了跨视角的置换不变性,又赋予网络自适应剔除视线遮挡、高光畸变及错误投影的能力。

3. 重叠分块联合流匹配与边界敏感同步:无缝缩放至大范围场景并消除拼接缝隙

单次前馈生成大场景会遭遇显存爆炸与分辨率稀释,而简单的独立分块去噪则不可避免地在分块交界处产生几何错位、撕裂与纹理断层。为了实现大场景的无缝重建,GenRecon 构建了基于 MultiDiffusion 的协同去噪机制,并在精细几何与纹理阶段设计了边界敏感同步策略。

算法在全局场景空间维护一个覆盖整个包围盒的连续噪声隐变量网格 \(z_t\)。在推理流匹配的每一步 \(t \in [T, \dots, 0]\)(共 12 步),从 \(z_t\) 中分别裁剪出各个分块对应的局部隐变量 \(z_t^{(c)}\),送入分块去噪器 \(\Phi\) 计算局部预测 \(\hat{z}_{t-1}^{(c)}\)。在分块重叠区域,全局隐变量通过归一化掩码加权平均进行同步更新: $\(z_{t-1}(x) = \frac{1}{\sum_{c\in\mathcal{C}} M_c(x)} \sum_{c\in\mathcal{C}} M_c(x) \hat{z}_{t-1}^{(c)}(x)\)$ 其中 \(M_c(x) \in \{0, 1\}\) 表示体素 \(x\) 是否位于分块 \(V_c\) 内部。在稀疏高分辨率的形状与 PBR 纹理阶段,由于分块外侧边缘体素缺乏上下文感受野,直接参与平均会引入系统性边缘偏差。GenRecon 引入边界敏感变体:强制令分块最外层 \(b=1\) 排的边界体素不贡献累加权重,但仍接收来自邻近重叠分块内部可靠体素的平均更新。通过这种“内部主导、边缘愈合”的同步策略,各分块在扩散过程中持续进行流形级通信,彻底消除了拼接接缝。

损失函数 / 训练策略

模型训练采用高效的参数微调方案(PEFT)。Trellis.2 主干参数全部冻结,仅在所有的注意力层上附加 rank 为 8、scaling 为 16 的 LoRA 适配器,并联合训练 3D 条件注入网络。训练数据包含来自合成室内数据集 SAGE-10k 的 5,000 个房间(生成 15,000 个随机旋转分块)以及 3D-FRONT 的 1,200 个户型(用于占用预测微调)。优化器选用 AdamW,全局批大小为 32;新引入的条件模块学习率设为 \(1 \times 10^{-4}\)(权重衰减 0.01),LoRA 参数学习率设为 \(3 \times 10^{-5}\)(无权重衰减)。训练时每分块随机采样 1 至 16 个输入视角,并以 0.1 的概率丢弃条件以支持无分类器引导(CFG)。推理时分块大小自适应设为 \(1.11 \times \text{场景高度}\),重叠率 \(m=0.25\),采用 12 步流匹配采样完成生成。

实验关键数据

主实验

评估在真实室内场景数据集 ScanNet++(25 个未见测试场景,每场景仅给 8 张稀疏输入视角)与合成数据集 3D-FRONT 上展开,基线涵盖 2DGS(无先验逐场景优化)、MonoSDF(单目先验引导神经隐式表面)、Depth Anything 3(DA3,单目深度基础模型 + TSDF 融合)、FineRecon(基于 DA3 的 3D 几何精细化)以及 Murre(扩散深度先验 + 3D 条件)。真实场景重建定量指标如下表所示:

方法 深度 MAE (m) ↓ 深度 RMSE (m) ↓ 法线角度误差 (°) ↓ 法线 LPIPS ↓ 完整度 (Completeness) ↑ 3D Chamfer (m) ↓ 3D F-score@10cm ↑ 3D 法线一致性 ↑
2DGS 0.2599 0.4535 35.9195 0.7047 0.7202 0.2181 0.4870 0.4903
MonoSDF 0.2496 0.3812 20.2113 0.4201 0.9716 0.1250 0.6317 0.7039
DA3 (TSDF) 0.1235 0.2428 26.2189 0.6562 0.8404 0.0998 0.6905 0.7140
FineRecon 0.0939 0.2147 19.1835 0.4888 0.8707 0.0819 0.7694 0.7851
Murre 0.1072 0.2524 23.2825 0.6141 0.7300 0.0978 0.7263 0.7125
Ours (GenRecon) 0.1024 0.2100 16.0179 0.3411 0.9904 0.0688 0.7771 0.7860

在合成数据集 3D-FRONT 上,GenRecon 的优势进一步扩大,3D Chamfer 误差达 0.0638 m(此前最优 Murre 为 0.1584 m,降低近 60%),3D F-score@10cm 达到 0.8655(比 Murre 提升 0.2641),表面完整度达到 0.9871。

消融实验

消融实验在 SAGE-10k 验证集上系统评估了 3D 条件注入通路与输入视角数量对生成质量的影响:

实验配置 输入视角数 3D 条件通路 3D Chamfer (m) ↓ F-score@10cm ↑ 法线一致性 (20cm截断) ↑ 说明
原生 Trellis.2 1 视角 无 0.3450 0.2067 0.2341 缺乏场景级可见性与位姿约束,生成崩溃
仅场景微调 1 视角 无 3D 条件 0.2578 0.3666 0.3549 局部结构合理但全局位姿完全错位
完整 GenRecon 1 视角 有 0.1345 0.6312 0.6349 单视角即可实现精准的位姿几何对齐
完整 GenRecon 2 视角 有 0.0812 0.7608 0.7688 观测增加显著提升几何精度与覆盖率
完整 GenRecon 4 视角 有 0.0420 0.9070 0.8411 进一步减少遮挡区域的幻觉补全
完整 GenRecon 8 视角 有 0.0291 0.9683 0.8804 完整配置,各指标达到极高精度

关键发现

  • 3D 条件通路是位姿控制的根本使能器:消融表表明,不引入投影 3D 条件通路的场景微调模型(1img no3Dcond)Chamfer 误差高达 0.2578 m,生成的几何仅是无序碎片;加入 3D 条件通路后误差立降至 0.1345 m,证实了显式反投影对空间位姿约束的关键作用。
  • 极端视角稀疏下的鲁棒性:在仅有 1 张输入图像的极端条件下,GenRecon 依靠强大的物体先验依然能重建出位姿正确的完整 3D 几何(F-score 达 0.6312);随着视角数增至 8 张,F-score 逼近 0.97,展示出平滑的可扩展性。
  • 超越确定性回归的完整性:ScanNet++ 真实测试中,GenRecon 取得了 0.9904 的超高完整度(Completeness),显著优于 2DGS(0.7202)和 Murre(0.7300),在家具底部、墙角等遮挡死角展现出生成先验特有的合理填补能力。

亮点与洞察

  • 将单图物体级生成先验泛化为多图场景级重建:无需从头预训练庞大的场景级生成模型,通过投影条件路径与 LoRA 微调巧妙复用了 Trellis.2 沉淀的高保真几何与 PBR 材质先验,极大地节省了算力并提升了资产生成品质。
  • 边界敏感的多分块流匹配同步机制:在 MultiDiffusion 聚合中主动剔除最外层感受野受限的体素贡献,让可信的内部预测平滑接合缝隙,为扩散模型拼接大尺度三维场景提供了可直接复用的 Engineering Trick。
  • 端到端原生输出 PBR 连续网格:摒弃了点云后处理 Poisson 重建或 Gaussian 离散基元难以编辑的缺点,直接输出包含 Albedo、Roughness、Metallic 属性的标准三角网格,可无缝对接标准游戏引擎并支持物理重光照。

局限与展望

  • 作者承认的局限:对非朗伯体表面(如大面积透明玻璃、高反光镜面)的重建质量有所下降,主因是训练集 SAGE-10k 中缺乏此类材质标注;当前分块划分主要针对垂直高度在 5 米以内的常规室内房间,拓展至多层或大跨度建筑需要自适应空间分块方案;在输入视角极度匮乏且缺乏几何线索时,强生成先验可能产生合理的幻觉(hallucination)。
  • 实验与现实差距:在真实场景 ScanNet++ 的纹理微调上,由于先验模型依赖 VLM 预测的伪材质进行监督,重建的 PBR 贴图在绝对物理测量精度(如绝对双向反射分布函数 SVBRDF)上与专用材质估计模型仍有差距。
  • 改进方向:可结合视觉基础模型提取的开集语义与材质法线先验增强镜面/玻璃鲁棒性,并在分块划分中引入八叉树或自适应层次结构以扩展至大规模室外场景。

相关工作与启发

  • vs 2DGS / NeuS (逐场景几何优化):传统方法依赖密集的射线相交与光度一致性约束,在稀疏视角与弱纹理区域容易产生大量空洞与漂移浮点;GenRecon 引入强生成先验,即便在单视角或大面积遮挡区域也能生成水密、完整的物体与场景结构。
  • vs Depth Anything 3 / FineRecon (前馈式深度回归与融合):回归模型仅能对视野内表面进行确定性估计,输出往往是非结构化的深度图或碎片化 TSDF,无法补全未观测几何;GenRecon 直接在 3D 潜空间中执行生成补全,输出带 PBR 材质的完整网格。
  • vs Pixal3D / ReconViaGen (多视角生成模型):并发工作 Pixal3D 虽然也采用了体素特征投影机制,但仍局限于单物体级别的无材质几何生成;GenRecon 突破了场景尺度的空间拓展难题,并实现了完整的 PBR 纹理联合生成。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 首次成功将前沿物体级 3D 生成先验通过投影条件与流匹配同步拓展至场景级 PBR 网格重建。
  • 实验充分度: ⭐⭐⭐⭐⭐ 兼顾 ScanNet++ 真实场景与 3D-FRONT 合成场景,包含详尽的 2D/3D 几何评估、光照渲染消融与视角规模分析。
  • 写作质量: ⭐⭐⭐⭐⭐ 动机与逻辑极其清晰,方法论严密,图文对照自洽。
  • 价值: ⭐⭐⭐⭐⭐ 成功连接了三维生成与多视角重建两大领域,为下游游戏制作、仿真与具身智能提供了极高实用价值的资产重建范式。