arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.01351cs.LGcs.AIcs.CV

AETTA:面向测试时自适应的无标签准确率估计

AETTA: Label-Free Accuracy Estimation for Test-Time Adaptation

KAIST(韩国科学技术院);Nokia Bell Labs(诺基亚贝尔实验室)

Taeckyung Lee, Sorn Chottananurak, Taesik Gong, Sung-Ju Lee

AI总结:

本文提出无标签准确率估计算法 AETTA,通过比较目标模型预测与 dropout 推理的预测分歧来估计测试时自适应的准确率,并在多个基线和 TTA 方法上验证其有效性。

摘要:

测试时自适应(TTA)已成为一种可行的解决方案,可利用无标签测试数据使预训练模型适应领域偏移。然而,TTA 面临自适应失败的挑战,因为它在动态场景中依赖于对未知测试样本的盲目自适应。传统的分布外性能估计方法在 TTA 语境下受到不现实假设的限制,例如需要带标签数据或重新训练模型。为解决这一问题,我们提出了 AETTA,一种面向 TTA 的无标签准确率估计算法。我们提出以预测分歧作为准确率估计,通过比较目标模型预测与 dropout 推理结果来计算。随后我们改进了预测分歧,以扩展 AETTA 在自适应失败情况下的适用性。我们使用四个基线和六种 TTA 方法进行了广泛评估,结果表明 AETTA 的估计准确率平均比基线高出 19.8 个百分点。我们进一步通过一个模型恢复案例研究证明了准确率估计的有效性,展示了我们基于准确率估计的模型恢复的实用性。源代码可在 https://github.com/taeckyung/AETTA 获取。

英文摘要:

Test-time adaptation (TTA) has emerged as a viable solution to adapt pre-trained models to domain shifts using unlabeled test data. However, TTA faces challenges of adaptation failures due to its reliance on blind adaptation to unknown test samples in dynamic scenarios. Traditional methods for out-of-distribution performance estimation are limited by unrealistic assumptions in the TTA context, such as requiring labeled data or re-training models. To address this issue, we propose AETTA, a label-free accuracy estimation algorithm for TTA. We propose the prediction disagreement as the accuracy estimate, calculated by comparing the target model prediction with dropout inferences. We then improve the prediction disagreement to extend the applicability of AETTA under adaptation failures. Our extensive evaluation with four baselines and six TTA methods demonstrates that AETTA shows an average of 19.8%p more accurate estimation compared with the baselines. We further demonstrate the effectiveness of accuracy estimation with a model recovery case study, showcasing the practicality of our model recovery based on accuracy estimation. The source code is available at https://github.com/taeckyung/AETTA.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑