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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Shape Generation | 3D Shape Generation Dataset (test) | log-WGAN ratio1.433 | 30 | |
| Unconditional 3D Medical Image Generation | LIDC-IDRI 128^3 | FID1.41 | 6 | |
| Unconditional 3D Medical Image Generation | RAD-ChestCT 128^3 | FID1.45 | 6 | |
| 3D Image Synthesis | LIDC-IDRI | Mean Latency (s)11.459 | 6 | |
| Unconditional 3D Image Generation | LIDC-IDRI 256^3 resolution | FID2.28 | 3 |