arXiv reaDer
画像分類のためのトランスフォーマーのロバスト性を理解する
Understanding Robustness of Transformers for Image Classification
ディープ畳み込みニューラルネットワーク(CNN)は、長い間、コンピュータービジョンタスクに最適なアーキテクチャでした。最近、Vision Transformer(ViT)のようなTransformerベースのアーキテクチャは、画像分類に関してResNetに匹敵するか、それを上回っています。ただし、重複しないパッチの使用など、Transformerアーキテクチャの詳細により、これらのネットワークが同じくらい堅牢であるかどうか疑問に思われるかもしれません。このホワイトペーパーでは、ViTモデルの堅牢性に関するさまざまな測定値について広範な調査を実施し、その結果をResNetベースラインと比較します。入力摂動に対するロバスト性とモデル摂動に対するロバスト性を調査します。十分な量のデータを事前にトレーニングすると、ViTモデルは、広範囲の摂動に対して少なくともResNetモデルと同じくらい堅牢であることがわかります。また、Transformerはほとんどすべての単一レイヤーの削除に対して堅牢であり、後のレイヤーからのアクティブ化は相互に高度に相関していますが、それでも分類において重要な役割を果たしていることがわかります。
Deep Convolutional Neural Networks (CNNs) have long been the architecture of choice for computer vision tasks. Recently, Transformer-based architectures like Vision Transformer (ViT) have matched or even surpassed ResNets for image classification. However, details of the Transformer architecture -- such as the use of non-overlapping patches -- lead one to wonder whether these networks are as robust. In this paper, we perform an extensive study of a variety of different measures of robustness of ViT models and compare the findings to ResNet baselines. We investigate robustness to input perturbations as well as robustness to model perturbations. We find that when pre-trained with a sufficient amount of data, ViT models are at least as robust as the ResNet counterparts on a broad range of perturbations. We also find that Transformers are robust to the removal of almost any single layer, and that while activations from later layers are highly correlated with each other, they nevertheless play an important role in classification.
updated: Fri Oct 08 2021 15:28:11 GMT+0000 (UTC)
published: Fri Mar 26 2021 16:47:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト