Scan-wise generalized PET denoising with contrastive adversarial learning

ObjectiveDeep learning (DL) has significantly advanced low-count PET denoising. However, models trained on specific distributions often yield biased outputs when applied to scans with different activity distributions caused by anatomical and physiological variations (distribution shifts). Existing methods fail to generalize well across these scan-wise variations. Our goal is to formulate PET denoising as a scan-wise Domain Generalization (DG) problem to mitigate these variations and achieve robust, unbiased denoising for unseen scans.ApproachWe propose a contrastive adversarial domain generalization framework to learn scaninvariant features efficiently. We leverage the property that multiple low-count noise realizations can be generated from a single raw list-mode PET scan to form a scan-wise domain distribution. Unlike conventional adversarial training with cross-entropy loss, we propose a contrastive adversarial framework that minimizes the mutual information between feature and scan-wise domains. Considering that each subject can have multiple scans in longitudinal studies, we further propose an ordered and memoryqueued contrastive adversarial framework. This method efficiently includes realizations from the same scan as positive pairs, different scans of the same subject as pseudopositive pairs, and different subjects as negative pairs in a memory batch. We explicitly exploit their ordered relationship as prior knowledge using a novel noisy-robust multipositive ordinal contrastive loss.Main resultsWe systematically validated the effectiveness of our methods using 1,920 noise realizations derived from 80 subjects with 192 longitudinal scans of MK-6240 tau PET data.The proposed contrastive adversarial domain generalization approach demonstrated superior denoising performance compared to cross-entropybased adversarial methods and standard baselines. The ordered contrastive loss successfully improved the peak signal-to-noise ratio and structural similarity index consistently while reducing bias and standard deviation in Alzheimer-related regions and the whole brain. Significance To our knowledge, this is the first attempt to alleviate the performance degradation in cross-scan denoising from the perspective of domain generalization. Our study is also a pioneering work in utilizing longitudinal scans as pseudo-positives within an ordered contrastive learning scheme to exploit fine-grained relationships for robust clinical PET imaging applications.