Deep J-Sense: Accelerated MRI Reconstruction via Unrolled Alternating Optimization

Overview

Accelerated multi-coil MRI acquisition is an ubiquitous approach in clinical use and is accompanied by reconstruction algorithms. In these systems, a single anatomical region is acquired by a set of coils, each with their own spatially varying sensitivity map. While multi-coil MRI provides additional degrees of freedom through simultaneous measurements of the same anatomical region, it comes with its own set of challenges related to accurately estimating these maps. Deep learning has been recently successfully leveraged to greatly improve reconstruction quality in multi-coil MRI acquisition. However, there are still major open questions concerning the robustness of these models when faced with distributional shifts (scan parameters at test time do not match the ones at training time) or robustness across different training conditions.

In this research project, we aim to solve some of the challenges of robust multi-coil MRI reconstruction and leverage the recent methodology of combining compressed sensing with deep learning, in the form of unrolled optimization to introduce Deep J-Sense: an algorithm that jointly estimates the image and set of sensitivity maps. We formulate multi-coil MRI reconstruction as an optimization problem with two unknown variables and a forward model based on k-space (spatial frequency) convolution. This optimization is unrolled in an alternating fashion, and a separate deep neural network is used for denoising each variable. The entire model is then trained end-to-end. A complete description is available in the paper.

Quantitative Results

We present statistical results obtained with methods trained and evaluate on a subset (five central slices from each scan) of the fastMRI [4] multi-coil knee dataset.

Test performance of Deep J-Sense, MoDL and E2E-VarNet
Figure 1. Test performance of Deep J-Sense (ours) [1], MoDL [2] and E2E-VarNet [3]. All models are trained at acceleration R=4 and tested at various acceleration factors, representing a distributional shift. Deep J-Sense has superior performance across the entire acceleration range, and can efficiently use more acquisitions at test-time.
A suite of Deep J-Sense and E2E-VarNet models trained and tested at different sizes of the fully-sampled auto-calibration signal
Figure 2. A suite of Deep J-Sense [1] and E2E-VarNet [3] models trained and tested at different sizes of the fully-sampled auto-calibration signal (ACS). A pair of models is trained and evaluated for each ACS size on the x-axis. This result highlights that Deep J-Sense can reliably learn a reconstruction function even with an extremely small ACS, whereas prior methods incur a performance gap.

Qualitative Results

Two randomly chosen reconstructions at acceleration R=4
Figure 3. Two randomly (not cherry-picked) chosen reconstructions at acceleration R=4. The first and last two rows correspond to each scan, respectively. The first and third rows show the reconstructed root sum-of-squares images, while the second and fourth show the magnitude of the estimated sensitivity map for a specific coil (no ground truth available). Deep J-Sense (ours) reconstructs images with a higher fidelity and estimates less noisy sensitivity maps outside of the anatomical region.

References

[1] Arvinte et al, J-Sense: Accelerated MRI Reconstruction via Unrolled Alternating Optimization, preprint, 2021.

[2] Aggarwal et al, MoDL: Model-based Deep Learning Architecture for Inverse Problems, IEEE Trans. Med. Imag., 2018.

[3] Sriram et al, End-to-end Variational Networks for Accelerated MRI Reconstruction, MICCAI, 2020.

[4] Zbontar et al, fastMRI: An open dataset and benchmarks for accelerated MRI, preprint, 2018.

Acknowledgements

Part of this work has been accepted for presentation at the ISMRM Annual Conference 2021. Supported by ONR grant N00014-19-1-2590, NIH Grant U24EB029240, and an AWS Machine Learning Research Award.