Stella Yu : Publications / Google Scholar

Memory-efficient Learning for High-Dimensional MRI Reconstruction
Ke Wang and Michael Kellman and Christopher M. Sandino and Kevin Zhangand Shreyas S. Vasanawala and Jonathan I. Tamir and Stella X. Yu and Michael Lustig
Magna Laude Award, International Society for Magnetic Resonance in Medicine, Online, 15-20 May 2021
Paper | arXiv

Abstract
Deep learning (DL) based unrolled reconstructions have shown state-of-the-art performance for under-sampled magnetic resonance imaging (MRI). Similar to compressed sensing, DL can leverage high-dimensional data (e.g. 3D, 2D+time, 3D+time) to further improve performance. However, network size and depth are currently limited by the GPU memory required for backpropagation. Here we use a memory-efficient learning (MEL) framework which favorably trades off storage with a manageable increase in computation during training. Using MEL with multi-dimensional data, we demonstrate improved image reconstruction performance for in-vivo 3D MRI and 2D+time cardiac cine MRI. MEL uses far less GPU memory while marginally increasing the training time, which enables new applications of DL to high-dimensional MRI.

Keywords
magnetic resonance imaging (MRI), unrolled reconstruction, memory-efficient learning