Diff-DTI: Fast Diffusion Tensor Imaging Using A Feature-Enhanced Joint Diffusion Model

Lang Zhang,Jinling He,Dong Liang,Hairong Zheng,Yanjie Zhu
2024-05-24
Abstract:Magnetic resonance diffusion tensor imaging (DTI) is a critical tool for neural disease diagnosis. However, long scan time greatly hinders the widespread clinical use of DTI. To accelerate image acquisition, a feature-enhanced joint diffusion model (Diff-DTI) is proposed to obtain accurate DTI parameter maps from a limited number of diffusion-weighted images (DWIs). Diff-DTI introduces a joint diffusion model that directly learns the joint probability distribution of DWIs with DTI parametric maps for conditional generation. Additionally, a feature enhancement fusion mechanism (FEFM) is designed and incorporated into the generative process of Diff-DTI to preserve fine structures in the generated DTI maps. A comprehensive evaluation of the performance of Diff-DTI was conducted on the Human Connectome Project dataset. The results demonstrate that Diff-DTI outperforms existing state-of-the-art fast DTI imaging methods in terms of visual quality and quantitative metrics. Furthermore, Diff-DTI has shown the ability to produce high-fidelity DTI maps with only three DWIs, thus overcoming the requirement of a minimum of six DWIs for DTI.
Image and Video Processing
What problem does this paper attempt to address?
### Problems the paper attempts to solve The paper aims to solve the problem of overly long scanning time in magnetic resonance diffusion tensor imaging (DTI). DTI is an important diagnostic tool for neurological diseases, but due to its long scanning time, it severely restricts its wide - scale clinical application. Theoretically, DTI requires at least 7 images, including 1 baseline image (b0 image) and 6 diffusion - weighted images (DWIs) in different directions to calculate the diffusion tensor \( \mathbf{D} \). In order to improve the estimation accuracy of \( \mathbf{D} \), it is usually necessary to obtain more than 30 DWIs in multiple diffusion directions, which further prolongs the scanning time. The paper proposes a feature - enhanced joint diffusion model (Diff - DTI) to accelerate image acquisition by generating accurate DTI parameter maps from a limited number of DWIs. Specifically, Diff - DTI introduces a joint diffusion model that directly learns the joint probability distribution of DWIs and DTI parameter maps for conditional generation. In addition, a feature - enhanced fusion mechanism (FEFM) is designed and integrated to preserve fine structures in the generated DTI maps. ### Main contributions 1. **Accelerating DTI imaging**: Diff - DTI can generate high - fidelity DTI maps using only 3 DWIs, significantly reducing the required amount of data and thus shortening the scanning time. 2. **High - quality image generation**: Diff - DTI outperforms existing fast DTI imaging methods in both visual quality and quantitative metrics. 3. **Feature enhancement**: By introducing FEFM, Diff - DTI can better capture and preserve detailed features in the image, improving the quality of the generated image. ### Experimental results - **Visual quality**: The DTI maps generated by Diff - DTI are very close to the reference maps visually and perform excellently in the reconstruction of complex texture details. - **Quantitative evaluation**: Diff - DTI shows the best performance in metrics such as peak signal - to - noise ratio (PSNR), structural similarity index (SSIM), and normalized mean - square error (NMSE). - **Performance under the condition of few DWIs**: Even when using only 3 DWIs, Diff - DTI can still generate high - quality DTI maps, showing its advantage in reducing the amount of data. ### Conclusion Diff - DTI successfully solves the problem of overly long DTI scanning time by introducing the feature - enhanced joint diffusion model, and significantly improves the generation quality and efficiency of DTI images. This method has important potential value in clinical applications.