Manifold-Constrained PET Reconstruction with Learned Flow-Matching Priors

Abstract

Image reconstruction for positron emission tomography (PET) is an ill-posed Poisson inverse problem that often suffers from severe noise amplification and artifacts. In this work, we introduce an unsupervised, optimization-based reconstruction framework that employs a flow-matching generative model as a learned manifold prior. We train the flow-matching model on high-quality PET images to learn a deterministic ordinary differential equation transport from a Gaussian latent distribution to the empirical PET image distribution, yielding a differentiable generator of anatomically plausible images. We incorporate this generator as an explicit manifold constraint into a regularized Poisson likelihood formulation. We solve the resulting optimization problem using an alternating direction method of multipliers algorithm, in which an expectation-maximization-type surrogate update enforces data consistency and a gradient-based latent-space projection enforces manifold proximity. We evaluate the proposed method on both simulated and real PET datasets, assessing dose-level robustness, lesion-insertion generalization, and cross-scanner transfer. Compared with conventional reconstruction methods and state-of-the-art deep learning baselines, the proposed method provides superior noise suppression, structural preservation, and quantitative accuracy, while maintaining high computational efficiency.