基于对比学习和卷积自注意力网络的减速机故障诊断方法

Contrastive Learning and Convolution Transformer Network-based Reducer Fault Diagnosis Method

  • 摘要: 【目的】针对传统神经网络在少标记样本下故障诊断准确率低的问题,提出一种基于对比学习和卷积自注意力网络方法。【方法】首先,原始监测数据经过数据增强得到相似样本对。同时,利用特征提取器将相似样本对映射到深层特征空间。然后,利用Transformer进行局部对比和全局对比,实现相似样本聚类。最后,通过少标记样本训练下游分类网络,提高模型的诊断性能。【结果】基于自建的减速机实验台,验证了所提方法的有效性。结果表明,所提方法能在少标记样本条件下实现减速机故障的精准诊断,准确率达到98.35%。相比现有的方法,准确率至少提升了5.29%。【结论】研究成果可为少标记样本下的故障诊断提供参考。

     

    Abstract: Objectives To address the issue of low fault diagnosis accuracy of traditional neural networks with few labeled samples, a method based contrastive learning and convolution transformer network is proposed. Methods First, raw monitoring data are transformed into similar sample pairs by data augmentation. At the same time, the similar sample pairs are mapped to the deep feature space by a feature extractor. Then, the transformer network is used to contrast locally and globally, thereby clustering representations with similar samples. Finally, the downstream classification network is trained with few labeled samples to improve the diagnostic performance of the proposed model. Results The effectiveness of the proposed method is verified by a self-built reducer test rig. The results show that the proposed method can accurately realize reducer faults with few labeled samples with an accuracy of 98.35%. Compared with the existing method, the accuracy is improved by at least 5.29%. Conclusions The research results can provide a reference for fault diagnosis with few labeled samples.

     

/

返回文章
返回