Equivariant conditional diffusion model for head and neck CT image synthesis from CBCT.
Authors
Affiliations (3)
Affiliations (3)
- School of Science and Engineering, University of Dundee, Dundee, Scotland, UK.
- Department of Applied Medical Sciences, Jordan University of Science and Technology, Irbid, Jordan.
- School of Cardiovascular and Metabolic Health, College of Medicine, Veterinary and Life Sciences, University of Glasgow, Glasgow, Scotland, UK.
Abstract
Cone-beam computed tomography (CBCT) is a commonly used modality for image guided radiotherapy (IGRT). It offers real time anatomical visualization with low acquisition cost and dose. Nevertheless, photon scattering and beam hindrance lead CBCT images to suffer from several artifacts. These involve inaccurate Hounsfield Unit (HU) values, which render a lower reliability towards the dose calculations and adaptive planning. Computed tomography (CT), on the contrary, offers better image quality and accurate HU calibration, yet is typically acquired using offline mode and fails to capture the intra-treatment anatomical changes. This renders a need for developing an accurate CBCT to CT synthesis to mitigate the gap in imaging quality in the adaptive radiotherapy workflow. To cater to this, we propose a novel diffusion based conditional generative model, coined EqDiff-CT, to synthesize high quality CT images from CBCT. EqDiff-CT employs a denoising diffusion probabilistic model (DDPM) to iteratively inject noise and learn latent representations that enable reconstruction of anatomically consistent CT images. A group equivariant conditional U-Net backbone, implemented with e2cnn steerable layers, enforces rotational equivariance (cyclic C4 symmetry), helping preserve fine structural details while minimizing noise and artifacts. The system was trained and validated on the SynthRAD2025 dataset, comprising CBCT and CT scans across multiple head and neck anatomical sites, and we compared it with advanced methods such as CycleGAN and DDPM. EqDiff-CT provided substantial gains in structural fidelity, HU accuracy and quantitative metrics. Visual findings further confirm the improved recovery, sharper soft tissue boundaries, and realistic bone reconstructions. The findings suggest that the diffusion model has offered a robust and generalizable framework for CBCT improvements. The proposed solution helps in improving the image quality as well as the clinical confidence in the CBCT guided treatment planning and dose calculations.