Intelligent fault diagnosis of worm gearbox based on adaptive CNN using amended gorilla troop optimization with quantum gate mutation strategy

The worm gearbox is a power transmission system that has various applications in industries. Being vital element of machinery, it becomes necessary to develop a robust fault diagnosis scheme for worm gearbox. Due to advancements in sensor technology, researchers from academia and industries prefer deep learning models for fault diagnosis. The optimal selection of hyperparameters (HPs) of deep learning models plays a significant role in stable performance. Existing methods mainly focused on manual tunning of these parameters, which is a troublesome process and sometimes leads to inaccurate results. Thus, exploring more sophisticated methods to optimize the HPs automatically is important. In this work, a novel optimization, i.e. amended gorilla troop optimization (AGTO), has been proposed to make the convolutional neural network (CNN) adaptive for extracting the features to identify the worm gearbox defects. Initially, the vibration and acoustic signals are converted into 2D images by the Morlet wavelet function. Then, the initial model of CNN is developed by setting hyperparameters. The search space of each HP is identified and optimized by the developed AGTO algorithm. The classification accuracy has been evaluated by AGTO-CNN, which is further validated by the confusion matrix. The performance of the developed model has also been compared with other models. It has been observed that the proposed AGTO not only achieved the highest degree of recognition accuracy i.e. 98.95 % but also achieved the least standard of deviation of 0.2145 than that of other classifiers. The AGTO algorithm is examined on twenty-three classical benchmark functions and the Wilcoxon test which demonstrates the effectiveness and dominance of the developed optimization algorithm. The results obtained suggested that the AGTO-CNN has the highest diagnostic accuracy, more stable while diagnosing the worm gearbox.(c) 2023 Elsevier B.V. All rights reserved.

Automated summary

This paper proposes a novel optimization method, Amended Gorilla Troop Optimization (AGTO), for the selection of hyperparameters in deep learning models for fault diagnosis. By converting vibration and acoustic signals into 2D images for feature extraction and optimizing the hyperparameters of the CNN model using the AGTO algorithm, stable performance is achieved. The results show that AGTO-CNN has the highest diagnostic accuracy.