Blocking the Leakage: Manifold-Aware Gradient Projection for Long-Horizon Test-Time Adaptation

Haoyu Xiong, Chengchao Wang, ZhongQiang Wang, He Huang, Qiuxia Yang, Zhengpeng Zhao, Yuanyuan Pu

Abstract: Test-Time Adaptation (TTA) empowers pre-trained models to adapt online to distribution shifts during inference, but such online updates often become unstable in long-horizon deployments. Prevailing approaches attribute this failure to error accumulation from noisy pseudo-labels, relying on heuristics to gate samples used for updates. We argue that this statistical view is insufficient: the problem lies not only in sample quality but also in the directionality of gradients. In this work, we identify a geometric failure mode termed manifold erosion. Through spectral analysis, we find that reliable gradients concentrate in a stable low-rank subspace, while gradients from confident mispredictions are high-rank yet exhibit a persistent directional leakage into this protected subspace. This leakage can accumulate coherently and gradually erode core representations, eventually leading to collapse. To address this, we propose Manifold-Aware Gradient Projection (MGP), a geometric intervention that tracks the dominant subspace online and projects gradients onto its orthogonal complement. By blocking the leakage path, MGP decouples stability from plasticity. Extensive experiments on diverse TTA benchmarks demonstrate its long-horizon stability, whereas prior methods often fail.

Lay Summary: AI models often face changing real-world conditions, such as noise, blur, weather, or new environments. Test-time adaptation helps models adjust during use without labels, but long-term updates can gradually damage important knowledge and cause performance collapse. This paper shows that the issue is not only wrong predictions, but harmful update directions that repeatedly affect sensitive parts of the model. We propose MGP, which blocks these risky updates while still allowing safe adaptation. Experiments show that MGP keeps models stable and robust over long periods under challenging distribution shifts.

Primary Area: Applications->Computer Vision

Keywords: test-time adaptation

Submission Number: 12374