Posterior Estimation for Dynamic PET imaging using Conditional Variational Inference

Xiaofeng Liu,Thibault Marin,Tiss Amal,Jonghye Woo,Georges El Fakhri,Jinsong Ouyang
2023-10-24
Abstract:This work aims efficiently estimating the posterior distribution of kinetic parameters for dynamic positron emission tomography (PET) imaging given a measurement of time of activity curve. Considering the inherent information loss from parametric imaging to measurement space with the forward kinetic model, the inverse mapping is ambiguous. The conventional (but expensive) solution can be the Markov Chain Monte Carlo (MCMC) sampling, which is known to produce unbiased asymptotical estimation. We propose a deep-learning-based framework for efficient posterior estimation. Specifically, we counteract the information loss in the forward process by introducing latent variables. Then, we use a conditional variational autoencoder (CVAE) and optimize its evidence lower bound. The well-trained decoder is able to infer the posterior with a given measurement and the sampled latent variables following a simple multivariate Gaussian distribution. We validate our CVAE-based method using unbiased MCMC as the reference for low-dimensional data (a single brain region) with the simplified reference tissue model.
Medical Physics,Artificial Intelligence,Signal Processing
What problem does this paper attempt to address?
The paper aims to address the problem of posterior estimation in dynamic positron emission tomography (PET) imaging. Specifically, the goal of the study is to efficiently estimate the posterior distribution of kinetic parameters given the time-activity curve (TAC) measurements of a target brain region. Due to information loss in the process from parameter imaging to measurement space, the inverse mapping is uncertain. Traditional solutions use Markov Chain Monte Carlo (MCMC) sampling methods, which, although capable of producing unbiased asymptotic estimates, are computationally expensive, especially for high-dimensional data. To address this issue, the authors propose a deep learning-based approach to efficiently estimate the posterior distribution. The specific method involves introducing latent variables to compensate for information loss in the forward process and using a Conditional Variational Autoencoder (CV AE) to optimize its Evidence Lower Bound (ELBO). The trained decoder can infer the posterior distribution based on given measurements and sampled latent variables (following a simple multivariate Gaussian distribution). To validate the effectiveness of the proposed CV AE method, the authors compared it with the MCMC method, particularly in the application of a simplified reference tissue model (SRTM) on low-dimensional data (a single brain region). Experimental results show that the CV AE method is consistent with the MCMC method in terms of posterior estimation and is more computationally efficient.