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].