Highly accelerated MRI via implicit neural representation guided posterior sampling of diffusion models

被引:1
作者
Chu, Jiayue [1 ]
Du, Chenhe [2 ]
Lin, Xiyue [2 ]
Zhang, Xiaoqun [3 ,4 ,5 ]
Wang, Lihui [6 ]
Zhang, Yuyao [2 ]
Wei, Hongjiang [1 ,7 ]
机构
[1] Shanghai Jiao Tong Univ, Sch Biomed Engn, Shanghai, Peoples R China
[2] ShanghaiTech Univ, Sch Informat Sci & Technol, Shanghai, Peoples R China
[3] Shanghai Jiao Tong Univ, Inst Nat Sci, Shanghai, Peoples R China
[4] Shanghai Jiao Tong Univ, Sch Math Sci, Shanghai, Peoples R China
[5] Shanghai Jiao Tong Univ, MOE LSC & SJTU GenSci Joint Lab, Shanghai, Peoples R China
[6] Guizhou Univ, Sch Comp Sci & Technol, Key Lab Intelligent Med Image Anal & Precise Diag, Guiyang, Peoples R China
[7] Shanghai Jiao Tong Univ, Natl Engn Res Ctr Adv Magnet Resonance Technol Dia, Shanghai, Peoples R China
基金
上海市科技启明星计划; 中国国家自然科学基金;
关键词
Diffusion model; Posterior sampling; Implicit neural representation; MRI acceleration; IMAGE-RECONSTRUCTION; INVERSE PROBLEMS; NETWORK;
D O I
10.1016/j.media.2024.103398
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Reconstructing high-fidelity magnetic resonance (MR) images from under-sampled k-space is a commonly used strategy to reduce scan time. The posterior sampling of diffusion models based on the real measurement data holds significant promise of improved reconstruction accuracy. However, traditional posterior sampling methods often lack effective data consistency guidance, leading to inaccurate and unstable reconstructions. Implicit neural representation (INR) has emerged as a powerful paradigm for solving inverse problems by modeling a signal's attributes as a continuous function of spatial coordinates. In this study, we present a novel posterior sampler for diffusion models using INR, named DiffINR. The INR-based component incorporates both the diffusion prior distribution and the MRI physical model to ensure high data fidelity. DiffINR demonstrates superior performance on in-distribution datasets with remarkable accuracy, even under high acceleration factors (up to R = 12 in single-channel reconstruction). Furthermore, DiffINR exhibits excellent generalizability across various tissue contrasts and anatomical structures with low uncertainty. Overall, DiffINR significantly improves MRI reconstruction in terms of accuracy, generalizability and stability, paving the way for further accelerating MRI acquisition. Notably, our proposed framework can be a generalizable framework to solve inverse problems in other medical imaging tasks.
引用
收藏
页数:12
相关论文
共 67 条
[1]   MoDL: Model-Based Deep Learning Architecture for Inverse Problems [J].
Aggarwal, Hemant K. ;
Mani, Merry P. ;
Jacob, Mathews .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2019, 38 (02) :394-405
[2]   High-Frequency Space Diffusion Model for Accelerated MRI [J].
Cao, Chentao ;
Cui, Zhuo-Xu ;
Wang, Yue ;
Liu, Shaonan ;
Chen, Taijin ;
Zheng, Hairong ;
Liang, Dong ;
Zhu, Yanjie .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2024, 43 (05) :1853-1865
[3]  
Cao CT, 2022, Arxiv, DOI arXiv:2212.11274
[4]   Accelerating multi-echo MRI in k-space with complex-valued diffusion probabilistic model [J].
Cao, Ying ;
Wang, Lihui ;
Zhang, Jian ;
Xia, Hui ;
Yang, Feng ;
Zhu, Yuemin .
2022 16TH IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING (ICSP2022), VOL 1, 2022, :479-484
[5]   Cluster Analysis of DSC MRI, Dynamic Contrast-Enhanced MRI, and DWI Parameters Associated with Prognosis in Patients with Glioblastoma after Removal of the Contrast-Enhancing Component: A Preliminary Study [J].
Chung, H. ;
Seo, H. ;
Choi, S. H. ;
Park, C. -k. ;
Kim, T. M. ;
Park, S. -h. ;
Won, J. K. ;
Lee, J. H. ;
Lee, S. -t. ;
Lee, J. Y. ;
Hwang, I. ;
Kang, K. M. ;
Yun, T. J. .
AMERICAN JOURNAL OF NEURORADIOLOGY, 2022, 43 (11) :1559-1566
[6]  
Chung H., 2023, INT C LEARN REPR
[7]  
Chung HYJ, 2024, Arxiv, DOI [arXiv:2303.05754, DOI 10.48550/ARXIV.2303.05754]
[8]   Come-Closer-Diffuse-Faster: Accelerating Conditional Diffusion Models for Inverse Problems through Stochastic Contraction [J].
Chung, Hyungjin ;
Sim, Byeongsu ;
Ye, Jong Chul .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :12403-12412
[9]   Score-based diffusion models for accelerated MRI [J].
Chung, Hyungjin ;
Ye, Jong Chul .
MEDICAL IMAGE ANALYSIS, 2022, 80
[10]  
Courant R., 1943, Bulletin of theAmerican Mathematical Society, DOI [10.1201/b16924-5, DOI 10.1201/B16924-5, DOI 10.1090/S0002-9904-1943-07818-4]