Few-Shot Traffic Prediction with Graph Networks using Locale as Relational Inductive Biases

by   Mingxi Li, et al.

Accurate short-term traffic prediction plays a pivotal role in various smart mobility operation and management systems. Currently, most of the state-of-the-art prediction models are based on graph neural networks (GNNs), and the required training samples are proportional to the size of the traffic network. In many cities, the available amount of traffic data is substantially below the minimum requirement due to the data collection expense. It is still an open question to develop traffic prediction models with a small size of training data on large-scale networks. We notice that the traffic states of a node for the near future only depend on the traffic states of its localized neighborhoods, which can be represented using the graph relational inductive biases. In view of this, this paper develops a graph network (GN)-based deep learning model LocaleGn that depicts the traffic dynamics using localized data aggregating and updating functions, as well as the node-wise recurrent neural networks. LocaleGn is a light-weighted model designed for training on few samples without over-fitting, and hence it can solve the problem of few-shot traffic prediction. The proposed model is examined on predicting both traffic speed and flow with six datasets, and the experimental results demonstrate that LocaleGn outperforms existing state-of-the-art baseline models. It is also demonstrated that the learned knowledge from LocaleGn can be transferred across cities. The research outcomes can help to develop light-weighted traffic prediction systems, especially for cities lacking in historically archived traffic data.


page 1

page 5


Adversarial Diffusion Attacks on Graph-based Traffic Prediction Models

Real-time traffic prediction models play a pivotal role in smart mobilit...

Domain Adversarial Spatial-Temporal Network: A Transferable Framework for Short-term Traffic Forecasting across Cities

Accurate real-time traffic forecast is critical for intelligent transpor...

A spatial-temporal short-term traffic flow prediction model based on dynamical-learning graph convolution mechanism

Short-term traffic flow prediction is a vital branch of the Intelligent ...

Traffic Prediction with Transfer Learning: A Mutual Information-based Approach

In modern traffic management, one of the most essential yet challenging ...

Minimalist Traffic Prediction: Linear Layer Is All You Need

Traffic prediction is essential for the progression of Intelligent Trans...

Diffusion Probabilistic Models for Graph-Structured Prediction

This paper studies graph-structured prediction for supervised learning o...

User Traffic Prediction for Proactive Resource Management: Learning-Powered Approaches

Traffic prediction plays a vital role in efficient planning and usage of...

Please sign up or login with your details

Forgot password? Click here to reset