Optimization of Hyperparameters in Object Detection Models Based on Fractal Loss Function

Abstract

Hyperparameters involved in neural networks (NNs) have a significant impact on the accuracy of model predictions. However, the values of the hyperparameters need to be manually preset, and finding the best hyperparameters has always puzzled researchers. In order to improve the accuracy and speed of target recognition by a neural network, an improved genetic algorithm is proposed to optimize the hyperparameters of the network by taking the loss function as the research object. Firstly, the role of all loss functions in object detection is analyzed, and a mathematical model is established according to the relationship between loss functions and hyperparameters. Secondly, an improved genetic algorithm is proposed, and the feasibility of the improved algorithm is verified by using complex fractal function and fractional calculus. Finally, the improved genetic algorithm is used to optimize the hyperparameters of the neural network, and the prediction accuracy of the model before and after the improvement is comprehensively analyzed. By comparing with state-of-the-art object detectors, our proposed method achieves the highest prediction accuracy in object detection. Based on an average accuracy rate of 95%, the detection speed is 20 frames per second, which shows the rationality and feasibility of the optimized model.