Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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.

Ruyi Zha, Tao Jun Lin, Yuanhao Cai, Jiwen Cao, Yanhao Zhang, Hongdong Li• 2024

Related benchmarks

TaskDatasetResultRank
Static CT ReconstructionSynthetic (test)
PSNR32.25
24
Static CT Reconstructionreal (test)
PSNR36.59
24
PET image reconstructionUDPET public (DRF 50)
PSNR36.74
8
PET image reconstructionUDPET DRF 20 public
PSNR41.72
8
PET image reconstructionClinical internal (DRF 4)
PSNR28.93
8
PET image reconstructionClinical DRF 10 internal
PSNR26.97
8
Tomographic reconstructionreal-world CT dataset
PSNR35.71
6
3D ReconstructionIn-house Real X-ray Dataset (test)
SSIM0.5236
5
Tomographic reconstructionNEMA phantom STIR v6.2 (simulation)
Time (min)12
3
PET Image Reconstruction Diameter MeasurementNEMA phantom
Diameter (Sphere 0) (mm)10
3
Showing 10 of 14 rows

Other info

Follow for update