Adaptive Mask-Guided k-Space Diffusion for Accelerated MRI Reconstruction.
Authors
Affiliations (4)
Affiliations (4)
- School of Information Engineering, Nanchang University, Nanchang, Jiangxi, China.
- School of Advanced Manufacturing, Nanchang University, Nanchang, Jiangxi, China.
- Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences, Shenzhen, Guangdong, China.
- Academy of Medical Engineering and Translational Medicine, Medical School, Faculty of Medicine, Tianjin University, Tianjin, China.
Abstract
Masked modeling has emerged as a powerful deep learning paradigm that reconstructs missing data by predicting masked regions, achieving strong performance across multiple domains. As a critical task in medical imaging, magnetic resonance imaging (MRI) reconstruction aims to recover high-quality images from under-sampled k-space data. However, existing methods usually optimize the full image domain or k-space, while overlooking the importance and structural roles of different frequency components. An adaptive mask-guided diffusion model (AMDM) was developed, which leverages adaptively estimated frequency distributions from k-space to construct a hybrid mask mechanism for handling diverse k-space inputs. It enables the effective separation of high-frequency and low-frequency components, producing multiple prior representations. In addition, the frequency distribution is used to generate adaptive masks, which subsequently guide the closed-loop diffusion process. Experimental results demonstrated improved reconstruction performance through frequency-aware modeling. AMDM achieved a PSNR of 36.89 dB on an independent public brain dataset without retraining, even at an under-sampling factor of <math xmlns="http://www.w3.org/1998/Math/MathML"><mrow><mi>R</mi> <mo>=</mo> <mn>12</mn></mrow> </math> . The proposed method shows that explicitly modeling k-space frequency heterogeneity provides an effective prior for MRI reconstruction, offering a promising direction for robust reconstruction under aggressive under-sampling.