Deep Network Pruning for Object Detection

With the increasing success of deep learning in various applications, there is an increasing need to have deep models that can be used for deployment in real-time and/or resource constrained scenarios. In this context, this paper analyzes the pruning of deep models for object detection in order to r...

Full description

Saved in:
Bibliographic Details
Published in2019 IEEE International Conference on Image Processing (ICIP) pp. 3915 - 3919
Main Authors Ghosh, Sanjukta, Srinivasa, Shashi K K, Amon, Peter, Hutter, Andreas, Kaup, Andre
Format Conference Proceeding
LanguageEnglish
Published IEEE 01.09.2019
Subjects
Biological system modeling
Clustering
CNN
Computational modeling
Computer architecture
Correlation
Deep Learning
Feature extraction
Mutual information
Network Pruning
Object detection
Online AccessGet full text

Cover

More Information
Summary:With the increasing success of deep learning in various applications, there is an increasing need to have deep models that can be used for deployment in real-time and/or resource constrained scenarios. In this context, this paper analyzes the pruning of deep models for object detection in order to reduce the number of weights and hence the number of computations. Very deep networks based on ResNet like architectures, like YOLOv3 have unique challenges when attempting to prune them. This paper proposes a network pruning technique based on agglomerative clustering for the feature extractor and using mutual information for the detector. The performance of the proposed techniques is also compared with that of a relatively shallow network, i.e., YOLOv2. A compression percentage of around 30% results in a 10% drop of mean average precision (mAP) in YOLOv3, whereas in YOLOv2 the drop was around 6% on the COCO dataset.
ISSN:2381-8549
DOI:10.1109/ICIP.2019.8803505