3 个月前

如果您的数据分布发生偏移,请使用自学习(self-learning)

如果您的数据分布发生偏移,请使用自学习(self-learning)

摘要

我们证明,诸如熵最小化和伪标签(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-aEfficientNet-L2 NoisyStudent + RPL
Top 1 Error: 14.8
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + DeepAug + Augmix + RPL
mean Corruption Error (mCE): 34.8
unsupervised-domain-adaptation-on-imagenet-cEfficientNet-L2+ENT
mean Corruption Error (mCE): 23.0
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + IG-3.5B + RPL
mean Corruption Error (mCE): 40.9
unsupervised-domain-adaptation-on-imagenet-cResNet50 + RPL
mean Corruption Error (mCE): 50.5
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + RPL
mean Corruption Error (mCE): 43.2
unsupervised-domain-adaptation-on-imagenet-cResNet50 + ENT
mean Corruption Error (mCE): 51.6
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + ENT
mean Corruption Error (mCE): 44.3
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + DeepAug + Augmix + ENT
mean Corruption Error (mCE): 35.5
unsupervised-domain-adaptation-on-imagenet-cResNeXt101 32x8d + IG-3.5B + ENT
mean Corruption Error (mCE): 40.8
unsupervised-domain-adaptation-on-imagenet-cEfficientNet-L2+RPL
mean Corruption Error (mCE): 22.0
unsupervised-domain-adaptation-on-imagenet-rResNet50 + RPL
Top 1 Error: 54.1
unsupervised-domain-adaptation-on-imagenet-rEfficientNet-L2 Noisy Student + ENT
Top 1 Error: 19.7
unsupervised-domain-adaptation-on-imagenet-rEfficientNet-L2 Noisy Student + RPL
Top 1 Error: 17.4
unsupervised-domain-adaptation-on-imagenet-rResNet50 + ENT
Top 1 Error: 56.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
如果您的数据分布发生偏移,请使用自学习(self-learning) | 论文 | HyperAI超神经