Hyperparameter Optimization of YOLO Model Based on Orthogonal Optimization Strategy

Qing-hua YANG , Guan-ci YANG , Shi-hao ZHONG

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (4) : 1573 -1579.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (4) : 1573-1579 DOI: 10.12404/j.issn.1671-1815.2309596
research-article
Hyperparameter Optimization of YOLO Model Based on Orthogonal Optimization Strategy
Author information +
History +
PDF

Abstract

In order to realize the automatic optimization of hyperparameters of YOLO model, the hyperparameter optimization of you only look once (YOLO) model based on orthogonal optimization strategy (OOS) was proposed. Firstly, based on the principle of statistical orthogonal test, the orthogonal search method of population and the hyperparameter contribution analysis strategy were proposed to improve the optimization efficiency of the algorithm. Then, the uniform orthogonal search strategy and the neighborhood orthogonal search strategy were designed to alleviate the problem of the YOLO model falling into the local optimum and premature convergence. Finally, YOLOv5, YOLOv5s-Transformer and YOLOv7 were used as optimization objects to test on two target detection datasets, NWPU VHR-10 and Pascal VOC. Test results show that the recognition accuracy of the YOLO model is improved by the OOS hyperparameter optimization method in all cases. The average recognition accuracy mAP@0.5 on two datasets is improved to 93.94%, 93.18%, 93.45%, and 85.81%, 84.59%, 89.96%. The mAP@0.5-0.95 is improved to 60.00%, 60.08%, 56.98%, and 62.27%, 58.89%, 70.77%. It can provide a new intelligent method for hyperparameter optimization of object detection model.

Keywords

object detection / hyperparameter optimization / regularization strategy / YOLO

Cite this article

Download citation ▾
Qing-hua YANG, Guan-ci YANG, Shi-hao ZHONG. Hyperparameter Optimization of YOLO Model Based on Orthogonal Optimization Strategy. Science Technology and Engineering, 2025, 25(4): 1573-1579 DOI:10.12404/j.issn.1671-1815.2309596

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (4153KB)

19

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉