Attention-Based Deep Ensemble Net for Large-Scale Online Taxi-Hailing Demand Prediction

Yang Liu, Zhiyuan Liu, Cheng Lyu, Jieping Ye

Research output: Contribution to journalArticlepeer-review

37 Scopus citations

Abstract

How to effectively ensemble different base models is a challenging but extremely valuable task. This study focuses on the construction of an ensemble framework designed for spatiooral data to predict large-scale online taxi-hailing demand, where an attention-based deep ensemble net is designed to enhance the prediction accuracy. We present three attention blocks to model the inter-channel relationship, inter-spatial relationship and position relationship of the feature maps. Then, the attention maps can be multiplied by the input feature map for adaptive feature refinement. The proposed method is a kind of commonly used ensemble method which applies to large-scale spatiooral prediction. Experimental results on city-wide online taxi-hailing demand predictions demonstrate that our proposed attention-based ensemble net is superior to the existing ensemble strategy in terms of the prediction accuracy.

Original languageEnglish
Article number8880638
Pages (from-to)4798-4807
Number of pages10
JournalIEEE Transactions on Intelligent Transportation Systems
Volume21
Issue number11
DOIs
StatePublished - Nov 2020
Externally publishedYes

Keywords

  • attention mechanism
  • demand prediction
  • Ensemble learning

Fingerprint

Dive into the research topics of 'Attention-Based Deep Ensemble Net for Large-Scale Online Taxi-Hailing Demand Prediction'. Together they form a unique fingerprint.

Cite this