QSMDiff: Unsupervised 3D Diffusion Models for Quantitative Susceptibility Mapping

Zhuang Xiong,Wei Jiang,Yang Gao,Feng Liu,Hongfu Sun
2024-03-21
Abstract:Quantitative Susceptibility Mapping (QSM) dipole inversion is an ill-posed inverse problem for quantifying magnetic susceptibility distributions from MRI tissue phases. While supervised deep learning methods have shown success in specific QSM tasks, their generalizability across different acquisition scenarios remains constrained. Recent developments in diffusion models have demonstrated potential for solving 2D medical imaging inverse problems. However, their application to 3D modalities, such as QSM, remains challenging due to high computational demands. In this work, we developed a 3D image patch-based diffusion model, namely QSMDiff, for robust QSM reconstruction across different scan parameters, alongside simultaneous super-resolution and image-denoising tasks. QSMDiff adopts unsupervised 3D image patch training and full-size measurement guidance during inference for controlled image generation. Evaluation on simulated and in-vivo human brains, using gradient-echo and echo-planar imaging sequences across different acquisition parameters, demonstrates superior performance. The method proposed in QSMDiff also holds promise for impacting other 3D medical imaging applications beyond QSM.
Image and Video Processing,Computer Vision and Pattern Recognition,Medical Physics
What problem does this paper attempt to address?
The paper aims to address the deconvolution problem in Quantitative Susceptibility Mapping (QSM), which is an inverse problem of quantifying susceptibility distribution from MRI tissue phase. Although supervised deep learning methods perform well in specific QSM tasks, their generalization ability under different acquisition scenarios remains limited. This paper proposes QSMDiff, a diffusion model based on 3D image patches, for robust QSM reconstruction under various scanning parameters, and it is capable of performing super-resolution processing and image denoising simultaneously. Specifically, QSMDiff adopts an unsupervised 3D image patch training approach, utilizing full-size measurement guidance to control image generation during inference. The method's evaluation on simulated and real human brain datasets demonstrates superior performance and high generalizability under different acquisition parameters (such as different image resolutions and acquisition directions). Additionally, QSMDiff has the potential to handle other 3D medical imaging applications.