Macro Micro attention Guided lstm for gear life prediction

Macro-Micro attention-guided LSTM for gear life prediction

In the mechanical transmission system, the gear is one of the most widely used transmission components. The fail-ure of the gear will cause serious accidents and huge economic loss. Therefore, the remaining life prediction of the gear is of great importance. In order to accurately predict the remaining life of the gear, a new type of long-short-term memory neural network with macro-micro attention is proposed. First, some typical time-domain and frequency-domain characteristics of vibration signals are calculated respectively, such as the maximum value, the absolute mean value, the standard deviation, the kurtosis and so on. Then, the principal component of these char-acteristics is extracted by the isometric mapping method. The importance of fusional characteristic information is filtered via a proposed macro-micro attention mechanism, so that the input weight of neural network data and recur-sive data can reach multi-level real-time amplification. With the new long short-term memory neural network, the health characteristics of gear vibration signals can be predicted based on the known fusion features. The ex-perimental results show that this method can predict the remaining life of gears and bearings well, and it has higher prediction accuracy than the conventional prediction methods.

For the detail, please see: Yi Qin, Sheng Xiang, Yi Chai, Haizhou Chen. Macroscopic-microscopic attention in LSTM networks based on fusion features for gear remaining life prediction. IEEE Transactions on Industrial Electronics, 2020, 67(12): 10865-10875.

Code Download