To main content

Hybrid graph convolution neural network and branch-and-bound optimization for traffic flow forecasting

Abstract

In this study, we combine graph optimization and prediction in a single pipeline to investigate an innovative convolutional graph-based neural network for urban traffic flow prediction in an edge IoT environment. Pre-processing of the linked graph is first performed to remove noise from the set of original road networks of urban traffic data. Outlier detection strategy is used to efficiently explore the road network and remove irrelevant patterns and noise. The resulting graph is then implemented to train an extended graph convolutional neural network to estimate the traffic flow in the city. To accurately tune the hyperparameter values of the proposed framework, a new optimization technique is developed based on branch and bound. For comparison, an intensive evaluation is conducted with multiple datasets and baseline methods. The results show that the proposed framework outperforms the baseline solutions, especially when the number of nodes in the graph is large.
Read the publication

Category

Academic article

Language

English

Author(s)

  • Youcef Djenouri
  • Asma Belhadi
  • Gautam Srivastava
  • Jerry Chun-Wei Lin

Affiliation

  • SINTEF Digital / Mathematics and Cybernetics
  • Kristiania University of Applied Sciences
  • Western Norway University of Applied Sciences
  • China Medical University
  • Lebanese American University
  • Brandon University

Year

2023

Published in

Future Generation Computer Systems

ISSN

0167-739X

Volume

139

Page(s)

100 - 108

View this publication at Norwegian Research Information Repository