Bearing Fault Reconstruction Diagnosis Method Based on ResNet-152 with Multi-Scale Stacked Receptive Field
Abstract
:1. Introduction
- (1)
- Three-layers stacked convolutional kernels are inserted into ultra-deep ResNets to replace large-size or less-layers convolutional kernels to improve the nonlinear representation of feature images;
- (2)
- The fault datasets are reconstructed to increase the data scale and retain the temporal features in the fault data, while reducing the difficulty of the convolution process;
- (3)
- Research on axle box bearings for subway trains to improve the efficiency and accuracy of diagnosis of this component.
2. Basic Components
2.1. Basic Structure of Residual Neural Networks
2.2. Insertion of Multi-Scale Superimposed Receptive Field
3. Design of Fundamental Architectures for ResNet-152-MSRF
4. Experiment Results
4.1. Data Collection and Processing
4.2. Hyperparameters Setup
4.3. Comparison of Six Target Diagnostic Models
4.4. Comparison between ResNet-152-MSRF and ResNet-50-MSRF
5. Conclusions
- (1)
- Evidenced by the experiments, ResNet-50 and ResNet-152 improved by 24.99%, 37.69%, 26.13% and 38.83% relative to VGG-16 and VGG-19, respectively. Additionally, the result indicates that networks with RBB are more suitable for large-scale deep feature extraction;
- (2)
- Evidenced by the data reconstruction, the scale of the obtained data is increased by about 42.88 times compared to the previous 1D time series signal, which is effective for data enhancement;
- (3)
- By embedding a multi-layered receptive field, the developed ReNet-152-MSRF enhances the accuracy by 9.07% compared to ResNet-152, and time cost increases non-significantly. ResNet-152-MSRF has a 2.67% and 1.87% higher average diagnostic accuracy, respectively, than ResNet-50-MSRF in different tasks. This suggests that deeper networks do not necessarily affect accuracy and perform well when trained on reconstructed fault data.
Author Contributions
Funding
Acknowledgments
Conflicts of Interest
References
- Liu, G.F.; Hou, Z.S. Adaptive Iterative Learning Control for Subway Trains Using Multiple-Point-Mass Dynamic Model under Speed Constraint. IEEE Trans. Intell. Transp. Syst. 2021, 22, 1388–1400. [Google Scholar] [CrossRef]
- Wang, Y.; Yang, Z.; Pang, J. Statistical analysis of urban rail transit lines in 2017 China-Express delivery of annual report on urban rail transit. Urban Mass Transit 2018, 21, 1–6. [Google Scholar]
- Sun, J.W.; Zhang, A.; Zhou, Z.X. Research on Urban Rail Transit Promoting the Process of Urbanization. China Soft Sci. 2020, 6, 96–111. [Google Scholar]
- Zhang, D.; Entezami, M.; Stewart, E.; Roberts, C.; Yu, D. Adaptive fault feature extraction from wayside acoustic signals from train bearings. J. Sound Vib. 2018, 425, 221–238. [Google Scholar] [CrossRef]
- Arturo, G.; Roberto, P.; Paul, B. Sustainable urban rail systems: Strategies and technologies for optimal management of regenerative braking energy. Energy Convers. Manag. 2013, 75, 1879–2227. [Google Scholar]
- Cao, H.; Fan, F.; Zhou, K.; He, Z. Wheel-bearing fault diagnosis of trains using empirical wavelet transform. Measurement 2016, 82, 439–449. [Google Scholar] [CrossRef]
- Zhang, Z.; Entezami, M.; Stewart, E.; Roberts, C. Enhanced fault diagnosis of roller bearing elements using a combination of empirical mode decomposition and minimum entropy deconvolution. J. Mech. Eng. Sci. 2017, 231, 655–671. [Google Scholar] [CrossRef]
- Lei, Y.; Lin, J.; Zuo, M.J.; He, Z. Condition monitoring and fault diagnosis of planetary gearboxes: A review. Measurement 2014, 48, 292–305. [Google Scholar] [CrossRef]
- Cui, M.; Wang, Y.; Lin, X.; Zhong, M. Fault Diagnosis of Rolling Bearings Based on an Improved Stack Autoencoder and Support Vector Machine. IEEE Sens. J. 2021, 21, 4927–4937. [Google Scholar] [CrossRef]
- Rai, A.; Upadhyay, S.H. A review on signal processing techniques utilized in the fault diagnosis of rolling element bearings. Tribol. Int. 2016, 96, 289–306. [Google Scholar] [CrossRef]
- Liu, R.; Yang, B.; Zio, E.; Chen, X. Artificial intelligence for fault diagnosis of rotating machinery: A review. Mech. Syst. Signal Processing 2018, 108, 33–47. [Google Scholar] [CrossRef]
- Roy, S.; Menapace, W.; Oei, S.; Luijten, B.; Fini, E.; Saltori, C.; Huijben, I.; Chennakeshava, N.; Mento, F.; Sentelli, A.; et al. Deep Learning for Classification and Localization of COVID-19 Markers in Point-of-Care Lung Ultrasound. IEEE Trans. Med. Imaging 2020, 39, 2676–2687. [Google Scholar] [CrossRef]
- Baltruschat, I.M.; Nickisch, H.; Grass, M.; Knopp, T.; Saalbach, A. Comparison of Deep Learning Approaches for Multi-Label Chest X-Ray Classification. Sci. Rep. 2019, 9, 6381. [Google Scholar] [CrossRef] [Green Version]
- Lee, K.B.; Shin, H.S. An Application of a Deep Learning Algorithm for Automatic Detection of Unexpected Accidents Under Bad CCTV Monitoring Conditions in Tunnels. In Proceedings of the 2019 International Conference on Deep Learning and Machine Learning in Emerging Application (Deep-ML), Istanbul, Turkey, 26–28 August 2019; pp. 7–11. [Google Scholar]
- Dikbayir, H.S.; Ïbrahim Bülbül, H. Deep Learning Based Vehicle Detection from Aerial Images. In Proceedings of the 2020 19th IEEE International Conference on Machine Learning and Applications (ICMLA), Miami, FL, USA, 14–17 December 2020; pp. 956–960. [Google Scholar]
- Shao, H.D.; Jiang, H.K.; Zhang, X.; Niu, M.G. Rolling bearing fault diagnosis using an optimization deep belief network. Meas. Sci. Technol. 2015, 26, 115002. [Google Scholar] [CrossRef]
- Goodfellow, I.; Benigo, Y.; Courville, A. Deep Learning; MIT Press: Cambridge, MA, USA, 2016. [Google Scholar]
- He, K.M.; Zhang, X.; Ren, S.; Sun, J. Deep Residual Learning for image recognition. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 27–30 June 2016; pp. 770–778. [Google Scholar]
- Gao, S.H.; Cheng, M.M.; Zhao, K.; Zhang, X.Y.; Yang, M.H.; Torr, P. Res2Net: A New Multi-Scale Backbone Architecture. IEEE Trans. Pattern Anal. Mach. Intell. 2021, 43, 652–662. [Google Scholar] [CrossRef] [Green Version]
- Paoletti, M.E.; Haut, J.M.; Fernandez-Beltran, R.; Plaza, J.; Plaza, A.J.; Pla, F. Deep Pyramidal Residual Networks for Spectral–Spatial Hyperspectral Image Classification. IEEE Trans. Geosci. Remote Sens. 2019, 57, 740–754. [Google Scholar] [CrossRef]
- Li, B.; He, Y. An Improved ResNet Based on the Adjustable Shortcut Connections. IEEE Access 2018, 6, 18967–18974. [Google Scholar] [CrossRef]
- Wen, L.; Li, X.; Gao, L. A transfer convolutional neural network for fault diagnosis based on ResNet-50. Neural Comput. Appl. 2020, 32, 6111–6124. [Google Scholar] [CrossRef]
- Rasp, S.; Thuerey, N. Data-Driven Medium-Range Weather Prediction with a Resnet Pretrained on Climate Simulations: A New Model for WeatherBench. J. Adv. Modeling Earth Syst. 2021, 13, 1942–2466. [Google Scholar] [CrossRef]
- Zhou, C.; Song, J.; Zhou, S.; Zhang, Z.; Xing, J. COVID-19 Detection Based on Image Regrouping and ResNet-SVM Using Chest X-Ray Images. IEEE Access 2021, 9, 81902–81912. [Google Scholar] [CrossRef]
- Zhao, M.; Kang, M.; Tang, B.; Pecht, M. Deep Residual Networks with Dynamically Weighted Wavelet Coefficients for Fault Diagnosis of Planetary Gearboxes. IEEE Trans. Ind. Electron. 2018, 65, 4290–4300. [Google Scholar] [CrossRef]
- Zhao, M.; Zhong, S.; Fu, X.; Tang, B.; Dong, S.; Pecht, M. Deep Residual Networks with Adaptively Parametric Rectifier Linear Units for Fault Diagnosis. IEEE Trans. Ind. Electron. 2021, 68, 2587–2597. [Google Scholar] [CrossRef]
- Zhao, M.; Zhong, S.; Fu, X.; Tang, B.; Pecht, M. Deep Residual Shrinkage Networks for Fault Diagnosis. IEEE Trans. Ind. Inform. 2020, 16, 4681–4690. [Google Scholar] [CrossRef]
- Wu, S.T.; Zhong, S.H.; Liu, Y. Deep residual learning for image steganalysis. Multimed. Tools Appl. 2018, 77, 10437–10453. [Google Scholar] [CrossRef]
- Loffe, S.; Szegedy, C. Batch normalization: Accelerating deep network training by reducing internal covariate shift. In Proceedings of the 32nd International Conference on Machine Learning, Lille, France, 6–11 July 2015; pp. 448–456. [Google Scholar]
- Wu, Y.X.; He, K.M. Group Normalization. Int. J. Comput. Vis. 2020, 128, 742–755. [Google Scholar] [CrossRef]
3 × 3 + 3 × 3 | 5 × 5 | 3 × 3 + 3 × 3 + 3 × 3 | 7 × 7 | |
---|---|---|---|---|
VGG-16 | 18,952,131 | 18,310,787 | 19,137,027 | 19,102,595 |
VGG-19 | 53,595,203 | 53,544,835 | 53,779,715 | 53,746,051 |
ResNet-50 | 762,691 | 714,703 | 1,234,093 | 1,197,084 |
ResNet-152 | 2,419,171 | 2,397,061 | 6,679,779 | 6,434,577 |
Category | Health Conditions | Sampling Points (per Series) | Series | Total Sampling Points | Rotation Speed (rpm) | Loads (KN) |
---|---|---|---|---|---|---|
1 | Roller fault | 122,581 | 16 | 1,961,296 | 1752 | 36/72 |
2 | Inner raceway fault | 125,049 | 16 | 2,000,784 | 1751 | 36/72 |
3 | Outer raceway fault | 122,514 | 16 | 1,960,224 | 1751 | 36/72 |
4 | Normal | 122,500 | 16 | 1,960,000 | 1750 | 36/72 |
5 | Total | - | 64 | 7,882,304 | - | 36/72 |
Fault Category | Roller Fault | Inner Raceway Fault | Outer Raceway Fault | Normal | Total |
---|---|---|---|---|---|
Sample size | 5620 | 5742 | 5505 | 4245 | 21,112 |
Image size | 128 × 128 | 128 × 128 | 128 × 128 | 128 × 128 | 128 × 128 |
Number of feature pixel points | 92,078,080 | 94,076,928 | 90,193,920 | 69,550,080 | 345,899,008 |
Components | VGG-16 | VGG-19 | ResNet-50 | ResNet-152 | ResNet-50-MSRF | ResNet-152-MSRF |
---|---|---|---|---|---|---|
Input | 128 × 128 × 3 | 128 × 128 × 3 | 128 × 128 × 3 | 128 × 128 × 3 | 128 × 128 × 3 | 128 × 128 × 3 |
Conv_2 | 13 | 16 | 50 | 152 | 50 | 152 |
Conv_kernel | (3 × 3, 1) | (3 × 3, 1) | (3 × 3, 2) | (3 × 3, 2) | (3 × 3 × 3, 2) | (3 × 3 × 3, 2) |
Strides | 1 | 1 | 1 | 1 | 1 | 1 |
BN | 15 | 18 | 48 | 150 | 48 | 150 |
Activation function_1 | ReLU | ReLU | ReLU | ReLU | ReLU | ReLU |
RBU | 0 | 0 | 1 | 1 | 1 | 1 |
Activation function_2 | softmax | softmax | softmax | softmax | softmax | softmax |
FC | 3 | 3 | 2 | 2 | 2 | 2 |
Loss function | Category_crossentropy | Category_crossentropy | Category_crossentropy | Category_crossentropy | Category_crossentropy | Category_crossentropy |
output | 4 | 4 | 4 | 4 | 4 | 4 |
Optimizer | Adam | Adam | Adam | Adam | Adam | Adam |
Lr | 0.0001 | 0.0001 | 0.0001 | 0.0001 | 0.0001 | 0.0001 |
Dropout | 0.5 | 0.5 | 0.5 | 0.5 | 0.5 | 0.5 |
Method | 2 Classification Accuracy (%) | 3 Classification Accuracy (%) | 4 Classification Accuracy (%) |
---|---|---|---|
VGG-16 | |||
VGG-19 | |||
ResNet-50 | |||
ResNet-152 | |||
ResNet-50-MSRF | |||
ResNet-152-MSRF |
Method | Training/Testing Batch Size | Epochs | Total Calculation Time (s) | Calculation Time per Step (s) |
---|---|---|---|---|
VGG-16 | 20/10 | 100 | 5019 | 50.19 |
VGG-19 | 20/10 | 100 | 6467 | 64.67 |
ResNet-50 | 20/10 | 100 | 2536 | 25.36 |
ResNet-152 | 20/10 | 100 | 6974 | 69.74 |
ResNet-50-MSRF | 20/10 | 100 | 2744 | 27.44 |
ResNet-152-MSRF | 20/10 | 100 | 7363 | 73.63 |
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations. |
© 2022 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
Share and Cite
Yu, H.; Miao, X.; Wang, H. Bearing Fault Reconstruction Diagnosis Method Based on ResNet-152 with Multi-Scale Stacked Receptive Field. Sensors 2022, 22, 1705. https://doi.org/10.3390/s22051705
Yu H, Miao X, Wang H. Bearing Fault Reconstruction Diagnosis Method Based on ResNet-152 with Multi-Scale Stacked Receptive Field. Sensors. 2022; 22(5):1705. https://doi.org/10.3390/s22051705
Chicago/Turabian StyleYu, Hu, Xiaodong Miao, and Hua Wang. 2022. "Bearing Fault Reconstruction Diagnosis Method Based on ResNet-152 with Multi-Scale Stacked Receptive Field" Sensors 22, no. 5: 1705. https://doi.org/10.3390/s22051705
APA StyleYu, H., Miao, X., & Wang, H. (2022). Bearing Fault Reconstruction Diagnosis Method Based on ResNet-152 with Multi-Scale Stacked Receptive Field. Sensors, 22(5), 1705. https://doi.org/10.3390/s22051705