论文标题
自我评分:基于分数的MRI重建模型的自我监督学习
Self-Score: Self-Supervised Learning on Score-Based Models for MRI Reconstruction
论文作者
论文摘要
最近,基于得分的扩散模型在MRI重建中表现出令人满意的性能。这些方法中的大多数都需要大量的完全采样的MRI数据作为培训集,有时在实践中很难获取。本文提出了用于MRI重建的完全采样的基于无DATA的分数扩散模型,该模型以未经言采样的数据以自我监督的方式学习了完全采样的MR图像。具体而言,我们首先通过贝叶斯深度学习从未采样的数据中推断出完全采样的MR图像分布,然后通过训练分数函数来扰动数据分布并近似其概率密度梯度。利用学到的分数函数为先验,我们可以通过执行条件的Langevin Markov链Monte Carlo(MCMC)采样来重建MR图像。公共数据集上的实验表明,所提出的方法优于现有的自我审查的MRI重建方法,并与常规(完全采样的数据训练)基于得分的扩散方法实现了可比的性能。
Recently, score-based diffusion models have shown satisfactory performance in MRI reconstruction. Most of these methods require a large amount of fully sampled MRI data as a training set, which, sometimes, is difficult to acquire in practice. This paper proposes a fully-sampled-data-free score-based diffusion model for MRI reconstruction, which learns the fully sampled MR image prior in a self-supervised manner on undersampled data. Specifically, we first infer the fully sampled MR image distribution from the undersampled data by Bayesian deep learning, then perturb the data distribution and approximate their probability density gradient by training a score function. Leveraging the learned score function as a prior, we can reconstruct the MR image by performing conditioned Langevin Markov chain Monte Carlo (MCMC) sampling. Experiments on the public dataset show that the proposed method outperforms existing self-supervised MRI reconstruction methods and achieves comparable performances with the conventional (fully sampled data trained) score-based diffusion methods.