Share your thoughts, 1 month free Claude Pro on usSee more
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
CBCT ReconstructionAAPM Clinical Dataset 60-view
PSNR40.29
42
CBCT ReconstructionAAPM Clinical Dataset 40-view
PSNR37.58
42
CBCT ReconstructionAAPM Clinical Dataset 20-view
PSNR34.4
42
Static CT ReconstructionSynthetic (test)
PSNR32.25
24
Static CT Reconstructionreal (test)
PSNR36.59
24
CT ReconstructionOOD Dataset
PSNR35.64
18
Sparse-View CT ReconstructionFUMPE
PSNR31.71
11
Sparse-View CT ReconstructionPENGWIN
PSNR34.7
11
Sparse-View CT ReconstructionFUMPE 10-view (cross-dataset)
PSNR25.11
11
Sparse-View CT ReconstructionPENGWIN 10-view (cross-dataset)
PSNR28.36
11
Showing 10 of 39 rows

Other info

Follow for update