A Novel Rotating Machinery Fault Diagnosis Method Based on Adaptive Deep Belief Network Structure and Dynamic Learning Rate Under Variable Working Conditions

Abstract

With the development of modern industries, the working environment of rotating machinery has become increasingly complicated. Therefore, it is very meaningful to accurately identify the type of equipment failure under variable operating conditions. This paper presents a rotating machinery fault diagnosis method based on dynamic learning rate deep belief network (DBN) with adaptive structure (PSO-DDBN). Firstly, the wavelet packet energy entropy principle was used to obtain the characteristic matrix of the original data, and then the characteristics of the data under variable conditions were distinguished. Secondly, in order to adjust the structure of DBN, the loss function of DBN was used to construct the convergence function in particle swarm optimization (PSO) adaptive process. The dynamic learning rate strategy was applied to the training process of the network. The network gradient value in each iteration was recorded and the dynamic learning rate function was constructed to achieve the purpose of dynamically adjusting the network learning rate and making the network convergence faster and more stable. Then, the performance of PSO-DDBN was verified by the data of bearing and gearbox under variable conditions. Finally, other intelligent diagnosis algorithms were compared with this method, and the results showed that this method had better universality and fault classification ability.

Keywords

• Deep belief network
• particle swarm optimization
• dynamic learning rate strategy
• multi condition fault diagnosis
• wavelet packet energy entropy