Quantum weighted long short-term memory neural network and its application in state degradation trend prediction of rotating machinery

Feng Li,Wang Xiang,Jiaxu Wang,Xueming Zhou,Baoping Tang
DOI: https://doi.org/10.1016/j.neunet.2018.07.004
IF: 7.8
2018-01-01
Neural Networks
Abstract:Classical long short-term memory neural network (LSTMNN) generally faces the challenges of poor generalization property and low training efficiency in state degradation trend prediction of rotating machinery. In this paper, a novel quantum neural network called quantum weighted long short-term memory neural network (QWLSTMNN) is proposed. First, quantum bits are introduced into the long short-term memory unit to express network weights and activity values. Then, a new learning algorithm based on quantum phase-shift gate and quantum gradient descent is presented to quickly update the quantum parameters of weight qubits and activity qubits. The above characteristics endow QWLSTMNN with better nonlinear approximation capability, higher generalization property and faster convergence speed than LSTMNN. State degradation trend prediction for rolling bearings demonstrates that higher prediction accuracy and higher computational efficiency can be obtained due to the advantages of QWLSTMNN in terms of nonlinear approximation capability, generalization property and convergence speed. It is believed that the proposed method based on QWLSTMNN is effective for state degradation trend prediction of rotating machinery.
What problem does this paper attempt to address?