So Hyun Kang*1, Jihoo Kim*1, JaeJin Cho2,3, Clarissa Z. Cooley2,3, Berkin Bilgic2,3,4, and Tae Hyung Kim1
1Department of Computer Engineering, Hongik University, Seoul, Korea, Republic of, 2Athinoula A. Martinos Center for Biomedical Imaging, Massachusetts General Hospital, Charlestown, MA, United States, 3Department of Radiology, Harvard Medical School, Boston, MA, United States, 4Harvard/MIT Health Sciences and Technology, Cambridge, MA, United States
Synopsis
Keywords: Machine Learning/Artificial Intelligence, Machine Learning/Artificial Intelligence, Image Reconstruction, Partial Fourier, Brain, Multi-echo MRI, Low-field MR
Motivation: We introduce a novel partial Fourier reconstruction method.
Goal(s): The objective is to enhance partial Fourier reconstruction by integrating the traditional phase constraint with the recent zero-shot deep learning approach.
Approach: The proposed method combines the virtual conjugate coils (VCC) phase constraint with zero-shot deep untrained generative prior (ZS-DUGP), assuming MRI can be nonlinearly represented by untrained networks, enabling simultaneous image reconstruction and prior learning without external training data. This approach enables robust partial Fourier reconstruction.
Results: Evaluation across diverse datasets, including the fastMRI, the QALAS multi-echo data, and the low-field MR data, validates its enhanced performance compared to existing techniques.
Impact: We propose a novel partial Fourier reconstruction combining virtual conjugate coils with a zero-shot untrained generative network prior. It provides robust reconstruction without external training dataset, evaluated across various scenarios (parallel imaging, multi-echo/contrast imaging, low-field MR) demonstrating its utility.
Introduction
We present a novel partial Fourier reconstruction method based on zero-shot deep learning. The partial Fourier method1–4 is a widely used accelerated MRI technique, where reconstruction is often performed by employing the phase smoothness constraint or conjugate symmetry in k-space5–12.
Other venues for accelerated imaging include the recently developed scan-specific deep learning reconstruction techniques13–15. Among these, zero-shot MRI reconstruction using a deep untrained generative network (ZS-DUGP)16 was proven to be powerful in multi-contrast imaging. This approach assumes a prior, suggesting that MRI images can be nonlinearly represented by an untrained generative neural network, enabling simultaneous image reconstruction and prior learning without the need for an external training dataset. It demonstrated improved reconstruction performance, especially in multi-echo/contrast scenarios.
In this study, we introduce a novel partial Fourier reconstruction method based on ZS-DUGP. The proposed method is built upon the same prior used in ZS-DUGP and incorporates a phase constraint by exploiting virtual conjugate coils. We evaluated the proposed method on various datasets, including a fastMRI17,18, a QALAS multi-echo dataset19–21, and a low-field dataset22,23, demonstrating improved performance compared to existing methods.
Code: https://anonymous.4open.science/r/Partial_Fourier_Reconstruction_With_Zero-Shot_Deep_Untrained_Generative_Prior-5495/Theory
We employ the following MRI data acquisition model,$$\mathbf{y}=\mathbf{U}\mathbf{F}\mathbf{S}\mathbf{x}=\mathbf{U}\mathbf{F}\mathbf{S}\mathbf{P}\mathbf{m}$$Here, $$$\mathbf{y}\in\mathbb{C}^m$$$ represents the acquired k-space data, $$$\mathbf{U}$$$ is undersampling, $$$\mathbf{F}$$$ is the Fourier transform, $$$\mathbf{S}$$$ represents coil sensitivity, and $$$\mathbf{x}\in\mathbb{C}^n$$$ represents the underlying image to be reconstructed. We decompose the image $$$\mathbf{x}$$$ as $$$\mathbf{x}=\mathbf{P}\mathbf{m}$$$, where $$$\mathbf{P}\in\mathbb{C}^{n\times n}$$$ is a diagonal matrix for the image phase, and $$$\mathbf{m}\in \mathbb{C}^n$$$ represents the magnitude. Assuming that the phase $$$\mathbf{P}$$$ is estimated in advance, we will employ the virtual conjugate coils (VCC)10 for the phase constraint,$$\tilde{\mathbf{y}}^*=\tilde{\mathbf{U}}\mathbf{F}\mathbf{S}^*\mathbf{P}^*\mathbf{m}$$where $$$\tilde{\color{white}{\cdot}}$$$,$$$\color{white}{\cdot}^*$$$ denote spatial-reversal and complex conjugation, respectively. Additionally, the ZS-DUGP employs the following prior assumption,$$\mathbf{x}\approx f_\theta(\mathbf{z})$$where $$$f_\theta(\cdot)$$$ represents an untrained generative neural network with trainable parameters $$$\theta$$$, and $$$\mathbf{z}$$$ is a fixed random input. The corresponding MAP estimator becomes,$$\hat{\mathbf{m}},\hat{\mathbf{\theta}}=\arg\min_{\mathbf{m},\mathbf{\theta}}\left\|\pmatrix{\mathbf{y}\\\tilde{\mathbf{y}}^*}-\pmatrix{\mathbf{U}\mathbf{F}\mathbf{S}\mathbf{P}\\\tilde{\mathbf{U}}\mathbf{F}\mathbf{S}^*\mathbf{P}^*}\mathbf{m}\right\|_2^2+\lambda\left\|\mathbf{m}-f_\theta(\mathbf{z})\right\|_2^2$$By applying alternating minimization, we obtain,$$\theta^{(i+1)}=\arg\min_{\theta}\left\|\mathbf{m}-f_\theta(\mathbf{z})\right\|_2^2$$$$\mathbf{m}^{(i+1)}=\left(\mathbf{B}^H\mathbf{B}+\mathbf{C}^H\mathbf{C}+\lambda\mathbf{I}\right)^{-1}\left(\mathbf{B}^H\mathbf{y}+\mathbf{C}^H\tilde{\mathbf{y}}^*+\lambda f_{\theta^{(i+1)}}(\mathbf{z})\right)$$where $$$\mathbf{B}=\mathbf{U}\mathbf{F}\mathbf{S}\mathbf{P}$$$, $$$\mathbf{C}=\tilde{\mathbf{U}}\mathbf{F}\mathbf{S}^*\mathbf{P}^*$$$. The first equation represents the loss function for network training, and the second one represents the image update through regularized constrained reconstruction. This iterative procedure allows for simultaneous network training (for prior learning) and image reconstruction without requiring external training data. Fig. 1 illustrates this procedure.
This process assumes that phase can be estimated in advance, often using a fully sampled autocalibration signal (ACS) in conventional methods. In our study, we performed phase estimation without fully sampled ACS. Fig. 2 shows the phase estimation process, where we conducted an initial SENSE reconstruction and extracted the central k-space region that was treated as synthetic ACS. This allowed us to determine the $$$\mathbf{P}$$$ matrix, which is integrated into the iterative process above.Methods
To evaluate the proposed method, three datasets were prepared. The first dataset was the fastMRI brain dataset17,18, from which we selected a fully-sampled data from the validation set and extracted a single slice. We removed the oversampled and zero-padded regions to generate a 2D dataset with a matrix size 224x320, and the original 16 channels were coil-compressed into 12 channels. For the experiments, we applied 3x uniform undersampling with 5/8 partial Fourier ($$$R_{eff}=4.8\times$$$). We compared the proposed method with SENSE24, VCC-SENSE10, SENSE+TV25, ZS-SSL15.
The second dataset was the 3D-QALAS dataset19–21, comprising 5 multi-echo data obtained through elliptical scanning. The matrix size was 208x208x176, with a resolution of 1.15mmx1.15mmx1.15mm. We selected one slice for reconstruction and applied a 12x variable density mask with 5/8 partial Fourier ($$$R_{eff}=15.5\times$$$). In the experiments, we compared SENSE24, Subspace with locally low rank constraint (Subspace+LLR)26,27, VCC-SENSE10, SENSE with joint TV (SENSE+jTV)28, and the proposed method.
The third dataset was the low-field dataset, acquired from a 73mT, 50 kg portable “Halbach dome” head scanner22,23. The data was acquired using a single coil and prospective partial Fourier sampling with the matrix size of 256x101. We applied the proposed method for reconstruction.Results
Fig. 3 shows the reconstruction results for the fastMRI dataset. Fig. 4 presents the results for the QALAS dataset. Compared with other methods, the proposed approach enables partial Fourier reconstruction and demonstrates improved performance with lower NRMSE values. Fig. 5 displays the reconstruction results for the low-field data, confirming that the proposed method can recover the missing k-space regions.Conclusion
In this study, we proposed a novel partial Fourier reconstruction using the zero-shot deep untrained generative prior. This method combines the phase constraint using virtual conjugate coils with an untrained generative neural network, allowing robust partial Fourier reconstruction without the need for external training data. We experimentally validated the proposed method to various scenarios, such as parallel imaging, multi-echo/contrast imaging, and low-field MR, demonstrating its utility.Acknowledgements
* These authors contributed equally to this work.
This work was supported in part by the National Research Foundation of Korea (NRF) grant funded by the Korean government (MSIT) (No. 2022R1F1A1074786), research grants NIH R01 EB028797, U01 EB025162, P41 EB030006, U01 EB026996, R03 EB031175, R01 EB032378, UG3 EB034875 and by Nvidia Corporation for computing support.
References
- Margosian P, Schmitt F, Purdy D. Faster MR imaging: imaging with half the data. Health Care Instrum. 1986;1(6):195.
- Noll DC, Nishimura DG, Macovski A. Homodyne detection in magnetic resonance imaging. IEEE Transactions on Medical Imaging. 1991;10(2):154-163. doi:10.1109/42.79473
- Huang F, Lin W, Li Y. Partial fourier reconstruction through data fitting and convolution in k-space. Magnetic Resonance in Medicine. 2009;62(5):1261-1269. doi:10.1002/mrm.22128
- Liang ZP, Boada F, Constable R, Haacke E, Lauterbur P, Smith M. Constrained reconstruction methods in MR imaging. Rev Magn Reson Med. 1992;4(2):67-185.
- Samsonov AA, Kholmovski EG, Parker DL, Johnson CR. POCSENSE: POCS-based reconstruction for sensitivity encoded magnetic resonance imaging. Magnetic Resonance in Medicine. 2004;52(6):1397-1406. doi:10.1002/mrm.20285
- Willig-Onwuachi JD, Yeh EN, Grant AK, Ohliger MA, McKenzie CA, Sodickson DK. Phase-constrained parallel MR image reconstruction. Journal of Magnetic Resonance. 2005;176(2):187-198. doi:10.1016/j.jmr.2005.06.004
- Bydder M, Robson MD. Partial fourier partially parallel imaging. Magnetic Resonance in Medicine. 2005;53(6):1393-1401. doi:10.1002/mrm.20492
- Lew C, Pineda AR, Clayton D, Spielman D, Chan F, Bammer R. SENSE phase-constrained magnitude reconstruction with iterative phase refinement. Magnetic Resonance in Medicine. 2007;58(5):910-921. doi:10.1002/mrm.21284
- Haldar JP. Low-Rank Modeling of Local $k$-Space Neighborhoods (LORAKS) for Constrained MRI. IEEE Transactions on Medical Imaging. 2014;33(3):668-681. doi:10.1109/TMI.2013.2293974
- Blaimer M, Heim M, Neumann D, Jakob PM, Kannengiesser S, Breuer FA. Comparison of phase-constrained parallel MRI approaches: Analogies and differences. Magnetic Resonance in Medicine. 2016;75(3):1086-1099. doi:10.1002/mrm.25685
- Chang HC, Guhaniyogi S, Chen N kuei. Interleaved diffusion-weighted improved by adaptive partial-Fourier and multiband multiplexed sensitivity-encoding reconstruction. Magnetic Resonance in Medicine. 2015;73(5):1872-1884. doi:10.1002/mrm.25318
- Kim TH, Setsompop K, Haldar JP. LORAKS makes better SENSE: Phase-constrained partial fourier SENSE reconstruction without phase calibration. Magnetic Resonance in Medicine. 2017;77(3):1021-1035. doi:10.1002/mrm.26182
- Akçakaya M, Moeller S, Weingärtner S, Uğurbil K. Scan-specific robust artificial-neural-networks for k-space interpolation (RAKI) reconstruction: Database-free deep learning for fast imaging. Magnetic Resonance in Medicine. 2019;81(1):439-453. doi:10.1002/mrm.27420
- Arefeen Y, Beker O, Cho J, Yu H, Adalsteinsson E, Bilgic B. Scan-specific artifact reduction in k-space (SPARK) neural networks synergize with physics-based reconstruction to accelerate MRI. Magnetic Resonance in Medicine. 2022;87(2):764-780. doi:10.1002/mrm.29036
- Yaman B, Hosseini SAH, Akcakaya M. Zero-Shot Self-Supervised Learning for MRI Reconstruction. In: International Conference on Learning Representations. ; 2022. https://openreview.net/forum?id=085y6YPaYjP.
- Kim TH, Cho J, Gagoski B, et al. Multi-echo MRI Reconstruction with Iteratively Refined Zero-shot Spatio-Temporal Deep Generative Prior. In: International Society for Magnetic Resonance in Medicine Annual Meeting (ISMRM). ; 2023:828.
- Knoll F, Zbontar J, Sriram A, et al. fastMRI: A Publicly Available Raw k-Space and DICOM Dataset of Knee Images for Accelerated MR Image Reconstruction Using Machine Learning. Radiology: Artificial Intelligence. 2020;2(1):e190007. doi:10.1148/ryai.2020190007
- Muckley MJ, Riemenschneider B, Radmanesh A, et al. Results of the 2020 fastMRI Challenge for Machine Learning MR Image Reconstruction. IEEE Transactions on Medical Imaging. 2021:1-1. doi:10.1109/TMI.2021.3075856
- Kvernby S, Warntjes MJB, Haraldsson H, Carlhäll CJ, Engvall J, Ebbers T. Simultaneous three-dimensional myocardial T1 and T2 mapping in one breath hold with 3D-QALAS. Journal of Cardiovascular Magnetic Resonance. 2014;16(1):102. doi:10.1186/s12968-014-0102-0
- Kvernby S, Warntjes M, Engvall J, Carlhäll CJ, Ebbers T. Clinical feasibility of 3D-QALAS – Single breath-hold 3D myocardial T1- and T2-mapping. Magnetic Resonance Imaging. 2017;38:13-20. doi:10.1016/j.mri.2016.12.014
- Fujita S, Hagiwara A, Hori M, et al. Three-dimensional high-resolution simultaneous quantitative mapping of the whole brain with 3D-QALAS: An accuracy and repeatability study. Magnetic Resonance Imaging. 2019;63:235-243. doi:10.1016/j.mri.2019.08.031
- Cooley CZ, Stockmann J, Wald LL. A portable brain MRI scanner based on a 72 mT, 35 kg “Halbach-bulb” magnet and external gradient coil. In: International Society for Magnetic Resonance in Medicine Annual Meeting (ISMRM). ; 2021:559.
- Cooley CZ, McDaniel PC, Stockmann JP, et al. A portable scanner for magnetic resonance imaging of the brain. Nat Biomed Eng. 2021;5(3):229-239. doi:10.1038/s41551-020-00641-5
- Pruessmann KP, Weiger M, Scheidegger MB, Boesiger P. SENSE: Sensitivity encoding for fast MRI. Magnetic Resonance in Medicine. 1999;42(5):952-962. doi:10.1002/(SICI)1522-2594(199911)42:5<952::AID-MRM16>3.0.CO;2-S
- Lustig M, Donoho D, Pauly JM. Sparse MRI: The application of compressed sensing for rapid MR imaging. Magnetic Resonance in Medicine. 2007;58(6):1182-1195. doi:10.1002/mrm.21391
- Zhang T, Pauly JM, Levesque IR. Accelerating parameter mapping with a locally low rank constraint. Magnetic Resonance in Medicine. 2015;73(2):655-661. doi:10.1002/mrm.25161
- Tamir JI, Uecker M, Chen W, et al. T2 shuffling: Sharp, multicontrast, volumetric fast spin-echo imaging. Magnetic Resonance in Medicine. 2017;77(1):180-195. doi:10.1002/mrm.26102
- Chen C, Li Y, Huang J. Calibrationless Parallel MRI with Joint Total Variation Regularization. In: Medical Image Computing and Computer-Assisted Intervention – MICCAI 2013. Berlin, Heidelberg: Springer; 2013:106-114. doi:10.1007/978-3-642-40760-4_14