基于贝叶斯优化与改进LeNet-5的滚动轴承故障诊断

doi:10.3969/j.issn.1000-1158.2022.07.12

摘要
图/表
参考文献(22)
相关文章 (15)

全文: PDF (1282 KB) HTML (1 KB)
输出: BibTeX | EndNote (RIS)

摘要考虑到卷积神经网络在滚动轴承故障诊断中存在网络结构难以确定、训练次数过多、时间过长等问题,设计了一种贝叶斯优化改进LeNet-5算法,以及采用该算法构建的轴承故障诊断模型。采用贝叶斯优化训练过程中学习率等超参数,多种故障轴承的振动信号直接作为改进LeNet-5网络的输入,对池化输出采用批归一化处理和改进池化层激活函数防止过拟合,利用全局平均池化层替代全连接层提高改进LeNet-5网络的泛化能力,用Softmax分类器实现滚动轴承故障的分类。通过轴承数据库开展实验,实验表明,该算法构建的轴承故障诊断模型在训练集上准确率为99.94%,验证集上的准确率为99.89%,测试集准确率也达到99.65%,与一维卷积神经网络和二维卷积神经网络对比分析,基于贝叶斯优化改进LeNet-5算法构建的轴承故障诊断模型在滚动轴承的故障诊断模型具有更高的准确率,更少的训练次数和训练时间。

	服务

	把本文推荐给朋友
	加入我的书架
	加入引用管理器
	E-mail Alert
	RSS
	作者相关文章
	汤亮
	凡焱峰
	徐适斐
	蔡凯翼

关键词 ：计量学, 滚动轴承, 故障诊断, 改进LeNet-5网络, 贝叶斯优化, 深度学习

Abstract：Considering that the network structure is difficult to determine, too many training times and too long training time exist in rolling bearing fault diagnosis by convolutional neural network (CNN), a Bayesian optimization algorithm for improving LeNet-5 is designed, and a bearing fault diagnosis model constructed by this algorithm is presented.The learning rate and other hyperparameters in the Bayesian optimization training process are adopted, the original vibration signals of various fault bearings are directly used as the input of the improved LeNet-5 network, batch normalization is adopted for the pooled output and the activation function of the improved pool layer is adopted to prevent over-fitting, and the global average pool layer is used to replace the full connection layer to improve the generalization ability of the improved LeNet-5 network, the fault classification of rolling bearing is realized by softmax classifier.The experimental results show the accuracy of bearing fault diagnosis model constructed by this algorithm training set is 99.94%, the accuracy of verification set is 99.89%, and the accuracy of test set is 99.65%, compared with 1D-CNN and 2D-CNN, bearing fault diagnosis model based on Bayesian optimization and improved LeNet-5 algorithm has higher accuracy, less training times and training time in the fault diagnosis model of rolling bearings.

Key words： metrology rolling bearings fault diagnosis improved LeNet-5 network Bayesian optimization deep learning

收稿日期: 2021-08-18 发布日期: 2022-07-18

PACS:	TB936
	TB973

基金资助:国家自然科学基金(61976083)

作者简介: 汤亮(1978-),男,湖北工业大学副教授,博士,主要研究方向为机械创新设计。Email:tangliang@hbut.edu.cn

引用本文:

汤亮,凡焱峰,徐适斐,蔡凯翼. 基于贝叶斯优化与改进LeNet-5的滚动轴承故障诊断[J]. 计量学报, 2022, 43(7): 913-919.
TANG Liang,FAN Yan-feng,XU Shi-fei,CAI Kai-yi. Fault Diagnosis of Rolling Bearing Based on Bayesian Optimization and Improved LeNet-5. Acta Metrologica Sinica, 2022, 43(7): 913-919.

链接本文:

http://jlxb.china-csm.org:81/Jwk_jlxb/CN/10.3969/j.issn.1000-1158.2022.07.12 或 http://jlxb.china-csm.org:81/Jwk_jlxb/CN/Y2022/V43/I7/913

［1］谢天雨, 董绍江. 基于改进CNN的噪声以及变负载条件下滚动轴承故障诊断方法［J］. 噪声与振动控制, 2021, 41 (2): 111-117.
Xie T Y, Dong S J. Fault Diagnosis Method of Rolling Bearings Based on Improved CNN under Noise and Variable Load Conditions ［J］. Noise and Vibration Control, 2021, 41 (2): 111-117.
［2］郑新. 基于EMD与PCA分析的滚动轴承故障特征研究［J］. 机械传动, 2016, 40 (1): 54-58 +63.
Zheng X. Fault Feature Research of Rolling Bearing based on Empirical Mode Decomposition and Principle Component Analysis［J］. Noise and Vibration Control, 2016, 40 (1): 54-58 +63.
［3］孟宗, 吕蒙, 殷娜, 等. 基于改进变分模态分解的滚动轴承故障诊断方法［J］. 计量学报, 2020, 41 (6): 717-723.
Meng Z, Lu M, Yin N, et al. Fault Diagnosis Method of Ｒolling Bearing Based on Improved Variational Mode Decomposition ［J］. Acta Metrologica Sinica, 2020, 41 (6): 717-723.
［4］周海军, 王超伟, 周国敬, 等. 基于随机森林的离心泵滚动轴承故障诊断［J］. 中国舰船研究, 2020, 15 (3): 129-135.
Zhou H J, Wang C W, Zhou G J, et al. Fault diagnosis of antifriction bearing of centrifugal pump based on random forest ［J］. Chinese Journal of Ship Research, 2020, 15 (3): 129-135.
［5］张晴晴, 刘勇, 潘接林, 等. 基于卷积神经网络的连续语音识别［J］. 工程科学学报, 2015, 37 (9): 1212-1217.
Zhang Q Q, Liu Y, Pan J L, et al. Continuous speech recognition based on convolutional neural network ［J］. Advanced Engineering Sciences, 2015, 37 (9): 1212-1217.
［6］郭保苏, 庄集超, 吴凤和, 等. 基于CT图像卷积神经网络处理的新冠肺炎检测［J］. 计量学报, 2021, 42 (4): 537-544.
Guo B S, Zhuang J C, Wu F H. Detection of Novel Coronavirus Pneumonia Based on CT Image with Convol-utional Neural Network Processing ［J］. Acta Metrologica Sinica, 2021, 42 (4): 537-544.
［7］熊剑, 邓松, 时大方. 基于改进残差网络的滚动轴承故障诊断［J］. 轴承, 2020 (11): 50-55.
Xiong J, Deng S, Shi D F. Fault Diagnosis for Rolling Bearings Based on Improved Residual Network ［J］. Bearing, 2020 (11): 50-55.
［8］Lu C, Wang Z Y, Zhou B. Intelligent fault diagnosis of rolling bearing using hierarchical convolutional network based health state classification ［J］. Advanced Engineering Informatics, 2017, 32:139-151.
［9］Wen L, Li X Y, Gao L, et al. A new convolutional neural network-based data-driven fault diagnosis method ［J］. IEEE Transactions on Industria Electronics, 2018, 65 (7): 5990-5998.
［10］刘星辰, 周奇才, 赵炯, 等. 一维卷积神经网络实时抗噪故障诊断算法［J］. 哈尔滨工业大学学报, 2019, 51 (7): 89-95.
Liu X C, Zhou Q C, Zhao J, et al. Real-time and anti-noise fault diagnosis algorithm based on 1-D convolu-tional neural network ［J］. Journal of Harbin Institute of Technology, 2019, 51 (7): 89-95.
［11］Zhang W, Li C H, Peng G L, et al. A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load ［J］. Mechanical Systems and Signal Processing, 2018, 100: 439-453.
［12］Tra V, Kim J, Khan S A, et al. Bearing Fault Diagnosis under Variable Speed Using Convolutional Neural Networks and the Stochastic Diagonal Levenberg-Marquardt Algorithm ［J］. Sensors, 2017, 17 (12): 2834.
［13］金江涛, 许子非, 李春, 等. 基于VMD能量熵与优化支持向量机的轴承故障诊断［J］. 计量学报, 2021, 42 (7): 898-905.
Jin J T, Xu Z F, Li C, et al. Bearing Fault Diagnosis Method Based on VMD and Convolutional Neural Net-work Undervarying Operation Conditions ［J］. Acta Metrologica Sinica, 2021, 42 (7): 898-905.
［14］张博, 张苗辉, 陈运忠. 基于空间金字塔池化和深度卷积神经网络的作物害虫识别［J］. 农业工程学报, 2019, 35 (19): 209-215.
Zhang B, Zhang M H, Chen Y Z. Crop pest identification based on spatial pyramid pooling and deep convolution neural network ［J］. Transactions of the Chinese Society of Agricultural Engineering, 2019, 35 (19): 209-215.
［15］张力超, 马蓉, 张垚鑫. 改进的LeNet-5模型在苹果图像识别中的应用［J］. 计算机工程与设计, 2018, 39 (11): 3570- 3575.
Zhang L C, Ma R, Zhang Y X. Application of improved LeNet-5 model in apple image recognition ［J］. Computer Engineering and Design, 2018, 39 (11): 3570-3575.
［16］崔佳旭, 杨博. 贝叶斯优化方法和应用综述［J］. 软件学报, 2018, 29 (10): 3068-3090.
Cui J X, Yang B. Survey on Bayesian Optimization Methodology and Applications ［J］. Journal of Software, 2018, 29 (10): 3068-3090.
［17］朱威, 屈景怡, 吴仁彪. 结合批归一化的直通卷积神经网络图像分类算法［J］. 计算机辅助设计与图形学学报, 2017, 29 (9): 1650-1657.
Zhu W, Qu J Y, Wu R B. Straight Convolutional Neural Networks Algorithm Based on Batch Normaliz-ation for Image Classification ［J］. Journal of Computer-Aided Design & Computer Graphics, 2017, 29 (9): 1650-1657.
［18］王秀席, 王茂宁, 张建伟, 等. 基于改进的卷积神经网络LeNet-5的车型识别方法［J］. 计算机应用研究, 2018, 35 (7): 2215-2218.
Wang X X, Wang M N, Zhang J W. Vehicle type reco-gnition based on improved convolutional neural network LeNet-5 ［J］. Application Research of Computers, 2018, 35 (7): 2215-2218.
［19］Zhang B X, Zhao Q, Feng W Q, et al. Alpha- MEX: A Smarter Global Pooling Method for Convolutional Neural Networks ［J］. Neurocomputing, 2018,321 (10): 36-48.
［20］Vincent P, Larochelle H, Lajoie I. Stacked denoising autoencoders: learning useful repre-sent ations in a deep network with a local denoising criterion ［J］. Journal of Machine Learning Research, 2010, 11 (12): 3371-34.
［21］曲建岭, 余路, 袁涛, 等. 基于一维卷积神经网络的滚动轴承自适应故障诊断算法［J］. 仪器仪表学报, 2018, 39 (7): 134-143.
Qu J L, Yu L, Yuan T. Adaptive fault diagnosis algorithm for rolling bearings based on one-Dimensional convolutional neural network ［J］. Chinese Journal of Scientific Instrument, 2018, 39 (7): 134-143.
［22］李益兵, 马建波, 江丽. 基于SFLA改进卷积神经网络的滚动轴承故障诊断［J］. 振动与冲击, 2020, 39 (24): 187-193.
Li Y B, Ma J B, Jiang L. Fault diagnosis of rolling bearing based on an improved convolutional neural network using SFLA ［J］. Journal of Vibration and Shock, 2020, 39 (24): 187-193.