DRACO: Differentiable Reconstruction for Arbitrary CBCT Orbits

Chengze Ye,Linda-Sophie Schneider,Yipeng Sun,Mareike Thies,Siyuan Mei,Andreas Maier
2024-10-19
Abstract:This paper introduces a novel method for reconstructing cone beam computed tomography (CBCT) images for arbitrary orbits using a differentiable shift-variant filtered backprojection (FBP) neural network. Traditional CBCT reconstruction methods for arbitrary orbits, like iterative reconstruction algorithms, are computationally expensive and memory-intensive. The proposed method addresses these challenges by employing a shift-variant FBP algorithm optimized for arbitrary trajectories through a deep learning approach that adapts to a specific orbit geometry. This approach overcomes the limitations of existing techniques by integrating known operators into the learning model, minimizing the number of parameters, and improving the interpretability of the model. The proposed method is a significant advancement in interventional medical imaging, particularly for robotic C-arm CT systems, enabling faster and more accurate CBCT reconstructions with customized orbits. Especially this method can also be used for the analytical reconstruction of non-continuous orbits like circular plus arc. The experimental results demonstrate that the proposed method significantly accelerates the reconstruction process compared to conventional iterative algorithms. It achieves comparable or superior image quality, as evidenced by metrics such as the mean squared error (MSE), the peak signal-to-noise ratio (PSNR), and the structural similarity index measure (SSIM). The validation experiments show that the method can handle data from different trajectories, demonstrating its flexibility and robustness across different scan geometries. Our method demonstrates a significant improvement, particularly for the sinusoidal trajectory, achieving a 38.6% reduction in MSE, a 7.7% increase in PSNR, and a 5.0% improvement in SSIM. Furthermore, the computation time for reconstruction was reduced by more than 97%.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: **How to provide an efficient and accurate reconstruction method for cone - beam computed tomography (CBCT) images of arbitrary orbits**. Traditional CBCT reconstruction methods, such as iterative reconstruction algorithms, are computationally very expensive and memory - intensive for processing data of non - circular orbits. In addition, existing analytical algorithms (such as the filtered back - projection method FBP) are only applicable to the case of assuming a circular trajectory and cannot be directly applied to arbitrary orbits. ### Specific problems and challenges 1. **Computational complexity and memory requirements**: Traditional methods (such as iterative reconstruction algorithms) require multiple iterations to gradually approximate the final reconstruction result, resulting in high computational complexity and long processing time. 2. **Applicability limitations**: The traditional FBP algorithm is based on the circular trajectory assumption and is not applicable to non - circular orbits, which limits its flexibility in practical applications. 3. **Reconstruction accuracy and speed**: Existing methods may produce large errors when dealing with complex orbits, and the reconstruction speed is slow, which affects their practicality and efficiency in the clinical environment. ### Solutions The paper proposes a new method, namely **differentiable shift - variant filtered back - projection (shift - variant FBP) neural network**, for CBCT image reconstruction of arbitrary orbits. This method optimizes the shift - variant FBP algorithm through deep learning to adapt to specific orbital geometries, thereby overcoming the limitations of traditional methods: - **Reducing the number of parameters**: By integrating known operators as prior knowledge, the number of parameters of the model is reduced, and the interpretability and reliability of the model are improved. - **Improving reconstruction speed and quality**: The experimental results show that this method significantly accelerates the reconstruction process and at the same time achieves image quality equivalent to or better than that of traditional iterative algorithms, especially performing excellently in indicators such as mean - square error (MSE), peak signal - to - noise ratio (PSNR) and structural similarity index (SSIM). - **Flexibility and robustness**: This method can process data from different trajectories, demonstrating its flexibility and robustness under different scanning geometries. In summary, this paper aims to solve the problems of high computational complexity, large memory requirements and insufficient reconstruction accuracy encountered by existing technologies when processing arbitrary - orbit data by introducing a new deep - learning - driven CBCT reconstruction method, thereby providing faster and more accurate solutions for the field of interventional medical imaging.