arXiv reaDer
ビジョントランスフォーマーの宝くじ仮説
The Lottery Ticket Hypothesis for Vision Transformers
従来の宝くじ仮説 (LTH) では、高密度のニューラル ネットワーク内に疎なサブネットワークが存在し、当選チケットと呼ばれる適切なランダム初期化手法が存在するため、最初から高密度の宝くじとほぼ同じレベルまでトレーニングできると主張されています。一方、ビジョン トランスフォーマー (ViTs) における LTH の研究はほとんど評価されていません。本稿ではまず、既存の方法では ViT の重みレベルで従来の当選チケットを見つけるのが難しいことを示します。次に、ViT の LTH を、ViT の入力依存性に触発された画像パッチで構成される入力画像に一般化します。つまり、入力画像パッチのサブセットが存在し、このパッチのサブセットのみを使用して ViT をゼロからトレーニングし、すべての画像パッチを使用してトレーニングした ViT と同様の精度を達成できます。この入力パッチのサブセットを当選チケットと呼びます。これは、入力内の大量の情報を表します。さらに、DeiT、LV-ViT、Swin Transformers など、さまざまな種類の ViT の入力パッチで当選チケットを見つけるためのシンプルかつ効果的な方法を紹介します。より具体的には、チケットセレクターを使用して、パッチの有益性に基づいて当選チケットを生成します。その間、比較のためにランダムに選択された別のパッチのサブセットを作成しました。実験では、勝ったチケットでトレーニングされたモデルとランダムに選択されたサブセットのパフォーマンスに明らかな違いがあることが示されています。
The conventional lottery ticket hypothesis (LTH) claims that there exists a sparse subnetwork within a dense neural network and a proper random initialization method, called the winning ticket, such that it can be trained from scratch to almost as good as the dense counterpart. Meanwhile, the research of LTH in vision transformers (ViTs) is scarcely evaluated. In this paper, we first show that the conventional winning ticket is hard to find at weight level of ViTs by existing methods. Then, we generalize the LTH for ViTs to input images consisting of image patches inspired by the input dependence of ViTs. That is, there exists a subset of input image patches such that a ViT can be trained from scratch by using only this subset of patches and achieve similar accuracy to the ViTs trained by using all image patches. We call this subset of input patches the winning tickets, which represent a significant amount of information in the input. Furthermore, we present a simple yet effective method to find the winning tickets in input patches for various types of ViT, including DeiT, LV-ViT, and Swin Transformers. More specifically, we use a ticket selector to generate the winning tickets based on the informativeness of patches. Meanwhile, we build another randomly selected subset of patches for comparison, and the experiments show that there is clear difference between the performance of models trained with winning tickets and randomly selected subsets.
updated: Wed Nov 02 2022 21:12:37 GMT+0000 (UTC)
published: Wed Nov 02 2022 21:12:37 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アソシエイト