Neural networks for parameter estimation in microstructural MRI: application to a diffusion-relaxation model of white matter

Abstract

Specific features of white matter microstructure can be investigated by using biophysical models to interpret relaxation-diffusion MRI brain data. Although more intricate models have the potential to reveal more details of the tissue, they also incur time-consuming parameter estimation that may converge to inaccurate solutions due to a prevalence of local minima in a degenerate fitting landscape. Machine-learning fitting algorithms have been proposed to accelerate the parameter estimation and increase the robustness of the attained estimates. So far, learning-based fitting approaches have been restricted to microstructural models with a reduced number of independent model parameters where dense sets of training data are easy to generate. Moreover, the degree to which machine learning can alleviate the degeneracy problem is poorly understood. For conventional least-squares solvers, it has been shown that degeneracy can be avoided by acquisition with optimized relaxation-diffusion-correlation protocols that include tensor-valued diffusion encoding. Whether machine-learning techniques can offset these acquisition requirements remains to be tested. In this work, we employ artificial neural networks to vastly accelerate the parameter estimation for a recently introduced relaxation-diffusion model of white matter microstructure. We also develop strategies for assessing the accuracy and sensitivity of function fitting networks and use those strategies to explore the impact of the acquisition protocol. The developed learning-based fitting pipelines were tested on relaxation-diffusion data acquired with optimal and sub-optimal acquisition protocols. Networks trained with an optimized protocol were observed to provide accurate parameter estimates within short computational times. Comparing neural networks and least-squares solvers, we found the performance of the former to be less affected by sub-optimal protocols; however, model fitting networks were still susceptible to degeneracy issues and their use could not fully replace a careful design of the acquisition protocol.