An evaluation of HTM and LSTM for short-term arterial traffic flow prediction

Jonathan Mackenzie, John Roddick, Rocco Zito

Research output: Contribution to journalArticlepeer-review

127 Citations (Scopus)

Abstract

Recent years have seen the emergence of two significant technologies: big data systems capable of storing, retrieving, and processing large amounts of data, and machine learning algorithms capable of learning and predicting complex sequences. In combination, these technologies present new opportunities to leverage the increasingly large amounts of traffic volume data to improve traffic flow prediction and the detection of anomalous traffic flows. In this paper, we investigate and evaluate the use of hierarchical temporal memory (HTM) for short-term prediction of traffic flows over real-world Sydney Coordinated Adaptive Traffic System data on arterial roads in the Adelaide metropolitan area in South Australia. Results are compared with those from long-short-term memory (LSTM). Extended experimentation with LSTM network configurations in both batch learning and online learning modes provide results with superior predictive performance over previous usage of LSTM and other deep learning techniques for short-term traffic flow prediction. In addition, we argue that HTM has potential as an effective tool for short term traffic flow prediction with results on par with LSTM and improvements when traffic flow distributions change.

Original languageEnglish
Article number8424074
Pages (from-to)1847-1857
Number of pages11
JournalIEEE Transactions on Intelligent Transportation Systems
Volume20
Issue number5
Early online date1 Aug 2018
DOIs
Publication statusPublished - 1 May 2019

Keywords

  • Arterial road networks
  • hierarchical temporal memory (HTM)
  • intelligent transportation systems
  • long short-term memory (LSTM)
  • traffic-flow prediction

Fingerprint

Dive into the research topics of 'An evaluation of HTM and LSTM for short-term arterial traffic flow prediction'. Together they form a unique fingerprint.

Cite this