中文

ImageNet 上的鲁棒性能否迁移到下游任务?

计算机视觉与模式识别 2022-04-11 v1 机器学习

摘要

随着干净 ImageNet 准确率接近其上限,研究界越来越关注分布偏移下的鲁棒准确率。尽管已提出多种方法来使神经网络鲁棒化,这些技术通常针对在 ImageNet 分类上训练的模型。同时,使用 ImageNet 预训练骨干网络用于下游任务(如目标检测、语义分割和来自不同领域的图像分类)是一种常见做法。这引发了一个问题:这些鲁棒的图像分类器能否将鲁棒性迁移到下游任务?对于目标检测和语义分割,我们发现一个原始的 Swin Transformer(一种为密集预测任务定制的 Vision Transformer 变体)比那些被训练为对损坏版 ImageNet 鲁棒的卷积神经网络迁移鲁棒性更好。对于 CIFAR10 分类,我们发现为 ImageNet 做了鲁棒化的模型在完全微调后无法保持鲁棒性。这些发现表明,当前的鲁棒化技术往往偏重 ImageNet 评估。此外,当我们考虑迁移学习时,网络架构是鲁棒性的一个重要来源。

关键词

引用

@article{arxiv.2204.03934,
  title  = {Does Robustness on ImageNet Transfer to Downstream Tasks?},
  author = {Yutaro Yamada and Mayu Otani},
  journal= {arXiv preprint arXiv:2204.03934},
  year   = {2022}
}

备注

CVPR 2022