Optimizing YOLOv5s Object Detection Through Knowledge Distillation Algorithm

This paper explores the application of knowledge distillation technology in target detection tasks, especially the impact of different distillation temperatures on the performance of student models. By using YOLOv5l as the teacher network and a smaller YOLOv5s as the student network, we found that with the increase of distillation temperature, the student's detection accuracy gradually improved, and finally achieved mAP50 and mAP50-95 indicators that were better than the original YOLOv5s model at a specific temperature. Experimental results show that appropriate knowledge distillation strategies can not only improve the accuracy of the model but also help improve the reliability and stability of the model in practical applications. This paper also records in detail the accuracy curve and loss function descent curve during the model training process and shows that the model converges to a stable state after 150 training cycles. These findings provide a theoretical basis and technical reference for further optimizing target detection algorithms.

Authors

Institutions

Publication Details

Published
2024-10-25
DOI
https://doi.org/10.1109/icbaie63306.2024.11117021
Citations
2
Primary Topic
Industrial Vision Systems and Defect Detection
Type
article
Field-Weighted Citation Impact
0.70
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Optimizing YOLOv5s Object Detection Through Knowledge Distillation Algorithm

Junliang Du, Jiacheng Hu, Aoran Shen, Yingbin Liang et al.
2 citations
Industrial Vision Systems and Defect Detection
0.70
article

Optimizing YOLOv5s Object Detection Through Knowledge Distillation Algorithm

Junliang Du, Jiacheng Hu, Aoran Shen, Yingbin Liang, Yuxiang Hu, Guanming Huang
article en
2 citations

Abstract

This paper explores the application of knowledge distillation technology in target detection tasks, especially the impact of different distillation temperatures on the performance of student models. By using YOLOv5l as the teacher network and a smaller YOLOv5s as the student network, we found that with the increase of distillation temperature, the student's detection accuracy gradually improved, and finally achieved mAP50 and mAP50-95 indicators that were better than the original YOLOv5s model at a specific temperature. Experimental results show that appropriate knowledge distillation strategies can not only improve the accuracy of the model but also help improve the reliability and stability of the model in practical applications. This paper also records in detail the accuracy curve and loss function descent curve during the model training process and shows that the model converges to a stable state after 150 training cycles. These findings provide a theoretical basis and technical reference for further optimizing target detection algorithms.

Tulane University (US), Northeastern University (US), Johns Hopkins University (US), Shanghai Jiao Tong University (CN), University of Michigan (US), University of Chicago (US), Seattle University (US)
Openalex Percentile: Top 23%
Industrial Vision Systems and Defect Detection
0.70
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.