arXiv reaDer
MTP:効率的なセマンティックセグメンテーションネットワークのためのマルチタスクプルーニング
MTP: Multi-Task Pruning for Efficient Semantic Segmentation Networks
このホワイトペーパーでは、セマンティックセグメンテーションネットワークのチャネルプルーニングに焦点を当てています。分類タスクでディープニューラルネットワークを圧縮および加速する以前の方法は、事前トレーニングを介した暗黙のマルチタスク学習問題を含むセマンティックセグメンテーションネットワークに直接適用することはできません。セグメンテーションネットワークの冗長性を識別するために、マルチタスクチャネルプルーニングアプローチを提示します。各畳み込みフィルターの重要性は、任意のレイヤーのチャネルで、分類タスクとセグメンテーションタスクによって同時に決定されます。さらに、ネットワーク全体でフィルターの重要度スコアを最適化するための代替スキームを開発します。いくつかのベンチマークでの実験結果は、最先端の剪定方法に対する提案されたアルゴリズムの優位性を示しています。特に、DeepLabv3では約2倍のFLOP削減が得られ、PASCAL VOC 2012データセットでは約1%mIoUの低下、Cityscapesデータセットでは約1.3%mIoUの低下が見られます。
This paper focuses on channel pruning for semantic segmentation networks. Previous methods to compress and accelerate deep neural networks in the classification task cannot be straightforwardly applied to the semantic segmentation network that involves an implicit multi-task learning problem via pre-training. To identify the redundancy in segmentation networks, we present a multi-task channel pruning approach. The importance of each convolution filter \wrt the channel of an arbitrary layer will be simultaneously determined by the classification and segmentation tasks. In addition, we develop an alternative scheme for optimizing importance scores of filters in the entire network. Experimental results on several benchmarks illustrate the superiority of the proposed algorithm over the state-of-the-art pruning methods. Notably, we can obtain an about 2× FLOPs reduction on DeepLabv3 with only an about 1% mIoU drop on the PASCAL VOC 2012 dataset and an about 1.3% mIoU drop on Cityscapes dataset, respectively.
updated: Tue Mar 15 2022 06:54:53 GMT+0000 (UTC)
published: Thu Jul 16 2020 15:03:01 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アソシエイト