简献忠, 张韬
控制工程. 2024, 31(4): 729-737.
在实际工程场景中,轴承故障数据远少于正常数据,不平衡数据下的轴承故障诊断方法存在参数多、模型复杂的问题。因此,提出了一种由带梯度惩罚的辅助分类瓦瑟斯坦生成对抗网络(auxiliary classier Wasserstein generative adversarial network with gradient
penalty, ACWGAN-GP)和神经回路策略故障诊断(neural circuit policy-fault diagnosis, NCP-FD)
网络组成的不平衡数据 NCP-FD(unbalanced data NCP-FD, UDNCP-FD)模型。首先,将轴
承故障信号数据转换为二维时频图,利用不平衡的训练集训练 ACWGAN-GP 生成高质量故障样本,并将其扩充到原始训练集中;然后,将扩充后的训练集输入到 NCP-FD 网络中进行学习;最后,将训练好的 NCP-FD 网络应用于测试集进行故障诊断。实验结果表明,
所提模型与其他不平衡数据下的轴承故障诊断模型相比,参数更少,存储空间更小,故障诊断准确率更高。