通过多假设框架提升测试时自适应的稳定性与鲁棒性
Multi-Hypothesis Test-Time Adaptation to Mitigate Underspecification

- 引入粒子化多轨迹探索,避免单一参数更新导致的歧义
- 在混合分布偏移下提升3-4%,单批次下提升2-3%
- 兼容现有方法,适合追求高可靠性的实际部署场景
测试时自适应(TTA)通过利用无标签目标数据调整模型参数来增强模型在分布偏移下的鲁棒性。然而,在缺乏监督的情况下,基于熵最小化的适应方法本质上是欠约束的:多个不同的参数更新可达到相似低熵,但导致截然不同的决策边界。这种现象称为欠定性,使标准TTA脆弱且易陷入虚假模式。本文从后验角度重新审视TTA,由熵最小化诱导出参数上的伪似然。不采用单一点估计,而是提出一种基于粒子的多样化框架,同时探索多个合理的适应轨迹。该方法通过输出、参数、优化器和输入四个层面的多层次多样化实现,可作为即插即用的封装兼容现有TTA方法。在挑战性基准上广泛实验表明,该方法在稳定性与鲁棒性上均有显著提升:在混合分布偏移下提升3-4%,单批次下提升2-3%,标签偏移下提升1-2.5%,优于当前最优基线。结果表明,将TTA视为多假设推断问题而非单点优化任务,是缓解欠定性、实现可靠实际部署的关键。
原文摘要 · Abstract (English)
Test-Time Adaptation (TTA) seeks to improve model robustness under distribution shifts by adapting parameters using unlabeled target data. However, in the absence of supervision, entropy-based adaptation is fundamentally underconstrained: multiple distinct parameter updates can achieve similarly low entropy while inducing drastically different decision boundaries. This phenomenon, known as underspecification, renders standard TTA brittle and prone to collapse into spurious modes. In this work, we reinterpret TTA through a posterior-inspired lens induced by entropy minimization, where low-entropy solutions define a pseudo-likelihood over parameters. Instead of committing to a single point estimate, we introduce a particle-based diversification framework that explores multiple plausible adaptation trajectories simultaneously. Our method can be viewed as a structured exploration of multiple plausible adaptation solutions, implemented through multi-level diversification at the output, parameter, optimizer, and input levels. Crucially, the framework acts as a plug-and-play wrapper compatible with existing TTA methods. Extensive experiments on challenging benchmarks demonstrate consistent gains in stability and robustness, achieving improvements of 3-4% under mixed shifts, 2-3% with batch size one, and 1-2.5% under label shifts, outperforming state-of-the-art baselines. Our results suggest that treating TTA as a multi-hypothesis inference problem, rather than a single-point optimization task, is key to mitigating underspecification and enabling reliable real-world deployment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。