Constrained Mixed-ARDE for Optimizing LSTM Hyperparameters in Traffic Flow Forecasting
摘要
This work contributes to the significant potential of Long Short-Term Memory (LSTM) for traffic flow prediction, which is highly dependent on complex; nonlinear and stochastic data. We propose an LSTM hyperparameters optimization approach using a variant of the Adaptive Reinitialized Differential Evolution (ARDE) algorithm; handling mixed types of genes and boundary constraints. Experiments of the proposed mixed-ARDE-based LSTM algorithm on three analyzed datasets demonstrate a considerable accuracy compared to baseline models, namely DE-LSTM, GA-LSTM and keras-tuner-LSTM.