GraphSparseNet: a Novel Method for Large Scale Traffic Flow Prediction
Traffic flow forecasting is a critical spatio-temporal data mining task with wide-ranging applications in intelligent route planning and dynamic traffic management. Recent advancements in deep learning, particularly through Graph Neural Networks (GNNs), have significantly enhanced the accuracy of these forecasts by capturing complex spatio-temporal dynamics. However, the scalability of GNNs remains a challenge due to their exponential growth in model complexity with increasing nodes in the graph. Existing methods to address this issue, including sparsification, decomposition, and kernel-based approaches, either do not fully resolve the complexity issue or risk compromising predictive accuracy. This paper introduces GraphSparseNet (GSNet), a novel framework designed to improve both the scalability and accuracy of GNN-based traffic forecasting models. GraphSparseNet is comprised of two core modules: the Feature Extractor and the Relational Compressor. These modules operate with linear time and space complexity, thereby reducing the overall computational complexity of the model to a linear scale. Our extensive experiments on multiple real-world datasets demonstrate that GraphSparseNet not only significantly reduces training time by 3.51x compared to state-of-the-art linear models but also maintains high predictive performance.
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
Energy-Guided Data Sampling for Traffic Prediction with Mini Training Datasets
Recent endeavors aimed at forecasting future traffic flow states through deep learning encounter various challenges and yield diverse outcomes. A notable obstacle arises from the substantial data requirements of deep lea…
Deep LearningTraffic PredictionSFADNet: Spatio-temporal Fused Graph based on Attention Decoupling Network for Traffic Prediction
In recent years, traffic flow prediction has played a crucial role in the management of intelligent transportation systems. However, traditional prediction methods are often limited by static spatial modeling, making it …
ManagementPredictionTime SeriesTraffic PredictionFederated Learning for Traffic Flow Prediction with Synthetic Data Augmentation
Deep-learning based traffic prediction models require vast amounts of data to learn embedded spatial and temporal dependencies. The inherent privacy and commercial sensitivity of such data has encouraged a shift towards …
Data AugmentationFederated LearningGraph AttentionPrediction+1ST-Mamba: Spatial-Temporal Selective State Space Model for Traffic Flow Prediction
Traffic flow prediction, a critical aspect of intelligent transportation systems, has been increasingly popular in the field of artificial intelligence, driven by the availability of extensive traffic data. The current c…
Computational EfficiencyMambaManagementPrediction+1Unlocking air traffic flow prediction through microscopic aircraft-state modeling
Short-term air traffic flow prediction in terminal airspace is essential for proactive air traffic management. Existing approaches predominantly model traffic flow as aggregated time series. However, traffic dynamics are…