Kerstin Hammernik1, Matthias Schloegl2, Erich Kobler1, Rudolf Stollberger2,3, and Thomas Pock1
1Institute of Computer Graphics and Vision, Graz University of Technology, Graz, Austria, 2Institute of Medical Engineering, Graz University of Technology, Graz, Austria, 3BioTechMed-Graz, Graz, Austria
Synopsis
In this work, we present a variational network for reconstructing dynamic multi-coil data. Incorporation of parallel imaging increases the acceleration potential due to additional spatial information, but was not considered so far in other learning-based reconstruction approaches for dynamic MRI. We show that variational network reconstructions with learned spatio-temporal regularization lead to further improvements in image quality compared to state-of-the-art Compressed Sensing approaches for different CINE cardiac datasets and acceleration factors with 10-times faster reconstruction time.
Introduction
Recent
developments in deep learning for accelerated MR image reconstruction have
shown improved results over Compressed Sensing (CS)-based approaches for both
static1 and dynamic2-4 imaging. However, most of current
learning-based approaches for dynamic MRI2-4 are limited to
single-coil data only, although MR data is always acquired with multiple
receiver coils. In parallel imaging (PI), coil-sensitivity information is
additionally included in the reconstruction,
which increases the spatial information and therefore the acceleration
potential. Thus, it is natural to include PI in every reconstruction framework,
which becomes already computationally demanding for current iterative CS
reconstruction for dynamic MRI and even more for learning-based network
architectures that involve many free parameters. In this work, we formulate a variational network (VN)1,5,6 that learns spatio-temporal regularization and dataterm weights from dynamic multi-coil MR
data. The highly efficient VN is tested on CINE cardiac data for different
contrasts and accelerations.Methods
To obtain a
spatio-temporal reconstruction $$$u\in{\mathbb{C}^{N_xN_yN_t}}$$$ of size $$$N_x\times{N_y}\times{N_t}$$$,
we define a VN1,5 by an unrolled incremental gradient descent (GD)
scheme with $$$J$$$ components and a fixed number of iterations $$$N$$$,
illustrated in Figure 1:
$$u^{n+1}=u^{n}-\sum\limits_{j=0}^J{\nabla}h_j(u^{n}),\quad0\leq{n}\leq{N-1}.$$
The term $$$\sum\limits_{j=0}^Jh_j(u^{n})$$$
is defined by a generalized CS model for dynamic MR image reconstruction,
including a data-fidelity
$$h_0(u)=\frac{\lambda}{2}\Vert{Au-f}\Vert_2^2,\quad\lambda>0$$
and a
regularization term
$$h_j(u)=\langle
1,\rho_j(k_j*u)\rangle,{\quad}j=1,\cdots,J.$$
The
data-fidelity $$$h_0(u)$$$ enforces consistency of the reconstruction $$$u$$$
to the acquired rawdata $$$f\in\mathbb{C}^{N_xN_yN_tN_c}$$$ for $$$N_c$$$
coils. The dynamic multi-coil forward operator $$$A$$$, i.e.
$$A:u=(u_t)_{t=1,\cdots,N_t}\mapsto(\mathcal{F}_t\left[c_iu_t\right])_{i=1,\cdots,N_c;\,t=1,\cdots,N_t},$$
involves
coil-sensitivity profiles $$$c_i$$$ and Fourier transforms with temporally
varying sampling masks $$$\mathcal{F}_t$$$. In the regularization part, prior
information in terms of convolutions with 2D+t complex filter kernels $$$k_j\in\mathbb{C}^{s{\times}s{\times}s_t}$$$
of size $$$s{\times}s{\times}s_t$$$, followed by non-linear potential functions
$$$\rho:\mathbb{C}^{N_xN_yN_t}\mapsto\mathbb{R}^{N_xN_yN_t}$$$ are applied to
the reconstruction $$$u$$$. All prior information including filter kernels,
activation functions $$$\phi=\rho^\prime$$$ and regularization parameters $$$\lambda$$$
are learned from pairs of undersampled data and corresponding fully-sampled
reference data using a mean-squared-error loss1.Experimental Setup
Fully
sampled retrospectively gated CINE cardiac data were acquired from four healthy
volunteers in breathhold using a 3T Siemens Magnetom Skyra and a 26-30-channel
spine-/body-coil. Each of these datasets consists of one two-chamber,
four-chamber, LVOT-view and four short-axis views, resulting in a total number
of 28 datasets. Two image contrasts with the common acquisition parameters,
matrix-size $$$192\times192$$$, voxel-size $$$1.8\times1.8\times6$$$mm$$$^3$$$,
were used: (1) FLASH TR/TE/FA=5.8ms/3.16ms/12$$$^{\circ}$$$, (2) bSSFP
TR/TE/FA=3.9ms/1.72ms/40$$$^{\circ}$$$. Time-frames were cropped to 17 frames
with similar temporal-resolution of $$$\Delta$$$t$$$\sim$$$50ms.
We trained
individual VNs for each contrast using a variable density sampling pattern with
8 reference-lines, also used for coil-sensitivity estimation7, and
acceleration factors $$$R\in\{8,12,16\}$$$ on 14 datasets. Sampling patterns
were chosen randomly out of a pool of 100 masks in each training iteration to
increase the variability of undersampling artifacts. The VN consists of $$$T=20$$$
GD steps, where $$$N_k=36$$$ filter kernels of size $$$7{\times}7{\times}5$$$
are learned along with their corresponding activation functions $$$\phi$$$.
Testing was performed on the remaining 14 datasets. The VN reconstructions were
compared to the CS-PI method Infimal-Convolution-Total-Generalized-Variation
(ICTGV)8 qualitatively and in terms of root-mean-squared-error (RMSE)
and structural similarity index (SSIM). Results
Figure 2
shows two-chamber, four-chamber and short-axis views for $$$R=12$$$. A
comparison for investigated acceleration factors is depicted in Figure 3. For
bSSFP, the VN results appear differently textured than the ICTGV
reconstructions. The VN reconstructions for FLASH appear sharper and with more
details and preserved texture compared to ICTGV (see detailed view in Figure 4).
The quantitative results in Figure 5 support our observations. While the RMSE
values for ICTGV and VN are similar, the SSIM results for VN are improved.
Besides the improvements in image quality, VN results can be reconstructed in
~6 sec, hence, $$$10\times$$$ faster than GPU-powered ICTGV reconstructions.Discussion and Conclusion
In this
work, we showed the efficiency of VNs for dynamic multi-coil data, exemplified
for CINE cardiac MRI. The proposed VNs allow us to reconstruct images $$$10\times$$$
faster than current PI-CS approaches. The additional information in the
temporal domain allows for higher accelerations as in the static case1.
The surprisingly small differences between the reconstructions for different
accelerations suggest to push the acceleration further in future work. In general,
CINE cardiac imaging is an important modality in dynamic multi-coil MR image reconstruction. However, it poses an interesting challenge for learning-based
approaches to generate high-quality reference data due to limitations in
breathhold capabilities and signal preparation. For bSSFP data, typical
acquisition artifacts such as banding seem to impact the quality of learning,
reflected in the results. In contrast, FLASH contains less acquisition
artifacts, but is characterized by a poorer SNR. This level of SNR leads to an
unnatural behaviour of ICTGV, while the VN results appear more natural and
sharper.Acknowledgements
We acknowledge grant support from the Austrian Science Fund (FWF) under the START project BIVISION, No. Y729, and ERC starting grant ”HOMOVIS”, No. 640156.References
[1] K.
Hammernik, T. Klatzer, E. Kobler, M. P. Recht, D. K. Sodickson, T. Pock and F.
Knoll. “Learning a Variational
Network for Reconstruction of Accelerated MRI Data”. Magnetic Resonance in
Medicine, vol. 79, no. 6, pp. 3055-3071, 2018.
[2] J. Schlemper, J. Caballero, J. Hajnal,
A. Price, and D. Rueckert. “A Deep Cascade of Convolutional Neural Networks for
Dynamic MR Image Reconstruction”. IEEE Transactions on Medical Imaging, vol.
37, no. 2, pp. 491-503, 2018.
[3] C. Qin,
J. Schlemper, J. Caballero, A. Price, J. V. Hajnal, and D. Rueckert.
“Convolutional Recurrent Neural Networks for Dynamic MR Image Reconstruction”.
IEEE Transactions on Medical Imaging, 2018 (in press).
[4] A.
Hauptmann, S. Arridge, F. Lucka, V. Muthurangu, and J. A. Steeden. “Real-time
cardiovascular MR with spatio-temporal artifact suppression using deep
learning-proof of concept in congenital heart disease”. Magnetic Resonance in
Medicine, 2018 (in press).
[5] E.
Kobler, T. Klatzer, K. Hammernik, and T. Pock. “Variational Networks:
Connecting Variational Methods and Deep Learning”. In German Conference on
Pattern Recognition (GCPR) 2017, Lecture Notes in Computer Science, vol. 10496,
pp. 281-293, 2017.
[6] Y. Chen, W. Yu, T. Pock. “On Learning Optimized Reaction Diffusion
Processes
for
Effective Image Restoration”. In Proceedings of the IEEE Conference on
Computer
Vision and Pattern Recognition (CVPR), pp. 5261-5269, 2015.
[7] M. Schloegl, M. Holler, K. Bredies, and R. Stollberger. “A Variational Approach for
Coil-Sensitivity Estimation for Undersampled Phase-Sensitive Dynamic MRI
Reconstruction”. In Proceedings of the International Society of Magnetic
Resonance in Medicine (ISMRM), vol. 23, p. 3692, Toronto, Canada, 2015.
[8] M. Schloegl, M. Holler, A. Schwarzl, K. Bredies, and R.
Stollberger. “Infimal
convolution of total generalized variation functionals for dynamic MRI”. Magnetic
Resonance in Medicine, vol. 78, no. 1, pp. 142-155, 2018.