Vibration Prediction of Flying IoT Based on LSTM and GRU

Abstract

Drones, flying Internet of Things (IoT), have been widely used in several industrial fields, including rescue, delivery, military, and agriculture. Motors connected to a drone’s propellers play a crucial role in its movement. However, once the motor is damaged, the drone is at risk of falling. Thus, to prevent the drone from falling, an accurate and reliable prediction of motor vibration is necessary. In this study, four types of time series vibration data collected in the time domain from motors are predicted using long short-term memory (LSTM) and gated recurrent unit (GRU), and the accuracy and time efficiency of the predicted and actual vibration waveforms are compared and examined. According to the simulation results, the coefficient of determination, R2, and the root mean square error values obtained from the actual and predicted vibrations by the LSTM and GRU are similar. Furthermore, both the LSTM and GRU show excellent performance in forecasting future motor vibration, but GRU can predict future vibration about 22.79% faster than LSTM.