Anatomy-preserving unpaired cone-beam CT refinement for image-guided radiotherapy using pseudo-label guided diffusion
cs.CV
Submitted: 2026-10-05
Updated: 2026-10-05
Code: https://github.com/Lai-KK/RefineCBCT
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Cone-beam computed tomography (CBCT) is widely used in image-guided radiotherapy, but scatter, beam hardening, noise, truncation, and other artifacts limit image quality and CT number accuracy.
Abstract
Cone-beam computed tomography (CBCT) is widely used in image-guided radiotherapy, but scatter, beam hardening, noise, truncation, and other artifacts limit image quality and CT number accuracy. Paired CBCT and CT data are difficult to obtain clinically because of motion, anatomical changes, and acquisition mismatch. We present RefineCBCT, an unpaired CBCT refinement framework that uses pseudo-label guidance and short-step diffusion to reduce artifacts while preserving patient-specific anatomy. RefineCBCT was trained and evaluated on unpaired CBCT and planning CT data from public LUNG TCIA and PELVIC TCIA datasets and compared with representative GAN and diffusion based methods. On LUNG TCIA, it achieved the best results across all metrics, with MAE 19.411, RMSE 62.758, PSNR 30.845 dB, and SSIM 0.931. On PELVIC TCIA, it achieved the best MAE, PSNR, and SSIM, with values of 14.905, 36.671 dB, and 0.876. The refined images showed fewer streaking and shading artifacts, clearer anatomical boundaries, and improved soft tissue uniformity, with line profile and ROI analyses showing closer agreement with planning CT. These results suggest that RefineCBCT provides efficient and effective CBCT refinement under clinically realistic unpaired training conditions and may support more reliable CBCT use in image-guided radiotherapy workflows. Code is publicly available on GitHub, and the evaluated datasets are available from The Cancer Imaging Archive.
Sources
- Latent Diffusion Model for Medical Image Standardization and Enhancement
- Zero-shot Medical Image Translation via Frequency-Guided Diffusion Models
- Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
- Blind CT Image Quality Assessment Using DDPM-derived Content and Transformer-based Evaluator
- Systematic Review on Learning-based Spectral CT
- Head-Neck Dual-energy CT Contrast Media Reduction Using Diffusion Models
- Diffusion Prior Regularized Iterative Reconstruction for Low-dose CT
- Diffusion Models for Image Restoration and Enhancement: A Comprehensive Survey
- Score-Based Generative Modeling through Stochastic Differential Equations
- Towards Principled Methods for Training Generative Adversarial Networks
- Tackling the Generative Learning Trilemma with Denoising Diffusion GANs
- Denoising Diffusion Implicit Models
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models