Solving ill-posed inverse problems using iterative deep neural networks
This addresses reconstruction challenges in medical imaging (e.g., CT scans) by combining classical regularization with deep learning, though it is incremental as it builds on existing methods.
The paper tackled ill-posed inverse problems, such as non-linear tomographic inversion, by proposing a partially learned gradient scheme using deep neural networks, resulting in a 5.4 dB PSNR improvement over TV reconstruction and reconstructions in about 0.4 seconds.
We propose a partially learned approach for the solution of ill posed inverse problems with not necessarily linear forward operators. The method builds on ideas from classical regularization theory and recent advances in deep learning to perform learning while making use of prior information about the inverse problem encoded in the forward operator, noise model and a regularizing functional. The method results in a gradient-like iterative scheme, where the "gradient" component is learned using a convolutional network that includes the gradients of the data discrepancy and regularizer as input in each iteration. We present results of such a partially learned gradient scheme on a non-linear tomographic inversion problem with simulated data from both the Sheep-Logan phantom as well as a head CT. The outcome is compared against FBP and TV reconstruction and the proposed method provides a 5.4 dB PSNR improvement over the TV reconstruction while being significantly faster, giving reconstructions of 512 x 512 volumes in about 0.4 seconds using a single GPU.