arXiv:2412.17378cs.CV2024-12被引 24

解决3D高斯溅射训练中的负载不均问题,提升渲染效率7.52倍

Balanced 3DGS: Gaussian-wise Parallelism Rendering with Fine-Grained Tiling

  • 按高斯粒度并行渲染,动态分配计算资源
  • 通过细粒度分块使所有SM负载均衡,性能提升7.52倍
  • 自适应选择渲染核函数,适合大规模3DGS训练

3D高斯溅射(3DGS)因出色的视觉质量和渲染速度受到学术界与工业界的关注。然而,3DGS模型训练仍耗时较长,尤其在像素与高斯球体间工作量差异大时,导致渲染CUDA核函数性能低下。本文提出Balanced 3DGS,一种基于高斯粒度并行的细粒度分块渲染方法,有效解决负载不均问题。首先,提出块间动态工作量分配技术,动态映射工作量至单个GPU的流式多处理器(SM)资源,奠定负载均衡基础;其次,首次提出高斯粒度并行渲染,显著降低线程束内工作量差异,是缓解负载不均的关键;在此基础上,提出细粒度联合负载均衡技术,实现各SM间工作量均匀分布,使前向渲染CUDA核函数性能最高提升7.52倍。此外,设计自适应渲染核函数选择策略,根据负载均衡情况动态调整,进一步提升训练效率。

原文摘要 · Abstract (English)

3D Gaussian Splatting (3DGS) is increasingly attracting attention in both academia and industry owing to its superior visual quality and rendering speed. However, training a 3DGS model remains a time-intensive task, especially in load imbalance scenarios where workload diversity among pixels and Gaussian spheres causes poor renderCUDA kernel performance. We introduce Balanced 3DGS, a Gaussian-wise parallelism rendering with fine-grained tiling approach in 3DGS training process, perfectly solving load-imbalance issues. First, we innovatively introduce the inter-block dynamic workload distribution technique to map workloads to Streaming Multiprocessor(SM) resources within a single GPU dynamically, which constitutes the foundation of load balancing. Second, we are the first to propose the Gaussian-wise parallel rendering technique to significantly reduce workload divergence inside a warp, which serves as a critical component in addressing load imbalance. Based on the above two methods, we further creatively put forward the fine-grained combined load balancing technique to uniformly distribute workload across all SMs, which boosts the forward renderCUDA kernel performance by up to 7.52x. Besides, we present a self-adaptive render kernel selection strategy during the 3DGS training process based on different load-balance situations, which effectively improves training efficiency.

3D高斯负载均衡渲染优化GPU并行

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