arXiv reaDer
再構築タスクは、ユニバーサル当選チケットを検索します
Reconstruction Task Finds Universal Winning Tickets
十分に訓練されたニューラルネットワークの剪定は、コンピュータービジョンレジームで有望な精度と効率のトレードオフを達成するのに効果的です。ただし、既存のプルーニングアルゴリズムのほとんどは、ソースドメインで定義された分類タスクにのみ焦点を当てています。元のモデルの強力な転送可能性とは異なり、プルーニングされたネットワークは、オブジェクト検出arXiv:arch-ive /2012.04643などの複雑なダウンストリームタスクに転送するのが困難です。この論文では、画像レベルの事前トレーニングタスクが多様なダウンストリームタスクのモデルを枝刈りすることができないことを示します。この問題を軽減するために、ピクセルレベルのタスクである画像再構成を従来の剪定フレームワークに導入します。具体的には、オートエンコーダーは元のモデルに基づいてトレーニングされ、その後、プルーニングプロセスはオートエンコーダーと分類損失の両方で最適化されます。ベンチマークダウンストリームタスクに関する実証的研究は、提案された方法が最先端の結果を明示的に上回ることができることを示しています。
Pruning well-trained neural networks is effective to achieve a promising accuracy-efficiency trade-off in computer vision regimes. However, most of existing pruning algorithms only focus on the classification task defined on the source domain. Different from the strong transferability of the original model, a pruned network is hard to transfer to complicated downstream tasks such as object detection arXiv:arch-ive/2012.04643. In this paper, we show that the image-level pretrain task is not capable of pruning models for diverse downstream tasks. To mitigate this problem, we introduce image reconstruction, a pixel-level task, into the traditional pruning framework. Concretely, an autoencoder is trained based on the original model, and then the pruning process is optimized with both autoencoder and classification losses. The empirical study on benchmark downstream tasks shows that the proposed method can outperform state-of-the-art results explicitly.
updated: Wed Feb 23 2022 13:04:32 GMT+0000 (UTC)
published: Wed Feb 23 2022 13:04:32 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アソシエイト