基于Nadam-TimeGAN和XGBoost的时序信号故障诊断方法  

Fault diagnosis method of timing signal based on Nadam-TimeGAN and XGBoost

作　　者：黑新宏[1,2] 高苗 张宽[1] 费蓉 邱原[1,2] 姬文江 HEI Xinhong;GAO Miao;ZHANG Kuan;FEI Rong;QIU Yuan;JI Wenjiang(School of Computer Science and Engineering,Xi'an University of Technology,Xi'an 710048,China;Shaanxi Key Laboratory of Network Computing and Security Technology,Xi'an 710048,China)

机构地区：[1]西安理工大学计算机科学与工程学院,陕西西安710048 [2]陕西省网络计算与安全技术重点实验室,陕西西安710048

出　　处：《通信学报》2024年第4期185-200,共16页Journal on Communications

基　　金：国家自然科学基金资助项目(No.62120106011,No.U20B2050);国家重点研发计划基金资助项目(No.2022YFB2602203)。

摘　　要：为了提高故障诊断模型在数据不平衡场景下的诊断性能和模型泛化能力,提出了一种基于Nadam-TimeGAN和XGBoost的时序信号故障诊断方法。首先对比基于LSTM和GRU的TimeGAN模型,选取性能更优的GRU网络作为TimeGAN模型的组成单元,然后采用Nadam优化算法对TimeGAN模型的各组件进行优化,即构建Nadam-TimeGAN模型用以数据扩充,最后构建一个平衡的数据集输入XGBoost集成学习模型进行分类训练。实验选取转辙机动作电流数据集进行验证性实验,选取MFPT轴承数据集和CWRU轴承数据集进行泛化性实验,并与8种方法进行对比,结果表明,所提方法在准确率、召回率以及F1-score这3种评价指标上均高于其他方法,从而验证了所提方法在不平衡数据故障诊断方面的有效性和泛化性。In order to improve the diagnostic performance and model generalization ability of the fault diagnosis model in data imbalance scenarios,a time series signal fault diagnosis method based on Nadam-TimeGAN and XGBoost was proposed.Firstly,the TimeGAN model based on LSTM and GRU was compared,and the GRU network with better performance was selected as the component unit of the TimeGAN model.The Nadam optimization algorithm was used to optimize the components of the TimeGAN model,that was,the Nadam-TimeGAN model was constructed for data expansion.After data expansion,a balanced data set was constructed and input into the XGBoost integrated learning model for classification training.In the experiment,the action current data set of switch machine was selected for verification experiment,the MFPT bearing data set and the CWRU bearing data set were selected for generalization experiment,and compared with eight methods.The results show that the proposed method is higher than other methods in accuracy,recall and F1-score.The experimental results validate the effectiveness and generalization of the proposed method for imbalanced data fault diagnosis.

关 键 词：时间序列生成对抗网络 Nesterov加速自适应矩估计 极致梯度提升 故障诊断 数据增强 

分 类 号：TP18[自动化与计算机技术—控制理论与控制工程]