We are sharing a MATLAB implementation of a machine learning algorithm for learning an optimized k-space sampling pattern in parallel MRI while also learning the parameters of a variational network used to reconstruct MR images.
The joint learning approach identifies effective pairs of k-space sampling locations and reconstruction parameters, resulting in better images than those reconstructed with conventional sampling patterns.
In experiments, combined learning has yielded observably improved images and better metrics of image quality. Improvements in root mean squared error (RMSE), structural similarity index measure (SSIM), peak signal-to-noise ratio (PSNR), and high frequency error norm (HFEN) have ranged from 1 percent to 62 percent. For more details about the advantages of this approach, see the related preprint below.
To learn the sampling pattern, our code employs an algorithm called bias-accelerated subset selection (BASS, available for download as Data-Driven Learning of MRI Sampling Pattern). To learn the variational network parameters, our code employs Adam, an algorithm for stochastic gradient-based optimization. For more information about these methods, see the references below.
Alternating Learning Approach for Variational Networks and Undersampling Pattern in Parallel MRI Applications.
arXiv. Preprint posted online October 27, 2021. arXiv:2110.14703 [eess.IV]
Please cite this work if you are using data-driven learning of MRI sampling pattern in your research.
Monotone FISTA with Variable Acceleration for Compressed Sensing Magnetic Resonance Imaging. IEEE Trans Comput Imaging.
IEEE Trans Comput Imaging. 2019 Mar;5(1):109-119. doi: 10.1109/TCI.2018.2882681
fastMRI: An Open Dataset and Benchmarks for Accelerated MRI.
arXiv. Preprint posted online November 21, 2018. arXiv:1811.08839 [cs.SV]
Fast data-driven learning of parallel MRI sampling patterns for large scale problems.
Sci Rep. 2021;11(1):19312. doi: 10.1038/s41598-021-97995-w
Fast Data-Driven Learning of MRI Sampling Pattern for Large Scale Problems.
arXiv. Preprint posted online November 4, 2020. arXiv:2011.02322 [eess.SP]
Adam: A Method for Stochastic Optimization.
arXiv. Preprint posted online December 22, 2014. arXiv:1412.6980 [cs.LG]
Get the Code
The software available on this page is provided free of charge and comes without any warranty. CAI²R and NYU Grossman School of Medicine do not take any liability for problems or damage of any kind resulting from the use of the files provided. Operation of the software is solely at the user’s own risk. The software developments provided are not medical products and must not be used for making diagnostic decisions.
The software is provided for non-commercial, academic use only. Usage or distribution of the software for commercial purpose is prohibited. All rights belong to the author (Marcelo Wust Zibetti) and NYU Grossman School of Medicine. If you use the software for academic work, please give credit to the author in publications and cite the related publications.
|2022-05-17||Version used for the published IEEE-TCI paper (reference pending). Includes VN and UNET reconstructions. Includes monotone and non-monotone BASS.|
|2022-01-14||Version used for the arXiv preprint.|
Questions about this resource may be directed to Marcelo Wust Zibetti, PhD.