Assessment of the generalization of learned image reconstruction and the potential for transfer learning

被引:159
作者
Knoll, Florian [1 ,2 ]
Hammernik, Kerstin [1 ,2 ,3 ]
Kobler, Erich [3 ]
Pock, Thomas [3 ,4 ]
Recht, Michael P. [1 ,2 ]
Sodickson, Daniel K. [1 ,2 ]
机构
[1] NYU, Dept Radiol, Sch Med, Ctr Biomed Imaging, 550 1st Ave, New York, NY 10016 USA
[2] NYU, Sch Med, Ctr Adv Imaging Innovat & Res CAI2R, 550 1st Ave, New York, NY 10016 USA
[3] Graz Univ Technol, Inst Comp Graph & Vis, Graz, Austria
[4] AIT Austrian Inst Technol GmbH, Ctr Vis Automat & Control, Vienna, Austria
基金
奥地利科学基金会; 欧洲研究理事会; 美国国家卫生研究院;
关键词
accelerated imaging; deep learning; iterative image reconstruction; machine learning; transfer learning; variational network; MRI;
D O I
10.1002/mrm.27355
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Purpose: Although deep learning has shown great promise for MR image reconstruction, an open question regarding the success of this approach is the robustness in the case of deviations between training and test data. The goal of this study is to assess the influence of image contrast, SNR, and image content on the generalization of learned image reconstruction, and to demonstrate the potential for transfer learning. Methods: Reconstructions were trained from undersampled data using data sets with varying SNR, sampling pattern, image contrast, and synthetic data generated from a public image database. The performance of the trained reconstructions was evaluated on 10 in vivo patient knee MRI acquisitions from 2 different pulse sequences that were not used during training. Transfer learning was evaluated by fine-tuning baseline trainings from synthetic data with a small subset of in vivo MR training data. Results: Deviations in SNR between training and testing led to substantial decreases in reconstruction image quality, whereas image contrast was less relevant. Trainings from heterogeneous training data generalized well toward the test data with a range of acquisition parameters. Trainings from synthetic, non-MR image data showed residual aliasing artifacts, which could be removed by transfer learning-inspired finetuning. Conclusion: This study presents insights into the generalization ability of learned image reconstruction with respect to deviations in the acquisition settings between training and testing. It also provides an outlook for the potential of transfer learning to fine-tune trainings to a particular target application using only a small number of training cases.
引用
收藏
页码:116 / 128
页数:13
相关论文
共 27 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], P 24 ANN M ISMRM SIN
[3]  
Cohen Ouri., 2017, Proceedings of the 25th Annual Meeting of the ISMRM, P688
[4]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[5]  
Gong E., 2017, Proceedings of the 25th Annual Meeting of the ISMRM, P5663
[6]  
Hammernik K., 2017, ISMRM 25th Annual Meeting, page, P0644
[7]  
Hammernik K., 2017, P 25 ANN M ISMRM HON
[8]  
Hammernik K., 2016, 24th Annual Meeting of ISMRM, P1088
[9]   Learning a variational network for reconstruction of accelerated MRI data [J].
Hammernik, Kerstin ;
Klatzer, Teresa ;
Kobler, Erich ;
Recht, Michael P. ;
Sodickson, Daniel K. ;
Pock, Thomas ;
Knoll, Florian .
MAGNETIC RESONANCE IN MEDICINE, 2018, 79 (06) :3055-3071
[10]  
Han YS, 2017, P 25 ANN M ISMRM HON, P690