Abstract:Neural ordinary differential equations (Neural ODEs) propose the idea that a sequence of layers in a neural network is just a discretisation of an ODE, and thus can instead be directly modelled by a parameterised ODE. This idea has had resounding success in the deep learning literature, with direct or indirect influence in many state of the art ideas, such as diffusion models or time dependant models. Recently, a continuous version of the U-net architecture has been proposed, showing increased performance over its discrete counterpart in many imaging applications and wrapped with theoretical guarantees around its performance and robustness. In this work, we explore the use of Neural ODEs for learned inverse problems, in particular with the well-known Learned Primal Dual algorithm, and apply it to computed tomography (CT) reconstruction.
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to use Neural Ordinary Differential Equations (Neural ODEs) to enhance the robustness of learning - based methods in Computed Tomography (CT) reconstruction, especially for CT reconstruction problems under low - dose and limited - angle geometric conditions. CT reconstruction is a typical inverse problem. Due to the incompleteness of measurement data and the influence of noise, this problem is ill - posed mathematically, that is, small input changes may lead to large output changes. Traditional CT reconstruction methods such as the Filtered Back - Projection (FBP) algorithm will have a significant performance degradation in the case of large noise or insufficient sampling. In recent years, data - driven methods, especially deep - learning techniques, have shown good performance in CT reconstruction, but these methods are usually more sensitive to noise.
The paper proposes a Continuous Learned Primal Dual algorithm (cLPD), which combines the advantages of the classic Learned Primal Dual (LPD) algorithm and Neural ODEs. By using Neural ODEs to replace the discrete convolution blocks in LPD, cLPD aims to improve the ability to handle noise, thereby providing higher - quality image reconstruction under low - dose and limited - angle conditions.
Specifically, the main contributions of the cLPD algorithm are as follows:
- **Improved Robustness**: The introduction of Neural ODEs makes cLPD more stable when dealing with data containing a large amount of noise, especially under low - dose and limited - angle conditions.
- **Theoretical Guarantee**: Neural ODEs provide theoretical support from the theory of ordinary differential equations, which helps to understand the internal mechanism and performance of the algorithm.
- **Performance Improvement**: Experimental results show that cLPD outperforms the traditional LPD and FBP methods in multiple settings (including clinical dose, extremely low dose, sparse angle, and limited angle).
Through these improvements, cLPD not only improves the image quality but also has higher reliability and practicality in practical applications, especially in clinical scenarios where it is necessary to reduce the radiation dose and limit the scanning range.