A geometric framework for ensemble average propagator reconstruction from diffusion MRI

Diffusion-weighted magnetic resonance imaging (dMRI) is a non-invasive technique to probe the complex micro-architecture of the tissue being imaged. The diffusional properties of the tissue at the imaged resolution are well captured by the ensemble average propagator (EAP), which is a probability density function characterizing the probability of water molecule diffusion. Many properties in the form of imaging 'stains' can then be computed from the EAP that can serve as bio-markers for a variety of diseases. This motivates the development of methods for the accurate estimation of the EAPs from dMRI, which is an actively researched area in dMRI analysis. To this end, in the recent past, dictionary learning (DL) techniques have been applied by many researchers for accurate reconstruction of the EAP fields from dMRI scans of the central nervous system (CNS). However, most of the DL-based methods did not exploit the geometry of the space of the EAPs, which are probability density functions. By exploiting the geometry of the space of probability density functions, it is possible to reconstruct EAPs that satisfy the mathematical properties of a density function and hence yield better accuracy in the EAP field reconstruction. Using a square root density parameterization, the EAPs can be mapped to a unit Hilbert sphere, which is a smooth manifold with well known geometry that we will exploit in our formulation of the DL problem. Thus, in this paper, we present a general formulation of the DL problem for data residing on smooth manifolds and in particular the manifold of EAPs, along with a numerical solution using an alternating minimization method. We then showcase the properties and the performance of our algorithm on the reconstruction of the EAP field in a patch-wise manner from the dMRI data. Through several synthetic, phantom and real data examples, we demonstrate that our non-linear DL-based approach produces accurate and spatially smooth estimates of the EAP field from dMRI in comparison to the state-of-the-art EAP reconstruction method called the MAPL method, as well as the linear DL-based EAP reconstruction approaches. To further demonstrate the accuracy and utility of our approach, we compute an entropic anisotropy measure (HA), that is a function of the well known Rényi entropy, from the EAP fields of control and injured rat spinal cords respectively. We demonstrate its utility as an imaging 'stain' via a quantitative comparison of HA maps computed from EAP fields estimated using our method and competing methods. The quantitative comparison is achieved using a two sample t-test and the results of significance are displayed for a visualization of regions of the spinal cord affected most by the injury.