Fang Liu1, Julia Velikina1, Richard Kijowski1, and Alexey Samsonov1
1Department of Radiology, University of Wisconsin-Madison, Madison, WI, United States
Synopsis
We introduced a novel reconstruction
framework by combining deep learning (DL) neural network with the Projections Onto
Convex Sets (POCS) algorithm, termed DL-POCS. The image restoration from
undersampled images was first performed by a convolutional encoder-decoder
network. Then the output from deep learning was used as initialization and
extra constraints were imposed to promote the POCS reconstruction. We evaluated
this approach on vastly undersampled knee MR data and found that this combined
approach is superior to each of individual components alone. Our study suggests
that deep learning regularized image reconstruction will have a substantial
impact on data-driven accelerated MR imaging.
Introduction
The Projections Onto
Convex Sets (POCS) algorithm presents an efficient way to utilize a wide range
of prior information in image restoration problems (1) such as restoration of partial k-space (2) and motion artifact correction(3). In our previous work, we adapted the
POCS formalism for parallel MRI data reconstruction, where we showed utilization
of prior information within the POCS framework is an efficient way to mitigate
aliasing and noise amplification, thereby affording higher acceleration factors
and improving image quality (4).
In recent years, there have been many efforts to develop deep learning (DL) based
methods for restoring undersampled image data for CT and MRI. These efforts
lead to sophisticated, computationally intensive reconstruction pipelines where
development of advanced deep learning networks with better performance is primary
focus. In this study, we explore an alternative approach, termed DL-POCS, for
the use of DL machinery for image reconstruction, in which DL-based pre-processing
of undersampled data is applied only once to generate prior information for
regularization and additional constraints of POCS framework. Evaluation of such
framework was performed for reconstruction of highly undersampled knee MR
images.Methods
Deep Learning
Network: A
convolutional encoder-decoder (CED) network was designed to restore MR images
from undersampled k-space data (Figure 1). The encoder consists of a set of
convolution layers followed by batch normalization (BN) and rectified-linear
unit (ReLU) activation. The decoder is a mirror network with the same structure
with the convolution layers replaced by deconvolution layers. Symmetric
shortcut connections between encoder and decoder layers were added by following
the fully pre-activation residual network strategy to forward transfer image
features. POCS Framework: The POCS processing pipeline follows POCSENSE
(4) framework (Figure
2). Instead of using zero-filled image as POCSENSE initial input, we initialize
it by prior image that is largely restored from learned latent aliases and
noise patterns during the CED training process. Additionally, the CED network output
can provide additional constraints such as object support in the POCS pipeline.
Evaluation: The evaluation was performed on 10 retrospectively
undersampled knee images from routine clinical scans. Images were acquired on
3T scanner (MR750, GE Healthcare, Waukesha, USA) using intermediate T2-weighted
coronal fast spin-echo (TR/TE=2125/20ms, 420×448 matrix, 32 slices). We
simulated 2D Cartesian parallel imaging sampling pattern with undersampling
factors R=3 and 10. The central 8% of k-space data was fully sampled and used
for calculating coil sensitivity. Eight knee images were used for training and
image augmentation through 2D translation, rotation and shearing was used to
generate three repeats. The network was trained using mean squared error as
image loss and an adaptive gradient-based optimization algorithm (ADAM) with an
initial learning rate of 0.0001 for a total of 100 epochs.Results
The total training phase took approximately 36 hours (computing hardware
included an Intel Xeon W3520 quad-core CPU, 32 GB DDR3 RAM, and two Nvidia
GeForce GTX 1080 Ti graphic cards with 7168 cores and 22GB GDDR5 RAM.).
Generating corrected images for one subject took approximately 30 seconds for
deep learning inference. Figure 3 demonstrates images reconstructed with several
methods for R=3. While the CED network and POCS applied independently substantially
reduced image artifacts, their combination within DL-POCS framework provided the
best performance with almost complete image restoration. The normalized root
mean squared error (nRMSE) for Zero Filling, DL, POCS and DL-POCS were 0.073, 0.045,
0.0023 and 0.0015, respectively. Figure 4 demonstrates the performance for
R=10. At this extreme undersampling
level, both DL and POCS retain visible image artifacts as indicated by the
arrows. The DL-POCS significantly improves reconstruction by efficiently
suppressing these artifacts. The nRMSE for Zero Filling, DL, POCS and DL-POCS were
0.095, 0.068, 0.078 and 0.056, respectively.Discussion
We have proposed a novel
image reconstruction framework by combining deep learning convolutional neural
network with POCS reconstruction for vastly undersampled MR data. We have
demonstrated that this combined reconstruction strategy offers superior
performance to the ones afforded by individual components. While most deep
learning-based studies to date focus on development and implementation of
better networks, our approach introduces a novel insight on integrating deep
learning into traditional well-studied reconstruction frameworks. The synergy of
group wise data-driven nature of deep learning and case specific data
consistency of POCS amplify the reconstruction performance. Our study suggests
that deep learning-regularized image reconstruction, such as DL-POCS, can have
a substantial impact on data-driven accelerated MRI.Acknowledgements
No acknowledgement found.References
1. Youla DC, Webb H: Image Restoration by the Method of Convex
Projections: Part 1 Theory. IEEE Trans Med Imaging 1982; 1:81–94.
2. ZP L, FE B, RT C, EM H, PC L, MR S: Constrained reconstruction methods
in MR imaging. Rev Magn Reson Med 1992; 4:67–185.
3. Samsonov AA, Velikina J, Jung Y, Kholmovski EG, Johnson CR, Block WF:
Pocs-enhanced correction of motion artifacts in parallel MRI. Magn Reson Med
2010; 63:1104–1110.
4. Samsonov AA, Kholmovski EG, Parker DL, Johnson CR: POCSENSE: POCS-based
reconstruction for sensitivity encoded magnetic resonance imaging. Magn
Reson Med 2004; 52:1397–1406.