提出多样性感知的记忆策略,提升测试时自适应的鲁棒性
GoTTA be Diverse: Rethinking Memory Policies for Test-Time Adaptation

- 设计基于类别均衡与特征空间多样性的记忆机制
- 在有限内存下,非独立同分布流中性能显著优于传统方法
- 适用于资源受限场景,可无缝集成现有自适应算法
测试时自适应(TTA)使预训练模型能在分布偏移下在线适应未标注测试流。现有研究多关注自适应目标,而忽略了记忆机制对实际测试流的关键影响。本文构建了去耦合记忆与自适应算法的系统性基准,统一评估不同记忆策略在独立同分布、非独立同分布、持续学习及真实测试流中的表现。结果表明,仅保留近期或类别均衡样本不足以应对复杂流;类内多样性是避免冗余缓冲、维持代表性适应信号的核心。为此,提出导向观测测试时自适应(GOTTA),通过结合类别均衡分配与特征空间多样性实现多样化记忆。该机制可作为即插即用组件,适配多种自适应目标。在图像噪声与视频流任务中,尤其在低内存和强非独立同分布条件下,多样性记忆表现最优,随容量增加仍具竞争力。这凸显了记忆管理在鲁棒性测试时自适应中的核心地位,并确立多样性为实用化自适应的关键原则。
原文摘要 · Abstract (English)
Test-time adaptation (TTA) enables a pre-trained model to adapt online to an unlabeled test stream under distribution shift. While most TTA research focuses on the adaptation objective, practical streams also depend critically on the memory used to select which test samples drive adaptation. Existing memory mechanisms are usually evaluated as components of specific TTA algorithms, making it difficult to isolate which memory design choices matter and when they matter. In this work, we provide a systematic benchmark that decouples memory from the adaptation algorithm and evaluates memory policies under unified conditions across i.i.d., non-i.i.d., continual, and practical test streams. Our study shows that effective memory management requires more than retaining recent or class-balanced samples. In particular, intra-class diversity is a key factor for avoiding redundant buffers and maintaining representative adaptation signals under temporally correlated and label-skewed streams. Motivated by this finding, we introduce Guided Observational Test-Time Adaptation (GOTTA), a family of diversity-aware memory policies that combine class-balanced allocation with feature-space diversity. GOTTA memories act as drop-in replacements for existing buffers and can be paired with different TTA objectives. Across corruption benchmarks and video-stream settings, diversity-aware memory improves adaptation most clearly under constrained memory budgets and challenging non-i.i.d. streams, while remaining competitive as memory capacity increases. These results highlight memory management as a first-class component of robust test-time adaptation and identify diversity as a central principle for practical TTA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。