Scalable Non-Cartesian Magnetic Resonance Imaging with R2D2
Yiwei Chen, Chao Tang, Amir Aghabiglou, Chung San Chu, Yves Wiaux
TL;DR
This work tackles scalable reconstruction for non-Cartesian MRI by decoupling data-consistency from denoisers through a residual-to-residual DNN series (R2D2), drawing on a Matching Pursuit perspective. The authors formalize the iteration $\boldsymbol{x}^i = \boldsymbol{x}^{i-1} + \mathbf{G}_{\boldsymbol{\theta}_i}(\boldsymbol{r}^{i-1}, \boldsymbol{x}^{i-1})$, with data residuals $\boldsymbol{r}$ computed from back-projected measurements and a normalized operator $\boldsymbol{\Phi}$. They propose two unrolled variants, R2D2-Net (NUFFT) and R2D2-Net (FFT), and show in radial 2D simulations that the scalable FFT version outperforms competitive baselines and the non-scalable NUFFT variant approaches, while still delivering high-quality reconstructions. The results indicate substantial gains over PnP and unrolled networks in terms of scalability and speed, with promising implications for 3D/4D dynamic MRI and multi-coil acquisitions. Future work will extend to complex-valued data, more advanced architectures, and larger-scale multi-coil settings to validate practical impact.
Abstract
We propose a new approach for non-Cartesian magnetic resonance image reconstruction. While unrolled architectures provide robustness via data-consistency layers, embedding measurement operators in Deep Neural Network (DNN) can become impractical at large scale. Alternative Plug-and-Play (PnP) approaches, where the denoising DNNs are blind to the measurement setting, are not affected by this limitation and have also proven effective, but their highly iterative nature also affects scalability. To address this scalability challenge, we leverage the "Residual-to-Residual DNN series for high-Dynamic range imaging (R2D2)" approach recently introduced in astronomical imaging. R2D2's reconstruction is formed as a series of residual images, iteratively estimated as outputs of DNNs taking the previous iteration's image estimate and associated data residual as inputs. The method can be interpreted as a learned version of the Matching Pursuit algorithm. We demonstrate R2D2 in simulation, considering radial k-space sampling acquisition sequences. Our preliminary results suggest that R2D2 achieves: (i) suboptimal performance compared to its unrolled incarnation R2D2-Net, which is however non-scalable due to the necessary embedding of NUFFT-based data-consistency layers; (ii) superior reconstruction quality to a scalable version of R2D2-Net embedding an FFT-based approximation for data consistency; (iii) superior reconstruction quality to PnP, while only requiring few iterations.
