Lune

ICML2026Top-tier venue

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

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

2026Year

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.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 52b8cd56-0408-4631-830c-e865b9d3a32d

Builds on17

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines