Populational and individual information based PET image denoising using conditional unsupervised learning

Abstract

Our study aims to improve the signal-to-noise ratio of positron emission tomography (PET) imaging using conditional unsupervised learning. The proposed method does not require low- and high-quality pairs for network training which can be easily applied to existing PET/computed tomography (CT) and PET/magnetic resonance (MR) datasets. This method consists of two steps: populational training and individual fine-tuning. As for populational training, a network was first pre-trained by a group of patients’ noisy PET images and the corresponding anatomical prior images from CT or MR. As for individual fine-tuning, a new network with initial parameters inherited from the pre-trained network was fine-tuned by the test patient's noisy PET image and the corresponding anatomical prior image. Only the last few layers were fine-tuned to take advantage of the populational information and the pre-training efforts. Both networks shared the same structure and took the CT or MR images as the network input so that the network output was conditioned on the patient’s anatomic prior information. The noisy PET images were used as the training and fine-tuning labels. The proposed method was evaluated on a 68Ga-PPRGD2 PET/CT dataset and a 18F-FDG PET/MR dataset. For the PET/CT dataset, with the original noisy PET image as the baseline, the proposed method has a significantly higher contrast-to noise ratio (CNR) improvement (71.85% ± 27.05%) than Gaussian (12.66% ± 6.19%, P = 0.002), nonlocal mean method (22.60% ± 13.11%, P = 0.002) and conditional deep image prior method (52.94% ± 21.79%, P = 0.0039). For the PET/MR dataset, compared to Gaussian (18.73% ± 9.98%, P < 0.0001), NLM (26.01% ± 19.40%, P < 0.0001) and CDIP (47.48% ± 25.36%, P < 0.0001), the CNR improvement ratio of the proposed method (58.07% ± 28.45%) is the highest. In addition, the denoised images using both datasets also showed that the proposed method can accurately restore tumor structures while also smoothing out the noise.