1. School of Mechanical Engineering, Hubei University of Technology, Wuhan, Hubei 430068, China
2. Hubei Engineering Research Center for Manufacturing Innovation Method, Wuhan, Hubei 430068, China
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.
［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.
2022, Vol. 43 
