R$^2$-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction

Abstract

3D Gaussian splatting (3DGS) has shown promising results in image rendering and surface reconstruction. However, its potential in volumetric reconstruction tasks, such as X-ray computed tomography, remains under-explored. This paper introduces R$^2$-Gaussian, the first 3DGS-based framework for sparse-view tomographic reconstruction. By carefully deriving X-ray rasterization functions, we discover a previously unknown \emph{integration bias} in the standard 3DGS formulation, which hampers accurate volume retrieval. To address this issue, we propose a novel rectification technique via refactoring the projection from 3D to 2D Gaussians. Our new method presents three key innovations: (1) introducing tailored Gaussian kernels, (2) extending rasterization to X-ray imaging, and (3) developing a CUDA-based differentiable voxelizer. Experiments on synthetic and real-world datasets demonstrate that our method outperforms state-of-the-art approaches in accuracy and efficiency. Crucially, it delivers high-quality results in 4 minutes, which is 12$\times$ faster than NeRF-based methods and on par with traditional algorithms.

Cite

Text

Zha et al. "R$^2$-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction." Neural Information Processing Systems, 2024. doi:10.52202/079017-1427

Markdown

[Zha et al. "R$^2$-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction." Neural Information Processing Systems, 2024.](https://mlanthology.org/neurips/2024/zha2024neurips-2gaussian/) doi:10.52202/079017-1427

BibTeX

@inproceedings{zha2024neurips-2gaussian,
  title     = {{R$^2$-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction}},
  author    = {Zha, Ruyi and Lin, Tao Jun and Cai, Yuanhao and Cao, Jiwen and Zhang, Yanhao and Li, Hongdong},
  booktitle = {Neural Information Processing Systems},
  year      = {2024},
  doi       = {10.52202/079017-1427},
  url       = {https://mlanthology.org/neurips/2024/zha2024neurips-2gaussian/}
}