arXiv reaDer
自動運転における深部視覚検出器のための視覚的顕著性ガイド付きチャネル プルーニング
Visual Saliency-Guided Channel Pruning for Deep Visual Detectors in Autonomous Driving
ディープ ニューラル ネットワーク (DNN) プルーニングは、推論中のメモリ要件と計算コストを削減できるため、リソースに制約のあるデバイスにデプロイするための事実上のコンポーネントになりました。特に、チャネルのプルーニングは、その構造化された性質と一般的なハードウェアの直接的な節約により、人気が高まりました。ただし、ほとんどの既存のプルーニング アプローチは、タスクの効用に直接関係しない重要度の尺度を使用します。さらに、視覚検出モデルに焦点を当てた文献はほとんどありません。これらのギャップを埋めるために、視覚的検出のための新しい勾配ベースの顕著性測定を提案し、それを使用してチャネルプルーニングをガイドします。 KITTI および COCO トラフィック データセットに関する実験は、最先端の競合するアプローチに対する私たちの剪定方法の有効性と優位性を示しています。元のモデルよりも少ないパラメーターで、より優れたパフォーマンスを達成することさえできます。私たちの剪定は、小規模なオブジェクトの処理にも大きな可能性を示しています。
Deep neural network (DNN) pruning has become a de facto component for deploying on resource-constrained devices since it can reduce memory requirements and computation costs during inference. In particular, channel pruning gained more popularity due to its structured nature and direct savings on general hardware. However, most existing pruning approaches utilize importance measures that are not directly related to the task utility. Moreover, few in the literature focus on visual detection models. To fill these gaps, we propose a novel gradient-based saliency measure for visual detection and use it to guide our channel pruning. Experiments on the KITTI and COCO traffic datasets demonstrate our pruning method's efficacy and superiority over state-of-the-art competing approaches. It can even achieve better performance with fewer parameters than the original model. Our pruning also demonstrates great potential in handling small-scale objects.
updated: Sat Mar 04 2023 22:08:22 GMT+0000 (UTC)
published: Sat Mar 04 2023 22:08:22 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アソシエイト