従来の宝くじ仮説 (LTH) では、高密度のニューラル ネットワーク内に疎なサブネットワークが存在し、当選チケットと呼ばれる適切なランダム初期化手法が存在するため、最初から高密度の宝くじとほぼ同じレベルまでトレーニングできると主張されています。一方、ビジョン トランスフォーマー (ViTs) における LTH の研究はほとんど評価されていません。本稿ではまず、既存の方法では ViTs の重みレベルで従来の当選チケットを見つけるのが難しいことを示します。次に、ViT の LTH を、ViT の入力依存性に触発された画像パッチで構成される入力データに一般化します。つまり、入力画像パッチのサブセットが存在し、このパッチのサブセットのみを使用して ViT をゼロからトレーニングし、すべての画像パッチを使用してトレーニングした ViT と同様の精度を達成できます。この入力パッチのサブセットを em ウィニング チケットと呼びます。これは、入力データ内の大量の情報を表します。チケット セレクターを使用して、DeiT、LV-ViT、Swin Transformers など、さまざまな種類の ViT のパッチの有益性に基づいて当選チケットを生成します。実験は、当選チケットとランダムに選択されたサブセットでトレーニングされたモデルのパフォーマンスに明確な違いがあることを示しており、提案された理論を検証しています。提案された Data-LTH-ViT と従来の LTH との間の類推的な類似性について詳しく説明し、理論の完全性をさらに検証します。コードは補足で提供されます。
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 code is provided in the supplementary.