Skip to yearly menu bar Skip to main content


Poster
in
Workshop: 2nd Workshop on Test-Time Adaptation: Putting Updates to the Test (PUT)

SteeringTTA: Guiding Diffusion Trajectories for Robust Test-Time-Adaptation

Jihyun Yu · Yoojin Oh · Won Bae · Mingyu Kim · Junhyug Noh

[ ] [ Project Page ]
Fri 18 Jul 2:30 p.m. PDT — 3:15 p.m. PDT

Abstract: Test-time adaptation (TTA) aims to correct performance degradation of deep models under distribution shifts by updating models or inputs using unlabeled test data. Input-only diffusion-based TTA methods improve robustness to corruptions but rely on gradient guidance, limiting exploration and generalization across distortion types. We propose SteeringTTA, an inference-only framework that adapted Feynman-Kac steering to guide diffusion-based input adaptation with pseudo-label driven reward. SteeringTTA maintains multiple particle trajectories, steered by a combination of cumulative top-$K$ probabilities and an entropy schedule, to balance exploration and confidence. On ImageNet‐C, SteeringTTA consistently outperforms the baseline without any model updates or source data.

Chat is not available.