arXiv reaDer
ConvNets対トランスフォーマー:誰の視覚的表現がより転送可能ですか?
ConvNets vs. Transformers: Whose Visual Representations are More Transferable?
ビジョントランスフォーマーは、ConvNetの空間誘導バイアスに制限されていないため、コンピュータービジョンの研究者から大きな注目を集めています。ただし、TransformerベースのバックボーンはImageNet分類で多くの進歩を遂げましたが、学習した表現がConvNetの機能と同じくらい転送可能であるか、それとも転送可能であるかはまだ不明です。この点に対処するために、15のシングルタスクおよびマルチタスクのパフォーマンス評価で、ConvNetとビジョントランスフォーマーの伝達学習能力を体系的に調査します。事前トレーニング済みモデルのパフォーマンスと転移学習の間の強い相関関係を考慮して、2つの残りのConvNet(つまり、R-101x3とR-152x4)と3つのTransformerベースのビジュアルバックボーン(つまり、ViT-B、ViT-L、 Swin-B)は、ImageNetでエラー率が高く、ダウンストリームデータセットで同様の転送学習パフォーマンスを示しています。トランスフォーマーベースのバックボーンは、13のダウンストリームタスク(15のうち)で一貫した利点があります。これには、きめ細かい分類、シーン認識(分類、セグメンテーション、深度推定)、オープンドメイン分類、顔認識などが含まれますが、これらに限定されません。具体的には、2つのViTモデルは、パフォーマンスの向上を実現するためにネットワーク全体の微調整に大きく依存していますが、SwinTransformerにはそのような要件はありません。さらに、ビジョントランスフォーマーは、マルチタスク学習でより堅牢に動作します。つまり、相互に有益なタスクを管理する際により多くの改善をもたらし、無関係なタスクに取り組む際のパフォーマンスの低下を減らします。私たちの発見が将来のビジョントランスフォーマーの探索と開発を促進することを願っています。
Vision transformers have attracted much attention from computer vision researchers as they are not restricted to the spatial inductive bias of ConvNets. However, although Transformer-based backbones have achieved much progress on ImageNet classification, it is still unclear whether the learned representations are as transferable as or even more transferable than ConvNets' features. To address this point, we systematically investigate the transfer learning ability of ConvNets and vision transformers in 15 single-task and multi-task performance evaluations. Given the strong correlation between the performance of pre-trained models and transfer learning, we include 2 residual ConvNets (i.e., R-101x3 and R-152x4) and 3 Transformer-based visual backbones (i.e., ViT-B, ViT-L and Swin-B), which have close error rates on ImageNet, that indicate similar transfer learning performance on downstream datasets. We observe consistent advantages of Transformer-based backbones on 13 downstream tasks (out of 15), including but not limited to fine-grained classification, scene recognition (classification, segmentation and depth estimation), open-domain classification, face recognition, etc. More specifically, we find that two ViT models heavily rely on whole network fine-tuning to achieve performance gains while Swin Transformer does not have such a requirement. Moreover, vision transformers behave more robustly in multi-task learning, i.e., bringing more improvements when managing mutually beneficial tasks and reducing performance losses when tackling irrelevant tasks. We hope our discoveries can facilitate the exploration and exploitation of vision transformers in the future.
updated: Wed Aug 11 2021 16:20:38 GMT+0000 (UTC)
published: Wed Aug 11 2021 16:20:38 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アソシエイト