Optimizing Sampling Patterns for Compressed Sensing MRI with Diffusion Generative Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ravula, Sriram, Levac, Brett, Arefeen, Yamin, Jalal, Ajil, Dimakis, Alexandros G., Tamir, Jonathan I.
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912898165506048
author Ravula, Sriram
Levac, Brett
Arefeen, Yamin
Jalal, Ajil
Dimakis, Alexandros G.
Tamir, Jonathan I.
author_facet Ravula, Sriram
Levac, Brett
Arefeen, Yamin
Jalal, Ajil
Dimakis, Alexandros G.
Tamir, Jonathan I.
contents Magnetic resonance imaging (MRI) is a powerful medical imaging modality, but long acquisition times limit throughput, patient comfort, and clinical accessibility. Diffusion-based generative models serve as strong image priors for reducing scan-time with accelerated MRI reconstruction and offer robustness across variations in the acquisition model. However, most existing diffusion-based approaches do not exploit the unique ability in MRI to jointly design both the sampling pattern and the reconstruction method. While prior learning-based approaches have optimized sampling patterns for end-to-end unrolled networks, analogous methods for diffusion-based reconstruction have not been established due to the computational burden of posterior sampling. In this work, we propose a method to optimize k-space sampling patterns for accelerated multi-coil MRI reconstruction using diffusion models as priors. We introduce a training objective based on a single-step posterior mean estimate that avoids backpropagation through an expensive iterative reconstruction process. Then we present a greedy strategy for learning Cartesian sampling patterns that selects informative k-space locations using gradient information from a pre-trained diffusion model while enforcing spatial diversity among samples. Experimental results across multiple anatomies and acceleration factors demonstrate that diffusion models using the optimized sampling patterns achieve higher-quality reconstructions in comparison to using fixed and learned baseline patterns.
format Preprint
id arxiv_https___arxiv_org_abs_2306_03284
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Optimizing Sampling Patterns for Compressed Sensing MRI with Diffusion Generative Models
Ravula, Sriram
Levac, Brett
Arefeen, Yamin
Jalal, Ajil
Dimakis, Alexandros G.
Tamir, Jonathan I.
Machine Learning
Image and Video Processing
Magnetic resonance imaging (MRI) is a powerful medical imaging modality, but long acquisition times limit throughput, patient comfort, and clinical accessibility. Diffusion-based generative models serve as strong image priors for reducing scan-time with accelerated MRI reconstruction and offer robustness across variations in the acquisition model. However, most existing diffusion-based approaches do not exploit the unique ability in MRI to jointly design both the sampling pattern and the reconstruction method. While prior learning-based approaches have optimized sampling patterns for end-to-end unrolled networks, analogous methods for diffusion-based reconstruction have not been established due to the computational burden of posterior sampling. In this work, we propose a method to optimize k-space sampling patterns for accelerated multi-coil MRI reconstruction using diffusion models as priors. We introduce a training objective based on a single-step posterior mean estimate that avoids backpropagation through an expensive iterative reconstruction process. Then we present a greedy strategy for learning Cartesian sampling patterns that selects informative k-space locations using gradient information from a pre-trained diffusion model while enforcing spatial diversity among samples. Experimental results across multiple anatomies and acceleration factors demonstrate that diffusion models using the optimized sampling patterns achieve higher-quality reconstructions in comparison to using fixed and learned baseline patterns.
title Optimizing Sampling Patterns for Compressed Sensing MRI with Diffusion Generative Models
topic Machine Learning
Image and Video Processing
url https://arxiv.org/abs/2306.03284