首页|改进掩码自编码器的滚动轴承半监督故障诊断

改进掩码自编码器的滚动轴承半监督故障诊断

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
针对滚动轴承在不同转速条件下数据分布不同以及实际工程应用中标签样本不足导致故障诊断精度低的问题，将领域适配模块融入掩码自编码器(MAE)中，提出了改进掩码自编码器(IMAE)的滚动轴承半监督故障诊断方法。首先，对滚动轴承振动信号进行连续小波变换(CWT)得到反应信号时频特征的二维时频图，然后对时频图随机掩码，利用无标签样本进行掩码自编码器预训练，获得数据中复杂的内在特征，减少对有标签样本的依赖;其次将领域适配模块引入到预训练后的编码器中，使用少量有标签源域数据对IMAE进行微调，在希尔伯特空间中利用最小化最大均值差异减小因转速不同造成的源域与目标域间数据分布差异;最后在Softmax分类层下实现滚动轴承半监督故障诊断。通过滚动轴承数据集实验验证，所提方法检测精度均达到 94%以上，证明了该方法的可行性与有效性。
Improved semi-supervised fault diagnosis of rolling bearings with mask autoencoder
To address the problems of different data distribution of rolling bearings under different speed conditions and low fault diagnosis accuracy caused by insufficient label samples in practical engineering applications,the domain adaptation modules are integrated into masked autoencoders(MAE).An improved masked autoencoders(IMAE)method for semi-supervised fault diagnosis of rolling bearings is proposed.Firstly,the two-dimensional time-frequency graph of the response signal is obtained by applying continuous wavelet transform(CWT)to the vibration signal of the rolling bearing.Then,the mask of the time-frequency graph is randomly masked,and the mask autoencoder is pre-trained with unlabeled samples to obtain the complex intrinsic features of the data.The reliance on labeled samples is reduced.Secondly,the domain adaptation module is introduced into the pre-trained encoder,and a small amount of labeled source domain data are used to fine-tune the IMAE,and the maximum mean difference is minimized in Hilbert space to reduce the data distribution difference between the source domain and the target domain caused by different rotational speeds.Finally,the semi-supervised fault diagnosis of rolling bearing is realized under the Softmax classification layer.Through the experimental evaluation of the rolling bearing data set,the detection accuracy of the proposed method is more than 94%,which proves the feasibility and effectiveness of the proposed method.

mask autoencoderrolling bearingdifferent speedsample labelsemi-supervised fault diagnosis

陈仁祥、张晓、张旭、赵玲、夏亮

展开 >

重庆交通大学交通工程应用机器人重庆市工程实验室 重庆 400074

重庆智能机器人研究院 重庆 400714

掩码自编码器 滚动轴承 不同转速 标签样本 半监督故障诊断

国家自然科学基金国家自然科学基金重庆市教委科学技术研究计划重庆市研究生联合培养基地项目重庆市自然科学基金创新发展联合基金重庆市专业学位研究生教学案例库项目重庆市研究生科研创新项目

5197507962073051KJZD-M202200701JDLHPYJD2021007CSTB2023NSCQ-LZX0127JDALK20220072023S0072

2024

10.19650/j.cnki.cjsi.J2311570

仪器仪表学报
中国仪器仪表学会

仪器仪表学报

CSTPCD北大核心
影响因子：2.372
ISSN：0254-3087
年,卷(期)：2024.45(1)
  • 17