摘要
测试时自适应 (TTA) 是一种有效的策略,用于减轻训练模型在测试时遇到输入分布偏移时出现的性能下降问题。然而,现有的 TTA 方法在面临额外类别分布偏移时,往往会遭受严重的性能下降。我们首先分析了标签分布偏移下的 TTA 方法,并确定了在不同协变量偏移中普遍存在的类别混淆模式。基于这一观察,我们引入了一种新的 TTA 方法——用于测试时自适应的标签分布偏移感知预测细化 (DART),该方法通过关注类别混淆模式来细化预测结果。DART 在中间阶段通过使用训练数据集将其暴露于具有不同类别分布的多个批次来训练预测细化模块。然后,该模块在测试时用于检测和纠正类别分布偏移,显著提高测试数据的伪标签精度。我们的方法在 CIFAR-10C 上的标签分布偏移情况下,精度提高了 5% 到 18%,并且在没有标签分布偏移的情况下,性能不会下降。在 CIFAR、PACS、OfficeHome 和 ImageNet 基准测试上的大量实验表明,DART 能够纠正由测试时分布偏移引起的不准确预测。这种改进导致现有 TTA 方法的性能增强,使 DART 成为一个有价值的插件工具。