Noise suppression in photon-counting computed tomography using unsupervised Poisson flow generative models

Deep learning (DL) has proven to be important for computed tomography (CT) image denoising. However, such models are usually trained under supervision, requiring paired data that may be difficult to obtain in practice. Diffusion models offer unsupervised means of solving a wide range of inverse problems via posterior sampling. In particular, using the estimated unconditional score function of the prior distribution, obtained via unsupervised learning, one can sample from the desired posterior via hijacking and regularization. However, due to the iterative solvers used, the number of function evaluations (NFE) required may be orders of magnitudes larger than for single-step samplers. In this paper, we present a novel image denoising technique for photon-counting CT by extending the unsupervised approach to inverse problem solving to the case of Poisson flow generative models (PFGM)++. By hijacking and regularizing the sampling process we obtain a single-step sampler, that is NFE = 1. Our proposed method incorporates posterior sampling using diffusion models as a special case. We demonstrate that the added robustness afforded by the PFGM++ framework yields significant performance gains. Our results indicate competitive performance compared to popular supervised, including state-of-the-art diffusion-style models with NFE = 1 (consistency models), unsupervised, and non-DL-based image denoising techniques, on clinical low-dose CT data and clinical images from a prototype photon-counting CT system developed by GE HealthCare.

Authors

Institutions

Publication Details

Journal
Visual Computing for Industry Biomedicine and Art
Published
2024-09-23
DOI
https://doi.org/10.1186/s42492-024-00175-6
Citations
5
Primary Topic
Medical Imaging Techniques and Applications
Type
article
Field-Weighted Citation Impact
2.02

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Noise suppression in photon-counting computed tomography using unsupervised Poisson flow generative models

Staffan Holmin, Timothy P. Szczykutowicz, Mats Danielsson, Dennis Hein et al.
5 citations
Visual Computing for Industry Biomedicine and Art
Medical Imaging Techniques and Applications
2.02
article

Noise suppression in photon-counting computed tomography using unsupervised Poisson flow generative models

Staffan Holmin, Timothy P. Szczykutowicz, Mats Danielsson, Dennis Hein, Jonathan S. Maltz, Ge Wang, Mats Persson
article en
5 citations

Abstract

Deep learning (DL) has proven to be important for computed tomography (CT) image denoising. However, such models are usually trained under supervision, requiring paired data that may be difficult to obtain in practice. Diffusion models offer unsupervised means of solving a wide range of inverse problems via posterior sampling. In particular, using the estimated unconditional score function of the prior distribution, obtained via unsupervised learning, one can sample from the desired posterior via hijacking and regularization. However, due to the iterative solvers used, the number of function evaluations (NFE) required may be orders of magnitudes larger than for single-step samplers. In this paper, we present a novel image denoising technique for photon-counting CT by extending the unsupervised approach to inverse problem solving to the case of Poisson flow generative models (PFGM)++. By hijacking and regularizing the sampling process we obtain a single-step sampler, that is NFE = 1. Our proposed method incorporates posterior sampling using diffusion models as a special case. We demonstrate that the added robustness afforded by the PFGM++ framework yields significant performance gains. Our results indicate competitive performance compared to popular supervised, including state-of-the-art diffusion-style models with NFE = 1 (consistency models), unsupervised, and non-DL-based image denoising techniques, on clinical low-dose CT data and clinical images from a prototype photon-counting CT system developed by GE HealthCare.

Visual Computing for Industry Biomedicine and ArtVol. 7(1)
Karolinska University Hospital (SE), University of Wisconsin–Madison (US), Rensselaer Polytechnic Institute (US), General Electric (Spain) (ES), Karolinska Institutet (SE), Applied Sciences Laboratory (United States) (US), KTH Royal Institute of Technology (SE)
Göran Gustafssons Stiftelser, Vetenskapsrådet, GE Healthcare
Openalex Percentile: Top 13%
Medical Imaging Techniques and Applications
2.02
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.