arXiv reaDer
プログレッシブ再トレーニングによる畳み込みニューラルネットワークの単純化
Convolutional Neural Network Simplification with Progressive Retraining
畳み込みニューラルネットワーク(CNN)モデルの説明を高速化、簡素化、および改善するために、カーネル剪定法が提案されています。ただし、簡略化されたモデルの有効性は、多くの場合、元のモデルを下回ります。この手紙では、カーネル除去のための客観的および主観的な関連性基準に基づいた新しい方法をレイヤーごとに提示します。プロセス中、CNNモデルは、次のレイヤーから最初のレイヤーへの重みを調整し、プロセスに関与しない後続のレイヤーの重みを保持することにより、現在のレイヤーが完全に単純化された場合にのみ再トレーニングされます。この戦略をプログレッシブ再トレーニングと呼びます。これは、通常、各単純化アクションの後にモデル全体を再トレーニングするカーネルプルーニング方法とは異なります。たとえば、1つまたはいくつかのカーネルを削除します。私たちの主観的な関連性基準は、視覚パターンを認識する人間の能力を活用し、単純化プロセスの設計者の理解を向上させます。適切な関連性基準と漸進的な再トレーニングの組み合わせは、私たちの方法がかなりのモデルの単純化で有効性を高めることができることを示しています。また、4つの難しい画像データセットを使用して、2つの人気のある方法や最先端の方法よりも優れた結果を提供できることも示しています。
Kernel pruning methods have been proposed to speed up, simplify, and improve explanation of convolutional neural network (CNN) models. However, the effectiveness of a simplified model is often below the original one. In this letter, we present new methods based on objective and subjective relevance criteria for kernel elimination in a layer-by-layer fashion. During the process, a CNN model is retrained only when the current layer is entirely simplified, by adjusting the weights from the next layer to the first one and preserving weights of subsequent layers not involved in the process. We call this strategy progressive retraining, differently from kernel pruning methods that usually retrain the entire model after each simplification action -- e.g., the elimination of one or a few kernels. Our subjective relevance criterion exploits the ability of humans in recognizing visual patterns and improves the designer's understanding of the simplification process. The combination of suitable relevance criteria and progressive retraining shows that our methods can increase effectiveness with considerable model simplification. We also demonstrate that our methods can provide better results than two popular ones and another one from the state-of-the-art using four challenging image datasets.
updated: Tue Jan 12 2021 19:05:42 GMT+0000 (UTC)
published: Tue Jan 12 2021 19:05:42 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アソシエイト