Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Pixel-Level Residual Diffusion Transformer: Scalable 3D CT Volume Generation

About

Generating high-resolution 3D CT volumes with fine details remains challenging due to substantial computational demands and optimization difficulties inherent to existing generative models. In this paper, we propose the Pixel-Level Residual Diffusion Transformer (PRDiT), a scalable generative framework that synthesizes high-quality 3D medical volumes directly at voxel-level. PRDiT introduces a two-stage training architecture comprising 1) a local denoiser in the form of an MLP-based blind estimator operating on overlapping 3D patches to separate low-frequency structures efficiently, and 2) a global residual diffusion transformer employing memory-efficient attention to model and refine high-frequency residuals across entire volumes. This coarse-to-fine modeling strategy simplifies optimization, enhances training stability, and effectively preserves subtle structures without the limitations of an autoencoder bottleneck. Extensive experiments conducted on the LIDC-IDRI and RAD-ChestCT datasets demonstrate that PRDiT consistently outperforms state-of-the-art models, such as HA-GAN, 3D LDM and WDM-3D, achieving significantly lower 3D FID, MMD and Wasserstein distance scores.

Zhenkai Zhang, Markus Hiller, Krista A. Ehinger, Tom Drummond• 2026

Related benchmarks

TaskDatasetResultRank
3D Shape Generation3D Shape Generation Dataset (test)
log-WGAN ratio1.433
30
Unconditional 3D Medical Image GenerationLIDC-IDRI 128^3
FID1.41
6
Unconditional 3D Medical Image GenerationRAD-ChestCT 128^3
FID1.45
6
3D Image SynthesisLIDC-IDRI
Mean Latency (s)11.459
6
Unconditional 3D Image GenerationLIDC-IDRI 256^3 resolution
FID2.28
3
Showing 5 of 5 rows

Other info

Follow for update