arXiv:2509.09251cs.LG2025-09

仅用1%标注数据,实现跨设备高精度故障诊断

Unsupervised Multi-Attention Meta Transformer for Rotating Machinery Fault Diagnosis

  • 用无监督方法从时频域数据中提取故障特征
  • 1%标注数据下准确率达99%,支持多设备通用
  • 适合工业场景中数据稀缺的故障诊断任务

旋转机械智能故障诊断通常需要大量标注样本,但在实际工业应用中,获取足够数据既困难又昂贵。且不同机械设备因结构差异,需为每类单独训练模型。为解决故障样本少、模型泛化能力差的问题,本文提出一种少样本无监督旋转机械故障诊断多注意力元变压器框架(MMT-FD)。该框架通过时频域编码器对未标注数据进行随机增强,生成状态表征,并输入元学习网络进行分类与泛化训练,再用少量标注数据微调。通过少量对比学习迭代实现模型优化,效率高。在轴承故障数据集和转子试验台数据上验证,仅用1%标注数据即达到99%诊断准确率,展现出强泛化能力。

原文摘要 · Abstract (English)

The intelligent fault diagnosis of rotating mechanical equipment usually requires a large amount of labeled sample data. However, in practical industrial applications, acquiring enough data is both challenging and expensive in terms of time and cost. Moreover, different types of rotating mechanical equipment with different unique mechanical properties, require separate training of diagnostic models for each case. To address the challenges of limited fault samples and the lack of generalizability in prediction models for practical engineering applications, we propose a Multi-Attention Meta Transformer method for few-shot unsupervised rotating machinery fault diagnosis (MMT-FD). This framework extracts potential fault representations from unlabeled data and demonstrates strong generalization capabilities, making it suitable for diagnosing faults across various types of mechanical equipment. The MMT-FD framework integrates a time-frequency domain encoder and a meta-learning generalization model. The time-frequency domain encoder predicts status representations generated through random augmentations in the time-frequency domain. These enhanced data are then fed into a meta-learning network for classification and generalization training, followed by fine-tuning using a limited amount of labeled data. The model is iteratively optimized using a small number of contrastive learning iterations, resulting in high efficiency. To validate the framework, we conducted experiments on a bearing fault dataset and rotor test bench data. The results demonstrate that the MMT-FD model achieves 99\% fault diagnosis accuracy with only 1\% of labeled sample data, exhibiting robust generalization capabilities.

故障诊断元学习无监督学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。