A Dynamic Discarding Technique to Increase Speed and Preserve Accuracy for YOLOv3

Authors

Ignacio Martinez-Alpiste, Gelayol Golcarenarenji, Qi Wang, Jose M. Alcaraz-Calero

Journal	Neural Computing and Applications
Impact Factor	4.774
Number of pages	14
Early online date	March 2021
Original language	English

This paper proposes an acceleration technique to minimise the unnecessary operations on a state-of-the-art machine learning model and thus to improve the processing speed while maintaining the accuracy. After the study of the main bottlenecks that negatively affect the performance of convolutional neural networks (CNN), this paper designs and implements a discarding technique for YOLOv3-based algorithms to increase the speed and maintain accuracy. After applying the discarding technique, YOLOv3 can achieve a 22% of improvement in terms of speed. Moreover, the results of this new discarding technique were tested on Tiny-YOLOv3 with three output layers on an autonomous vehicle for pedestrian detection and it achieved an improvement of 48.7% in speed. The dynamic discarding technique just needs one training process to create the model and thus execute the approach, which preserves accuracy. The improved detector based on the discarding technique is able to readily alert the operator of the autonomous vehicle to take the emergency brake of the vehicle in order to avoid collision and consequently save lives.

DOI: 10.1007/s00521-021-05764-7

Go to publication Cite BIBTEX