Browse > Article
http://dx.doi.org/10.13104/imri.2020.24.4.232

A Self-Supervised Learning Framework for Under-Sampling Pattern Design Using Graph Convolution Network  

Li, Yuze (Center for Biomedical Imaging Research, Department of Biomedical Engineering, School of Medicine, Tsinghua University)
Chen, Huijun (Center for Biomedical Imaging Research, Department of Biomedical Engineering, School of Medicine, Tsinghua University)
Publication Information
Investigative Magnetic Resonance Imaging / v.24, no.4, 2020 , pp. 232-240 More about this Journal
Abstract
Purpose: To generate the under-sampling pattern using a self-supervised learning framework based on a graph convolutional network. Materials and Methods: We first decoded the k-space data into the graph and put it into the network. After the processing of graph convolution layers and graph pooling layers, the network generated the under-sampling pattern for MR reconstruction. We trained the network on the simulated brain dataset enabled by the self-supervised learning strategy. We did simulation along with the in vivo brain and liver experiments under different noise levels and accelerating factors to compare the performance between the proposed method and traditional methods using the PSNR and SSIM index. Results: The simulation experiments showed that the proposed method can achieve the best performance with low accelerating factors (2 and 3) at all noise levels and in high accelerating factors (4 and 5) at high noise levels (50 and 70 dB). In in vivo experiments, the proposed method attained the highest PSNR and SSIM in the brain dataset as well as in the liver dataset after fine tuning on a small liver dataset. Conclusion: The self-supervised learning framework based on a graph convolutional network was able to design the under-sampling mask for MR reconstruction. The superior performance in the simulation and in vivo experiments demonstrated the feasibility and flexibility of the proposed method and its potential in clinical use.
Keywords
Undersampling Pattern Design; Graph Convolutional Network; Self-supervised Learning;
Citations & Related Records
연도 인용수 순위
  • Reference
1 Griswold MA, Jakob PM, Heidemann RM, et al. Generalized autocalibrating partially parallel acquisitions (GRAPPA). Magn Reson Med 2002;47:1202-1210   DOI
2 Knoll F, Clason C, Diwoky C, Stollberger R. Adapted random sampling patterns for accelerated MRI. MAGMA 2011;24:43-50   DOI
3 Vellagoundar J, Machireddy RR. A robust adaptive sampling method for faster acquisition of MR images. Magn Reson Imaging 2015;33:635-643   DOI
4 Litjens G, Kooi T, Bejnordi BE, et al. A survey on deep learning in medical image analysis. Med Image Anal 2017;42:60-88   DOI
5 Kipf TN, Welling M. Semi-supervised classification with graph convolutional networks. ICLR 2017
6 Gal Y, Ghahramani Z. Dropout as a Bayesian approximation: representing model uncertainty in deep learning. ICML 2016
7 https://brainweb.bic.mni.mcgill.ca/brainweb/. Accessed November 22, 2019
8 Zhang Y, Peterson BS, Ji G, Dong Z. Energy preserved sampling for compressed sensing MRI. Comput Math Methods Med 2014;2014:546814   DOI
9 Pruessmann KP, Weiger M, Scheidegger MB, Boesiger P. SENSE: sensitivity encoding for fast MRI. Magn Reson Med 1999;42:952-962   DOI
10 Lustig M, Donoho D, Pauly JM. Sparse MRI: the application of compressed sensing for rapid MR imaging. Magn Reson Med 2007;58:1182-1195   DOI
11 Seeger M, Nickisch H, Pohmann R, Scholkopf B. Optimization of k-space trajectories for compressed sensing by Bayesian experimental design. Magn Reson Med 2010;63:116-126   DOI
12 Zhang Z, Cui P, Zhu W. Deep learning on graphs: a survey. arXiv, 2018;1812.04202
13 Gao H, Ji S. Graph U-nets. ICML 2019
14 Wang Z, Bovik AC, Sheikh HR, Simoncelli EP. Image quality assessment: from error visibility to structural similarity. IEEE Trans Image Process 2004;13:600-612   DOI
15 Goldstein T, Osher S. The split Bregman method for L1-regularized problems. Siam J Imaging Science 2009;2:323-343   DOI