Exploring the Potential of Machine Learning Algorithms to Improve Diffusion Nuclear Magnetic Resonance Imaging Models Analysis.

Abstract

Purpose: This paper explores different machine learning (ML) algorithms for analyzing diffusion nuclear magnetic resonance imaging (dMRI) models when analytical fitting shows restrictions. It reviews various ML techniques for dMRI analysis and evaluates their performance on different b-values range datasets, comparing them with analytical methods.

Materials and methods: After standard fitting for reference, four sets of diffusion-weighted nuclear magnetic resonance images were used to train/test various ML algorithms for prediction of diffusion coefficient (D), pseudo-diffusion coefficient (D*), perfusion fraction (f), and kurtosis (K). ML classification algorithms, including extra-tree classifier (ETC), logistic regression, C-support vector, extra-gradient boost, and multilayer perceptron (MLP), were used to determine the existence of diffusion parameters (D, D*, f, and K) within single voxels. Regression algorithms, including linear regression, polynomial regression, ridge, lasso, random forest (RF), elastic-net, and support-vector machines, were used to estimate the value of the diffusion parameters. Performance was evaluated using accuracy (ACC), area under the curve (AUC) tests, and cross-validation root mean square error (RMSE_CV). Computational timing was also assessed.

Results: ETC and MLP were the best classifiers, with 94.1% and 91.7%, respectively, for the ACC test and 98.7% and 96.3% for the AUC test. For parameter estimation, RF algorithm yielded the most accurate results The RMSE_CV percentages were: 8.39% for D, 3.57% for D*, 4.52% for f, and 3.53% for K. After the training phase, the ML methods demonstrated a substantial decrease in computational time, being approximately 232 times faster than the conventional methods.

Conclusions: The findings suggest that ML algorithms can enhance the efficiency of dMRI model analysis and offer new perspectives on the microstructural and functional organization of biological tissues. This paper also discusses the limitations and future directions of ML-based dMRI analysis.