PET Denoising and Uncertainty Estimation Based on NVAE Model

Jianan Cui,Yutong Xie,Kuang Gong,Kyungsang Kim,Jaewon Yang,Peder Larson,Thomas Hope,Spencer Behr,Youngho Seo,Huafeng Liu,Quanzheng Li
DOI: https://doi.org/10.1109/nss/mic44867.2021.9875487
2021-01-01
Abstract:The structure of the deep neural network is constantly changing, and its performance is constantly breaking through. Recently, a new network, Nouveau variational auto-encoder (NVAE), has been proposed and gained great attention. In addition to the ability to generate high-quality images, the more important nature of NVAE is that it can generate a distribution that makes it possible to measure the uncertainty. In this work, we proposed to use NVAE for PET image denoising and estimate the uncertainty from both training data and model at the same time. 2.5D training based on 28 patients was performed and quantification based on 7 real patient data showed that NVAE has a good performance for PET denoising, which outperforms the Unet. The variance of 1000 sample output was calculated to show the uncertainty map.
What problem does this paper attempt to address?