arXiv reaDer
ビジョンと言語で宝くじをプレイする
Playing Lottery Tickets with Vision and Language
大規模な事前トレーニングは、最近、視覚と言語(VL)の研究に革命をもたらしました。 LXMERTやUNITERなどのモデルは、幅広いVLタスクで最先端技術を大幅に向上させました。ただし、このようなモデルのパラメーターの数が多いと、実際のアプリケーションの妨げになります。並行して、宝くじの仮説(LTH)での作業により、ディープニューラルネットワークには、単独でトレーニングした場合に高密度ネットワークと同等またはそれ以上のパフォーマンスを達成できる小さなマッチングサブネットワークが含まれていることが示されました。この作業では、このようなトレーニング可能なサブネットワークが事前トレーニングされたVLモデルにも存在するかどうかを評価するための最初の実証的研究を実行します。 UNITERをメインテストベッドとして使用し(LXMERTとViLTでもテスト)、視覚的な質問応答、視覚的な常識的な推論、視覚的な含意、表現の理解の参照、画像テキスト検索、GQA、NLVRなどの実験用の7つの代表的なVLタスクを統合します^ 2。総合的な分析を通じて、主な調査結果を以下のように要約します。 (i)完全なモデルのパフォーマンスに厳密に一致するサブネットワークを見つけることは困難です。ただし、完全な精度の99%を維持する、50%〜70%のスパース性で「リラックスした」当選チケットを見つけることができます。 (ii)タスク固有のプルーニングによって検出されたサブネットワークは他のタスクに適度に転送されますが、60%/ 70%のスパース性で事前トレーニングタスクで検出されたサブネットワークは普遍的に転送され、平均して完全な精度の98%/ 96%に一致します。すべてのタスク。 (iii)UNITERの他に、LXMERTやViLTなどの他のモデルも宝くじをプレイできます。ただし、ViLTで達成できる最高のスパース性は、LXMERTおよびUNITERよりもはるかに低くなります(30%対70%)。 (iv)LTHは、他の訓練方法(例、敵対的訓練)を使用する場合にも関連性があります。
Large-scale pre-training has recently revolutionized vision-and-language (VL) research. Models such as LXMERT and UNITER have significantly lifted the state of the art over a wide range of VL tasks. However, the large number of parameters in such models hinders their application in practice. In parallel, work on the lottery ticket hypothesis (LTH) has shown that deep neural networks contain small matching subnetworks that can achieve on par or even better performance than the dense networks when trained in isolation. In this work, we perform the first empirical study to assess whether such trainable subnetworks also exist in pre-trained VL models. We use UNITER as the main testbed (also test on LXMERT and ViLT), and consolidate 7 representative VL tasks for experiments, including visual question answering, visual commonsense reasoning, visual entailment, referring expression comprehension, image-text retrieval, GQA, and NLVR^2. Through comprehensive analysis, we summarize our main findings as follows. (i) It is difficult to find subnetworks that strictly match the performance of the full model. However, we can find "relaxed" winning tickets at 50%-70% sparsity that maintain 99% of the full accuracy. (ii) Subnetworks found by task-specific pruning transfer reasonably well to the other tasks, while those found on the pre-training tasks at 60%/70% sparsity transfer universally, matching 98%/96% of the full accuracy on average over all the tasks. (iii) Besides UNITER, other models such as LXMERT and ViLT can also play lottery tickets. However, the highest sparsity we can achieve for ViLT is far lower than LXMERT and UNITER (30% vs. 70%). (iv) LTH also remains relevant when using other training methods (e.g., adversarial training).
updated: Tue Dec 14 2021 23:04:45 GMT+0000 (UTC)
published: Fri Apr 23 2021 22:24:33 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アソシエイト