UniPET:面向多种剂量降低因子的通用PET图像去噪网络
阅读原文· arxiv.org现有基于深度学习的PET图像去噪方法通常假设低剂量图像的剂量降低因子(DRF)固定且已知,实际中DRF变化时性能大幅下降。UniPET将域泛化引入PET图像去噪,通过风格对齐网络(SAN)对齐并恢复不同DRF下的风格,同时提出区域感知学习策略(RALS),区分平坦区域与风格化区域并对后者进行对抗学习,防止过度平滑。实验表明,UniPET在特定DRF下性能与单DRF专用模型相当,在通用PET图像去噪任务上达到定量、感知和临床层面的最先进水平。
Most existing deep learning-based PET image denoising methods assume a fixed and known dose reduction factor (DRF) for low-dose PET images. However, these methods encounter significant performance degradation when the DRF varies beyond the assumed one in practical applications. To address the challenge posed by varied DRFs, several preliminary studies focus on the task of universal PET image denoising, aiming to train a universal model over low-dose data across DRFs. Nonetheless, these vanilla universal models often struggle with misaligned styles present in different DRF data, leading to the style elimination issue with a significant over-smoothing effect. To deal with this issue, we innovatively introduce domain generalization to PET image denoising and propose a universal PET image denoising network (UniPET) to achieve high-quality PET image denoising across diverse DRFs. UniPET comprises two primary innovations: a style alignment network (SAN) and a region-aware learning strategy (RALS). Specifically, SAN utilizes style alignment techniques derived from domain generalization to align and recover styles across different DRFs, ensuring the model's generalizability across various DRFs while effectively preserving styles. Furthermore, to enhance style recovery, RALS distinguishes between flat and stylized regions, exclusively conducting adversarial learning on the latter, thereby more effectively guiding the model's focus towards learning stylized regions. It is demonstrated that our proposed UniPET can adaptively recover different DRF styles and achieve high-quality PET image denoising across DRFs. Comprehensive experiments show that UniPET exhibits comparable performance to individual DRF-specific models at specific DRFs and realizes state-of-the-art performance in universal PET image denoising quantitatively, perceptually, and clinically.