U-TTT让PET图像去噪模型在测试时自适应调整,提升临床实用性和泛化能力。
U-TTT: Towards Generalizable PET Image Denoising via Test-Time Training

- 测试时通过自监督动态调整参数,实现对不同扫描条件的实时适应。
- 在未见剂量和未见扫描仪上均达到领先去噪效果,峰值信噪比超基准模型3.2dB。
- 适合需要高鲁棒性的医学影像处理场景,尤其适用于多设备、多剂量环境。
现有的深度学习PET图像去噪模型在分布偏移下性能严重下降,难以在临床中可靠部署。其根本原因在于固定参数模型无法在训练后适应测试数据的变化(如剂量水平或扫描仪类型)。为突破此限制并实现强泛化能力,我们提出U-TTT,一种集成测试时训练(TTT)层的U型网络,通过自监督机制在推理阶段动态调整参数,以适配每个测试样本的特性。此外,为全面捕捉3D PET数据的复杂退化,U-TTT采用双域自适应机制:空间测试时训练(S-TTT)层校正空间结构退化,频率测试时训练(F-TTT)层抑制全局噪声谱并恢复高频细节。大量实验表明,U-TTT在挑战性分布偏移下(包括未见剂量与未见扫描仪)均实现当前最优去噪性能,显著优于现有方法。
原文摘要 · Abstract (English)
Existing deep learning models for Positron Emission Tomography (PET) image denoising often suffer from severe performance degradation under distribution shifts, fundamentally restricting their robust clinical deployment. This lack of generalization stems from the conventional paradigm of fixed-parameter models that cannot adapt to variations in test data (e.g., dose levels or scanner types) after training. To overcome this limitation and achieve robust generalization, we introduce U-TTT, a novel U-shaped model that integrates Test-Time Training (TTT) layers to dynamically adjust model parameters during inference through self-supervision, thereby adapting to the specific characteristics of each test instance. Furthermore, to comprehensively capture the complex degradations of 3D PET data, U-TTT features a dual-domain adaptation mechanism comprising a Spatial Test-Time Training (S-TTT) layer and a Frequency Test-Time Training (F-TTT) layer. The S-TTT layer captures and corrects spatial structural degradations, while the F-TTT layer suppresses global noise spectra and restores delicate high-frequency details. Extensive experiments demonstrate that U-TTT achieves state-of-the-art PET denoising performance and exhibits superior generalization under challenging distribution shifts, including both unseen dose levels and unseen scanners. Our code will be available at https://github.com/Yaziwel/U-TTT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。