Sequential Diffusion-Guided Deep Image Prior For Medical Image Reconstruction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liang, Shijun, Alkhouri, Ismail, Qu, Qing, Wang, Rongrong, Ravishankar, Saiprasad
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929643104239616
author Liang, Shijun
Alkhouri, Ismail
Qu, Qing
Wang, Rongrong
Ravishankar, Saiprasad
author_facet Liang, Shijun
Alkhouri, Ismail
Qu, Qing
Wang, Rongrong
Ravishankar, Saiprasad
contents Deep learning (DL) methods have been extensively applied to various image recovery problems, including magnetic resonance imaging (MRI) and computed tomography (CT) reconstruction. Beyond supervised models, other approaches have been recently explored including two key recent schemes: Deep Image Prior (DIP) that is an unsupervised scan-adaptive method that leverages the network architecture as implicit regularization but can suffer from noise overfitting, and diffusion models (DMs), where the sampling procedure of a pre-trained generative model is modified to allow sampling from the measurement-conditioned distribution through approximations. In this paper, we propose combining DIP and DMs for MRI and CT reconstruction, motivated by (i) the impact of the DIP network input and (ii) the use of DMs as diffusion purifiers (DPs). Specifically, we propose a sequential procedure that iteratively optimizes the DIP network with a DM-refined adaptive input using a loss with data consistency and autoencoding terms. We term the approach Sequential Diffusion-Guided DIP (uDiG-DIP). Our experimental results demonstrate that uDiG-DIP achieves superior reconstruction results compared to leading DM-based baselines and the original DIP for MRI and CT tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2410_04482
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Sequential Diffusion-Guided Deep Image Prior For Medical Image Reconstruction
Liang, Shijun
Alkhouri, Ismail
Qu, Qing
Wang, Rongrong
Ravishankar, Saiprasad
Image and Video Processing
Deep learning (DL) methods have been extensively applied to various image recovery problems, including magnetic resonance imaging (MRI) and computed tomography (CT) reconstruction. Beyond supervised models, other approaches have been recently explored including two key recent schemes: Deep Image Prior (DIP) that is an unsupervised scan-adaptive method that leverages the network architecture as implicit regularization but can suffer from noise overfitting, and diffusion models (DMs), where the sampling procedure of a pre-trained generative model is modified to allow sampling from the measurement-conditioned distribution through approximations. In this paper, we propose combining DIP and DMs for MRI and CT reconstruction, motivated by (i) the impact of the DIP network input and (ii) the use of DMs as diffusion purifiers (DPs). Specifically, we propose a sequential procedure that iteratively optimizes the DIP network with a DM-refined adaptive input using a loss with data consistency and autoencoding terms. We term the approach Sequential Diffusion-Guided DIP (uDiG-DIP). Our experimental results demonstrate that uDiG-DIP achieves superior reconstruction results compared to leading DM-based baselines and the original DIP for MRI and CT tasks.
title Sequential Diffusion-Guided Deep Image Prior For Medical Image Reconstruction
topic Image and Video Processing
url https://arxiv.org/abs/2410.04482