ContextMRI: Enhancing Compressed Sensing MRI through Metadata Conditioning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chung, Hyungjin, Lee, Dohun, Wu, Zihui, Kim, Byung-Hoon, Bouman, Katherine L., Ye, Jong Chul
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916558099447808
author Chung, Hyungjin
Lee, Dohun
Wu, Zihui
Kim, Byung-Hoon
Bouman, Katherine L.
Ye, Jong Chul
author_facet Chung, Hyungjin
Lee, Dohun
Wu, Zihui
Kim, Byung-Hoon
Bouman, Katherine L.
Ye, Jong Chul
contents Compressed sensing MRI seeks to accelerate MRI acquisition processes by sampling fewer k-space measurements and then reconstructing the missing data algorithmically. The success of these approaches often relies on strong priors or learned statistical models. While recent diffusion model-based priors have shown great potential, previous methods typically ignore clinically available metadata (e.g. patient demographics, imaging parameters, slice-specific information). In practice, metadata contains meaningful cues about the anatomy and acquisition protocol, suggesting it could further constrain the reconstruction problem. In this work, we propose ContextMRI, a text-conditioned diffusion model for MRI that integrates granular metadata into the reconstruction process. We train a pixel-space diffusion model directly on minimally processed, complex-valued MRI images. During inference, metadata is converted into a structured text prompt and fed to the model via CLIP text embeddings. By conditioning the prior on metadata, we unlock more accurate reconstructions and show consistent gains across multiple datasets, acceleration factors, and undersampling patterns. Our experiments demonstrate that increasing the fidelity of metadata, ranging from slice location and contrast to patient age, sex, and pathology, systematically boosts reconstruction performance. This work highlights the untapped potential of leveraging clinical context for inverse problems and opens a new direction for metadata-driven MRI reconstruction.
format Preprint
id arxiv_https___arxiv_org_abs_2501_04284
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ContextMRI: Enhancing Compressed Sensing MRI through Metadata Conditioning
Chung, Hyungjin
Lee, Dohun
Wu, Zihui
Kim, Byung-Hoon
Bouman, Katherine L.
Ye, Jong Chul
Computer Vision and Pattern Recognition
Machine Learning
Compressed sensing MRI seeks to accelerate MRI acquisition processes by sampling fewer k-space measurements and then reconstructing the missing data algorithmically. The success of these approaches often relies on strong priors or learned statistical models. While recent diffusion model-based priors have shown great potential, previous methods typically ignore clinically available metadata (e.g. patient demographics, imaging parameters, slice-specific information). In practice, metadata contains meaningful cues about the anatomy and acquisition protocol, suggesting it could further constrain the reconstruction problem. In this work, we propose ContextMRI, a text-conditioned diffusion model for MRI that integrates granular metadata into the reconstruction process. We train a pixel-space diffusion model directly on minimally processed, complex-valued MRI images. During inference, metadata is converted into a structured text prompt and fed to the model via CLIP text embeddings. By conditioning the prior on metadata, we unlock more accurate reconstructions and show consistent gains across multiple datasets, acceleration factors, and undersampling patterns. Our experiments demonstrate that increasing the fidelity of metadata, ranging from slice location and contrast to patient age, sex, and pathology, systematically boosts reconstruction performance. This work highlights the untapped potential of leveraging clinical context for inverse problems and opens a new direction for metadata-driven MRI reconstruction.
title ContextMRI: Enhancing Compressed Sensing MRI through Metadata Conditioning
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2501.04284