融合元学习与图神经网络的稀有故障诊断方法

本发明设计融合元学习与图神经网络的稀有故障诊断方法

背景技术：

1、航空发附机匣是飞机传动系统中的重要部件，随着机匣的服役时间延长，性能缓慢退化，发生事故风险逐渐增大，如何准确诊断航空发动机附件机匣故障对军用与民用飞机的运维至关重要。尽管状态监测系统可以不断地采集机器的运行监测数据，但由于故障较少发生，从航空发动机附件机匣系统采集到的标记监测数据有限、采集到的数据大部分是健康数据，故障数据较少，并且在实际应用中训练数据域与目标数据域存在差异，导致模型诊断性能下降。近年来迁移学习在解决样本不足的问题上逐渐受到重视，迁移学习的本质是建立网络模型，通过从任务相似的源数据域中学习信息，指导目标数据域的分类任务。目前流行的迁移学习方法大部分是假设来自不同域的任务数据分布相同，包括工况一致或标签空间分布一致等。在实际工业环境中，同一机台可能出现新的故障类别，机台的运行状态也会随着工作场景做出切换，导致源域和目标域之间存在分布差异，这时训练好的模型就会失效。为解决这个问题，本文使用元学习方法让网络模型从源域中学习通用元知识，快速适应新任务。

2、同时，图神经网络(graph neural networks,gnn)正被广泛应用于小样本分类问题上。具体来说，图神经网络中的每个样本结点都会通过边与其他数据样本建立联系，这些信息能够用来捕获不同实例之间的相互依赖关系。所以，可以利用gnn学习未知样本与样本组之间的距离，增加类间距离，缩小类内距离，提高样本间的区分性。综上，本文提出基于元学习的融合图神经网络稀有故障诊断方法(attention graph taskmeta learning,agtml)来有效解决样本不足以及不同类别的故障样本难以区分的问题。模型训练时采用元学习的训练策略，随机抽取一定数量的元任务，对每个任务中的支持集(已知样本)和查询集(未知样本)的故障特征向量构建局部任务图，图的节点为样本特征，边为样本特征之间的距离，利用gnn学习图节点的边权分布规律，将任务中的已知样本的标签信息传播到未知样本中并预测未知样本的故障类别。对比实验表明该方法在cwru数据集和航空发附机匣数据集上均取得了较好的实验效果。

技术实现思路

1、本发明的目的是为了解决航空发附机匣标记监测数据有限且在实际应用中训练数据域与目标数据域存在差异导致模型诊断性能下降的问题，而提出的一种基于元学习的融合图神经网络的稀有故障诊断方法。

2、上述发明目的主要是通过以下技术方案实现的：

3、s1、通过短时傅里叶变换(short-time fourier transform,stft)将采集到的振动信号转换为时频图像，融合多通道图像数据；

4、s2、利用数据量充足的数据集作为源域，将来自源域的标记数据输入到编码器中，得到样本的特征表示，再利用解码器重构输入数据，通过优化损失函数led，更新编码器的每层网络参数；

5、本方法利用自编码器进行预训练，上游网络模型包括卷积编码器和卷积解码器，通过重构后的数据和原始数据相比，让编码器不断学习，直到其最大可能的代表输入数据，然后迁移至下游网络。卷积编码器由卷积层和最大池化层组成，输入是二维时频图xi，输出为提取的图像特征，特征向量是m e，第l层卷积核是偏置是输出是其中e表示编码器输出，h表示隐藏层，具体计算见式(1)：

6、

7、pel-1是第l-1层的最大池化层的输出，具体算式如下：

8、

9、其中ρ是池化窗口的大小，设置为4，β是下采样因子，设置为2。

10、卷积解码器是由卷积层和上采样层组成。卷积解码器的输入是特征，输出是重构的样本x'(i)，特征向量是m d，第l层卷积核是是偏置是输出是d表示解码器输出，具体计算见式(3)：

11、

12、是第l-1层的最大池化层的输出，具体算式如下：

13、

14、其中ε是上采样因子，设置为2。

15、在预训练过程中，通过优化损失函数led重构输入信号样本，具体计算见式(5)：

16、

17、其中c表示输入样本的总数，n表示输入数据维度。

18、s3、将源域划分为训练集和测试集，根据元学习协议随机抽取训练数据，形成100个元任务，每个元任务中的类别数为n；

19、元学习可以解决在有限的数据情况下的跨域故障诊断问题，具体来说，元学习是让模型从多个相关任务中学习元知识，在目标任务中表现出更强的泛化能力和更快的适应能力，根据n-way,k-shot,u-test协议组织元任务t，每次从总类别中随机抽取n个类别，每个类别中分别随机抽取k个标记样本组成支持集s，抽取u个未标记样本作为查询集q。s用于训练故障分类模型并提供相应损失的反馈，q用于评估训练后模型的性能，具体算式如下：

20、t＝s∪q        (6)

21、

22、

23、

24、其中xi和yi是第i个输入数据及标签，属于集合c，ci是每个类别下的数据集的集合，c是训练或测试数据集的所有类的集合。训练集和查询集的标签是互斥的，即：

25、ctrain∩ctest＝0          (10)

26、s4、上游网络使用源域训练样本训练加权自编码器，将其网络参数和权重传递到下游网络；

27、s5、利用注意力加权机制网络(attention weighted encoder network，awen)提取目标域样本特征并构建任务图；

28、使用少量标记样本微调网络权值，筛选并抑制无用的故障特征，以此来获得具有判别性的故障特征。首先将原始特征me输入到二维卷积中做卷积操作，得到特征的重要程度，sigmoid函数调整生成的特征权重。具体计算见式(11)：

29、

30、其中，σ代表sigmoid函数，α为生成的特征调整权重。为了能够避免注意力过度的问题，进行特征重标定操作，如式(13)所示，这里直接使用特征权值α与输出特征m进行匹配相乘得到输出o。具体计算见式(12)和(13)：

31、

32、

33、最后在保留重标定特征o的同时，引入残差连接来提升特征提取方法的性能，如式(14)所示，用原始输入特征me与重标定特征o相加，得到注意力机制下最终学习到的新特征向量me′。

34、

35、s6、利用基于局部任务的图学习方法(graph learning methods for localtasks，ltgl)进行图学习并更新局部图的节点之间的边权，得到查询集的预测标签；

36、基于任务的图学习方法除了考虑到样本和样本之间的关系，还兼顾了样本组(包含所有故障类别的样本组)之间的全局相关性，减少模型误判，其利用cnn更新任务图结构的节点和边权，使支持集s中的已知样本之间更具区分性，查询集q的预测标签更精准。

37、图中包含|t|＝n×k+u个特征节点。g＝(vg,eg；t)表示任务图的节点和边权结构。vg＝{vi}i＝1,...,|t|表示图中节点集，其初始值来自于awen网络模型的输出特征eg＝{eij}i,j＝1,...,|t|表示图中节点间的边权集合，其初始值算式如下：

38、

39、其中，yi表示i的标签类别，[e1||e2]中的e1代表相似性，e2代表相异性，||是连接操作。

40、首先通过相邻节点之间的聚合更新节点特征信息利用l-1层的边缘特征对节点的更新进行加权，具体算式如下：

41、

42、之后利用图神经网络计算下一层网络的输入节点之间的边权，具体算式如下：

43、

44、

45、其中表示l层的节点i和j之间的相似权重和相异权重，是参数的相似性度量网络，用于计算2个节点的相似程度，网络构成为一个简单的多层感知网络。

46、最终以边权重为主要分类依据，通过不断更新边权最终得到样本间的相关度，实现故障分类。损失函数具体算式如下：

47、

48、其中et表示为每轮的任务数量，u表示查询集样本个数，表示第i个样本的预测标签，le定义为交叉熵损失。

49、s7、将任务中的已知样本的标签信息传播到未知样本中，保存训练好的网络模型并应用至目标域的数据集上。

50、发明效果

51、本发明提供了一种基于元学习的融合图神经网络的稀有故障诊断方法。首先，将原始信号转换为时频图像，对多通道的时频图像进行融合，作为网络模型的输入。然后，利用源数据域训练一个自编码器，并将编码器作为特征提取网络。接下来，以提取到的样本特征为节点，特征之间距离为边权，对已知样本和未知样本建图。最后，利用基于元任务的图学习方法，学习图节点的边权分布规律，将任务中的已知样本的标签信息传播到未知样本中。