Blended-Transfer Learning for Compressed-Sensing Cardiac CINE MRI
10.13104/imri.2021.25.1.10
- Author:
Seong Jae PARK
1
;
Chang-Beom AHN
Author Information
1. Department of Electrical Engineering, Kwangwoon University, Seoul, Korea
- Publication Type:Original Article
- From:Investigative Magnetic Resonance Imaging
2021;25(1):10-22
- CountryRepublic of Korea
- Language:English
-
Abstract:
Purpose:To overcome the difficulty in building a large data set with a high-quality in medical imaging, a concept of 'blended-transfer learning' (BTL) using a combination of both source data and target data is proposed for the target task.
Materials and Methods:Source and target tasks were defined as training of the source and target networks to reconstruct cardiac CINE images from undersampled data, respectively. In transfer learning (TL), the entire neural network (NN) or some parts of the NN after conducting a source task using an open data set was adopted in the target network as the initial network to improve the learning speed and the performance of the target task. Using BTL, an NN effectively learned the target data while preserving knowledge from the source data to the maximum extent possible. The ratio of the source data to the target data was reduced stepwise from 1 in the initial stage to 0 in the final stage.
Results:NN that performed BTL showed an improved performance compared to those that performed TL or standalone learning (SL). Generalization of NN was also better achieved. The learning curve was evaluated using normalized mean square error (NMSE) of reconstructed images for both target data and source data. BTL reduced the learning time by 1.25 to 100 times and provided better image quality. Its NMSE was 3% to 8% lower than with SL.
Conclusion:The NN that performed the proposed BTL showed the best performance in terms of learning speed and learning curve. It also showed the highest reconstructed-image quality with the lowest NMSE for the test data set. Thus, BTL is an effective way of learning for NNs in the medical-imaging domain where both quality and quantity of data are always limited.
