1673-159X

CN 51-1686/N

基于GRU神经网络的柴油机排放特性预测

Prediction of Diesel Engine Emission Characteristics Based on GRU Neural Network

  • 摘要: 为提高柴油机排放特性预测精度及减少试验成本,提出一种基于门控循环单元网络(GRU)的柴油机排放特性预测方法。首先利用随机森林算法进行输入变量的重要性分析,确定各排放参数的输入变量,然后通过粒子群优化算法对GRU预测模型进行超参数自动寻优,基于寻优后的最优超参数对模型进行训练和验证,在训练集上NOx、CO、CO2、THC、排气温度和排气压力的决定系数(R2)分别为0.985、0.994、0.991、0.986、0.979和0.998,在验证集上分别为 0.962、0.969、0.987、0.954、0.975和0.992,最后对GRU预测模型进行泛化性测试,并与其他神经网络预测模型进行对比。结果表明在测试集上其R2平均值为0.930,相较于训练集和验证集分别下降了5.9%和4.3%,说明GRU预测模型具有极强的泛化能力。在与其他预测模型的对比中,GRU预测模型也表现出更好的性能特性。

     

    Abstract: To improve the accuracy of diesel engine emission predictions and reduce experimental costs, a method based on a gated recurrent unit (GRU) network was proposed. First, the importance of input variables was analyzed using the random forest algorithm, and the input variables for each emission parameter were determined. Then, the hyperparameters of the GRU prediction model were automatically optimized using the particle swarm optimization algorithm, and the model was trained and validated using the optimal hyperparameters. In the training set, the coefficients of determination (R2) for NOx, CO, CO2, THC, exhaust temperature, and exhaust pressure are 0.985, 0.994, 0.991, 0.986, 0.979, and 0.998, respectively. The corresponding values in the validation set are 0.962, 0.969, 0.987, 0.954, 0.975, and 0.992, respectively. Finally, the generalization performance of the GRU prediction model was tested, and the model was compared with other neural network prediction models. The results show that the average R2 value is 0.930 in the test set, which is 5.9% and 4.3% lower than the average values for the training and validation sets, respectively, indicating that the GRU prediction model has excellent generalization ability. Compared with other prediction models, the GRU model also demonstrates superior predictive performance.

     

/

返回文章
返回