一种基于CNN和LSTM的滚动轴承故障分析

技术领域

本申请属于机器学习和故障识别技术领域，特别是涉及深度学习模型CNN与深度学习模型LSTM的滚动轴承故障分析方法。

背景技术

滚动轴承故障诊断技术一直以来都是机械故障诊断领域中的重要研究课题。列车作为一种复杂机械设备广泛应用于公共出行。列车机械系统出现故障，干扰交通系统的运作，严重时更会危及人员的安全，造成巨大的经济损失。滚动轴承是列车的核心零部件。据统计，30％的机械设备故障是由轴承故障引起的，对滚动轴承微弱故障的早期检测分析具有重要的现实意义和实用价值。不同的故障类型具有不同的信号特征，振动信号能够直观地反映出轴承的健康状态，因此应用较为广泛。信号特征主要包表现形式分为两种：时域和频域，然而单纯的通过时域分析或频域分析无法完全地表现振动信号中的故障特征。近年来，卷积神经网络(convolutional neural network，CNN) 被广泛应用于图像处理、语音识别、故障分析等领域中。在轴承故障诊断中，将一维时序数据或二维时频图作为输入，通过神经网络进行训练提取故障特征被证明可以在有噪声变工况的场景下对故障进行分析。

Jing等利用CNN从振动信号的频谱中学习特征，实现变速箱的健康状态监测。Qu等提出了一维卷积神经网络故障诊断算法，实现了基于深度网络的自适应特征提取和故障诊断。SE等从电动机的频谱中通过快速傅里叶变换使用频率分量训练CNN。Zhuang等使用长短时记忆网络(Long Short-Term Memory，LSTM)网络通过栅极结构提取时序数据的时间特征，增强模型的泛化能力。Ren等结合时域和频域特征提出一种多轴承剩余使用寿命协同预测的集成深度学习方法，获取了良好的预测效果。Liu 使用CNN+LSTM模型并通过一个激活层映射成一个寿命指标，进而对轴承寿命进行预测。

虽然CNN模型能够充分提取数据的空间特征，但不能提取数据的时间特征。在数据量较大的情况下，LSTM网络又难以提取数据的非线性特征，数据特征提取不足，收敛速度也较慢。且相较于将时域特征或频域特征作为输入的模型，结合时域和频域特征的模型会表现出更优秀的分类效果。

发明内容

本发明的目的是提供一种基于CNN和LSTM的滚动轴承故障分析方法。充分利用CNN的空间特征提取能力和LSTM的时序特征学习能力，充分提取振动图像与时间依赖的关系，通过全连接层和 softmax层对滚动轴承振动特征进行分类识别和故障判断，增强分类精度。且LSTM相比于传统的RNN，解决了梯度消失的问题，能降低训练模型的难度。

根据本申请的一个方面，提供了一种基于CNN和LSTM的轴承故障分析方法，包括：

获取轴承振动数据，对含有噪声的振动数据采用巴特沃思滤波器 (Butterworthfilter)处理呼吸频率信号。

其中巴特沃斯滤波器的公式为：

其中n＝滤波器的阶数，ω

通过快速傅里叶变换(fast Fourier transform，FFT)将预处理信号的描述从时域转换到频域；

FFT利用离散信号计算信号X(k)的频谱。其中FFT的公式为：

即使用FFT算法可以降低频域转换计算量、提高转换速度，其时间复杂度为o(nolg

对处理后得到的时域图、频域图作为输入，分别通过预设好的 CNN网络进行训练，CNN只对一维时间轴上的数据进行卷积运算，沿时域信号和时频图的时间t轴移动提取图像特征。CNN通过3个卷积层来进行特征提取。池化层位于每个卷积层的后面对特征图进行降维，这里是通过使用最大池化操作减少输出的复杂度和防止数据的过拟合，保留数据的时间特征；

其中最大池化层计算公式为：

其中w、d为最大池化滤波器预设的长和宽。

将CNN学习得到的时域图、频域图特征通过add层进行特征融合，图像的维度本身并没有增加，每一维下的信息量增加了，通过add 层进行特征融合可以保留时间特征。将融合特征作为长短时记忆网络层的输入；

将add层特征融合后得到的融合特征输入到长短时记忆网络层，长短时记忆网络的各层之间以串联的方式连接，以含有时序的数据作为输入，获取时空特征；长短时记忆网络层内部存在3个门：遗忘门、输入门、输出门，每个时刻t门的更新公式如下：

遗忘门f

f

输入门i

c

c

输出门o

o

h

其中，W

将获取的时空特征作为输入，在全连接层中将其映射到样本标记空间，并通过后续的softmax层运算得到一个分类概率的结果，对滚动轴承故障进行识别和分类。

其中softmax函数的公式为：

表示当存在K个线性函数时，样本向量x

附图说明

后文将参照附图以示例性而非限制性的方式详细描述本申请的一些具体实施例。附图中相同的附图标记标示了相同或类似的部件或部分。本领域技术人员应该理解，这些附图未必是按比例绘制的。附图中：

图1是本申请方法一种基于CNN和LSTM的滚动轴承故障分析方法的的流程示意图；

图2是根据本申请一个实施例的一种基于CNN和LSTM的滚动轴承故障分析方法的示意性结构框图。

图3是本申请实施例提供的一种计算设备。

图4是本申请实施例还提供的一种计算机可读存储介质。

具体实施方式

图1是根据本申请实施例的一种基于CNN和LSTM的滚动轴承故障分析方法流程示意图。参见图1所知，本申请实施例提供的一种基于CNN和LSTM的滚动轴承故障分析方法及系统可以包括：

步骤S1：对处理后的振动数据使用巴特沃斯滤波器(Butterworth Filter)去噪声，并进行快速傅里叶变换(FFT)将预处理的时域信号转化为频域信号；

步骤S2：利用CNN网络进行学习来获取时域图、频域图的图像特征；

步骤S3：通过add层进行图像特征融合；

步骤S4：将add层获得的融合图像特征输入到LSTM网络中，通过LSTM网络进一步学习特征中包含的时序特征；

步骤S5：通过一个全连接层和Softmax函数实现分类功能，用训练好的网络对测试样本进行故障分类；

本发明的目的是提供一种基于CNN和LSTM的滚动轴承故障分析方法。首先对数据集进行噪声过滤，将振动数据用时域图和时频图两种方式表示，更好的进行特征提取。利用CNN的空间特征提取能力对时频图和时域图分别提取振动特征，并进行特征融合，利用LSTM的时序特征学习能力，充分提取振动特征与时间依赖的关系，通过全连接层和Softmax层对滚动轴承振动特征进行分类识别和故障判断，增强分类精度。

所述方法采用的实验数据集来源为CWRU(凯斯西储大学轴承数据中心)，该数据集是目前国际上使用最为广泛的滚动轴承故障数据集。该数据集记录了电动机的实际测试条件以及轴承故障状态，使用放电加工技术(EDM)向电机轴承植入故障。在轴承内圈，滚动体和轴承外圈上分别引入直径范围从0.007英寸到0.040英寸的断层。将有故障的轴承重新安装到测试电动机中，轴承实验以12,000个样本/秒和48,000个样本/秒的速度记录0到3马力(电动机速度为1797 到1720RPM)的振动数据。

S1、获取轴承振动数据，对含有噪声的振动数据采用巴特沃思滤波器(Butterworth filter)处理呼吸频率信号。

其中巴特沃斯滤波器的公式为：

其中n＝滤波器的阶数，ω

通过快速傅里叶变换(fast Fourier transform，FFT)将预处理信号的描述从时域转换到频域；

FFT利用离散信号计算信号X(k)的频谱。其中FFT可以表示为：

即使用FFT算法可以降低频域转换计算量、提高转换速度，其时间复杂度为o(nlog

S2、对处理后得到的时域图、频域图作为输入，分别通过预设好的CNN网络进行训练，CNN只对一维时间轴上的数据进行卷积运算，沿时域信号和时频图的时间t轴移动提取图像特征。CNN通过3 个卷积层来进行特征提取。池化层位于每个卷积层的后面对特征图进行降维，这里是通过使用最大池化操作减少输出的复杂度和防止数据的过拟合，保留数据的时间特征；

其中最大池化层计算公式为：

其中w、d为最大池化滤波器预设的长和宽。

S3、将CNN学习得到的时域图、频域图特征，通过add层进行特征融合，图像的维度本身并没有增加，每一维下的信息量增加了，通过add层进行特征融合可以保留时间特征。将融合特征作为长短时记忆网络层的输入；

S4、将Add层特征融合后得到的融合特征输入到长短时记忆网络层，长短时记忆网络的各层之间以串联的方式连接，以含有时序的数据作为输入，获取时空特征；长短时记忆网络层内部存在3个门：遗忘门、输入门、输出门，每个时刻t门的更新公式如下：

遗忘门f

f

输入门i

c

c

输出门o

o

h

其中，W

S5、将获取的时空特征作为输入，在全连接层中将其映射到样本标记空间，并通过后续的softmax层运算得到一个分类概率的结果，对滚动轴承故障进行识别和分类。

其中softmax函数的公式为：

表示当存在K个线性函数时，样本向量x

图2是根据本申请一个实施例的一种基于CNN和LSTM的滚动轴承故障分析方法的示意性结构框图。

图3中本申请实施例还提供了一种计算设备，该计算设备包括存储器320、处理器310和存储在所述存储器320内并能由所述处理器 310运行的计算机程序，该计算机程序存储于存储器320中的用于程序代码的空间330，该计算机程序在由处理器310执行时实现用于执行任一项根据本发明的方法步骤331。

图4中本申请实施例还提供了一种计算机可读存储介质。该计算机可读存储介质包括用于程序代码的存储单元，该存储单元设置有用于执行根据本发明的方法步骤的程序331′，该程序被处理器执行。

本申请实施例还提供了一种包含指令的计算机程序产品。当该计算机程序产品在计算机上运行时，使得计算机执行根据本发明的方法步骤。

在上述实施例中，可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时，可以全部或部分地以计算机程序产品的形式实现。所述计算机程序产品包括一个或多个计算机指令。在计算机加载和执行所述计算机程序指令时，全部或部分地产生按照本申请实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、获取其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中，或者从一个计算机可读存储介质向另一个计算机可读存储介质传输，例如，所述计算机指令可以从一个网站站点、计算机、服务器或数据中心通过有线(例如同轴电缆、光纤、数字用户线(DSL))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、服务器或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存取的任何可用介质或者是包含一个或多个可用介质集成的服务器、数据中心等数据存储设备。所述可用介质可以是磁性介质，(例如，软盘、硬盘、磁带)、光介质(例如，DVD)、或者半导体介质(例如固态硬盘Solid State Disk(SSD)) 等。

专业人员应该还可以进一步意识到，结合本文中所公开的实施例描述的各示例的单元及算法步骤，能够以电子硬件、计算机软件或者二者的结合来实现，为了清楚地说明硬件和软件的可互换性，在上述说明中已经按照功能一般性地描述了各示例的组成及步骤。这些功能究竟以硬件还是软件方式来执行，取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能，但是这种实现不应认为超出本申请的范围。

本领域普通技术人员可以理解实现上述实施例方法中的全部或部分步骤是可以通过程序来指令处理器完成，所述的程序可以存储于计算机可读存储介质中，所述存储介质是非短暂性(英文：non- transitory)介质，例如随机存取存储器，只读存储器，快闪存储器，硬盘，固态硬盘，磁带(英文：magnetic tape)，软盘(英文：floppy disk)，光盘(英文：optical disc)及其任意组合。

以上所述，仅为本申请较佳的具体实施方式，但本申请的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本申请揭露的技术范围内，可轻易想到的变化或替换，都应涵盖在本申请的保护范围之内。因此，本申请的保护范围应该以权利要求的保护范围为准。