arXiv reaDer
物体認識のための宝くじの仮説
The Lottery Ticket Hypothesis for Object Recognition
近年、物体認識やキーポイント推定などの認識タスクが広く採用されています。これらのタスクのほとんどの最先端の方法は、計算コストが高く、メモリフットプリントが大きいディープネットワークを使用します。このため、これらのシステムを低電力の組み込みデバイスに展開することは非常に困難です。したがって、このようなモデルでは、ストレージ要件と計算量を減らすことが最も重要です。最近提案された宝くじチケット仮説(LTH)は、大きなデータセットでトレーニングされたディープニューラルネットワークには、密なネットワークと同等のパフォーマンスを実現する小さなサブネットワークが含まれていると述べています。この作業では、オブジェクト検出、インスタンスセグメンテーション、およびキーポイント推定のコンテキストでモデルプルーニングのLTHを調査する最初の実証的研究を実行します。私たちの調査によると、ImageNetの事前トレーニングから取得した宝くじは、ダウンストリームのタスクにうまく転送されません。パフォーマンスを低下させることなく、さまざまなサブタスクで全体のスパース性が最大80%の宝くじを見つける方法についてのガイダンスを提供します。最後に、オブジェクトのサイズ、頻度、検出の難しさなど、さまざまなタスク属性に関して、トレーニングされたチケットの動作を分析します。
Recognition tasks, such as object recognition and keypoint estimation, have seen widespread adoption in recent years. Most state-of-the-art methods for these tasks use deep networks that are computationally expensive and have huge memory footprints. This makes it exceedingly difficult to deploy these systems on low power embedded devices. Hence, the importance of decreasing the storage requirements and the amount of computation in such models is paramount. The recently proposed Lottery Ticket Hypothesis (LTH) states that deep neural networks trained on large datasets contain smaller subnetworks that achieve on par performance as the dense networks. In this work, we perform the first empirical study investigating LTH for model pruning in the context of object detection, instance segmentation, and keypoint estimation. Our studies reveal that lottery tickets obtained from ImageNet pretraining do not transfer well to the downstream tasks. We provide guidance on how to find lottery tickets with up to 80% overall sparsity on different sub-tasks without incurring any drop in the performance. Finally, we analyse the behavior of trained tickets with respect to various task attributes such as object size, frequency, and difficulty of detection.
updated: Tue Dec 08 2020 18:59:13 GMT+0000 (UTC)
published: Tue Dec 08 2020 18:59:13 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アソシエイト