Reliable Deep Diffusion Tensor Estimation: Rethinking the Power of Data-Driven Optimization Routine

Jialong Li,Zhicheng Zhang,Yunwei Chen,Qiqi Lu,Ye Wu,Xiaoming Liu,QianJin Feng,Yanqiu Feng,Xinyuan Zhang
2024-09-04
Abstract:Diffusion tensor imaging (DTI) holds significant importance in clinical diagnosis and neuroscience research. However, conventional model-based fitting methods often suffer from sensitivity to noise, leading to decreased accuracy in estimating DTI parameters. While traditional data-driven deep learning methods have shown potential in terms of accuracy and efficiency, their limited generalization to out-of-training-distribution data impedes their broader application due to the diverse scan protocols used across centers, scanners, and studies. This work aims to tackle these challenges and promote the use of DTI by introducing a data-driven optimization-based method termed DoDTI. DoDTI combines the weighted linear least squares fitting algorithm and regularization by denoising technique. The former fits DW images from diverse acquisition settings into diffusion tensor field, while the latter applies a deep learning-based denoiser to regularize the diffusion tensor field instead of the DW images, which is free from the limitation of fixed-channel assignment of the network. The optimization object is solved using the alternating direction method of multipliers and then unrolled to construct a deep neural network, leveraging a data-driven strategy to learn network parameters. Extensive validation experiments are conducted utilizing both internally simulated datasets and externally obtained in-vivo datasets. The results, encompassing both qualitative and quantitative analyses, showcase that the proposed method attains state-of-the-art performance in DTI parameter estimation. Notably, it demonstrates superior generalization, accuracy, and efficiency, rendering it highly reliable for widespread application in the field.
Computer Vision and Pattern Recognition,Machine Learning,Image and Video Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the accuracy of parameter estimation in diffusion tensor imaging (DTI) in clinical diagnosis and neuroscience research. Specifically: 1. **Noise Sensitivity**: Traditional model - based fitting methods (such as the least - squares method) are very sensitive to noise when processing diffusion - weighted (DW) images, resulting in a decline in the estimation accuracy of the diffusion tensor and its related parameters (such as fractional anisotropy (FA), mean diffusivity (MD), axial diffusivity (AD), and radial diffusivity (RD)). 2. **Limited Generalization Ability**: Although traditional data - driven deep - learning methods show potential in terms of accuracy and efficiency, they have limited generalization ability when dealing with data outside the training distribution, which limits their possibility of being widely used in different centers, scanners, and studies. To meet these challenges, the paper proposes a data - driven optimization method - DoDTI (Data - driven optimization - based Diffusion Tensor Imaging). DoDTI combines the weighted linear least - squares fitting algorithm and the denoising regularization technique, aiming to improve the accuracy and generalization ability of DTI parameter estimation. Specifically, DoDTI achieves this goal in the following ways: - **Weighted Linear Least - Squares Fitting (WLLS)**: It is used to fit DW images to the diffusion tensor field from diverse acquisition settings. - **Denoising Regularization**: A deep - learning - based denoiser is applied to regularize the diffusion tensor field, rather than the DW images, thereby avoiding the limitation of network fixed - channel allocation. - **Alternating Direction Method of Multipliers (ADMM)**: It is used to solve the optimization objective and expand it to construct a deep neural network, using a data - driven strategy to learn network parameters. Through extensive validation experiments, including internally simulated data sets and externally obtained in - vivo data sets, the results show that DoDTI has achieved state - of - the - art performance in DTI parameter estimation, demonstrating excellent generalization ability, accuracy, and efficiency, making it highly reliable in this field.