CN112577747B - 一种基于空间池化网络的滚动轴承故障诊断方法 - Google Patents
一种基于空间池化网络的滚动轴承故障诊断方法 Download PDFInfo
- Publication number
- CN112577747B CN112577747B CN202011436394.7A CN202011436394A CN112577747B CN 112577747 B CN112577747 B CN 112577747B CN 202011436394 A CN202011436394 A CN 202011436394A CN 112577747 B CN112577747 B CN 112577747B
- Authority
- CN
- China
- Prior art keywords
- pooling
- samples
- neural network
- training
- network
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Classifications
-
- G—PHYSICS
- G01—MEASURING; TESTING
- G01M—TESTING STATIC OR DYNAMIC BALANCE OF MACHINES OR STRUCTURES; TESTING OF STRUCTURES OR APPARATUS, NOT OTHERWISE PROVIDED FOR
- G01M13/00—Testing of machine parts
- G01M13/04—Bearings
- G01M13/045—Acoustic or vibration analysis
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
- G06F18/2413—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on distances to training or reference patterns
- G06F18/24133—Distances to prototypes
- G06F18/24137—Distances to cluster centroïds
- G06F18/2414—Smoothing the distance, e.g. radial basis function networks [RBFN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
- G06F18/253—Fusion techniques of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/08—Feature extraction
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F2218/00—Aspects of pattern recognition specially adapted for signal processing
- G06F2218/12—Classification; Matching
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02T—CLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
- Y02T90/00—Enabling technologies or technologies with a potential or indirect contribution to GHG emissions mitigation
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Computational Linguistics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Health & Medical Sciences (AREA)
- Software Systems (AREA)
- Evolutionary Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Acoustics & Sound (AREA)
- Testing Of Devices, Machine Parts, Or Other Structures Thereof (AREA)
Abstract
本发明提供了一种基于空间池化网络的滚动轴承故障诊断方法,采集滚动轴承故障状态和正常状态的振动信号,并对采集到的轴承振动信号进行切割形成样本,将样本分为训练集、验证集和测试集;然后将训练集和验证集中的样本输入到卷积神经网络中进行训练并调整其结构;在确定结构的卷积神经网络最后一个“卷积+池化”单元后加入空间池化注意力实现对特征的加权,并加入两层空间池化层和softmax分类器,完成空间池化模型的构建;将训练集和验证集的样本输入到空间池化网络中进行参数更新,并将测试集的样本输入到已经训练好的空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度。
Description
技术领域
本发明涉及轴承故障智能诊断技术领域,尤其是涉及一种基于空间池化网络的滚动轴承故障诊断方法。
背景技术
由于长期运行在重负载、高转速的环境下,滚动轴承成为了旋转机械设备中最易发生故障的零件之一。因此对滚动轴承进行早期故障诊断具有重要意义。滚动轴承的故障诊断实质上是一个模式识别过程,当轴承发生故障时,振动信号各频带能量发生变化。近年来,随着机器学习研究的不断深入,采用数据驱动的诊断方法取得较大进展,如支持向量机、随机森林、深度信念网络等。这种故障诊断方法的主要思想是先利用信号处理方法提取数据中的相关特征,然后通过特定的分类模型实现轴承故障诊断。然而人工提取特征工作量大、需要领域内专家知识、带有主观因素,在实际应用中,模型的诊断准确率和泛化能力往往会受到影响。为了摆脱人工提取特征带来的限制,深度学习方法在轴承故障诊断中逐步得到应用。常见的方法是利用卷积神经网络或者循环神经网络自适应地提取轴承振动信号的有效特征,最后由Softmax分类器实现故障分类。这类方法不需要人工提取特征,可以自动学习特征提取而不需要大量的先验知识,但是在特征选择以及训练稳定性上仍然存在缺陷,无法实现所提取特征的层次性和差异性的区分。
发明内容
本发明提出了一种基于空间池化网络的滚动轴承故障诊断方法,是一种将卷积神经网络与注意力模块相结合的滚动轴承故障诊断模型,实现了自适应特征提取和筛选的滚动轴承故障诊断。
本发明采用的技术方案如下:
一种基于空间池化网络的滚动轴承故障诊断方法,包括以下步骤:
步骤S1:采集滚动轴承在各故障状态和正常状态下的振动信号,不同类型的轴承振动信号数据作为一个域赋予不同的标签,对轴承振动信号数据进行切割形成样本并划分为训练集、验证集和测试集;
步骤S2:将所述训练集和所述验证集中的样本输入到卷积神经网络中进行训练,所述卷积神经网络由多个“卷积+池化”组成,根据验证集诊断准确率和损失函数值调整所述卷积神经网络结构;
步骤S3:在调整后确定结构的卷积神经网络最后一个“卷积+池化”单元后,设置空间池化注意力机制实现对特征的加权,加权后的特征继续通过两层空间池化层实现对特征信息的整合,最后加入softmax分类器完成空间池化网络的构建;
步骤S4:将所述训练集和所述验证集中的样本输入到所述空间池化网络中进行参数更新,当目标函数满足迭代终止条件时停止更新;
步骤S5:将所述测试集的样本输入到已经训练好的所述空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度。
所述步骤S2具体包括以下步骤:
步骤S21:将步骤S1中的所述训练集和所述验证集样本输入到初始化的卷积神经网络模型,所述初始化的卷积神经网络模型为典型卷积神经网络;典型卷积神经网络由输入层,卷积层,池化层,全连接层和输出层组成,卷积层和池化层两者之间交替设置,通常取多个“卷积+池化”单元提取输入的多种特征;卷积层使用一定大小的卷积核对输入特征面的局部区域进行卷积运算,通过非线性激活函数后输出多个特征面,同一输入特征面与同一输出特征面中共用相同的卷积核,以此实现权值共享,易于训练。池化层通常紧跟卷积层之后,同样有多个特征面组成,它的每一个特征面与上一层卷积层输出的特征面唯一对应,不改变特征面的个数,池化层对局部接受域进行下采样操作。其数学模型可表述为:
式中,f()为池化函数,本方案选用最大池化方法,down()表示下采样函数;
步骤S22:根据所述测试集和所述验证集的诊断准确率及损失函数调整网络结构,最终选择的卷积神经网络结构由四个“卷积+池化”单元和一个全连接层和softmax分类器组成。
所述步骤S3具体包括以下步骤:
步骤S31:分别在步骤S22中确定好的卷积神经网络的不同层加入空间池化注意力机制,由训练集和验证集的诊断准确率和损失函数确定空间池化注意力机制设置在卷积神经网络最后一个“卷积+池化”单元后,效果最佳。
所述注意力机制是通过计算注意力的概率分布,对关键性输入分配更多的权重,从而起到对模型的优化作用。
所述空间池化注意力机制是通过最大池化和均值池化实现特征在空间上的压缩,并通过sigmoid函数计算得出注意力的概率分布,设F为m×n维特征矩阵,其中m为空间维度,n为通道维度,空间池化注意力机制的数学模型可以表述为:
式中,AvgP()和MaxP()分别代表平均池化和最大池化,MLP()为多层感知机,σ为sigmoid激活函数,M(F)是由特征矩阵F学习得到的空间池化注意力概率矩阵,其大小为n×1,学习得到的注意力概率矩阵与原特征融合的计算公式为:
步骤S32:经过空间池化注意力机制加权后的特征又通过两层空间池化层进行特征信息整合;通常经过多个“卷积+池化”后得到的是高维特征,典型的卷积神经网络通过全连接层进行特征降维和特征信息的再次整合,但是全连接层所需训练参数较多,本文通过两层空间池化层代替全连接层,训练参数大大减少,训练速度和稳定性得到很大提升。
步骤S33:通过两层空间池化的特征输入到Softmax层实现滚动轴承状态分类。
所述步骤S4中,训练时采用Adam优化器,损失函数采用交叉熵损失函数;当训练集和验证集损失函数值都较小且趋于稳定时停止迭代。
所述步骤S5中,将同负载下的测试集的样本输入到已经训练好的空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度,同时用不同负载下的样本进行验证并与其他模型进行对比分析。
所述步骤S1具体包括以下步骤:
步骤S11:首先是采集滚动轴承在滚动体故障B、外圈故障OR和内圈故障IR状态下,选取至少三种不同故障直径尺寸和正常状态下的振动信号,并加入高斯白噪声,不同类型的轴承振动数据作为一个域赋予不同的标签;
步骤S12:取连续的多个数据点为一个样本,采用数据重叠分割的方法进行数据增强。
步骤S13:将得到的样本划分为训练集、验证集和测试集,其中训练集和验证集样本的振动信号用于训练过程,所述测试集用于计算模型精度,不参与训练。
本发明的有益效果如下:
本发明能够在不需要大量先验知识的情况下自动提取特征,并且通过空间池化注意力机制对所提取的特征分配相应的权重,滚动轴承故障诊断准确率更高;
本发明用两层空间池化层代替传统卷积神经网络结构中的全连接层,需要训练的参数大大减少,具有训练速度快、训练过程稳定的优点;
本发明独特的网络结构设计实现了多层次的特征提取和特征选择,在变工况下故障诊断效果依然很好。
附图说明
图1是本发明的流程图。
图2是本发明的空间池化注意力原理图。
图3是本发明的空间池化网络结构图。
图4是本发明的卷积神经网络与空间池化网络在训练过程中的损失函数图。
图5是本发明的卷积神经网络与空间池化网络在训练过程中的诊断准确率图。
图6是本发明的卷积神经网络对各个类别的诊断准确的混淆矩阵率。
图7是本发明的“卷积神经网络+LSTM”模型对各个类别的诊断准确的混淆矩阵率。
图8是本发明的空间池化网络对各个类别的诊断准确的混淆矩阵率。
图9是本发明的卷积神经网络模型、“卷积神经网络+LSTM”模型和空间池化网络模型在不同负载下的诊断准确率图。
具体实施方式
以下结合附图说明本发明的具体实施方式。
如图1所示,本实施例的一种基于空间池化网络的滚动轴承故障诊断方法,包括以下步骤:
步骤S1:采集滚动轴承在9种故障状态和1种正常状态下的振动信号,不同类型的轴承振动数据作为一个域赋予不同的标签,对其进行切割形成样本并划分为训练集、验证集和测试集;具体地包括如下步骤:
步骤S11:采集滚动轴承在滚动体故障B、外圈故障OR和内圈故障IR状态下,故障直径尺寸分别为0.18mm、0.36mm和0.54mm和正常状态下的振动信号,为了提高模型的泛化能力,加入高斯白噪声,信噪比SNR=2,不同类型的轴承振动数据作为一个域赋予不同的标签;
步骤S12:取连续的2048个数据点为一个样本,采用数据重叠分割的方法进行数据增强,计算公式如下:
上式中,L为数据总长度,L1为单个样本数据长度,D为步长,N获得的样本数。本实施例取D=200,总共获得6000个样本;
步骤S13:将步骤S12得到的样本划分为训练集、测试集和验证集,其比例为65%、25%、10%;其中训练集和验证集样本的振动信号用于训练过程,测试集用于计算模型精度,不参与训练。数据集汇总如表1所示:
表1
步骤S2:将训练集和验证集中的样本输入到卷积神经网络中进行训练,卷积神经网络由多个“卷积+池化”组成,根据验证集诊断准确率和损失函数值调整卷积神经网络结构;
具体地包括如下步骤:
步骤S21:将训练集和验证集样本输入到初始化的卷积神经网络模型。
典型卷积神经网络由输入层,卷积层,池化层,全连接层和输出层组成,卷积层和池化层两者之间交替设置,通常取多个“卷积+池化”单元提取输入的多种特征。卷积层使用一定大小的的卷积核对输入特征面的局部区域进行卷积运算,通过非线性激活函数后输出多个特征面,同一输入特征面与同一输出特征面中共用相同的卷积核,以此实现权值共享,易于训练。其数学模型可描述为:
式中,为第l层的第j个特征映射,f()为激活函数,M为输入特征映射个数,/>为l-1层第i个特征映射个数,*为卷积运算符,/>为可训练的卷积核,/>为偏置;
池化层通常紧跟卷积层之后,同样有多个特征面组成,它的每一个特征面与上一层卷积层输出的特征面唯一对应,不改变特征面的个数,池化层对局部接受域进行下采样操作,其数学模型可表述为:
式中,f()为池化函数,本实施例选用最大池化方法,down()表示下采样函数。
步骤S22:根据测试集和验证集的诊断准确率及损失函数调整网络结构,最终选择的卷积神经网络由四个“卷积+池化”单元和一个全连接层和softmax分类器组成。
步骤S3:在调整后确定结构的卷积神经网络最后一个“卷积+池化”单元后设置空间池化注意力机制实现对特征的加权,加权后的特征继续通过两层空间池化层实现对特征信息的整合,最后加入softmax分类器完成空间池化网络的构建;具体地包括如下步骤:
步骤S31:分别在确定好的卷积神经网络的不同层加入空间池化注意力机制,由训练集和验证集的诊断准确率和损失函数确定空间池化注意力机制设置在卷积神经网络最后一个“卷积+池化”单元后效果最佳。
注意力机制的实质是通过计算注意力的概率分布,对关键性输入分配更多的权重,从而起到对模型的优化作用。空间池化注意力机制结构如图2所示,它是通过最大池化和均值池化实现特征在空间上的压缩,并通过sigmoid函数计算得出注意力的概率分布,设F为m×n维特征矩阵,其中m为空间维度,n为通道维度,则空间池化注意力机制的数学模型可以表述为:
式中,AvgP()和MaxP()分别代表平均池化和最大池化,MLP()为多层感知机,σ为sigmoid激活函数,M(F)是由特征矩阵F学习得到的空间池化注意力概率矩阵,其大小为n×1。学习得到的注意力概率矩阵与原特征融合的计算公式为:
步骤S32:经过空间池化注意力机制加权后的特征又通过两层空间池化层进行特征信息整合。通常经过多个“卷积+池化”后得到的是高维特征,典型的卷积神经网络通过全连接层进行特征降维和特征信息的再次整合,但是全连接层所需训练参数较多,以本专利所设计的模型为例,在最后一个“卷积+池化”模块后,数据大小为13×60,Softmax分类器的输入为1×60,若使用全连接层与Softmax分类器进行连接,则全连接层需要训练的参数个数为13×60×60,而本实施例用两层空间池化层代替全连接层的需要训练的参数总个数7×60,由此可见,训练参数数量大大减少,训练速度显著提升,同时也起到防止过拟合的作用,训练过程更加稳定;
步骤S33:通过两层空间池化的特征输入到Softmax层实现滚动轴承状态分类,空间池化网络结构如图3所示。
步骤S4:将训练集和验证集中的样本输入到空间池化网络中进行参数更新,当目标函数满足迭代终止条件时停止更新;
具体地,训练时采用Adam优化器,损失函数采用交叉熵损失函数。当训练集和验证集损失函数值都较小且趋于稳定时停止迭代。
步骤S5:将测试集的样本输入到已经训练好的空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度。
具体地,将同负载下的测试集的样本输入到已经训练好的空间池化网络模型中得到轴承状态类型并与标签进行对比,计算得出诊断精度。同时用不同负载下的样本进行验证并与其他模型进行对比分析。
对比实验选取在轴承故障诊断领域应用较为广泛的四个机器学***均频率、频率标准偏差、信息熵、分形维数等,经过筛选后共计120维特征。SVM的分类策略采用“onevsrest”的,核函数选用高斯核,惩罚系数C取6;RF模型树最大深度取8,数量为400;1D-CNN使用了4个“卷积+池化”单元、两个全连接层和一个Softmax分类层;CNN-LSTM使用4个“卷积+池化”单元提取多维特征,然后输入到双向LSTM模型中,最后用全连接层和Softmax层进行故障分类。在“卷积+池化”单元中,1D-CNN、CNN-LSTM的设置方式与空间池化网络相同。
每个模型诊断准确率的详细结果如表2所示。可以看出,基于人工提取特征的两个模型SVM与RF其故障诊断准确率相对较低,都在94%以下,原因主要在于该类模型的学习能力对所提取特征有较强的依赖而人工提取特征又难以挖掘出信号更全面、更深层的信息。这也突出了基于深度学习自动提取特征的优势,例如:一维卷积神经网络1D-CNN利用多个卷积核以及卷积核之间的叠加提取信号的多角度、多层次的特征,在测试集上实现了97.86%的诊断准确率;CNN-LSTM用“卷积+池化”单元提取短时特征,然后输入到LSTM中挖掘时序数据的内在联系,诊断准确率达到94.90%,相比人工提取特征的模型也有了显著提升。空间池化网络模型将空间池化注意力与1D-CNN结合,实现对特征的权重分配,训练数据集和测试数据集的诊断准确率分别达到99.89%和99.00%,此外,由于用两层空间池化层代替了传统卷积神经网络中的全连接层,训练参数大大减少,在单步训练消耗的时间上也有一定的提升。
表2
图4为本实施例的一维卷积神经网路1D-CNN与空间池化网络训练过程中的损失函数图;图5为本发明的一维卷积神经网路1D-CNN与空间池化网络训练过程中的诊断准确率图。
本实施例中1D-CNN与空间池化网络两者的优化器、mini-batch和学习率设置均相同。可以看出:空间池化网络的诊断准确率在训练刚开始便呈直线式上升,迭代次数为97时,准确率已经达到了90%,迭代次数为975时,模型完成收敛,诊断准确率不再发生变化;而1D-CNN诊断准确率在迭代次数小于1000时增长速度较快(仍显著慢于空间池化网络模型),大于1000时以较为缓慢的速度增长,迭代次数大约为2700时模型完成收敛。由损失函数的变化情况可以看出,相比1D-CNN,空间池化网络不仅损失函数下降速度快,而且训练过程的随机波动较小,训练的更加稳定。由此可知,将空间池化注意力加入到1D-CNN中,强调关键的局部特征,抑制不必要的特征,极大的增强了对特征的表达能力和模型的学习能力;同时用两层空间池化层代替传统卷积神经网络中的全连接层使得空间池化网络训练速度更快、训练过程更加稳定。
为了进一步分析模型在各个类别的预测情况,图6、图7和图8分别展示了1D-CNN、CNN-LSTM和空间池化网络的诊断准确率的混淆矩阵,横纵坐标分别为样本的预测标签和真实标签,将故障表述简化,如“Normal”表示正常状态,“B007”表示滚动体故障且故障尺寸为0.18mm即0.007英寸。
对比可以看出,1D-CNN模型在预测故障状态为B014、B021、IR021时出现错误,其中对B021的诊断准确率最差,仅为80%;
CNN-LSTM模型在预测故障状态为B007、B014、B021、IR014、OR014时均出现错误,其中诊断准确率最低的是B014和B021,分别为81%、76%;
空间池化网络模型仅在预测故障状态为B021时出现错误,准确率为90%,其余的轴承状态均预测正确。
因此,在预测轴承状态为B021时三个模型的诊断准确率均为最差,空间池化网络将该状态10%的样本预测为B007和B014,1D-CNN将该状态12%的样本预测为B007,其余错误预测为IR007、IR014。
图9是本发明的卷积神经网络模型、“卷积神经网络+LSTM”模型和空间池化网络模型在不同负载下的诊断准确率图。分别在负载为1hp、2hp、3hp的轴承数据进行测试,各取1500个样本作为测试集,每个样本同样由连续的2048个数据点组成,并与1D-CNN和CNN-LSTM进行对比。由图可以看出,三个模型在0负载测试集上的故障准确率表现均较好,但在变负载的情况下,CNN-LSTM诊断准确率显著下降,在负载为2hp、3hp时下降到85%以下,难以应用在变负载运行情况的轴承故障诊断;而1D-CNN和空间池化网络相对较稳定,在三个测试集诊断准确率均在91%以上,在负载为1hp、2hp时,空间池化网络模型准确率明显高于1D-CNN,在负载为3hp时低于1D-CNN,但从整体诊断效果来看,空间池化网络在变工况下表现更优。
Claims (3)
1.一种基于空间池化网络的滚动轴承故障诊断方法,其特征在于,包括以下步骤:
步骤S1:采集滚动轴承在故障状态和正常状态下的振动信号,不同类型的轴承振动信号数据作为一个域赋予不同的标签,对轴承振动信号数据进行切割形成样本并划分为训练集、验证集和测试集,具体包括:
步骤S11:首先是采集滚动轴承在至少三种不同故障直径尺寸下滚动体故障B、外圈故障OR、内圈故障IR状态和正常状态下的振动信号,并加入高斯白噪声,不同类型的轴承振动数据作为一个域赋予不同的标签;
步骤S12:取连续的多个数据点为一个样本,采用数据重叠分割的方法进行数据增强;
步骤S13:将得到的样本划分为训练集、验证集和测试集,其中训练集和验证集样本的振动信号用于训练过程,测试集用于计算模型精度,不参与训练;
步骤S2:将所述训练集和所述验证集中的样本输入到卷积神经网络中进行训练,所述卷积神经网络由多个“卷积+池化”组成,根据验证集诊断准确率和损失函数值调整所述卷积神经网络结构,具体包括:
步骤S21:将步骤S1中的所述训练集和所述验证集样本输入到初始化的卷积神经网络模型,所述初始化的卷积神经网络模型为典型卷积神经网络;
步骤S22:根据所述测试集和所述验证集的诊断准确率及损失函数调整网络结构,最终选择的卷积神经网络结构由四个“卷积+池化”单元和一个全连接层和softmax分类器组成;
步骤S3:在调整后确定结构的卷积神经网络最后一个“卷积+池化”单元后,设置空间池化注意力机制实现对特征的加权,加权后的特征继续通过两层空间池化层实现对特征信息的整合,最后加入softmax分类器完成空间池化网络的构建,具体包括:
步骤S31:分别在步骤S22中确定好的卷积神经网络的不同层加入空间池化注意力机制,由训练集和验证集的诊断准确率和损失函数确定空间池化注意力机制设置在卷积神经网络最后一个“卷积+池化”单元之后;所述空间池化注意力机制是通过最大池化和均值池化实现特征在空间上的压缩,并通过sigmoid函数计算得出注意力的概率分布,设为维特征矩阵,其中/>为空间维度,/>为通道维度,空间池化注意力机制的数学模型可以表述为:
式中,和/>分别代表平均池化和最大池化,/>为多层感知机,为/>激活函数,/>是由特征矩阵/>学习得到的空间池化注意力概率矩阵,其大小为/>,学习得到的注意力概率矩阵与原特征融合的计算公式为:
步骤S32:经过空间池化注意力机制加权后的特征又通过两层空间池化层进行特征信息整合;
步骤S33:通过两层空间池化的特征输入到Softmax层实现滚动轴承状态分类;
步骤S4:将所述训练集和所述验证集中的样本输入到所述空间池化网络中进行参数更新,当目标函数满足迭代终止条件时停止更新;
步骤S5:将所述测试集的样本输入到已经训练好的所述空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度。
2.根据权利要求1所述的基于空间池化网络的滚动轴承故障诊断方法,其特征在于,所述步骤S4中,训练时采用Adam优化器,损失函数采用交叉熵损失函数;当训练集和验证集损失函数值都较小且趋于稳定时停止迭代。
3.根据权利要求2所述的基于空间池化网络的滚动轴承故障诊断方法,其特征在于,所述步骤S5中,将同负载下的测试集的样本输入到已经训练好的空间池化网络中得到轴承状态类型并与标签进行对比,计算得出诊断精度,同时用不同负载下的样本进行验证并与其他模型进行对比分析。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011436394.7A CN112577747B (zh) | 2020-12-07 | 2020-12-07 | 一种基于空间池化网络的滚动轴承故障诊断方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011436394.7A CN112577747B (zh) | 2020-12-07 | 2020-12-07 | 一种基于空间池化网络的滚动轴承故障诊断方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112577747A CN112577747A (zh) | 2021-03-30 |
CN112577747B true CN112577747B (zh) | 2023-08-08 |
Family
ID=75131062
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011436394.7A Active CN112577747B (zh) | 2020-12-07 | 2020-12-07 | 一种基于空间池化网络的滚动轴承故障诊断方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112577747B (zh) |
Families Citing this family (25)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113155464B (zh) * | 2021-03-31 | 2022-03-18 | 燕山大学 | 面向轴承故障识别的cnn模型可视化优化方法 |
CN112884093B (zh) * | 2021-04-29 | 2021-08-31 | 四川大学 | 基于dscrn模型的旋转机械故障诊断方法、设备和存储介质 |
CN113255882B (zh) * | 2021-04-30 | 2022-11-04 | 南通大学 | 一种基于改进卷积胶囊网络的轴承故障诊断方法 |
CN113281029B (zh) * | 2021-06-09 | 2022-03-15 | 重庆大学 | 一种基于多尺度网络结构的旋转机械故障诊断方法及*** |
CN113469060A (zh) * | 2021-07-02 | 2021-10-01 | 浙大城市学院 | 多传感器融合卷积神经网络航空发动机轴承故障诊断方法 |
CN113469281B (zh) * | 2021-07-22 | 2023-11-24 | 西北工业大学 | 一种工业齿轮箱多源信息融合故障诊断方法 |
CN113567131B (zh) * | 2021-07-29 | 2024-05-03 | 沈阳建筑大学 | 基于s变换与微型卷积神经网络模型的轴承故障诊断方法 |
CN113705424A (zh) * | 2021-08-25 | 2021-11-26 | 浙江工业大学 | 基于时间卷积降噪网络的演艺装备故障诊断模型构建方法 |
CN113852507A (zh) * | 2021-09-29 | 2021-12-28 | 济南浪潮数据技术有限公司 | 一种预测网络健康状态的方法、***、设备和存储介质 |
CN114065804A (zh) * | 2021-10-25 | 2022-02-18 | 上海电力大学 | 电励磁双凸极电机绕组匝间短路故障诊断方法和*** |
CN114118142A (zh) * | 2021-11-05 | 2022-03-01 | 西安晟昕科技发展有限公司 | 一种雷达脉内调制类型的识别方法 |
CN114088405A (zh) * | 2021-11-10 | 2022-02-25 | 中国人民解放军陆军炮兵防空兵学院 | 一种基于声谱图的cnn故障诊断模型的发动机故障诊断方法 |
CN114048682B (zh) * | 2021-11-22 | 2024-05-03 | 昆明理工大学 | 一种基于优选小波基与多维深度特征融合的滚动轴承声发射智能诊断方法 |
CN114295377B (zh) * | 2021-12-13 | 2023-04-18 | 南京工业大学 | 一种基于遗传算法的cnn-lstm轴承故障诊断方法 |
CN114441173B (zh) * | 2021-12-28 | 2023-11-24 | 东南大学 | 基于改进深度残差收缩网络的滚动轴承故障诊断方法 |
CN114048787B (zh) * | 2022-01-11 | 2022-04-22 | 杭州云智声智能科技有限公司 | 一种基于Attention CNN模型的轴承故障实时智能诊断方法与*** |
CN114897138A (zh) * | 2022-05-06 | 2022-08-12 | 东南大学 | 基于注意力机制和深度残差网络的***故障诊断方法 |
CN114896733B (zh) * | 2022-05-20 | 2024-02-13 | 合肥工业大学 | 一种基于深度强化学习的滚动轴承故障类型识别方法 |
CN115017945A (zh) * | 2022-05-24 | 2022-09-06 | 南京林业大学 | 基于增强型卷积神经网络的机械故障诊断方法和诊断*** |
CN114662680B (zh) * | 2022-05-24 | 2022-08-30 | 杭州谨煜科技有限公司 | 一种基于Gaussian Noise CNN模型多变工况下轴承故障诊断方法与*** |
CN114970044B (zh) * | 2022-06-20 | 2023-05-19 | 华北电力大学 | 一种基于阈值卷积神经网络的滚动轴承故障诊断方法及*** |
CN115795131B (zh) * | 2023-02-10 | 2023-05-16 | 山东能源数智云科技有限公司 | 基于人工智能的电子档案分类方法、装置及电子设备 |
CN116010805B (zh) * | 2023-03-24 | 2023-06-16 | 昆明理工大学 | 基于卷积神经网络的滚动轴承故障特征提取方法、装置 |
CN116701912B (zh) * | 2023-06-14 | 2023-11-14 | 盐城工学院 | 基于一维卷积神经网络的轴承故障诊断方法及*** |
CN117807509B (zh) * | 2024-02-29 | 2024-04-30 | 南京工业大学 | 基于平行注意力的轴承故障诊断方法、设备及存储介质 |
Citations (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107526853A (zh) * | 2016-06-22 | 2017-12-29 | 北京航空航天大学 | 基于层叠卷积网络的滚动轴承故障模式识别方法及装置 |
CN109596326A (zh) * | 2018-11-30 | 2019-04-09 | 电子科技大学 | 基于优化结构卷积神经网络的旋转机械故障诊断方法 |
CN110084794A (zh) * | 2019-04-22 | 2019-08-02 | 华南理工大学 | 一种基于注意力卷积神经网络的皮肤癌图片识别方法 |
CN110110642A (zh) * | 2019-04-29 | 2019-08-09 | 华南理工大学 | 一种基于多通道注意力特征的行人重识别方法 |
CN110163302A (zh) * | 2019-06-02 | 2019-08-23 | 东北石油大学 | 基于正则化注意力卷积神经网络的示功图识别方法 |
CN110516596A (zh) * | 2019-08-27 | 2019-11-29 | 西安电子科技大学 | 基于Octave卷积的空谱注意力高光谱图像分类方法 |
CN110672343A (zh) * | 2019-09-29 | 2020-01-10 | 电子科技大学 | 基于多注意力卷积神经网络的旋转机械故障诊断方法 |
CN111458148A (zh) * | 2020-04-26 | 2020-07-28 | 上海电机学院 | 一种基于cbam的卷积神经网络滚动轴承故障诊断方法 |
CN111956212A (zh) * | 2020-07-29 | 2020-11-20 | 鲁东大学 | 基于频域滤波-多模态深度神经网络的组间房颤识别方法 |
-
2020
- 2020-12-07 CN CN202011436394.7A patent/CN112577747B/zh active Active
Patent Citations (9)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107526853A (zh) * | 2016-06-22 | 2017-12-29 | 北京航空航天大学 | 基于层叠卷积网络的滚动轴承故障模式识别方法及装置 |
CN109596326A (zh) * | 2018-11-30 | 2019-04-09 | 电子科技大学 | 基于优化结构卷积神经网络的旋转机械故障诊断方法 |
CN110084794A (zh) * | 2019-04-22 | 2019-08-02 | 华南理工大学 | 一种基于注意力卷积神经网络的皮肤癌图片识别方法 |
CN110110642A (zh) * | 2019-04-29 | 2019-08-09 | 华南理工大学 | 一种基于多通道注意力特征的行人重识别方法 |
CN110163302A (zh) * | 2019-06-02 | 2019-08-23 | 东北石油大学 | 基于正则化注意力卷积神经网络的示功图识别方法 |
CN110516596A (zh) * | 2019-08-27 | 2019-11-29 | 西安电子科技大学 | 基于Octave卷积的空谱注意力高光谱图像分类方法 |
CN110672343A (zh) * | 2019-09-29 | 2020-01-10 | 电子科技大学 | 基于多注意力卷积神经网络的旋转机械故障诊断方法 |
CN111458148A (zh) * | 2020-04-26 | 2020-07-28 | 上海电机学院 | 一种基于cbam的卷积神经网络滚动轴承故障诊断方法 |
CN111956212A (zh) * | 2020-07-29 | 2020-11-20 | 鲁东大学 | 基于频域滤波-多模态深度神经网络的组间房颤识别方法 |
Non-Patent Citations (1)
Title |
---|
AR-Net: Adaptive Attention and Residual Refinement Network for Copy-Move Forgery Detection;Ye Zhu 等;《IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS》;20201031;第16卷(第10期);第6714-6723页 * |
Also Published As
Publication number | Publication date |
---|---|
CN112577747A (zh) | 2021-03-30 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112577747B (zh) | 一种基于空间池化网络的滚动轴承故障诊断方法 | |
CN109918752B (zh) | 基于迁移卷积神经网络的机械故障诊断方法、设备及介质 | |
CN111967486A (zh) | 一种基于多传感器融合的复杂装备故障诊断方法 | |
CN112557034B (zh) | 一种基于pca_cnns的轴承故障诊断方法 | |
CN108344564A (zh) | 一种基于深度学习的主轴特性试验台状态识别及预测方法 | |
CN113988126B (zh) | 一种基于少标签数据特征迁移的滚动轴承故障诊断方法 | |
CN105913450A (zh) | 基于神经网络图像处理的轮胎橡胶炭黑分散度评价方法及*** | |
CN111046961B (zh) | 基于双向长短时记忆单元和胶囊网络的故障分类方法 | |
CN110880369A (zh) | 基于径向基函数神经网络的气体标志物检测方法及应用 | |
CN112836604A (zh) | 一种基于vmd-ssae的滚动轴承故障诊断分类方法、***、设备及其存储介质 | |
CN110987436B (zh) | 基于激励机制的轴承故障诊断方法 | |
CN106980761A (zh) | 一种滚动轴承运行状态退化趋势预测方法 | |
CN112765890A (zh) | 基于动态域适应网络的多工况旋转机械剩余寿命预测方法 | |
CN107274016A (zh) | 随机蛙跳算法优化对称极限学习机的带钢出口厚度预测方法 | |
CN109543818A (zh) | 一种基于深度学习模型的链路评估方法和*** | |
CN116542382A (zh) | 基于混合优化算法的污水处理溶解氧浓度预测方法 | |
CN111695607A (zh) | 基于lstm增强模型的电子设备故障预测方法 | |
CN112307927A (zh) | 基于bp网络针对非合作通信中mpsk信号的识别研究 | |
CN112581264A (zh) | 一种基于蝗虫算法优化mlp神经网络的信用风险预测方法 | |
CN113240527A (zh) | 基于可解释机器学习的债券市场违约风险预警方法 | |
CN115392333A (zh) | 一种基于改进端到端ResNet-BiLSTM双通道模型的设备故障诊断方法 | |
CN115545070A (zh) | 基于综合平衡网络的类不平衡轴承智能诊断方法 | |
CN114905335A (zh) | 一种结合域对抗和卷积神经网络的刀具磨损预测方法 | |
CN107038450A (zh) | 基于深度学习的无人机警察*** | |
CN112819087A (zh) | 一种基于模块化神经网络的出水bod传感器异常检测方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |