Intelligent classification framework for gear surface damage and gear type using CNN transfer learning

Abstract

In the mechanical transmission system gear has a very important role, and Gear surface damage is an important factor affecting gear transmission. Regular inspection of each gear and its surface damage is of great significance for ensuring the stable operation of a whole mechanical system. To improve the identification efficiency and accuracy for gears and their surface damage, In this paper, we propose a gear and its surface damage recognition method based on PyTorch deep learning library, expand the dataset using data augmentation and data extension techniques, compare the performance of gear and its surface damage recognition under three typical models, namely AlexNet, VGG16, and ResNet-101, and optimize each training model through migration learning and hyperparameter comparison. The results show that the speed and accuracy of model training can be improved by applying transfer learning. By extending the dataset using data augmentation techniques, the robustness of the network is improved considerably. When the batch size is 6 and the initial learning rate is set to 0.01, the model training effect is the best, and ResNet has higher recognition accuracy and stability than AlexNet and VGG16, which is more suitable for the classification of gears and their surface damage. In order to find the optimal ResNet model, four ResNet models with different number of layers, ResNet-34, ResNet-50, ResNet-101 and ResNet-152, were compared, among which the ResNet-101 model showed the optimal performance in gear surface damage recognition.

Keywords

Convolutional neural network gear surface damage deep learning transfer learning

Get full access to this article

View all access options for this article.

References

Nishiguchi

Yamamura

Hiroi

, et al. Design and fabrication of prestressed concrete box girder bridge with U-shaped segments, 2018.

LeCun

Bengio

Hinton

Deep learning. Nature 2015; 521(7553): 436–444.

Lecun

Bottou

Bengio

, et al. Gradient-based learning applied to document recognition. Proc IEEE 1998; 86(11): 2278–2324.

Krizhevsky

Sutskever

Hinton

GE.

Imagenet classification with deep convolutional neural networks. Advances in neural information processing systems. 2012; pp.84–90.

Simonyan

Zisserman

Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:14091556. 2014.

Zhang

Ren

, et al. Deep residual learning for image recognition. New York, NY: IEEE, 2016.

Singh

Desai

KA.

Automated surface defect detection framework using machine vision and convolutional neural networks. J Intell Manuf 2023; 34(4): 1995–2011.

Yang

Yao

, et al. Extreme Learning Machine with local connections. Neurocomputing 2019; 368: 146–152.

Rozantsev

Salzmann

Fua

Beyond sharing weights for deep domain adaptation. IEEE Trans Pattern Anal Mach Intell 2019; 41: 801–814.

10.

Zhao

Liu

Han

, et al. Pooling the convolutional layers in deep convnets for action recognition. arXiv preprint arXiv:151102126. 2015.

11.

Ama

, et al. Transfer learning using a multi-scale and multi-network ensemble for skin lesion classification - ScienceDirect. Computer Methods and Programs in Biomedicine 2020; 193: 105475.

12.

Miltenović

Rakonjac

Oarcea

, et al. Detection and monitoring of pitting progression on gear tooth flank using Deep Learning. Appl Sci 2022; 12(11): 5327.

13.

Kandel

Castelli

The effect of batch size on the generalizability of the convolutional neural networks on a histopathology dataset. ICT Express 2020; 6: 312–315.

14.

Ren

Cao

Wen

, et al. A modified Elman neural network with a new learning rate scheme. Neurocomputing 2018; 286: 11–18.

15.

Xie

Zhang

Lim

CP.

Evolving CNN-LSTM models for time series prediction using Enhanced Grey Wolf Optimizer. IEEE Access 2020; 8: 161519–161541.

16.

Pan

Yang

A survey on Transfer Learning. IEEE Trans Knowl Data Eng 2010; 22(10): 1345–1359.

17.

Yang

Xiao

Zhang

, et al. Image Data Augmentation for Deep Learning: A Survey. arXiv e-prints, 2022.

18.

Liu

Guo

Bidirectional LSTM with attention mechanism and convolutional layer for text classification. Neurocomputing 2019; 337: 325–338.

19.

Liao

, et al. An image retrieval method for binary images based on DBN and Softmax classifier. IETE Tech Rev 2015; 32(4): 294–303.

20.

Wang

Shi

, et al. Adaptive batch normalization for practical domain adaptation. Pattern Recognit 2018; 80: 109–117.

21.

Wang

Giannakis

Chen

Learning ReLU networks on linearly separable data: algorithm, optimality, and generalization. IEEE Trans Signal Process 2019; 67: 2357–2370.

22.

Wookey

The real-world-weight cross-entropy loss function: modeling the costs of Mislabeling. IEEE Access 2020; 8: 4806–4813.

23.

Weiss

Khoshgoftaar

Wang

A survey of transfer learning. J Big Data 2016; 3(1): 1–40.

24.

Singh

Kumar

Desai

KA.

Comparative assessment of common pre-trained CNNs for vision-based surface defect detection of machined components. Expert Syst Appl 2023; 218: 119623.