Optimization of YOLOv8 model based on pruning

Dengji Zhang; Sunyan Hong; Weiwei Gao; Jing Yu; Zhenting Chen; Xutian Shen; Jianxiong Wu

doi:10.1117/12.3049687

8 November 2024 Optimization of YOLOv8 model based on pruning

Dengji Zhang, Sunyan Hong, Weiwei Gao, Jing Yu, Zhenting Chen, Xutian Shen, Jianxiong Wu

Author Affiliations +

Proceedings Volume 13416, Fourth International Conference on Advanced Algorithms and Neural Networks (AANN 2024); 134162B (2024) https://doi.org/10.1117/12.3049687
Event: 2024 4th International Conference on Advanced Algorithms and Neural Networks, 2024, Qingdao, China

Abstract

YOLOv8 is one of the most commonly used object detection algorithms. However, its network model has a large number of parameters, resulting in slow performance on embedded devices. One of the challenges in industrial applications of this algorithm is reducing the parameter size of the YOLOv8 model without significantly compromising its detection accuracy, thus enabling it to run efficiently on embedded devices. To address this, a structured pruning strategy based on Torch-Pruning was designed specifically for medium-sized YOLOv8 models like YOLOv8m.In this study, the model was trained on the COCO dataset, resulting in a computational workload of 39.6G and a parameter count of 25.9M. Thirteen pruning iterations were conducted with different pruning rates to systematically reduce the model's parameter count and identify the optimal pruned model. Comparative analysis with the unpruned model showed promising results: the computational workload decreased from 39.6G to 33.7G, a reduction of 14.9%; the parameter count decreased from 25.9M to 22.0M, a reduction of 15%; the average precision improved from 0.6 before pruning to 0.7 after fine-tuning the pruned model parameters; and the inference time per image decreased from 10.2ms before pruning to 9.5ms afterward.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Dengji Zhang, Sunyan Hong, Weiwei Gao, Jing Yu, Zhenting Chen, Xutian Shen, and Jianxiong Wu "Optimization of YOLOv8 model based on pruning", Proc. SPIE 13416, Fourth International Conference on Advanced Algorithms and Neural Networks (AANN 2024), 134162B (8 November 2024); https://doi.org/10.1117/12.3049687

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
7 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Performance modeling

Data modeling

Mathematical optimization

Education and training

Object detection

Instrument modeling

Neural networks

Show All Keywords

Keywords/Phrases

Search In:

Publication Years