arXiv:2504.11946cs.CV2025-04

用强化学习和扩散模型提升稀疏视角下的三维网格重建质量

R-Meshfusion: Reinforcement Learning Powered Sparse-View Mesh Reconstruction with Diffusion Priors

  • 通过一致性扩散模块过滤噪声,生成可靠伪监督信号
  • 基于UCB的在线强化学习策略自适应选择最优观测视角
  • 适合需要高精度3D重建的稀疏视图场景,如逆向工程

从多视角图像中进行网格重建是计算机视觉中的基础问题,但在稀疏视角条件下性能显著下降,尤其在无真实观测的未见区域。尽管扩散模型在有限输入下合成新视角方面表现强劲,但其输出常含视觉伪影且缺乏3D一致性,难以支撑可靠的网格优化。本文提出一种新型框架,以扩散模型为驱动,实现稳健的稀疏视角网格重建。为缓解扩散输出不稳定性,设计了一致性扩散模块,通过四分位距(IQR)分析过滤不可靠生成结果,并采用方差感知图像融合生成鲁棒伪监督信号。在此基础上,构建基于上置信界(UCB)的在线强化学习策略,根据扩散损失自适应选择最具信息量的视角以增强重建。最终,融合图像与稀疏视角真值联合监督基于NeRF的模型,确保几何与外观的一致性。大量实验表明,该方法在几何质量与渲染质量上均有显著提升。

原文摘要 · Abstract (English)

Mesh reconstruction from multi-view images is a fundamental problem in computer vision, but its performance degrades significantly under sparse-view conditions, especially in unseen regions where no ground-truth observations are available. While recent advances in diffusion models have demonstrated strong capabilities in synthesizing novel views from limited inputs, their outputs often suffer from visual artifacts and lack 3D consistency, posing challenges for reliable mesh optimization. In this paper, we propose a novel framework that leverages diffusion models to enhance sparse-view mesh reconstruction in a principled and reliable manner. To address the instability of diffusion outputs, we propose a Consensus Diffusion Module that filters unreliable generations via interquartile range (IQR) analysis and performs variance-aware image fusion to produce robust pseudo-supervision. Building on this, we design an online reinforcement learning strategy based on the Upper Confidence Bound (UCB) to adaptively select the most informative viewpoints for enhancement, guided by diffusion loss. Finally, the fused images are used to jointly supervise a NeRF-based model alongside sparse-view ground truth, ensuring consistency across both geometry and appearance. Extensive experiments demonstrate that our method achieves significant improvements in both geometric quality and rendering quality.

三维重建扩散模型强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。