제목 |
DEVELOPMENT OF A LIGHT AND ACCURATE NOX PREDICTION MODEL FOR DIESEL ENGINES USING MACHINE LEARNING AND XAI METHODS
|
분야 |
Electric, Fuel Cell, and Hybrid Vehicle |
언어 |
English |
저자 |
Jeong Jun Park(Hansung University), Sangyul Lee(Sungkyunkwan University), Seunghyup Shin(Korea Railroad Research Institute), Minjae Kim( Myongji University), Ji Hwan Park( Seoul National University) |
Key Words |
Diesel engines, NOx prediction model, Machine learning, Random forest, XAI |
초록 |
In the previous research, we had developed a Nitrogen Oxides emissions (NOx) prediction model using deep learning methodology. However, it is necessary to develop a lightweight model for practical application. We developed a machine learning model for predicting NOx using Random Forest method to reduce the negligible input features. For comparison, a Base model was developed with all features. The Shapley Additive Explanations method, which can show the influence of the input features, and the Pearson Correlation Coefficient method, which can show the proportional relationship between the output and the input values, were used to choose the dominant features. The final model was determined using the Shapley Additive Explanations method. The final model shows similar prediction performance to the base model, though it has only 11 features (30 % of the Base model). The final model shows 15.76 for the root mean square error and 0.965 for R2 with the test data. By extracting the dominant features influencing NOx prediction, we could develop a lightweight and accurate NOx prediction model.
|
미리보기 |
다운로드
|
원문(PDF) |
다운로드
|