DM4CT: Benchmarking Diffusion Models for Computed Tomography Reconstruction (ICLR 2026)

Jiayang Shi, Daniel M. Pelt, K. Joost Batenburg

Abstract: Diffusion models have recently emerged as powerful priors for solving inverse problems. While computed tomography (CT) is theoretically a linear inverse problem, it poses many practical challenges. These include correlated noise, artifact structures, reliance on system geometry, and misaligned value ranges, which make the direct application of diffusion models more difficult than in domains like natural image generation. To systematically evaluate how diffusion models perform in this context and compare them with established reconstruction methods, we introduce DM4CT, \textit{a comprehensive benchmark for CT reconstruction}. DM4CT includes datasets from both medical and industrial domains with sparse-view and noisy configurations. To explore the challenges of deploying diffusion models in practice, we additionally acquire a high-resolution CT dataset at a high-energy synchrotron facility and evaluate all methods under real experimental conditions. We benchmark ten recent diffusion-based methods alongside seven strong baselines, including model-based, unsupervised, and supervised approaches. Our analysis provides detailed insights into the behavior, strengths, and limitations of diffusion models for CT reconstruction. The real-world dataset is publicly available at https://zenodo.org/records/15420527, and the codebase is open-sourced at https://github.com/DM4CT/DM4CT.

• Table of Contents
• Environment requirements
• Datasets
• Preprocessing
• Pretrained diffusion models
• CT Reconstruction Methods
• Training Diffusion Models from Scratch
• Citation

• At least one Nvidia GPU for inference.
• Main dependencies are pytorch, diffuers, astra-toolbox, tifffile.

We provide the conda configuration to create the same environment for the benchmark.

Create and activate the main Conda environment:

conda env create -f environment.yml
conda activate diffusers-ct

For MBIR reconstruction (due to version conflicts), use the separate environment:

conda env create -f mbir.yml
conda activate mbir

• The medical CT dataset: Low Dose Grand Challenge
• The industrial CT dataset: LoDoInd
• Real-world Synchrotron CT Dataset: Rocks Dataset

We provide the preprocessing code to perform

• train/test split
• rescale intensity values to the range (-1, 1)
• flat-field correction and ring reduction for the synchrotron dataset

All pretrained models are available here.

@inproceedings{
shi2026dmct,
title={{DM}4{CT}: Benchmarking Diffusion Models for Computed Tomography Reconstruction},
author={Shi, Jiayang and Pelt, Dani{\"e}l M and Batenburg, K Joost},
booktitle={The Fourteenth International Conference on Learning Representations},
year={2026},
url={https://openreview.net/forum?id=YE5scJekg5}
}