arXiv reaDer
The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks
ニューラルネットワークのプルーニング技術は、トレーニング済みネットワークのパラメーター数を90%以上削減し、ストレージ要件を減らし、精度を損なうことなく推論の計算パフォーマンスを向上させることができます。ただし、現在の経験では、プルーニングによって生成される疎なアーキテクチャは最初からトレーニングするのが難しく、同様にトレーニングパフォーマンスが向上します。標準のプルーニング手法では、初期化により効果的にトレーニングできるサブネットワークが自然に発見されることがわかります。これらの結果に基づいて、「宝くじチケット仮説」を明確にします:密に、ランダムに初期化されたフィードフォワードネットワークは、サブネットワーク(「勝ちチケット」)を含みます-単独でトレーニングされると、同様の元のネットワークに匹敵するテスト精度に達します反復回数。私たちが見つけた当選チケットは、初期化宝くじに当選しました。それらの接続には、トレーニングを特に効果的にする初期重みがあります。当選チケットを識別するアルゴリズムと、宝くじチケットの仮説とこれらの偶発的な初期化の重要性をサポートする一連の実験を紹介します。 MNISTおよびCIFAR10の完全に接続された畳み込みフィードフォワードアーキテクチャのサイズの10〜20%未満の入賞チケットが常に見つかります。このサイズを超えると、当選したチケットは元のネットワークよりも早く学習し、より高いテスト精度に達します。
Neural network pruning techniques can reduce the parameter counts of trained networks by over 90%, decreasing storage requirements and improving computational performance of inference without compromising accuracy. However, contemporary experience is that the sparse architectures produced by pruning are difficult to train from the start, which would similarly improve training performance. We find that a standard pruning technique naturally uncovers subnetworks whose initializations made them capable of training effectively. Based on these results, we articulate the "lottery ticket hypothesis:" dense, randomly-initialized, feed-forward networks contain subnetworks ("winning tickets") that - when trained in isolation - reach test accuracy comparable to the original network in a similar number of iterations. The winning tickets we find have won the initialization lottery: their connections have initial weights that make training particularly effective. We present an algorithm to identify winning tickets and a series of experiments that support the lottery ticket hypothesis and the importance of these fortuitous initializations. We consistently find winning tickets that are less than 10-20% of the size of several fully-connected and convolutional feed-forward architectures for MNIST and CIFAR10. Above this size, the winning tickets that we find learn faster than the original network and reach higher test accuracy.
updated: Mon Mar 04 2019 15:51:11 GMT+0000 (UTC)
published: Fri Mar 09 2018 18:51:28 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト