基于改进的粒子群优化神经网络粗糙度预测模型

Roughness Prediction Model Based on Improved Particle Swarm Optimization Neural Network

神经网络算法在金属零件加工表面粗糙度预测中有着广泛的运用,但是尚存网络初始化及梯度弥散这类“黑盒”问题。提出一种基于传统PSO-BP框架的改进模型,使用Xavier替代传统高斯分布初始化粒子群,在算法优化器方面使用最新的自适应矩限制取代随机梯度下降算法进行模型参数更新,经实际测试发现其训练集均方误差及其测试集绝对误差相对传统PSO-BP算法均有所降低,可以更有效担任预测任务。

Neural network algorithm is widely used in the prediction of the surface roughness of metal parts,but there are still"black box"problems such as network initialization.An improved model based on the traditional PSO-BP framework is proposed.Xavier is used to replace the traditional standard Gaussian distribution to initialize the particle swarm,and the latest adaptive moment limit is used to replace the random gradient descent algorithm in the algorithm optimizer to update the model parameters.The actual test shows that the mean square error of the training set and the absolute error of the test set are lower than those of the traditional PSO-BP algorithm To be more effective in forecasting tasks.