Intelligent fault diagnosis of rotating machinery using lightweight network with modified tree‐structured parzen estimators

Abstract

Deep learning-based methods have been widely used in the field of rotating machinery fault diagnosis. It is of practical significance to improve the calculation speed of the model on the premise of ensuring accuracy, so as to realise real-time fault diagnosis. However, designing an efficient and lightweight fault diagnosis network requires expert knowledge to determine the network structure and adjust the hyperparameters of the network, which is time-consuming and laborious. In order to design fault diagnosis networks considering both time and accuracy effortlessly, a novel lightweight network with modified treestructured parzen estimators (LN-MT) is proposed for intelligent fault diagnosis of rotating machinery. Firstly, a lightweight framework based on global average pooling and group convolution is proposed, and a hyperparameter optimisation (HPO) method based on Bayesian optimisation called tree-structured parzen estimator is utilised to automatically search the optimal hyperparameters for the fault diagnosis task. The objective of the HPO algorithm is the weighting of accuracy and calculating time, so as to find models that balance both time and accuracy. The results of comparison experiments indicate that LN-MT can achieve superior fault diagnosis accuracies with few trainable parameters and less calculating time. K E Y W O R D S convolutional neural network, fault diagnosis, hyperparameter optimisation, neural architecture search | INTRODUCTIONModern electromechanical equipment is playing an important role in manufacturing and industry. Gears and bearings are key components of rotating machinery [1,2], the failure of which will greatly affect the performance of the machine, resulting in serious security risk and economic loss [3][4][5]. In order to ensure the continuous and stable operation of the equipment, an algorithm that can timely diagnose the failure of rotating machinery is required [6][7][8], so as to reduce economic losses. Hence, it is of great significance to develop a fault diagnosis algorithm for rotating machinery [9].

Machine Learning Based Precision Agriculture using Ensemble Classification with TPE Model

“…Training data set (𝑋𝑋 𝑖𝑖 , 𝑌𝑌 𝑖𝑖 ), 𝑖𝑖 = 1,2,3, … … , 𝑁𝑁 where 𝑋𝑋𝑖𝑖 refers to a continuous-valued vector in 𝑛𝑛 dimensions, and 𝑌𝑌 𝑖𝑖 = {0,1} The corresponding class identification is denoted by "0" for normal and "1" for abnormalities. The proposed method consists of two parts: training and testing [25]. Following method, the training space is partitioned into k separate clusters C_1, C_2, C_3,..., C_K.…”

Turbulence along the Runway Glide Path: The Invisible Hazard Assessment Based on a Wind Tunnel Study and Interpretable TPE-Optimized KTBoost Approach

“…The TPE algorithm is a sequential model-based global optimization algorithm that effectively determines the hyperparameters of a machine learning model. It was developed to overcome the shortcomings of the traditional Bayesian Optimization approaches when dealing with categorical and conditional hyperparameters by introducing Parzen window estimators, thereby enhancing the performance of the hyperparameter search strategy [26,27]. Employing the Parzen-window density estimation, the TPE algorithm produces probability density functions within a hyperparametric search space.…”