arXiv reaDer
堅牢なビジョントランスフォーマーに向けて
Towards Robust Vision Transformer
Vision Transformer(ViT)とその改良されたバリアントに関する最近の進歩は、自己注意ベースのネットワークがほとんどのビジョンタスクで従来の畳み込みニューラルネットワーク(CNN)を上回っていることを示しています。ただし、既存のViTは標準の精度と計算コストに重点を置いており、モデルの堅牢性と一般化に対する本質的な影響の調査が不足しています。この作業では、ViTのコンポーネントについて、敵対的な例に対する堅牢性、一般的な破損、および分布の変化への影響の観点から体系的な評価を行います。一部のコンポーネントは堅牢性に悪影響を与える可能性があることがわかりました。堅牢なコンポーネントをViTの構成要素として使用および組み合わせることにより、新しいビジョントランスであり、強力な堅牢性を備えた優れたパフォーマンスを備えたRobust Vision Transformer(RVT)を提案します。さらに、位置認識アテンションスケーリングとパッチワイズ拡張と呼ばれる2つの新しいプラグアンドプレイ技術を提案して、RVTを拡張します。これをRVT *と略します。 ImageNetと6つの堅牢性ベンチマークの実験結果は、以前のViTや最先端のCNNと比較して、RVTの高度な堅牢性と一般化能力を示しています。さらに、RVT-S *は、ImageNet-CやImageNet-Sketchなどの複数の堅牢性リーダーボードでもトップ1ランクを獲得しています。コードはhttps://git.io/Jswdkで入手できます。
Recent advances on Vision Transformer (ViT) and its improved variants have shown that self-attention-based networks surpass traditional Convolutional Neural Networks (CNNs) in most vision tasks. However, existing ViTs focus on the standard accuracy and computation cost, lacking the investigation of the intrinsic influence on model robustness and generalization. In this work, we conduct systematic evaluation on components of ViTs in terms of their impact on robustness to adversarial examples, common corruptions and distribution shifts. We find some components can be harmful to robustness. By using and combining robust components as building blocks of ViTs, we propose Robust Vision Transformer (RVT), which is a new vision transformer and has superior performance with strong robustness. We further propose two new plug-and-play techniques called position-aware attention scaling and patch-wise augmentation to augment our RVT, which we abbreviate as RVT*. The experimental results on ImageNet and six robustness benchmarks show the advanced robustness and generalization ability of RVT compared with previous ViTs and state-of-the-art CNNs. Furthermore, RVT-S* also achieves Top-1 rank on multiple robustness leaderboards including ImageNet-C and ImageNet-Sketch. The code will be available at https://git.io/Jswdk.
updated: Wed May 26 2021 09:01:06 GMT+0000 (UTC)
published: Mon May 17 2021 15:04:15 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アソシエイト