用稀疏性加速3D高斯点的二阶优化,训练快4倍
Matrix-free Second-order Optimization of Gaussian Splats with Residual Sampling
- 利用高斯点影响像素有限的稀疏性,设计免矩阵二阶优化
- 在低数量高斯点下比Adam快5倍,中等数量下快1.3倍
- 适合追求快速训练的3D重建与渲染研究者
3D高斯点绘图(3DGS)因其高质量渲染和快速推理被广泛用于新视角合成。然而,3DGS主要依赖Adam等一阶优化器,导致训练时间长。为此,我们提出一种基于Levenberg-Marquardt(LM)和共轭梯度(CG)的新型二阶优化策略,专为高斯点绘图设计。关键洞察是:3DGS中的雅可比矩阵具有显著稀疏性,因为每个高斯点仅影响有限像素。我们利用这一稀疏性,提出一种免矩阵且支持GPU并行的LM优化方法。为进一步提升效率,我们设计了针对相机视角和损失函数的采样策略,从而显著降低计算复杂度。此外,通过引入启发式学习率确定方法,避免了代价高昂的线搜索,有效提升了二阶近似的收敛速度。实验表明,该方法相比标准LM实现4倍加速,在低高斯数量下优于Adam约5倍,中等数量下提速约1.3倍。同时,免矩阵实现相较同期二阶优化器3DGS-LM提升2倍速度,内存占用减少3.5倍。项目页面:https://vcai.mpi-inf.mpg.de/projects/LM-RS/
原文摘要 · Abstract (English)
3D Gaussian Splatting (3DGS) is widely used for novel view synthesis due to its high rendering quality and fast inference time. However, 3DGS predominantly relies on first-order optimizers such as Adam, which leads to long training times. To address this limitation, we propose a novel second-order optimization strategy based on Levenberg-Marquardt (LM) and Conjugate Gradient (CG), specifically tailored towards Gaussian Splatting. Our key insight is that the Jacobian in 3DGS exhibits significant sparsity since each Gaussian affects only a limited number of pixels. We exploit this sparsity by proposing a matrix-free and GPU-parallelized LM optimization. To further improve its efficiency, we propose sampling strategies for both camera views and loss function and, consequently, the normal equation, significantly reducing the computational complexity. In addition, we increase the convergence rate of the second-order approximation by introducing an effective heuristic to determine the learning rate that avoids the expensive computation cost of line search methods. As a result, our method achieves a 4x speedup over standard LM and outperforms Adam by ~5x when the Gaussian count is low while providing ~1.3x speed in moderate counts. In addition, our matrix-free implementation achieves 2x speedup over the concurrent second-order optimizer 3DGS-LM, while using 3.5x less memory. Project Page: https://vcai.mpi-inf.mpg.de/projects/LM-RS/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。