3D CBCT Challenge 2024: Improved Cone Beam CT Reconstruction using SwinIR-Based Sinogram and Image Enhancement

TL;DR

The proposed methodology uses Nesterov Accelerated Gradient Descent (NAG) to solve the least squares (NAG-LS) problem in CT image reconstruction, offering a promising solution for both low dose and clinical dose CBCT reconstruction.

Abstract

In this paper, we present our approach to the 3D CBCT Challenge 2024, a part of ICASSP SP Grand Challenges 2024. Improvement in Cone Beam Computed Tomography (CBCT) reconstruction has been achieved by integrating Swin Image Restoration (SwinIR) based sinogram and image enhancement modules. The proposed methodology uses Nesterov Accelerated Gradient Descent (NAG) to solve the least squares (NAG-LS) problem in CT image reconstruction. The integration of sinogram and image enhancement modules aims to enhance image clarity and preserve fine details, offering a promising solution for both low dose and clinical dose CBCT reconstruction. The averaged mean squared error (MSE) over the validation dataset has decreased significantly, in the case of low dose by one-fifth and clinical dose by one-tenth. Our solution is one of the top 5 approaches in this challenge.

Figures (3)

• Figure 1: Proposed CBCT reconstruction pipeline.
• Figure 2: Block diagram of SwinIR architecture liang2021swinir.
• Figure 3: (a), (d) are CT images reconstructed using FDK (Baseline). (b), (e) are CT images reconstructed using NAG-LS with SEM. (c), (f) are CT images reconstructed using NAG-LS with SEM and IEM. Upper and lower row images are low dose and clinical dose respectively. (g) is the clean CT image.