一种基于DCGAN样本补充的滚动轴承故障诊断方法技术

本发明专利技术公开了一种基于DCGAN样本补充的滚动轴承故障诊断方法，所述方法包括以下步骤：首先将一维滚动轴承故障信号通过遗传算法优化后的变分模态分解框架中进行降噪预处理，并对分解后的信号依据峭度准则进行重构，然后对重构信号进行连续小波变换得到二维滚动轴承故障信号，最后，将二维滚动轴承振动信号放入深度生成对抗网络模型中生成辅助样本数据集，并通过卷积神经网络实现小样本、非均衡数据集下的滚动轴承智能诊断。本发明专利技术实现了变分模态分解和卷积神经网络参数的自适应确定，并通过深度生成对抗网络的方式解决了扩充小样本、非平衡数据集导致样本多样性差的问题，该方法可以快速、准确的实现滚动轴承故障类型的识别。准确的实现滚动轴承故障类型的识别。准确的实现滚动轴承故障类型的识别。

【技术实现步骤摘要】
一种基于DCGAN样本补充的滚动轴承故障诊断方法


[0001]本专利技术涉及机械故障
，尤其涉及一种基于DCGAN样本补充的滚动轴承故障诊断方法。

技术介绍

[0002]滚动轴承作为旋转机械应用广泛且容易损坏的零部件，对其进行故障诊断有着重要的理论和实际意义，现有的故障诊断方法在对滚动轴承实测得到的数据往往存在不同类型样本累积数据量不同导致的数据非平衡问题，导致现有方法在非平衡小样本故障诊断场景下识别准确率有限，甚至存在状态误识别，而解决这种问题的方法通常是使用平衡数据集训练卷积神经网络(CNN)与循环神经网络(GAN)相结合的深度生成对抗网络(DCGAN)模型来扩充小样本、非平衡数据集，最终根据得到的实测数据对滚动轴承故障类型进行识别和判断。
[0003]扩充小样本、非平衡数据集最终得到的样本多样性差，从而导致无法快速、准确的实现对滚动轴承故障类型的识别，降低了滚动轴承故障诊断的工作效率。

技术实现思路

[0004]基于
技术介绍
存在的技术问题，本专利技术提出了一种基于DCGAN样本补充的滚动轴承故障诊断方法。...

【技术保护点】

【技术特征摘要】
1.一种基于DCGAN样本补充的滚动轴承故障诊断方法，其特征在于，所述方法包括以下步骤：S1，运用加速度传感器对滚动轴承各故障部位进行一维信号采集，然后按照内圈故障、外圈故障、滚动体故障和正常状态进行标签标记，得到具有样本标签的原始一维滚动轴承数据集；S2，建立最小信息熵函数作为遗传算法(GA)的适应度函数，通过选择、交叉和变异等遗传操作对变分模态分解中分解个数K和二次惩罚因子a进行参数优化，实现参数的自适应确定；S3，对S1中得到的具有样本标签的原始一维滚动轴承数据集放到遗传算法优化后的变分模性分解框架中进行信号降噪处理；S4，将S3得到的若干IMF分量依据峭度准则原理进行信号重构，得到具有明显故障特征的滚动轴承一维故障信号；S5，对S4中的一维重构信号进行连续小波变换，将其转换为二维信号，并作为原始数据集；S6，建立用于滚动轴承故障诊断辅助样本的深度生成对抗神经网络(DCGAN)模型，该模型由判别器和生成器两部分组成，将滚动轴承故障二维数据输入判别器中进行分类打分标准，然后在生成器中输入多维噪声信号，输出滚动轴承故障二维数据的辅助样本，判别器与生成器之间相互博弈，最终达到“纳什均衡”，生成器与判别器博弈原理为：式中，z为生成网络的输入噪声，x为真实样本，G(z)为生成网络的虚假样本，D(G(z))为判别样本真实程度的概率；S7，将S5中的原始二维信号数据集作为原始样本输入S6中的判别器内，深度生成对抗神经网络通过相互对抗“博弈”生成辅助样本；S8，建立用于滚动轴承故障智能诊断的卷积神经网络(CNN)模型，该模型基于Alexnet网络；S9，建立最小交叉熵函数作为遗传算法(GA)的适应度函数，通过选择、交叉和变异等遗传操作对卷积神经网络中迭代次数epoch和批处理大小Batch_size进行参数优化，实现参数的自适应确定；S10，设计通过遗传算法改进卷积神经网络有效性的试验；S11，设计通过深度生成对抗神经网络解决小样本的试验；S12，设计通过深度生成对抗神经网络解决故障类间样本不均衡...

【专利技术属性】
技术研发人员：胡益嘉，顾英，徐振南，张溟晨，
申请(专利权)人：云南农业大学，
类型：发明
国别省市：