DCGAN-CNN在滚动轴承故障诊断的应用研究

Application Research of DCGAN-CNN in Fault Diagnosis of Rolling Bearings

  • 摘要: 针对滚动轴承数据集样本稀缺导致的模型过拟合问题,本研究提出一种融合深度卷积生成对抗网络与卷积神经网络的两阶段诊断模型框架(DCGAN-CNN)。首先构建具有梯度惩罚机制的DCGAN架构,通过对抗训练生成符合真实数据分布的一维振动信号,其次设计多尺度特征融合CNN分类器,利用生成数据增强模型泛化能力。实验表明,在数据样本不足的条件下,本模型诊断准确率达97.9%,较传统重叠采样方法提升1.2%,证明该模型框架在滚动轴承故障诊断领域的优势。

     

    Abstract: Aiming at the problem of model overfitting caused by scarce samples in rolling bearing datasets, this study proposes a two-stage fault diagnosis model framework (DCGAN-CNN) that integrates deep convolutional generative adversarial networks and convolutional neural networks. First, a DCGAN structure with a gradient penalty mechanism is constructed to generate one-dimensional vibration signals consistent with the real data distribution via adversarial training. Then, a multi-scale feature fusion CNN classifier is designed to improve the generalization ability of the model using the generated data. Experimental results show that the diagnostic accuracy of the proposed model reaches 97.9% under insufficient data samples, which is 1.2% higher than that of the traditional overlapping sampling method. This verifies the advantages of the model framework in the field of rolling bearing fault diagnosis.

     

/

返回文章
返回