Artificial Intelligence and Data Science
Online ISSN : 2435-9262
COMPARISON OF GRADIENT BOOSTING DECISION TREE AND GRAPH NEURAL NETWORK FOR SHORT-TIME SPEED PREDICTION
Riku OGATAToshiyuki MIYAZAKIYoshikazu KIKUCHIYutaro MURANOHiroaki SUGAWARA
Author information
JOURNAL OPEN ACCESS

2023 Volume 4 Issue 2 Pages 154-162

Details
Abstract

By constructing a digital twin at key locations, real-time traffic flow forecasting and dynamic traffic control can be performed to avoid traffic congestion. In this paper, short-time speed prediction was conducted using open data from England in anticipation of the above applications. Gradient Boosting Decision Tree (GBDT) and Graph Neural Network (GNN) were used for the model, and a comparison was made between the two. The comparison results for the entire 170 target locations showed that the GNN was superior, but the evaluation of individual locations revealed that there were several locations where GBDT was superior. The results also confirmed the GNN was superior at the points where time contributed significantly, and confirmed that the addition of data from other points, which was judged to be valid based on the GNN adjacency matrix, contributed to improving the GBDT accuracy at these points. Finally, the use of GBDT and GNN is discussed.

Content from these authors
© 2023 Japan Society of Civil Engineers
Previous article Next article
feedback
Top