English
Related papers

Related papers: Efficient Prompt Learning for Traffic Forecasting

200 papers

Urban flow prediction is a classic spatial-temporal forecasting task that estimates the amount of future traffic flow for a given location. Though models represented by Spatial-Temporal Graph Neural Networks (STGNNs) have established…

Machine Learning · Computer Science 2024-12-10 Haiyang Jiang , Tong Chen , Wentao Zhang , Nguyen Quoc Viet Hung , Yuan Yuan , Yong Li , Lizhen Cui

Accurately forecasting dynamic processes on graphs, such as traffic flow or disease spread, remains a challenge. While Graph Neural Networks (GNNs) excel at modeling and forecasting spatio-temporal data, they often lack the ability to…

Machine Learning · Computer Science 2024-08-30 Zakaria Elabid , Lena Sasal , Daniel Busby , Abdenour Hadid

GNN-based recommenders have excelled in modeling intricate user-item interactions through multi-hop message passing. However, existing methods often overlook the dynamic nature of evolving user-item interactions, which impedes the adaption…

Information Retrieval · Computer Science 2024-02-20 Yuhao Yang , Lianghao Xia , Da Luo , Kangyi Lin , Chao Huang

Accurate traffic prediction is essential for Intelligent Transportation Systems (ITS), yet current methods struggle with the inherent complexity and non-linearity of traffic dynamics, making it difficult to integrate spatial and temporal…

Machine Learning · Computer Science 2025-07-02 Ruiyuan Jiang , Dongyao Jia , Eng Gee Lim , Pengfei Fan , Yuli Zhang , Shangbo Wang

Multivariate Time Series (MTS) forecasting plays a vital role in a wide range of applications. Recently, Spatial-Temporal Graph Neural Networks (STGNNs) have become increasingly popular MTS forecasting methods. STGNNs jointly model the…

Machine Learning · Computer Science 2022-08-17 Zezhi Shao , Zhao Zhang , Fei Wang , Yongjun Xu

Intelligent Transportation System (ITS) is crucial for improving traffic congestion, reducing accidents, optimizing urban planning, and more. However, the complexity of traffic networks has rendered traditional machine learning and…

Machine Learning · Computer Science 2024-09-20 Hourun Li , Yusheng Zhao , Zhengyang Mao , Yifang Qin , Zhiping Xiao , Jiaqi Feng , Yiyang Gu , Wei Ju , Xiao Luo , Ming Zhang

Spatio-temporal modeling is foundational for smart city applications, yet it is often hindered by data scarcity in many cities and regions. To bridge this gap, we propose a novel generative pre-training framework, GPD, for spatio-temporal…

Machine Learning · Computer Science 2024-03-26 Yuan Yuan , Chenyang Shao , Jingtao Ding , Depeng Jin , Yong Li

Real-time and precise traffic flow prediction is vital for the efficiency of intelligent transportation systems. Traditional methods often employ graph neural networks (GNNs) with predefined graphs to describe spatial correlations among…

Machine Learning · Computer Science 2024-06-18 Ben-Ao Dai , Bao-Lin Ye , Lingxi Li

Traffic prediction is critical for optimizing travel scheduling and enhancing public safety, yet the complex spatial and temporal dynamics within traffic data present significant challenges for accurate forecasting. In this paper, we…

Machine Learning · Computer Science 2025-02-19 Lingxiao Cao , Bin Wang , Guiyuan Jiang , Yanwei Yu , Junyu Dong

Traffic prediction is a cornerstone of modern intelligent transportation systems and a critical task in spatio-temporal forecasting. Although advanced Spatio-temporal Graph Neural Networks (STGNNs) and pre-trained models have achieved…

Machine Learning · Computer Science 2026-01-01 Weilin Ruan , Xilin Dang , Ziyu Zhou , Sisuo Lyu , Yuxuan Liang

Graph Neural Networks (GNNs) are powerful in learning semantics of graph data. Recently, a new paradigm "pre-train and prompt" has shown promising results in adapting GNNs to various tasks with less supervised data. The success of such…

Machine Learning · Computer Science 2024-06-04 Qingqing Ge , Zeyuan Zhao , Yiding Liu , Anfeng Cheng , Xiang Li , Shuaiqiang Wang , Dawei Yin

Traffic data exhibits complex temporal, spatial, and spatial-temporal correlations. Most of models use either independent modules to separately extract temporal and spatial correlations or joint modules to synchronously extract them,…

Computer Vision and Pattern Recognition · Computer Science 2025-04-14 Kai Hu , Zhidan Zhao , Zhifeng Hao

Signed Graph Neural Networks (SGNNs) are effective in learning expressive representations for signed graphs but typically require substantial task-specific labels, limiting their applicability in label-scarce industrial scenarios. In…

Machine Learning · Computer Science 2025-08-19 Zian Zhai , Sima Qing , Xiaoyang Wang , Wenjie Zhang

Graph Neural Networks (GNNs) have advanced spatiotemporal forecasting by leveraging relational inductive biases among sensors (or any other measuring scheme) represented as nodes in a graph. However, current methods often rely on Recurrent…

Machine Learning · Computer Science 2024-05-30 Aref Einizade , Fragkiskos D. Malliaros , Jhony H. Giraldo

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…

Machine Learning · Computer Science 2022-11-11 Mingxi Li , Yihong Tang , Wei Ma

Modeling complex spatial and temporal correlations in the correlated time series data is indispensable for understanding the traffic dynamics and predicting the future status of an evolving traffic system. Recent works focus on designing…

Machine Learning · Computer Science 2020-10-23 Lei Bai , Lina Yao , Can Li , Xianzhi Wang , Can Wang

Accurate prediction of network-wide traffic conditions is essential for intelligent transportation systems. In the last decade, machine learning techniques have been widely used for this task, resulting in state-of-the-art performance. We…

Machine Learning · Computer Science 2021-11-23 Lei Lin , Weizi Li , Lei Zhu

Recently, graph prompt learning has garnered increasing attention in adapting pre-trained GNN models for downstream graph learning tasks. However, existing works generally conduct prompting over all graph elements (e.g., nodes, edges, node…

Machine Learning · Computer Science 2024-10-30 Bo Jiang , Hao Wu , Beibei Wang , Jin Tang , Bin Luo

Despite the recent success of Graph Neural Networks, it remains challenging to train a GNN on large graphs with millions of nodes and billions of edges, which are prevalent in many graph-based applications. Traditional sampling-based…

Machine Learning · Computer Science 2022-10-04 Zheng Chai , Guangji Bai , Liang Zhao , Yue Cheng

The criticality of prompt and precise traffic forecasting in optimizing traffic flow management in Intelligent Transportation Systems (ITS) has drawn substantial scholarly focus. Spatio-Temporal Graph Neural Networks (STGNNs) have been…

Machine Learning · Computer Science 2023-08-16 Zepu Wang , Yuqi Nie , Peng Sun , Nam H. Nguyen , John Mulvey , H. Vincent Poor