CNN-Based Projected Gradient Descent for Consistent CT Image Reconstruction
H. Gupta, K.H. Jin, H.Q. Nguyen, M.T. McCann, M. Unser
IEEE Transactions on Medical Imaging, vol. 37, no. 6, pp. 1440-1453, June 2018.
We present a new image reconstruction method that replaces the projector in a projected gradient descent (PGD) with a convolutional neural network (CNN). Recently, CNNs trained as image-to-image regressors have been successfully used to solve inverse problems in imaging. However, unlike existing iterative image reconstruction algorithms, these CNN-based approaches usually lack a feedback mechanism to enforce that the reconstructed image is consistent with the measurements. We propose a relaxed version of PGD wherein gradient descent enforces measurement consistency, while a CNN recursively projects the solution closer to the space of desired reconstruction images. We show that this algorithm is guaranteed to converge and, under certain conditions, converges to a local minimum of a non-convex inverse problem. Finally, we propose a simple scheme to train the CNN to act like a projector. Our experiments on sparse-view computed tomography reconstruction show an improvement over total variation-based regularization, dictionary learning, and a state-of-the-art deep learning-based direct reconstruction technique.
@ARTICLE(http://bigwww.epfl.ch/publications/gupta1802.html, AUTHOR="Gupta, H. and Jin, K.H. and Nguyen, H.Q. and McCann, M.T. and Unser, M.", TITLE="{CNN}-Based Projected Gradient Descent for Consistent {CT} Image Reconstruction", JOURNAL="{IEEE} Transactions on Medical Imaging", YEAR="2018", volume="37", number="6", pages="1440--1453", month="June", note="")