Parameter estimation for WMTI-Watson model of white matter using encoder-decoder recurrent neural network

Abstract

Biophysical modelling of the diffusion MRI signal provides estimates of specific microstructural tissue properties. Although nonlinear optimization such as non-linear least squares (NLLS) is the most widespread method for model estimation, it suffers from local minima and high computational cost. Deep Learning approaches are steadily replacing NL fitting, but come with the limitation that the model needs to be retrained for each acquisition protocol and noise level. The White Matter Tract Integrity (WMTI)-Watson model was proposed as an implementation of the Standard Model of diffusion in white matter that estimates model parameters from the diffusion and kurtosis tensors (DKI). Here we proposed a deep learning approach based on the encoder-decoder recurrent neural network (RNN) to increase the robustness and accelerate the parameter estimation of WMTI-Watson. We use an embedding approach to render the model insensitive to potential differences in distributions between training data and experimental data. This RNN-based solver thus has the advantage of being highly efficient in computation and more readily translatable to other datasets, irrespective of acquisition protocol and underlying parameter distributions as long as DKI was pre-computed from the data. In this study, we evaluated the performance of NLLS, the RNN-based method and a multilayer perceptron (MLP) on synthetic and in vivo datasets of rat and human brain. We showed that the proposed RNN-based fitting approach had the advantage of highly reduced computation time over NLLS (from hours to seconds), with similar accuracy and precision but improved robustness, and superior translatability to new datasets over MLP.

References in corpus (7)

• Rotationally-invariant mapping of scalar and orientational metrics of neuronal microstructure with diffusion MRI
• Neurite Exchange Imaging (NEXI): A minimal model of diffusion in gray matter with inter-compartment water exchange
• Deep Learning How to Fit an Intravoxel Incoherent Motion Model to Diffusion-Weighted MRI
• Diffusion time dependence of microstructural parameters in fixed spinal cord
• Transfer Learning in Magnetic Resonance Brain Imaging: a Systematic Review
• Double Diffusion Encoding Prevents Degeneracy in Parameter Estimation of Biophysical Models in Diffusion MRI
• Intra- and extra-axonal axial diffusivities in the white matter: which one is faster?

Cited by in corpus (1)

• Neurite Exchange Imaging (NEXI): A minimal model of diffusion in gray matter with inter-compartment water exchange