SEA-Net: Structure-Enhanced Attention Network for Limited-Angle CBCT Reconstruction of Clinical Projection Data

Abstract

Limited-angle CBCT (LA-CBCT) is of great clinical interest, because the scanning time and the patient dose are proportional to the scanning range of gantry rotation angles of CBCT. However, the image reconstruction for LA-CBCT remains technically challenging, which suffers from severe wedge artifacts and image distortions. This work aims to improve LA-CBCT by developing deep learning (DL) methods for real clinical CBCT projection data, which is the first feasibility study of clinical-projection-data-based LA-CBCT, to the best of our knowledge. Targeting at real clinical projection data, we have explored various DL methods such as image/data/hybrid-domain methods and finally developed a so-called Structure-Enhanced Attention Network (SEA-Net) method that has the best image quality from clinical projection data among the DL methods we have implemented. Specifically, the proposed SEA-Net employs a specialized structure enhancement sub-network to promote texture preservation. Based on the observation that the distribution of wedge artifacts in reconstruction images is non-uniform, the spatial attention module is utilized to emphasize the relevant regions while ignores the irrelevant ones, which leads to more accurate texture restoration. SEA-Net was validated in comparison with analytic (FDK), iterative (TV), image-domain DL (DDNet and FED-INet, data-domain DL (DCAR), dual-domain DL (Sam'Net), and various unrolling DL (hdNet, CTNet, FSR-Net, CasRedSCAN) methods. Among all methods, the SEA-Net had the best image reconstruction quality as quantified by root-mean-square error (RMSE), peak signal-to-noise ratio (PSNR), and structural similarity index (SSIM), for various LA-CBCT problems of 90°-180° projection data. In addition, LA-CBCT via SEA-Net provided comparable accuracy for both patient setup (quantified by image registration accuracy from planning CT (pCT) to CBCT) and dose calculation (see the table), with full-view CBCT. We explored various DL methods and developed an image-domain-based method termed SEA-Net that provided the best image quality for clinical projection data. To the best of our knowledge, this is the first feasibility study of the real clinical-projection-data-based LA-CBCT. Moreover, LA-CBCT via SEA-Net can potentially provide comparable accuracy for patient setup and dose calculation, with full-view CBCT.