中文

探究 corruption 鲁棒性:视觉 Transformer 与 MLP-Mixer 中的归纳偏置

计算机视觉与模式识别 2021-07-06 v2 机器学习

摘要

近来,视觉 Transformer 与基于 MLP 的模型被提出,以解决卷积神经网络中一些普遍存在的弱点。由于 Transformer 在该领域应用的新颖性以及自注意力机制,这些架构在多大程度上对 corruption 具有鲁棒性仍不清楚。尽管一些工作提出数据增强对于模型抵御 corruption 的鲁棒性仍然必不可少,我们主张探究架构本身对 corruption 鲁棒性的影响。我们发现视觉 Transformer 架构本质上比 ResNet-50 和 MLP-Mixer 对 corruption 更具鲁棒性。我们还发现,参数量比 ResNet-50 少 5 倍的视觉 Transformer 具有更强的形状偏置。我们的代码已公开以供复现。

关键词

引用

@article{arxiv.2106.13122,
  title  = {Exploring Corruption Robustness: Inductive Biases in Vision Transformers and MLP-Mixers},
  author = {Katelyn Morrison and Benjamin Gilby and Colton Lipchak and Adam Mattioli and Adriana Kovashka},
  journal= {arXiv preprint arXiv:2106.13122},
  year   = {2021}
}

备注

Under review at the Uncertainty and Robustness in Deep Learning workshop at ICML 2021. Our appendix is attached to the last page of the paper