arXiv reaDer
フィルタ内の重み共有による微分可能なチャネルスパース性検索
Differentiable Channel Sparsity Search via Weight Sharing within Filters
この論文では、畳み込みニューラルネットワークの微分可能チャネルスパース性検索(DCSS)を提案します。ユーザーが畳み込み層ごとに手動でプルーニング比率を設定する必要がある従来のチャネルプルーニングアルゴリズムとは異なり、DCSSはスパース性の最適な組み合わせを自動的に検索します。微分可能アーキテクチャ検索(DARTS)に触発されて、継続的な緩和から教訓を引き出し、勾配情報を活用して計算コストとメトリックのバランスを取ります。 DARTSのスキームを直接適用すると、形状の不一致や過剰なメモリ消費が発生するため、フィルター内での重み共有と呼ばれる新しい手法を紹介します。この手法は、ごくわずかな追加リソースとの形状の不一致の問題をエレガントに排除します。 DCSSの有効性を検証するために、画像分類だけでなく、セマンティックセグメンテーションや画像超解像などの詳細なタスクについても包括的な実験を行います。以前のネットワークプルーニングアプローチと比較して、DCSSは画像分類のための最先端の結果を達成します。セマンティックセグメンテーションと画像の超解像の実験結果は、タスク固有の検索がスリムなモデルを転送するよりも優れたパフォーマンスを達成することを示しており、DCSSの幅広い適用性と高効率を示しています。
In this paper, we propose the differentiable channel sparsity search (DCSS) for convolutional neural networks. Unlike traditional channel pruning algorithms which require users to manually set prune ratios for each convolutional layer, DCSS automatically searches the optimal combination of sparsities. Inspired by the differentiable architecture search (DARTS), we draw lessons from the continuous relaxation and leverage the gradient information to balance the computational cost and metrics. Since directly applying the scheme of DARTS causes shape mismatching and excessive memory consumption, we introduce a novel technique called weight sharing within filters. This technique elegantly eliminates the problem of shape mismatching with negligible additional resources. We conduct comprehensive experiments on not only image classification but also find-grained tasks including semantic segmentation and image super resolution to verify the effectiveness of DCSS. Compared with previous network pruning approaches, DCSS achieves state-of-the-art results for image classification. Experimental results of semantic segmentation and image super resolution indicate that task-specific search achieves better performance than transferring slim models, demonstrating the wide applicability and high efficiency of DCSS.
updated: Wed Jan 05 2022 13:39:06 GMT+0000 (UTC)
published: Wed Oct 28 2020 02:49: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アソシエイト