YOLO-Tight: an Efficient Dynamic Compression Method for YOLO Object Detection Networks
Wei Yan, Ting Liu, Yuzhuo Fu · 2021
Deep learning algorithms perform well in the field of object detection. Object detection networks represented by YOLO, SSD and faster-RCNN have achieved excellent performance on public datasets such as VOC and COCO. However, deep learning models are difficult to deploy on the edge computing platform with less computing resources due to its huge amount of parameters and computation. In this paper, we propose an efficient dynamic sparsity method to help the network quickly mine important parameters, and then prune the unimportant weight channels, which makes the network model more compact and consumes less computation. In the case of high sparsity, our method is more robust than L1 regularization and other regularization forms, and can achieve better sparsity and pruning effects. Through this method, we can prune the YOLOv3 network and the enhanced YOLOv3-SPP3 network by up to 90%. This allows the network to achieve 5× reduction in FLOPs and maintain an accuracy loss of less than 1% on the BDD100k dataset.