Skip to main navigation Skip to search Skip to main content

Real-time Traffic Congestion Prediction: A Novel Online Learning Method with Multi-Head Attention Mechanism and LSTM-Based Integrated Learning

  • Chuanyun Fu
  • , Jiaming Liu
  • , Zhaoyou Lu
  • , Ayinigeer Wumaierjiang
  • , Huahua Liu
  • , Wei Bai
  • Harbin Institute of Technology
  • Sichuan Police College

Research output: Contribution to journalConference articlepeer-review

Abstract

Real-time traffic congestion prediction is essential for proactive traffic management, as it enhances the responsiveness of traffic systems, including route guidance, control, and enforcement. However, the heavy reliance on extensive historical data presents a significant challenge for real-time model updates. To overcome this limitation, this study proposes an advanced online learning framework that integrates a multi-head attention mechanism with LSTM-based ensemble learning. This approach incorporates traffic congestion factors as input features and employs average delay per kilometer as the predictive output. The experimental findings indicate that: 1) the proposed approach successfully enables real-time traffic congestion forecasting, and 2) it demonstrates strong adaptability in dynamic traffic environments.

Original languageEnglish
JournalSAE Technical Papers
DOIs
StatePublished - 10 Dec 2025
Event2025 International Conference on Intelligent Transportation and Future Mobility, ITFM 2025 - Guilin, China
Duration: 11 Apr 202513 Apr 2025

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure

Fingerprint

Dive into the research topics of 'Real-time Traffic Congestion Prediction: A Novel Online Learning Method with Multi-Head Attention Mechanism and LSTM-Based Integrated Learning'. Together they form a unique fingerprint.

Cite this