Search papers, labs, and topics across Lattice.
This paper introduces U-TTT, a novel U-shaped model designed for Positron Emission Tomography (PET) image denoising that employs Test-Time Training (TTT) layers to adaptively adjust model parameters during inference. By integrating a dual-domain adaptation mechanism with Spatial and Frequency TTT layers, U-TTT effectively addresses spatial structural degradations and suppresses global noise, leading to enhanced performance under distribution shifts. Experimental results show that U-TTT achieves state-of-the-art denoising performance while demonstrating robust generalization across various test conditions, including different dose levels and scanner types.
U-TTT adapts dynamically during inference, achieving state-of-the-art PET image denoising even under challenging distribution shifts.
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.
Tsinghua AI