Prior Frequency Guided Diffusion Model for Limited Angle (LA)-CBCT Reconstruction

Jiacheng Xie,Hua-Chieh Shao,Yunxiang Li,You Zhang

2024-04-09

Abstract:Cone-beam computed tomography (CBCT) is widely used in image-guided radiotherapy. Reconstructing CBCTs from limited-angle acquisitions (LA-CBCT) is highly desired for improved imaging efficiency, dose reduction, and better mechanical clearance. LA-CBCT reconstruction, however, suffers from severe under-sampling artifacts, making it a highly ill-posed inverse problem. Diffusion models can generate data/images by reversing a data-noising process through learned data distributions; and can be incorporated as a denoiser/regularizer in LA-CBCT reconstruction. In this study, we developed a diffusion model-based framework, prior frequency-guided diffusion model (PFGDM), for robust and structure-preserving LA-CBCT reconstruction. PFGDM uses a conditioned diffusion model as a regularizer for LA-CBCT reconstruction, and the condition is based on high-frequency information extracted from patient-specific prior CT scans which provides a strong anatomical prior for LA-CBCT reconstruction. Specifically, we developed two variants of PFGDM (PFGDM-A and PFGDM-B) with different conditioning schemes. PFGDM-A applies the high-frequency CT information condition until a pre-optimized iteration step, and drops it afterwards to enable both similar and differing CT/CBCT anatomies to be reconstructed. PFGDM-B, on the other hand, continuously applies the prior CT information condition in every reconstruction step, while with a decaying mechanism, to gradually phase out the reconstruction guidance from the prior CT scans. The two variants of PFGDM were tested and compared with current available LA-CBCT reconstruction solutions, via metrics including PSNR and SSIM. PFGDM outperformed all traditional and diffusion model-based methods. PFGDM reconstructs high-quality LA-CBCTs under very-limited gantry angles, allowing faster and more flexible CBCT scans with dose reductions.

Medical Physics,Machine Learning

What problem does this paper attempt to address?

The paper aims to address the reconstruction problem of Cone-beam computed tomography (CBCT) under Limited-Angle (LA) acquisition. Specifically, the paper proposes a new framework based on a diffusion model—Prior Frequency-Guided Diffusion Model (PFGDM)—for robust and structure-preserving CBCT reconstruction under limited angles. ### Main Issues 1. **Background and Challenges**: - CBCT is widely used in image guidance for radiotherapy, but under limited-angle acquisition, traditional methods (such as the Filtered Back Projection algorithm FDK) produce severe undersampling artifacts, leading to poor image quality. - Existing iterative reconstruction methods (such as Model-Based Iterative Reconstruction, MBIR) have shown some improvements but still fail to achieve high-quality reconstruction results under extremely limited angles. 2. **Proposed Method**: - The paper develops two variants of PFGDM (PFGDM-A and PFGDM-B), utilizing high-frequency information extracted from patient-specific prior CT scans as conditions to guide the LA-CBCT reconstruction process. - PFGDM-A stops introducing high-frequency conditions after a preset step, while PFGDM-B continuously introduces conditions by gradually reducing the high-frequency information. ### Experimental Results - Tests were conducted on multiple anatomical sites (lung, pelvis, head and neck) and different angles (30°, 90°, 120°). - Results show that PFGDM-A and PFGDM-B significantly outperform other existing methods in terms of Peak Signal-to-Noise Ratio (PSNR) and Structural Similarity Index (SSIM), especially under extremely small angles (e.g., 30°). ### Significance - PFGDM can reconstruct high-quality LA-CBCT images under very limited gantry angles, allowing for faster and more flexible CBCT scans and reducing radiation dose. ### Summary The PFGDM method proposed in this paper achieves high-quality CBCT reconstruction under limited angles by introducing high-frequency prior information, making it a promising tool for clinical applications.

Prior Frequency Guided Diffusion Model for Limited Angle (LA)-CBCT Reconstruction

Prior frequency guided diffusion model for limited angle (LA)-CBCT reconstruction

CBCT-Based Synthetic CT Image Generation Using Conditional Denoising Diffusion Probabilistic Model

CT Reconstruction using Diffusion Posterior Sampling conditioned on a Nonlinear Measurement Model

HC$^3$L-Diff: Hybrid conditional latent diffusion with high frequency enhancement for CBCT-to-CT synthesis

CBCT-based synthetic CT image generation using a diffusion model for CBCT-guided lung radiotherapy

Diffusion Prior Regularized Iterative Reconstruction for Low-dose CT

Diffusion Posterior Sampling for Nonlinear CT Reconstruction

Energy-guided diffusion model for CBCT-to-CT synthesis

Using a patient‐specific diffusion model to generate CBCT‐based synthetic CTs for CBCT‐guided adaptive radiotherapy

Physics-informed Score-based Diffusion Model for Limited-angle Reconstruction of Cardiac Computed Tomography

PrideDiff: Physics-Regularized Generalized Diffusion Model for CT Reconstruction

Prior Image-Constrained Iterative Reconstruction with Adaptive Step Size for Limited-Angle CBCT

Texture-preserving diffusion model for CBCT-to-CT synthesis

CT Reconstruction using Nonlinear Diffusion Posterior Sampling with Detector Blur Modeling

Unsupervised Bayesian Generation of Synthetic CT from CBCT Using Patient-Specific Score-Based Prior

WE-G-18A-04: 3D Dictionary Learning Based Statistical Iterative Reconstruction for Low-Dose Cone Beam CT Imaging

Image reconstruction based on nonlinear diffusion model for limited-angle computed tomography

Optimization-Based Image Reconstruction Regularized with Inter-Spectral Structural Similarity for Limited-Angle Dual-Energy Cone-Beam CT

LBP-based penalized weighted least-squares approach to low-dose cone-beam computed tomography reconstruction

Parallel Diffusion Model-based Sparse-view Cone-beam Breast CT