Stacking LSTMs to Extract Features in Two-Dimensional Data for Prediction Tasks on Travel Time and Crimes Frequency
Data-based methods have some advantages over other methods for prediction tasks, with domain knowledge is not required. In these methods, Long-short Term Memory has achieved tremendous success in recent years, because its capable of feature extraction of time series. However, there are fewer studies on using Long-short Term Memory to extract the features in spatial dimension. In our work, we attempt to stack Long-short Term Memorys to extract the spatial dimensional features while completing feature extraction of time series for prediction tasks. An multilayer stacking method with multiway inputs and an grid mode stacking method were proposed to extract the features in two dimensions. In the grid mode stacking method, an memory block was proposed to merge the gated cell vectors and each Long-short Term Memory is for one embedding vector separately. Based on two public datasets, we evaluated the proposed methods on travle time prediction and crime prediction. Travel time dataset is provided by Highways England, and crime dataset is provided by the Home Office. Experimental results on the datasets demonstrate that the proposed methods perform well compared to the baseline methods.
Authors
- Xiangdong Ran (ORCID: https://orcid.org/0000-0002-7128-0949)
- Fuyun Deng (ORCID: https://orcid.org/0009-0004-3060-9738)
- Kai Niu (ORCID: https://orcid.org/0009-0009-3357-6130)
Institutions
- Beijing Academy of Artificial Intelligence (CN)
- Beijing Information Science & Technology University (CN)
Publication Details
- Published
- 2024-12-20
- DOI
- https://doi.org/10.1109/acai63924.2024.10899699
- Primary Topic
- Traffic Prediction and Management Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00