Compression Method of Rail Surface Defect Detection Model Combined with Bayesian Optimization
Aiming at the problem of cumbersome manual parameter adjustment and large model capacity of rail surface defect detection model,a compression method of rail surface defect detection model incorporating Bayesian optimiza-tion is proposed.Firstly,the Bayesian optimization algorithm is used to obtain the optimal hyperparameters that bal-ance the sparsity and accuracy of the model;secondly,the obtained hyperparameters are introduced into the sparse training and the detection model is compressed under two strategies focusing on the compression ratio and the accu-racy;finally,the parameters of the compressed model are corrected through the knowledge distillation to compensate for the loss of accuracy caused by the sparse training.The experimental results show that the compression ratio of the lightweight track surface detection model obtained by this method under the two sparse training strategies can reach 96.35％and 93.22％,respectively,and the detection speed is improved by more than two times after the hard-ware is deployed,which can avoid the negative impact of manual parameterization on the compression accuracy.
rail surface defectsdefect detection modelmodel compressionBayesian optimizationsparse training
万方数据
维普
