SamRobNODDI:q-space sampling-augmented continuous representation learning for robust and generalized NODDI

Objective. Neurite orientation dispersion and density imaging (NODDI) microstructure estimation from diffusion magnetic resonance imaging (dMRI) is of great significance for the discovery and treatment of various neurological diseases. Current deep learning-based methods accelerate the speed of NODDI parameter estimation and improve the accuracy. However, most methods require the number and coordinates of gradient directions during testing and training to remain strictly consistent, significantly limiting the generalization and robustness of these models in NODDI parameter estimation. Therefore, it is imperative to develop methods that can perform robustly under varying diffusion gradient directions.Approach. In this paper, we propose aq-space sampling augmentation-based continuous representation learning framework (SamRobNODDI) to achieve robust and generalized NODDI. Specifically, a continuous representation learning method based onq-space sampling augmentation is introduced to fully explore the information between different gradient directions inq-space. Furthermore, we design a sampling consistency loss to constrain the outputs of different sampling schemes, ensuring that the outputs remain as consistent as possible, thereby further enhancing performance and robustness to varyingq-space sampling schemes. SamRobNODDI is also a flexible framework that can be applied to different backbone networks.Main results. SamRobNODDI was compared against seven state-of-the-art methods across 19 diverseq-space sampling schemes. Extensive experimental validations have been conducted under both identical and diverse sampling schemes for training and testing, as well as across varying sampling rates, different loss functions, and multiple network backbones.Significance. Results demonstrate that the proposed SamRobNODDI has better performance, robustness, generalization, and flexibility in the face of varyingq-space sampling schemes.