Fill the K-Space and Refine the Image: Prompting for Dynamic and Multi-Contrast MRI Reconstruction

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The key to dynamic or multi-contrast magnetic resonance imaging (MRI) reconstruction lies in exploring inter-frame or inter-contrast information. Currently, the unrolled model, an approach combining iterative MRI reconstruction steps with learnable neural network layers, stands as the best-performing method for MRI reconstruction. However, there are two main limitations to overcome: firstly, the unrolled model structure and GPU memory constraints restrict the capacity of each denoising block in the network, impeding the effective extraction of detailed features for reconstruction; secondly, the existing model lacks the flexibility to adapt to variations in the input, such as different contrasts, resolutions or views, necessitating the training of separate models for each input type, which is inefficient and may lead to insufficient reconstruction. In this paper, we propose a two-stage MRI reconstruction pipeline to address these limitations. The first stage involves filling the missing k-space data, which we approach as a physics-based reconstruction problem. We first propose a simple yet efficient baseline model, which utilizes adjacent frames/contrasts and channel attention to capture the inherent inter-frame/-contrast correlation. Then, we extend the baseline model to a prompt-based learning approach, PromptMR, for all-in-one MRI reconstruction from different views, contrasts, adjacent types, and acceleration factors. The second stage is to refine the reconstruction from the first stage, which we treat as a general video restoration problem to further fuse features from neighboring frames/contrasts in the image domain. Extensive experiments show that our proposed method significantly outperforms previous state-of-the-art accelerated MRI reconstruction methods.

Original languageAmerican English
Title of host publicationStatistical Atlases and Computational Models of the Heart. Regular and CMRxRecon Challenge Papers - 14th International Workshop, STACOM 2023, Held in Conjunction with MICCAI 2023, Revised Selected Papers
EditorsOscar Camara, Esther Puyol-Antón, Avan Suinesiaputra, Alistair Young, Maxime Sermesant, Qian Tao, Chengyan Wang
PublisherSpringer Science and Business Media Deutschland GmbH
Pages261-273
Number of pages13
ISBN (Print)9783031524479
DOIs
StatePublished - 2024
Event14th International Workshop on Statistical Atlases and Computational Models of the Heart, STACOM 2023 held in Conjunction with MICCAI 2023 - Vancouver, Canada
Duration: Oct 12 2023Oct 12 2023

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume14507 LNCS

Conference

Conference14th International Workshop on Statistical Atlases and Computational Models of the Heart, STACOM 2023 held in Conjunction with MICCAI 2023
Country/TerritoryCanada
CityVancouver
Period10/12/2310/12/23

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Keywords

  • Dynamic
  • MRI reconstruction
  • Multi-contrast
  • Prompt-based learning
  • Two-stage approach

Fingerprint

Dive into the research topics of 'Fill the K-Space and Refine the Image: Prompting for Dynamic and Multi-Contrast MRI Reconstruction'. Together they form a unique fingerprint.

Cite this