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
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 52b8cd56-0408-4631-830c-e865b9d3a32dBuilds on17
- The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution GeneralizationDan Hendrycks, Steven Basart, Norman Mu, Saurav Kadavath et al.ICCV 2021 · 2,294 citations
- Tent: Fully Test-Time Adaptation by Entropy MinimizationDequan Wang, Evan Shelhamer, Shaoteng Liu, Bruno A. Olshausen et al.ICLR 2021 · 1,731 citations
- Test-Time Training with Self-Supervision for Generalization under Distribution ShiftsYu Sun, Xiaolong Wang, Zhuang Liu, John Miller et al.ICML 2020 · 1,220 citations
- Efficient Test-Time Model Adaptation without ForgettingShuaicheng Niu, Jiaxiang Wu, Yifan Zhang, Yaofo Chen et al.ICML 2022 · 579 citations
- Continual Test-Time Domain AdaptationQin Wang, Olga Fink, Luc Van Gool, Dengxin DaiCVPR 2022 · 383 citations
Related papers
- Geometry-aware Test-Time Adaptation on GraphsLingwei Wei, Dou Hu, Li Sun, Chengze Li et al.KDD 2026
- Lifelong Test-Time Adaptation via Online Learning in Tracked Low-Dimensional SubspaceDexin Duan, Rui Xu, Peilin Liu, Fei WenNeurIPS 2025 · 7 citations
- Neural Collapse in Test-Time AdaptationXiao Chen, Zhongjing Du, Jiazhen Huang, Jiang Xu et al.CVPR 2026 · 2 citations
- Towards Stable Test-time Adaptation in Dynamic Wild WorldShuaicheng Niu, Jiaxiang Wu, Yifan Zhang, Zhiquan Wen et al.ICLR 2023 · 62 citations
- Test-Time Ensemble via Linear Mode Connectivity: A Path to Better AdaptationByungjai Kim, Chanho Ahn, Wissam J. Baddar, Kikyung Kim et al.ICLR 2025
