arXiv reaDer
ニューラルネットワークのタスク依存のディープLDAプルーニング
Task dependent Deep LDA pruning of neural networks
ディープラーニングの成功により、限られた数の人気のあるディープネットがさまざまなビジョンタスクに広く採用されています。ただし、これは通常、不必要に高い複雑さをもたらし、タスクの有用性が低いという多くの機能をもたらす可能性があります。この論文では、フィッシャーの線形判別分析(LDA)に基づくタスク依存のディーププルーニングフレームワークを導入することにより、この問題に対処します。このアプローチは、畳み込み、完全に接続された、モジュールベースのディープネットワーク構造に適用できます。すべての場合において、事前決定空間で見られるニューロンモチーフの高い無相関化とクロスレイヤーデコンバー依存性を活用します。さらに、特定のタスクのネットワークアーキテクチャ検索におけるアプローチの可能性を検証し、ノイズや敵対的攻撃に対するモデルの堅牢性に対する剪定の影響を分析します。ジェネリックオブジェクト(ImageNet、CIFAR100)およびドメイン固有のタスク(Adience、およびLFWA)のデータセットに関する実験結果は、最先端のプルーニングアプローチおよび固定コンパクトネット(SqueezeNet、MobileNetなど)よりも優れたフレームワークのパフォーマンスを示しています。提案された方法は、ほとんどのパラメーター(VGG16では98%〜99%、すでにコンパクトなInceptionNetでは最大82%)を破棄し、FLOPを大幅に削減(VGG16では83%、InceptionNetでは最大64%)した後でも、同等の精度を維持できます。剪定により、タスクに適した、より小さく、より正確でより堅牢なモデルを導出することもできます。
With deep learning's success, a limited number of popular deep nets have been widely adopted for various vision tasks. However, this usually results in unnecessarily high complexities and possibly many features of low task utility. In this paper, we address this problem by introducing a task-dependent deep pruning framework based on Fisher's Linear Discriminant Analysis (LDA). The approach can be applied to convolutional, fully-connected, and module-based deep network structures, in all cases leveraging the high decorrelation of neuron motifs found in the pre-decision space and cross-layer deconv dependency. Moreover, we examine our approach's potential in network architecture search for specific tasks and analyze the influence of our pruning on model robustness to noises and adversarial attacks. Experimental results on datasets of generic objects (ImageNet, CIFAR100) as well as domain specific tasks (Adience, and LFWA) illustrate our framework's superior performance over state-of-the-art pruning approaches and fixed compact nets (e.g. SqueezeNet, MobileNet). The proposed method successfully maintains comparable accuracies even after discarding most parameters (98%-99% for VGG16, up to 82% for the already compact InceptionNet) and with significant FLOP reductions (83% for VGG16, up to 64% for InceptionNet). Through pruning, we can also derive smaller, but more accurate and more robust models suitable for the task.
updated: Tue Nov 24 2020 05:56:56 GMT+0000 (UTC)
published: Wed Mar 21 2018 20:52: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アソシエイト