arXiv:2411.15204v2 宣告类型: replace-cross 摘要: 测试时适应（TTA）是一种在模型遇到测试时输入分布变化时减轻训练模型性能下降的有效方法。然而，现有的TTA方法在面对额外的类别分布变化时往往会遭受显著的性能下降。我们首先在标签分布变化下分析TTA方法，并识别出在不同协变量变化中常见的类别间的混淆模式。基于这一观察，我们引入了标签分布变化意识的预测精炼方法（DART），这是一种新颖的TTA方法，通过关注类别间的混淆模式来细化预测。DART 在中间时间通过使用训练数据集将多个具有不同类别分布的批次暴露给预测精炼模块进行训练。然后，在测试时使用该模块来检测和纠正类别分布变化，显著提高测试数据的伪标签准确性。在没有标签分布变化的条件下，该方法表现出色，而仅在CIFAR-10C的标签分布变化下实现5-18%的准确度提升。广泛的实验表明，DART 能够纠正由测试时分布变化引起的不准确预测，从而增强现有TTA方法的性能。这一改进使DART成为有价值的插件工具。