Four-Dimensional Cone-Beam CT Reconstruction Via Diffusion Model and Motion Compensation

Xianghong Wang,Zhengwei Ou,Peng Jin,Jiayi Xie,Ze Teng,Lei Xu,Jichen Du,Mingchao Ding,Yang Chen,Tianye Niu
DOI: https://doi.org/10.1109/trpms.2024.3449155
2024-01-01
IEEE Transactions on Radiation and Plasma Medical Sciences
Abstract:Four-dimensional Cone-beam Computed Tomography (4D CBCT) has recently been recognized as a proficient technique in mitigating motion artifacts attributed to respiratory organ movement. The primary challenges in 4D CBCT reconstruction encompass the precision in projection grouping, the efficacy in reconstructing from sparsely sampled data, and the accuracy in deformation field estimation. To surmount these challenges, we propose an innovative approach that integrates meticulous respiratory curve extraction for projection grouping and utilizes a diffusion model network with motion compensation (MoCo) techniques targeted at significantly enhancing image quality. An object detection network is employed to ascertain the exact position of the diaphragm, which is then normalized to formulate the respiratory curve. Further, we employ a U-Net architecture-based diffusion model, which integrates attention mechanisms to enhance sparse-view reconstruction and reduce artifacts through Guided-Diffusion. Deviating from conventional optical flow methods, our approach introduces an unsupervised registration network for deformation vector field (DVF) in phase-enhanced images. This DVF is then utilized in a motion-compensated, ordered-subset, simultaneous algebraic reconstruction technique, culminating in the generation of 4D CBCT images. The efficacy of this method has been substantiated through validation on both simulated and clinical datasets, with the results from comparative experiments indicating promising outcomes.
What problem does this paper attempt to address?