跳转至

RPBG: Towards Robust Neural Point-based Graphics in the Wild

会议: ECCV 2024
arXiv: 2405.05663
代码: https://github.com/QT-Zhu/RPBG
领域: 3D视觉 / 神经渲染
关键词: 点云渲染, 神经重渲染, 鲁棒性, 退化感知卷积, 新视角合成

一句话总结

本文针对Neural Point-based Graphics (NPBG)在真实场景中的鲁棒性不足问题,提出RPBG,通过退化感知卷积模块、注意力驱动的点可见性校正、轻量级背景建模和点云增强,在不修改点栅格化流程的前提下显著提升了点云神经重渲染在多种wild数据集上的质量和稳定性。

研究背景与动机

领域现状:点云表示因直观的几何表达、易操作性和快速收敛等优势在新视角合成(NVS)中越来越流行。NPBG通过将学习到的神经纹理栅格化后用U-Net渲染为RGB图像,展示了灵活简洁的流程。

现有痛点:NPBG只在理想条件(合成数据、精心拍摄的人头)下表现良好。面对真实wild场景时存在三大问题:(1) 无法处理背景(原始方法需要庞大的环境贴图);(2) 稀疏、有缺口的点云导致栅格化不完整;(3) 简单的z-buffer可见性判断无法正确处理复杂遮挡关系。NeRF方法虽可处理多种场景但需针对不同场景类型定制参数化策略。

核心矛盾:要保持点云方法的内存效率和可扩展性优势(不使用可微栅格化),同时提升神经渲染器的能力以应对各种退化情况。

本文目标:在不改变高效的硬点z-buffer栅格化流程的前提下,通过增强神经渲染器和辅助策略使点云重渲染在各类真实数据集上稳健工作。

切入角度:从图像修复(image restoration)领域借鉴思路——将不完整的栅格化结果视为退化图像,用退化感知的神经网络来"修复"。

核心 idea:设计Downgrade-aware Convolution (DAC)模块,在卷积层中注入退化类型信息(背景/前景/遮挡),使渲染器能自适应处理不同退化模式;同时用视觉self-attention实现伪逐点背面剔除。

方法详解

整体框架

标准NPBG流程:三角测量获取3D点→赋予可学习神经纹理→硬z-buffer栅格化为2D特征图→增强的CNN渲染器输出RGB。RPBG的改进集中在渲染器、背景建模、点云增强和训练策略。

关键设计

  1. 退化感知卷积(DAC)+ 注意力可见性校正:

    • 功能:让渲染器识别并正确处理不同类型的退化区域
    • 核心思路:将栅格化结果的每个像素标记为"有效前景"、"背景"或"潜在遮挡",生成退化类型掩码。DAC模块在卷积运算中conditioning on这个掩码,使得对不同退化区域采用不同的处理逻辑。同时引入视觉self-attention机制,在特征空间中根据上下文推断正确的点可见性,实现伪背面剔除
    • 设计动机:vanilla U-Net无法区分"该像素为空是因为背景"还是"该像素为空是因为点云稀疏"——DAC提供了这个关键区分能力
  2. 轻量级背景建模:

    • 功能:用极低成本建模场景背景
    • 核心思路:用一个可学习的默认特征向量作为所有背景像素的神经纹理,而非ADOP那样需要庞大的环境贴图。配合更强的渲染器,这种简单方案能达到相似的量化效果
    • 设计动机:环境贴图内存消耗大且不够通用;单一默认向量极其轻量级
  3. 基于伪密度的点云增强:

    • 功能:改善三角测量不充分区域的点云覆盖
    • 核心思路:从训练后的神经纹理计算伪密度分数(纹理向量的范数),低密度区域的3D位置可能是错误三角测量。通过迭代地向这些区域添加新点来增强点云,改善栅格化覆盖
    • 设计动机:SfM/COLMAP的三角测量在纹理缺失、重复纹理等区域常常失败,导致点云有大片空洞

损失函数 / 训练策略

标准photometric loss(L1 + LPIPS感知损失)。与NPBG的分阶段训练不同,RPBG端到端联合优化神经纹理和渲染器参数,简化训练流程。

实验关键数据

主实验

方法 360° scenes Inside-out Large-scale Sparse-view 整体鲁棒性
NPBG
mip-NeRF 360 好(特定场景) 需特殊参数化
F2-NeRF
RPBG 最高

消融实验

配置 PSNR变化 说明
无DAC 显著下降 退化类型区分至关重要
无attention可见性 下降 遮挡处理不佳
无点云增强 下降 稀疏区域质量差
环境贴图替代默认向量 相近 轻量方案同样有效
完整RPBG 最优 所有改进互补

关键发现

  • RPBG在4类典型挑战场景上(360°、inside-out、大规模、稀疏视角)都显著优于NPBG基线
  • 相比NeRF方法,RPBG的最大优势在于统一参数化——无需针对不同场景类型手动配置
  • DAC模块的退化感知能力是性能提升的最大贡献因素
  • 点云方法的内存效率使其在大规模场景上相比3DGS和NeRF有天然优势

亮点与洞察

  • 图像修复视角处理渲染退化:将不完整栅格化视为"退化图像"是一个巧妙的跨领域类比,引入了成熟的退化处理技术
  • 统一参数化的鲁棒性:在所有数据集上用完全相同的超参数取得一致良好结果,这在NVS领域极为罕见
  • 保持可扩展性:在增强渲染器的同时不触碰栅格化流程,保留了点云方法对大规模场景的可扩展性

局限与展望

  • 渲染质量仍不及3DGS等最新方法在其擅长的场景上的表现
  • 依赖SfM/COLMAP的初始点云质量——极端稀疏或无纹理场景仍然困难
  • 推理速度受限于CNN渲染器,不及实时方法
  • 可结合3DGS的splatting技术进一步提升前景质量

相关工作与启发

  • vs NPBG/NPBG++: 直接改进的基线,RPBG通过渲染器增强大幅修复了原方法的脆弱性
  • vs 3DGS: 3DGS用可微splatting但内存消耗大;RPBG用硬z-buffer+强渲染器平衡质量和可扩展性
  • vs mip-NeRF 360/F2-NeRF: NeRF方法需场景特定参数化,RPBG统一处理所有场景类型
  • 退化感知卷积的思路可迁移到任何"有噪声/不完整中间表示→最终输出"的渲染管线

评分

  • 新颖性: ⭐⭐⭐⭐ 图像修复思路处理渲染退化,DAC设计有创意
  • 实验充分度: ⭐⭐⭐⭐⭐ 4类场景类型全面覆盖,含详尽消融
  • 写作质量: ⭐⭐⭐⭐ 问题分析深入,改进动机明确
  • 价值: ⭐⭐⭐⭐ 鲁棒的点云渲染方案对实际应用有重要意义

相关论文