Transferable Optimization Network for Cross-Domain Image Reconstruction
About
We develop a novel transfer learning framework to tackle the challenge of limited training data in image reconstruction problems. The proposed framework consists of two training steps, both of which are formed as bi-level optimizations. In the first step, we train a powerful universal feature-extractor that is capable of learning important knowledge from large, heterogeneous data sets in various domains. In the second step, we train a task-specific domain-adapter for a new target domain or task with only a limited amount of data available for training. Then the composition of the adapter and the universal feature-extractor effectively explores feature which serve as an important component of image regularization for the new domains, and this leads to high-quality reconstruction despite the data limitation issue. We apply this framework to reconstruct under-sampled MR images with limited data by using a collection of diverse data samples from different domains, such as images of other anatomies, measurements of various sampling ratios, and even different image modalities, including natural images. Experimental results demonstrate a promising transfer learning capability of the proposed method.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Reconstruction | ImageNet | PSNR36.66 | 56 | |
| Image Reconstruction | CIFAR-10 | -- | 25 | |
| Image Reconstruction | Stanford2D | PSNR (dB)38.17 | 7 | |
| MRI Reconstruction | fastMRI | PSNR (dB)33.58 | 7 | |
| MRI Reconstruction | Brain anatomy MRI | PSNR (dB)37.88 | 7 | |
| MRI Reconstruction | Knee anatomy MRI | PSNR (dB)32.47 | 7 | |
| MRI Reconstruction | Cardiac anatomy MRI | PSNR (dB)37.26 | 7 | |
| MRI Reconstruction | Prostate anatomy MRI | PSNR (dB)33.31 | 7 | |
| MRI Reconstruction | MRI 10% sampling ratio | PSNR (dB)32.13 | 7 | |
| MRI Reconstruction | MRI 15% sampling ratio small (train) | PSNR (dB)34.9 | 7 |