従来の宝くじ仮説 (LTH) は、高密度のニューラル ネットワーク内に疎なサブネットワークと当たりチケットと呼ばれる適切なランダム初期化メソッドが存在し、それをゼロからトレーニングして高密度のニューラル ネットワークとほぼ同じレベルまでトレーニングできると主張しています。一方、ビジョントランスフォーマー (ViT) における LTH の研究はほとんど評価されていません。本稿ではまず、既存の方法では従来の当たりチケットをViTの重みレベルで見つけることが難しいことを示す。次に、ViT の入力依存性からインスピレーションを得た画像パッチからなるデータを入力するために、ViT の LTH を一般化します。つまり、入力画像パッチのサブセットが存在し、このパッチのサブセットのみを使用して ViT を最初からトレーニングでき、すべての画像パッチを使用してトレーニングされた ViT と同様の精度を達成できます。入力パッチのこのサブセットを em 勝ちチケットと呼びます。これは、入力データ内の大量の情報を表します。チケット セレクターを使用して、DeiT、LV-ViT、Swin Transformers などのさまざまなタイプの ViT のパッチの情報に基づいて当選チケットを生成します。実験では、当選チケットを使用してトレーニングされたモデルとランダムに選択されたサブセットのパフォーマンスの間に明らかな違いがあることが示されており、これにより私たちが提案した理論が検証されます。私たちは、理論の完全性をさらに検証するために、提案した Data-LTH-ViT と従来の LTH の間の類似点について詳しく説明します。ソース コードは https://github.com/shawnricecake/vit-lottery-ticket-input で入手できます。
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 the weight level of ViTs by existing methods. Then, we generalize the LTH for ViTs to input data 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 em winning tickets, which represent a significant amount of information in the input data. We use a ticket selector to generate the winning tickets based on the informativeness of patches for various types of ViT, including DeiT, LV-ViT, and Swin Transformers. The experiments show that there is a clear difference between the performance of models trained with winning tickets and randomly selected subsets, which verifies our proposed theory. We elaborate on the analogical similarity between our proposed Data-LTH-ViTs and the conventional LTH to further verify the integrity of our theory. The Source codes are available at https://github.com/shawnricecake/vit-lottery-ticket-input.