Choice of training label matters: how to best use deep learning for quantitative MRI parameter estimation
arXiv:2205.05587 · doi:10.59275/j.melba.2024-geb5
Abstract
Deep learning (DL) is gaining popularity as a parameter estimation method for quantitative MRI. A range of competing implementations have been proposed, relying on either supervised or self-supervised learning. Self-supervised approaches, sometimes referred to as unsupervised, have been loosely based on auto-encoders, whereas supervised methods have, to date, been trained on groundtruth labels. These two learning paradigms have been shown to have distinct strengths. Notably, self-supervised approaches have offered lower-bias parameter estimates than their supervised alternatives. This result is counterintuitive - incorporating prior knowledge with supervised labels should, in theory, lead to improved accuracy. In this work, we show that this apparent limitation of supervised approaches stems from the naive choice of groundtruth training labels. By training on labels which are deliberately not groundtruth, we show that the low-bias parameter estimation previously associated with self-supervised methods can be replicated - and improved on - within a supervised learning framework. This approach sets the stage for a single, unifying, deep learning parameter estimation framework, based on supervised learning, where trade-offs between bias and variance are made by careful adjustment of training label.
arXiv commentAccepted for publication at the Journal of Machine Learning for Biomedical Imaging (MELBA) https://melba-journal.org/2024:002
References in corpus (5)
- Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)
- Deep Learning How to Fit an Intravoxel Incoherent Motion Model to Diffusion-Weighted MRI
- SANDI: a compartment-based model for non-invasive apparent soma and neurite imaging by diffusion MRI
- Model-Informed Machine Learning for Multi-component T2 Relaxometry
- Task-driven assessment of experimental designs in diffusion MRI: a computational framework