结合SE-VAE与M1DCNN的小样本数据下轴承故障诊断
Bearing Fault Diagnosis Under Small Sample Data Based on SE-VAE and M1DCNN
李梦男 1李琨 1叶震 1高宏宇1
作者信息
- 1. 昆明理工大学信息工程与自动化学院,昆明 650500
- 折叠
摘要
针对轴承故障诊断中故障样本数量少导致诊断正确率低的问题,提出了一种基于注意力机制变分自编码器(SE-VAE)和多尺度一维卷积神经网络(M1DCNN)的轴承故障诊断方法.将轴承数据集的训练集输入到SE-VAE中进行训练,生成与训练样本分布相似的生成样本,并添加到训练集中增加训练集的样本数量.将扩充后的训练集输入到M1DCNN中进行训练,随后将训练好的模型应用于测试集,输出故障诊断结果.实验结果表明,所提方法能够在不同负载的小样本轴承故障数据集上取得较好的故障诊断准确率.
Abstract
Aiming at the problem of low diagnostic accuracy caused by the small number of fault samples in bearing fault diagnosis,a new bearing fault diagnosis method based on attention mechanism variation autoencoder(SE-VAE)and multi-scale one-dimensional convolutional neural network(M1DCNN)was proposed.Firstly,the training set of bearing data set is input into SE-VAE for training,generated samples with similar distribution to the training samples are obtained and added to the training set to increase the number of samples in the training set.Then,the extended training set is input into M1DCNN for training,and finally the trained model is applied to the test set to output the fault diagnosis results.Experimental results show that the proposed method can achieve better fault diagnosis accuracy on small sample bearing fault data sets with different loads.
关键词
轴承故障诊断/变分自编码器/注意力机制/多尺度一维卷积神经网络/小样本Key words
bearing fault diagnosis/variation autoencoder(VAE)/attention mechanism/multiscale one-dimensional convolutional neural network/small sample引用本文复制引用
基金项目
国家自然科学基金(82160787)
昆明理工大学科技园有限公司下达项目(2018KF3)
出版年
2024
国家科技期刊平台
NSTL
万方数据