arXiv reaDer
Localization-aware Channel Pruning for Object Detection
 チャネルプルーニングは、ディープモデル圧縮の重要な方法の1つです。既存のプルーニング方法のほとんどは、主に分類に焦点を合わせています。それらのいくつかは、オブジェクト検出に関する体系的な研究を行っています。ただし、オブジェクト検出は分類とは異なり、分類には意味情報だけでなくローカリゼーション情報も必要です。本論文では、分類のための最先端の枝刈り方法である識別を意識したチャネル枝刈り(DCP)に基づいて、分類と回帰のための重要な情報を持つチャネルを見つけるためのローカライズを意識した補助ネットワークを提案します。オブジェクトを検出するためにチャネルのプルーニングを直接実行でき、多くの時間とコンピューティングリソースを節約できます。ローカリゼーション情報をキャプチャするために、最初にピクセルアラインメントによってデフォルトボックスの正確なローカリゼーション情報を取得し、浅いレイヤーをプルーニングするときにデフォルトボックスの受容フィールドを拡大できるコンテキストROIAlignレイヤーで補助ネットワークを設計します。次に、オブジェクト検出タスクの損失関数を作成します。この関数は、分類と回帰の重要な情報を含むチャネルを保持する傾向があります。広範な実験により、この方法の有効性が実証されています。 MS COCOでは、ResNet-50に基づいてSSDの70%のパラメーターをわずかな精度低下で除去します。これは、最先端の方法よりも優れています。
Channel pruning is one of the important methods for deep model compression. Most of existing pruning methods mainly focus on classification. Few of them conduct systematic research on object detection. However, object detection is different from classification, which requires not only semantic information but also localization information. In this paper, based on discrimination-aware channel pruning (DCP) which is state-of-the-art pruning method for classification, we propose a localization-aware auxiliary network to find out the channels with key information for classification and regression so that we can conduct channel pruning directly for object detection, which saves lots of time and computing resources. In order to capture the localization information, we first design the auxiliary network with a contextual ROIAlign layer which can obtain precise localization information of the default boxes by pixel alignment and enlarges the receptive fields of the default boxes when pruning shallow layers. Then, we construct a loss function for object detection task which tends to keep the channels that contain the key information for classification and regression. Extensive experiments demonstrate the effectiveness of our method. On MS COCO, we prune 70% parameters of the SSD based on ResNet-50 with modest accuracy drop, which outperforms the-state-of-art method.
updated: Sat Feb 29 2020 08:29:03 GMT+0000 (UTC)
published: Wed Nov 06 2019 07:41:21 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト