Fault Diagnosis Method Based on Xgboost and LR Fusion Model under Data Imbalance

Liling Ma,Tianyi Wang,Xiaoran Liu,Junzheng Wang,Wei Shen
DOI: https://doi.org/10.15918/j.jbit1004-0579.2021.050
2022-01-01
Journal of Beijing Institute of Technology
Abstract:Diagnosis methods based on machine learning and deep learning are widely used in the field of motor fault diagnosis. However, due to the fact that the data imbalance caused by the high cost of obtaining fault data will lead to insufficient generalization performance of the diagnosis method. In response to this problem, a motor fault monitoring system is proposed, which includes a fault diagnosis method (Xgb_LR) based on the optimized gradient boosting decision tree (Xgboost) and logistic regression (LR) fusion model and a data augmentation method named data simulation neighborhood interpolation(DSNI). The Xgb_LR method combines the advantages of the two models and has positive adaptability to imbalanced data. Simultaneously, the DSNI method can be used as an auxiliary method of the diagnosis method to reduce the impact of data imbalance by expanding the original data (signal). Simulation experiments verify the effectiveness of the proposed methods.
What problem does this paper attempt to address?