基于低秩模态融合与对抗度量的滚动轴承故障诊断

Rolling Bearing Fault Diagnosis Based on Low-Rank Multimodal Fusion and Adversarial Metric

  • 摘要: 针对矿用滚动轴承故障特征微弱、高质量样本稀缺及跨工况分布偏移导致传统深度学习模型泛化性能不足的问题,提出了一种基于低秩模态融合与对抗度量的轻量化故障诊断模型(MTSFCL)。利用超小波变换对振动信号进行转换,构建时序与时频图双模态输入数据,多维度增强矿用滚动轴承的故障表征。设计轻量化双分支特征提取层,时序分支采用通道注意力机制(ECA)增强的双向门控单元(BiGRU),在捕获时序信号中长期依赖关系的同时,有效抑制冗余信息的干扰。空间分支基于改进的StarNet架构,使用多尺度卷积与选择性核融合机制,提取时频图中的多尺度故障特征,并利用元素乘法在不增加网络深度的前提下实现高维空间特征的映射。设计了一种低秩多模态融合模块(LMF),利用低秩因子将时序与空间特征投影至公共子空间,并通过逐元素乘法进行非线性融合,在低计算成本下实现双模态特征的深度交互。结合条件域对抗(CDAN)与作为度量约束的局部最大均值差异(LMMD),构建了对抗度量域适应模块,联合减少源域与目标域之间的边缘与条件分布差异,提高了模型的泛化性能。选用凯斯西储大学(CWRU)轴承数据集进行故障诊断实验,实验结果表明:MTSFCL参量仅为0.3221M。在单一工况下MTSFCL平均诊断准确率为99.91%;在每类仅有5个故障样本的小样本工况下,平均诊断准确率为94.12%;在跨工况下平均诊断准确率为99.28%;在噪声干扰下同样保持高准确率。

     

    Abstract: To address the issues of insufficient generalization performance in traditional deep learning models caused by weak fault features, scarcity of high-quality samples, and cross-condition distribution shifts in mining rolling bearings, a lightweight fault diagnosis model based on Low-rank Multimodal Fusion and Adversarial Metric (MTSFCL) is proposed. The Superlet Transform (SLT) is utilized to convert vibration signals, constructing dual-modal input data of time-series and time-frequency diagrams to enhance the fault representation of mining rolling bearings from multiple dimensions. A lightweight dual-branch feature extraction layer is designed, where the temporal branch employs a Bidirectional Gated Recurrent Unit (BiGRU) enhanced by the Efficient Channel Attention (ECA) mechanism, effectively suppressing redundant information interference while capturing long-term dependencies in time-series signals. The spatial branch is based on an improved StarNet architecture, using multi-scale convolution and a Selective Kernel (SK) fusion mechanism to extract multi-scale fault features from time-frequency diagrams, and utilizing element-wise multiplication to achieve the mapping of high-dimensional spatial features without increasing network depth. A Low-rank Multimodal Fusion (LMF) module is designed, utilizing low-rank factors to project temporal and spatial features into a common subspace and performing non-linear fusion via element-wise multiplication, achieving deep interaction of dual-modal features at a low computational cost. Combining Conditional Domain Adversarial Network (CDAN) with Local Maximum Mean Discrepancy (LMMD) as a metric constraint, an adversarial metric domain adaptation module is constructed to jointly reduce the marginal and conditional distribution discrepancies between the source and target domains, improving the model's generalization performance. The Case Western Reserve University (CWRU) bearing dataset is selected for fault diagnosis experiments, and the experimental results show that: the parameter size of MTSFCL is only 0.3221 M; the average diagnostic accuracy of MTSFCL is 99.91% under single operating conditions; the average diagnostic accuracy is 94.12% under small-sample conditions with only 5 fault samples per class; the average diagnostic accuracy is 99.28% under cross-condition scenarios; and high accuracy is maintained under noise interference.

     

/

返回文章
返回