Keywords: multi-task learning, homoscedastic uncertainty, dynamic weight average, deep inverse, domain shift, MRI, convolutional neural network, variational network
TL;DR: We propose a novel multi-task learning scheme that can jointly reconstruct various MRI datasets using a single network.
Abstract: Model-based accelerated MRI reconstruction methods leverage large datasets to reconstruct diagnostic-quality images from undersampled k-space. These networks require matching training and test time distributions to achieve high quality reconstructions. However, there is inherent variability in MR datasets, including different contrasts, orientations, anatomies, and institution-specific protocols. The current paradigm is to train separate models for each dataset. However, this is a demanding process and cannot exploit information that may be shared amongst datasets. To address this issue, we propose multi-task learning (MTL) schemes that can jointly reconstruct multiple datasets. We test multiple MTL architectures and weighted loss functions against single task learning (STL) baselines. Our quantitative and qualitative results suggest that MTL can outperform STL across a range of dataset ratios for two knee contrasts.
Conference Poster: pdf