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
Static CT ReconstructionSynthetic (test)
PSNR32.25
24
Static CT Reconstructionreal (test)
PSNR36.59
24
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
CT Reconstruction512^2-projection dataset 75-view (test)
PSNR39.04
8
CT Reconstruction512^2-projection dataset 50-view (test)
PSNR38.09
8
CT Reconstruction512^2-projection dataset 25-view (test)
PSNR35.23
8
PET image reconstructionUDPET public (DRF 50)
PSNR36.74
8
Showing 10 of 33 rows

Other info

Follow for update