# High-dimensional Embedding Prior for Noisy K-space Domain MRIReconstruction

arXiv: https://arxiv.org/abs/2607.01176 · HTML: https://mlsanity.com/paper/2607.01176/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 30 of its 57 references that are in the map.
- First posted on arXiv 1 Jul 2026.

## Builds on (most cited first, 30 of 30)

- [Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2006.11239/) (2020)
- [Denoising Diffusion Implicit Models](https://mlsanity.com/paper/2010.02502/) (2020)
- [Improved Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2102.09672/) (2021)
- [Diffusion Posterior Sampling for General Noisy Inverse Problems](https://mlsanity.com/paper/2209.14687/) (2022)
- [Variational Diffusion Models](https://mlsanity.com/paper/2107.00630/) (2021)
- [Improving Diffusion Models for Inverse Problems using Manifold Constraints](https://mlsanity.com/paper/2206.00941/) (2022)
- [fastMRI: An Open Dataset and Benchmarks for Accelerated MRI](https://mlsanity.com/paper/1811.08839/) (2018)
- [MoDL: Model Based Deep Learning Architecture for Inverse Problems](https://mlsanity.com/paper/1712.02862/) (2017)
- [A Deep Cascade of Convolutional Neural Networks for Dynamic MR Image Reconstruction](https://mlsanity.com/paper/1704.02422/) (2017)
- [A Variational Perspective on Solving Inverse Problems with Diffusion Models](https://mlsanity.com/paper/2305.04391/) (2023)
- [A Survey on Diffusion Models for Inverse Problems](https://mlsanity.com/paper/2410.00083/) (2024)
- [Score Approximation, Estimation and Distribution Recovery of Diffusion Models on Low-Dimensional Data](https://mlsanity.com/paper/2302.07194/) (2023)
- [Adaptive Diffusion Priors for Accelerated MRI Reconstruction](https://mlsanity.com/paper/2207.05876/) (2022)
- [Convolutional Recurrent Neural Networks for Dynamic MR Image Reconstruction](https://mlsanity.com/paper/1712.01751/) (2017)
- [Deep learning for undersampled MRI reconstruction](https://mlsanity.com/paper/1709.02576/) (2017)
- [Beyond First-Order Tweedie: Solving Inverse Problems using Latent Diffusion](https://mlsanity.com/paper/2312.00852/) (2023)
- [Diffusion Model Based Posterior Sampling for Noisy Linear Inverse Problems](https://mlsanity.com/paper/2211.12343/) (2022)
- [Diffusion models for inverse problems](https://mlsanity.com/paper/2508.01975/) (2025)
- [Correlated and Multi-frequency Diffusion Modeling for Highly Under-sampled MRI Reconstruction](https://mlsanity.com/paper/2309.00853/) (2023)
- [One-shot Generative Prior in Hankel-k-space for Parallel Imaging Reconstruction](https://mlsanity.com/paper/2208.07181/) (2022)
- [CDiffMR: Can We Replace the Gaussian Noise with K-Space Undersampling for Fast MRI?](https://mlsanity.com/paper/2306.14350/) (2023)
- [Joint Frequency and Image Space Learning for MRI Reconstruction and Analysis](https://mlsanity.com/paper/2007.01441/) (2020)
- [WKGM: Weight-K-space Generative Model for Parallel Imaging Reconstruction](https://mlsanity.com/paper/2205.03883/) (2022)
- [Noise Level Adaptive Diffusion Model for Robust Reconstruction of Accelerated MRI](https://mlsanity.com/paper/2403.05245/) (2024)
- [Sampling possible reconstructions of undersampled acquisitions in MR imaging](https://mlsanity.com/paper/2010.00042/) (2020)
- [High-Dimensional MR Reconstruction Integrating Subspace and Adaptive Generative Models](https://mlsanity.com/paper/2306.08630/) (2023)
- [Learning to Reconstruct Accelerated MRI Through K-space Cold Diffusion without Noise](https://mlsanity.com/paper/2311.10162/) (2023)
- [Deep Data Consistency: a Fast and Robust Diffusion Model-based Solver for Inverse Problems](https://mlsanity.com/paper/2405.10748/) (2024)
- [Sub-DM:Subspace Diffusion Model with Orthogonal Decomposition for MRI Reconstruction](https://mlsanity.com/paper/2411.03758/) (2024)
- [Understanding Generalization from Embedding Dimension and Distributional Convergence](https://mlsanity.com/paper/2601.22756/) (2026)

## Abstract (arXiv)

Magnetic resonance imaging (MRI) reconstruction under realistic acquisition conditions can be fundamentally viewed as estimating the underlying k-space distribution from incomplete and noise-corrupted measurements. While diffusion models have recently shown strong potential as generative prior for inverse problems,existingapproachesstruggletohandlenoisyreconstruction settings, especially when operating directly in k-space domain. In this work, we propose a unified high-dimensional k-space reconstruction framework tailored for noisy inverse problems, whichenhancesdiffusion-based solversthroughrepresentation lifting.Ratherthanmodifyingthe underlying optimization procedures, the proposed framework augments the data representation space, enabling existing diffusion-based solvers to operate on enriched k-space embeddings with improved expressiveness. Extensive experiments on both in-house and public datasets across varying noise levels and undersampled factors demonstrate that the proposed frame work consistently improves reconstruction quality for multiple diffusion-based inverse solvers. Notably, the largest gains are observed in high-noise regimes, which is consistent with our theoretical analysis of error propagation under high-dimensional representation. These results suggest that high-dimensional representation provides a general and model-agnostic mechanism for improving diffusion-based MRI reconstruction in noisy settings, offering a new perspective on robust k-space generative modeling for practical inverse problems. The code will be available at https://github.com/yqx7150/HEP-MRIRec.