一种基于部分域对抗的滚动轴承迁移学习故障诊断方法技术

本发明专利技术公开了一种基于部分域对抗的滚动轴承迁移学习故障诊断方法，包括以下步骤：搭建滚动轴承故障数据样本库，划分源域和目标域故障数据；提取源域和目标域故障数据的隐含特征；构建标签预测器；构建加权域分类器，得到样本特征来自源域分布的概率和权值；将加权的源域样本特征和不加权的目标域样本特征，送入另一域分类器，判别样本特征来自源域还是目标域，构造梯度反转层；优化模型；将测试数据输入特征提取器获得样本特征，并将获得的样本特征输入标签预测器，获得预测标签，并计算分类精度。本发明专利技术将对抗思想融于部分迁移网络，提出了源域样本加权后再进行域分类的策略，提升了样本域自适应能力，解决了在目标域进行无监督标签预测的问题。

【技术实现步骤摘要】
一种基于部分域对抗的滚动轴承迁移学习故障诊断方法
本专利技术涉及轴承故障诊断领域，特别涉及一种基于部分域对抗的滚动轴承迁移学习故障诊断方法。
技术介绍
滚动轴承是各种机械化领域的高端部件，如高铁领域、风电领域、航空领域等，有着更精确的传动性能、更小的体积和更轻的重量。但是，滚动轴承的维修也将产生更多的花费。这主要是因为滚动轴承的工作环境一般都很恶劣，尤其是在一些重要领域，滚动轴承容易损坏。如果没有及时处理出现问题的轴承，就会出现很多事故，轻则停止运行，重则发生重大安全事故，从而产生无法预计的后果。因此，对滚动轴承故障进行更加精确的诊断有着重要意义。域适应理论主要指的是在源域和目标域之间取得公共特征，使这些特征更加符合源域分布，通过迁移学习将该特征用于源域数据训练的分类器。浅层学习方法具有所需样本量小、训练快速等特点，因此与域适应理论结合进行故障诊断获得了发展。但浅层方法因为特征提取不充分、表达能力低下，使得用于故障诊断的域适应模型可移植性弱。目前，深度学习因其层次深、计算力强等特点，被广泛应用于滚动轴承的故障诊断领域。利用深度学习来提取源域故障数据的深度特征，并应用于目标域数据，能提高诊断的精确度。深度学习结合域适应理论能针对不同工况建立更精确的诊断模型。源域数据的标签可以看做是包括所有故障类别，但目标域数据仅包含某些故障类别。在现有大多数方法中，只靠减少源域和目标域的分布之间的距离来减小域偏移是行不通的。一种可行的方法是通过构造源域样本的权值剔除异常类。然而，目标域样本一般是没有标签的，通过简单的深度学习方法没办法训练一个合适的加权模型。
技术实现思路
为了解决上述技术问题，本专利技术提供一种算法简单、诊断精度高的基于部分域对抗的滚动轴承迁移学习故障诊断方法。本专利技术解决上述问题的技术方案是：一种基于部分域对抗的滚动轴承迁移学习故障诊断方法，包括以下步骤：(1)搭建不同工况下滚动轴承故障数据样本库，划分源域和目标域故障数据，并将源域样本作为训练数据，将目标域样本作为测试数据；(2)使用堆栈自动编码SAE网络作为特征提取器F，提取源域和目标域故障数据的隐含特征；(3)使用源域样本特征训练softmax分类器作为标签预测器C；(4)构建加权域分类器D，得到样本特征来自源域分布的概率，并计算样本特征的权值；(5)将加权的源域样本特征和不加权的目标域样本特征，送入另一域分类器D0，判别样本特征来自源域还是目标域，构造梯度反转层用于实现特征提取器和域分类器的对抗学习；(6)使用随机梯度下降算法SGD优化特征提取器F、标签预测器C、加权域分类器D和域分类器D0的总损失；(7)将测试数据输入特征提取器获得样本特征，并将获得的样本特征输入标签预测器，获得预测标签，并计算分类精度。上述基于部分域对抗的滚动轴承迁移学习故障诊断方法，所述步骤(1)中，源域数据为目标域数据为其中x(i)(i＝1,2,…,ns)为源域带标签样本，y(i)(i＝1,2,…,ns)为源域样本的标签，x(j)(j＝1,2,…,nt)为目标域无标签样本；ns和nt分别为源域和目标域样本数量。样本标签种类其中表示目标域样本标签种类数，表示源域样本标签种类数。上述基于部分域对抗的滚动轴承迁移学习故障诊断方法，所述步骤(2)中，堆栈自编码SAE网络由两个自编码器AE堆叠组成，AE网络由编码器和解码器组成；其中，编码器用来压缩输入数据为代表性特征：Y＝f(W(1)X+b(1))其中，X为输入数据，Y为代表性特征，W(1)为权重矩阵，b(1)为偏置向量，f为激活函数；然后，解码器将得到的代表性特征重构回与输入数据相同维度的向量：X′＝f(W(2)Y+b(2))其中，X′为重构后的向量，W(2)为重构过程的权重矩阵，b(2)为重构过程的偏置向量，f为激活函数；整个自编码过程的损失函数为：其中，X′(i)为第i个重构数据，X(i)为第i个输入数据，m为输入数据的个数；W和b分别表示该损失中要更新的权重和偏置参数，最小化压缩损失函数得到优化后的AE网络；SAE在AE的基础上，将AE得到的代表性特征作为输入数据，输入到第二个AE网络中，用同样的方法训练网络参数，最终得到所需代表性特征。上述基于部分域对抗的滚动轴承迁移学习故障诊断方法，所述步骤(3)中，softmax分类器是一种多标签分类器，用于得到多种标签的概率，概率最大的标签即为样本标签；对于输入源域数据对应k类标签y(i)＝{1,2,…k}，i＝1,2,…ns，则对于样本x(i)属于每种标签的概率hθ(x(i))为：其中，θ＝{θ1,θ2,…θk}表示模型参数，T表示转置，为假设函数；损失函数Jθ为：其中，1{真表达式}＝1，1{假表达式}＝0；由于目标域样本没有标签，因此标签预测器由源域样本训练，并直接用于目标域样本。上述基于部分域对抗的滚动轴承迁移学习故障诊断方法，所述步骤(4)中，加权域分类器D的输出值给出了样本特征属于源域分布的概率；加权域分类器的优化目标为：其中，F(x)表示特征提取器F得到的样本特征，x为原样本，D(F(x))表示加权域分类器D中样本特征的概率；指取期望，x～ps(x)表示样本x来自源域分布ps(x)，x～pt(x)表示样本x来自目标域分布pt(x)；对于输入序列z，加权域分类器输出z属于源域的概率D(z)：D(z)＝p(y＝1|z)＝σ(F(z))其中，p(y＝1|z)表示在输入z的条件下，y＝1的概率，σ为sigmoid函数，F(z)为SAE网络提取的深度特征；D*(z)表示域分类器最优化后的结果，当D*(z)＝1时，表示z来自源域分布的异常类，这些样本根本没有出现在目标域，因此这些样本被赋予更小的权重；当D*(z)＝0时，表示这些样本为源域和目标域共有的类别，因此这些样本被赋予更大的权重，从而减少样本在这些共有类上的域偏移；因此，权重函数为：其中，ps(z)表示源域分布的z，pt(z)表示目标域分布的z；由权重函数可知，D*(z)越大，越小，ps(z)/pt(z)越大；所以，源域的异常类样本的权重比共有类样本小；然后，权重被归一化为：上述基于部分域对抗的滚动轴承迁移学习故障诊断方法，所述步骤(5)中，域分类器D0与特征提取器F进行最小最大博弈，以抑制共有类的偏移；域分类器D0优化目标如下：其中，ω(z)独立于D0，被看作一个常数，D0(F(x))表示域分类器D0中样本特征的概率；由上式可知，该优化目标首先实现域分类器损失的最大化，以判别来自源域分布的样本特征，然后实现特征提取器损失的最小化，使特征提取器提取的目标域样本特征更加接近于源域样本特征；两者进行对抗学习，使域分类器无法判别样本特征是来自源域分布还是目标域分布，从而目标域样本特征更适用于源本文档来自技高网...

【技术保护点】
1.一种基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于，包括以下步骤：/n(1)搭建不同工况下滚动轴承故障数据样本库，划分源域和目标域故障数据，并将源域样本作为训练数据，将目标域样本作为测试数据；/n(2)使用堆栈自动编码SAE网络作为特征提取器F，提取源域和目标域故障数据的隐含特征；/n(3)使用源域样本特征训练softmax分类器作为标签预测器C；/n(4)构建加权域分类器D，得到样本特征来自源域分布的概率，并计算样本特征的权值；/n(5)将加权的源域样本特征和不加权的目标域样本特征，送入另一域分类器D

【技术特征摘要】
1.一种基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于，包括以下步骤：
(1)搭建不同工况下滚动轴承故障数据样本库，划分源域和目标域故障数据，并将源域样本作为训练数据，将目标域样本作为测试数据；
(2)使用堆栈自动编码SAE网络作为特征提取器F，提取源域和目标域故障数据的隐含特征；
(3)使用源域样本特征训练softmax分类器作为标签预测器C；
(4)构建加权域分类器D，得到样本特征来自源域分布的概率，并计算样本特征的权值；
(5)将加权的源域样本特征和不加权的目标域样本特征，送入另一域分类器D0，判别样本特征来自源域还是目标域，构造梯度反转层用于实现特征提取器和域分类器的对抗学习；
(6)使用随机梯度下降算法SGD优化特征提取器F、标签预测器C、加权域分类器D和域分类器D0的总损失；
(7)将测试数据输入特征提取器获得样本特征，并将获得的样本特征输入标签预测器，获得预测标签，并计算分类精度。


2.根据权利要求1所述的基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于：所述步骤(1)中，源域数据为目标域数据为其中x(i)(i＝1,2,…,ns)为源域带标签样本，y(i)(i＝1,2,…,ns)为源域样本的标签，x(j)(j＝1,2,…,nt)为目标域无标签样本；ns和nt分别为源域和目标域样本数量。样本标签种类其中表示目标域样本标签种类数，表示源域样本标签种类数。


3.根据权利要求2所述的基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于：所述步骤(2)中，堆栈自编码SAE网络由两个自编码器AE堆叠组成，AE网络由编码器和解码器组成；其中，编码器用来压缩输入数据为代表性特征：
Y＝f(W(1)X+b(1))
其中，X为输入数据，Y为代表性特征，W(1)为权重矩阵，b(1)为偏置向量，f为激活函数；
然后，解码器将得到的代表性特征重构回与输入数据相同维度的向量：
X′＝f(W(2)Y+b(2))
其中，X′为重构后的向量，W(2)为重构过程的权重矩阵，b(2)为重构过程的偏置向量，f为激活函数；
整个自编码过程的损失函数为：



其中，X′(i)为第i个重构数据，X(i)为第i个输入数据，m为输入数据的个数；W和b分别表示该损失中要更新的权重和偏置参数，最小化压缩损失函数得到优化后的AE网络；
SAE在AE的基础上，将AE得到的代表性特征作为输入数据，输入到第二个AE网络中，用同样的方法训练网络参数，最终得到所需代表性特征。


4.根据权利要求3所述的基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于：所述步骤(3)中，softmax分类器是一种多标签分类器，用于得到多种标签的概率，概率最大的标签即为样本标签；
对于输入源域数据对应k类标签y(i)＝{1,2,…k}，i＝1,2,…ns，则对于样本x(i)属于每种标签的概率hθ(x(i))为：



其中，θ＝{θ1,θ2,…θk}表示模型参数，T表示转置，为假设函数；
损失函数Jθ为：



其中，1{真表达式}＝1，1{假表达式}＝0；
由于目标域样本没有标签，因此标签预测器由源域样本训练，并直接用于目标域样本。


5.根据权利要求4所述的基于部分域对抗的滚动轴承迁移学习故障诊断方法，其特征在于：所述步骤(4)中，加权域分类器D的输出值给出了样本特征属于源域分布的概率；
加权域分类器的优化目标为：



其中，F(x)表示特征提取器F得到的样本特征，x为原样本，...

【专利技术属性】
技术研发人员：刘朝华，陆碧良，王畅通，陈磊，李小花，张红强，
申请(专利权)人：湖南科技大学，
类型：发明
国别省市：湖南;43