Evgenia RusakSteffen SchneiderGeorge PachitariuLuisa EckPeter GehlerOliver BringmannWieland BrendelMatthias Bethge

摘要
我们证明,诸如熵最小化和伪标签(pseudo-labeling)等自学习技术在应对系统性领域偏移时,能够简单而有效地提升已部署计算机视觉模型的性能。我们开展了大规模的广泛实验,结果表明,无论模型架构、预训练方法或分布偏移类型如何,这些方法均能带来一致的性能提升。与此同时,自学习技术在实际应用中也极为简便:它无需访问原始训练数据或训练方案,对超参数选择具有较强的鲁棒性,实现方式直接明了,且仅需少量适应轮次(adaptation epochs)即可完成。这使得自学习技术对所有在真实世界中应用机器学习算法的实践者而言都极具吸引力。我们在多个基准数据集上取得了当前最先进的自适应性能:CIFAR10-C 上误差为 8.5%,ImageNet-C 上 mCE 为 22.0%,ImageNet-R 上误差为 17.4%,ImageNet-A 上误差为 14.8%。此外,我们还对自监督适应方法的动力学特性进行了理论分析,并提出了一种新的分类数据集——ImageNet-D,该数据集即使在采用自适应方法的情况下依然极具挑战性。
代码仓库
bethgelab/robustness
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| unsupervised-domain-adaptation-on-imagenet-a | EfficientNet-L2 NoisyStudent + RPL | Top 1 Error: 14.8 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + DeepAug + Augmix + RPL | mean Corruption Error (mCE): 34.8 |
| unsupervised-domain-adaptation-on-imagenet-c | EfficientNet-L2+ENT | mean Corruption Error (mCE): 23.0 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + IG-3.5B + RPL | mean Corruption Error (mCE): 40.9 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNet50 + RPL | mean Corruption Error (mCE): 50.5 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + RPL | mean Corruption Error (mCE): 43.2 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNet50 + ENT | mean Corruption Error (mCE): 51.6 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + ENT | mean Corruption Error (mCE): 44.3 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + DeepAug + Augmix + ENT | mean Corruption Error (mCE): 35.5 |
| unsupervised-domain-adaptation-on-imagenet-c | ResNeXt101 32x8d + IG-3.5B + ENT | mean Corruption Error (mCE): 40.8 |
| unsupervised-domain-adaptation-on-imagenet-c | EfficientNet-L2+RPL | mean Corruption Error (mCE): 22.0 |
| unsupervised-domain-adaptation-on-imagenet-r | ResNet50 + RPL | Top 1 Error: 54.1 |
| unsupervised-domain-adaptation-on-imagenet-r | EfficientNet-L2 Noisy Student + ENT | Top 1 Error: 19.7 |
| unsupervised-domain-adaptation-on-imagenet-r | EfficientNet-L2 Noisy Student + RPL | Top 1 Error: 17.4 |
| unsupervised-domain-adaptation-on-imagenet-r | ResNet50 + ENT | Top 1 Error: 56.1 |