R$^2$-Gaussian: Rectifying Radiative Gaussian Splatting for Tomographic Reconstruction
About
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 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. Code and models are available on the project page https://github.com/Ruyi-Zha/r2_gaussian.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Static CT Reconstruction | Synthetic (test) | PSNR32.25 | 24 | |
| Static CT Reconstruction | real (test) | PSNR36.59 | 24 | |
| PET image reconstruction | UDPET public (DRF 50) | PSNR36.74 | 8 | |
| PET image reconstruction | UDPET DRF 20 public | PSNR41.72 | 8 | |
| PET image reconstruction | Clinical internal (DRF 4) | PSNR28.93 | 8 | |
| PET image reconstruction | Clinical DRF 10 internal | PSNR26.97 | 8 | |
| Tomographic reconstruction | real-world CT dataset | PSNR35.71 | 6 | |
| 3D Reconstruction | In-house Real X-ray Dataset (test) | SSIM0.5236 | 5 | |
| Tomographic reconstruction | NEMA phantom STIR v6.2 (simulation) | Time (min)12 | 3 | |
| PET Image Reconstruction Diameter Measurement | NEMA phantom | Diameter (Sphere 0) (mm)10 | 3 |