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

Purpose: Biophysical modeling of the diffusion MRI (dMRI) signal provides estimates of specific microstructural tissue properties. Although non-linear least squares (NLLS) is the most widespread fitting method, it suffers from local minima and high computational cost. Deep learning approaches are steadily replacing NLLS, but come with the limitation that the model needs to be retrained for each acquisition protocol and noise level. In this study, a novel fitting approach was proposed based on the encoder-decoder recurrent neural network (RNN) to accelerate model estimation with good generalization to various datasets.

Methods: The white matter tract integrity (WMTI)-Watson model as an implementation of the Standard Model of diffusion in white matter derives its parameters indirectly from the diffusion and kurtosis tensors (DKI). The RNN-based solver, which estimates the WMTI-Watson model from DKI, is therefore more readily translatable to various data, irrespective of acquisition protocols as long as the DKI was pre-computed from the signal. An embedding approach was also used to render the model insensitive to potential differences in distributions between training data and experimental data. The analytical solution, NLLS, RNN-, and a multilayer perceptron (MLP)-based methods were evaluated on synthetic and in vivo datasets of rat and human brain.

Results: The proposed RNN solver showed highly reduced computation time over the analytical solution and NLLS, with similar accuracy but improved robustness, and superior generalizability over MLP.

Conclusion: The RNN estimator can be easily applied to various datasets without retraining, which shows great potential for a widespread use.