A Feature Selection-Based XGBoost Model for Fault Prediction

Laixin Zhou,You Li,Baohua Qiang,Ran Duan
DOI: https://doi.org/10.1109/CIS54983.2021.00056
2021-11-01
Abstract:At present, more and more researches focus on fault prediction of industrial equipments and medical equipments. But the existing approaches cannot effectively deal with the problem of large amount of noisy data and low value density from massive equipments status information, so the prediction accuracy is lower. To cope with these problems, we propose an extreme gradient boosting (XGBoost) algorithm based on random forest feature selection to predict the fault of wind turbine of industrial equipments. Firstly, we establish a complete data processing flow for wind turbine status data, which includes data acquisition, data imbalance processing, feature selection and other operations. Then, the XGBoost model for fault prediction of wind turbine is constructed based on feature selection. Compared with the traditional machine learning algorithms, our proposed approach performs best on the dataset and the fault prediction accuracy of wind turbine can reach 98%, which demonstrates the effectiveness of our proposed algorithm. In the following research, we will conduct in-depth study on fault prediction for medical equipments of Southwest Hospital of AMU based on our proposed approach.
Engineering,Computer Science
What problem does this paper attempt to address?