用流匹配模型约束PET重建,提升图像质量与效率
Manifold-Constrained PET Reconstruction with Learned Flow-Matching Priors

- 用流匹配学习高质量PET图像的潜在流形,生成解剖合理图像
- 在低剂量、跨扫描仪数据上表现优异,噪声抑制强且结构保留好
- 无需标注数据,适合临床低剂量PET重建与跨设备应用
正电子发射断层成像(PET)图像重建是一个病态的泊松逆问题,常伴随严重噪声放大和伪影。本文提出一种无监督、基于优化的重建框架,采用流匹配生成模型作为学习到的流形先验。通过在高质量PET图像上训练流匹配模型,学习从高斯潜空间到实际PET图像分布的确定性常微分方程映射,得到可微分的解剖合理图像生成器。将该生成器作为显式流形约束,嵌入正则化泊松似然模型中。利用交替方向乘子法求解,其中期望最大化型代理更新保证数据一致性,梯度驱动的潜空间投影确保流形接近性。在模拟与真实PET数据集上评估了剂量鲁棒性、病灶插入泛化性和跨扫描仪迁移能力。相比传统方法与先进深度学习基线,本方法在噪声抑制、结构保持和定量准确性方面均更优,同时保持高计算效率。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。