买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：哈尔滨理工大学

摘要：本发明公开了一种有限数据条件下的滚动轴承故障诊断方法，所述方法通过使用连续小波变换进行预处理，有效避免了轴承振动信号具有的非线性和非平稳性的缺点，同时其具有一定的去噪效果；通过特征映射重构模块构建生成器，可使上采样过程不再基于像素点的空间位置而是基于特征的内容，可以有效避免棋盘效应的产生。同时，局部区域中特征点的重要信息可以得到更多的关注，所以重组生成的特征图的语义信息可以比原始特征图更强。再在鉴别器中加入坐标注意力机制，使得模型可以考虑特征图通道信息和空间位置信息之间的关系，能够提升整体的特征提取能力，且根据特征的重要性赋予权重，保证提取的特征信息的完整性，最终提升生成样本的质量。

主权项：1.一种有限数据条件下的滚动轴承故障诊断方法，其特征在于所述方法包括如下步骤：步骤一、构建基于CWT的有限样本数据集，将振动信号数据集转变为时频图数据集；步骤二、以DCGAN网络为基础模型，使用特征映射重构模块构建全新的生成器，同时在其鉴别器中引入坐标注意力机制，构建基于FMRGAN的数据增强网络，具体步骤如下：步骤二一、使用四个特征映射重构模块构成特征映射重构生成器主体：特征映射重构生成器由一个反卷积层和四个特征映射重构模块组成，特征映射重构模块包括两个部分：其一是根据每个目标位置的内容预测生成自适应重组内核的自适应内核模块，其二是用预测生成的自适应重组内核进行重组特征操作，其中：自适应内核模块由通道压缩器、内容编码器和内核归一化器三个子模块组成，首先，在通道压缩器中使用1×1的卷积层将输入特征通道数从C压缩到Cm；然后，内容编码器将压缩后的特征图作为输入，在内容编码器中基于输入特征的内容使用内核大小为kencoder的卷积层生成自适应重组内核，生成自适应重组内核后再进行reshape操作将编码的通道信息扩充到同一维度；最后，内核归一化器将Softmax函数应用于每个自适应重组内核，对每个上采样大小为kup*kup的自适应重组核进行归一化，使得内核值之和为1；自适应内核模块在χl的区域为每个位置l'预测该位置的自适应内核Wl'： 对于每个自适应重组内核Wl'，特征映射重组模块在局部区域内重组特征，目标位置l'和以l＝i,j为中心的相应区域Nχl,kup重组操作如下式所示： 其中，Nχl,k表示特征图χ上以位置l为中心，k*k大小的区域；特征映射重构模块将χl区域和内核Wl'进行重组得到最终的输出；步骤二二、在鉴别器中引入坐标注意力模块：步骤二二、在鉴别器中引入坐标注意力模块：坐标注意力模块由坐标信息嵌入和坐标注意力生成两部分组成，对于给定的输入特征张量X＝{x1,x2,x3……xc}，xc代表第c个通道的输入特征张量，第c个通道的全局信息嵌入如下式所示： 其中，zc是与第c个通道相关联的输出，xci,j表示输入序列在通道c、位置i,j处的值，在CA中使用两个池化核H，1和1，W将上式转换为两个一维特征编码操作，H和W分别表示在水平坐标方向和竖直坐标方向上进行大小为H和W的池化，则高度h处的第c个通道的输出写为： 其中，xch,i表示输入序列在通道c、位置h,i处的值；宽度为w的第c个通道的输出可以写为： 其中，xcj,w表示输入序列在通道c、位置j,w处的值；上述两个变换分别沿着两个不同的空间方向聚合特征，以生成一对方向感知特征图，上述操作被称为Coordinateinformationembedding；在获得一对方向感知特征图之后，首先级联两个特征图，然后使用一个共享的1×1卷积进行F1变换：f＝δF1[zw,zw]其中，[·,·]表示沿空间维度的级联运算，δ是非线性激活函数，生成的是空间信息在水平方向和垂直方向的中间特征图，C是通道数，r是用于控制模块大小的减小率，将f分解为两个独立的张量和再利用两个1×1卷积变换Fh和Fw将特征图fh和fw变换为和输入X同样通道数的张量，其过程如下：gh＝σFhfhgw＝σFwfw其中，σ代表Sigmoid函数，输出gh和gw分别被展开并用作注意力权重，CA模块的最终输出如下式所示： 其中，yci,j表示在通道c上的位置i,j处的输出值；表示在通道c上对行维度的注意力权重，用于调整行维度的重要性；表示在通道c上对列维度的注意力权重，用于调整列维度的重要性；i和j分别表示矩阵中的行和列索；步骤二三、通过对抗性学习机制训练FMRGAN的生成器和鉴别器：对于数据增强模型，满足Pz分布的噪声z与类标签C相结合，并输入到特征映射重构生成器G用来生成信号Sg，生成的信号Sg和原始信号Sreal一起用作鉴别器D的输入，G和D以对抗性的方式交替训练，直到模型达到纳什均衡状态；步骤二四、使用经过训练的FMRGAN生成伪样本，并将伪样本添加到步骤一构建的有限样本数据集中以构建增强数据集；步骤三、将增强数据集输入到VisionTransformer分类网络中得到故障诊断结果。

全文数据：

权利要求：

百度查询： 哈尔滨理工大学 一种有限数据条件下的滚动轴承故障诊断方法