Self-updating DT model for UAV energy prediction by online LSTM


This is a GitHub repo for the submitted paper "A Self-Evolving Digital Twin System Architecture for Indoor UAV Management based on Online Deep Learning" to IROS 2024. In this repo, we will release the realization of the proposed method called online LSTM, collected UAV data, and other comparison methods.

The parameter settings of the comparison methods

Models Model setting
MLP An input layer with dimensions [11, 50], two hidden layers with dimensions [50, 50], and one output layer with dimensions [50, 1]. The Adam optimization method is employed with a learning rate of 0.001.
LSTM An LSTM layer with dimensions [11, 50] and a linear layer with dimensions [50, 1]. The Adam optimization method is employed with a learning rate of 0.001.
Transformer An input embedding layer with dimensions [11, 500], a positional encoding layer with dimensions 400, two transformer encoding layers (400 dimensions, 10 attention heads, and a dropout probability of 0.1), and a linear decoder layer with dimensions [400, 1]. The AdamW optimization method is employed with a learning rate of 0.005.
Online MLP The neural network is aligned with the MLP method.
OSELM The random layer comprises 70 hidden units, employing the "sigmoid" activation function.
SRPRegressor The number of members in the ensemble is 20.
OXTRegressor The number of trees in the ensemble is 20.
Online LSTM The neural network is aligned with LSTM method.