Do We Really Need Graph Neural Networks for Traffic Forecasting?

التفاصيل البيبلوغرافية
العنوان: Do We Really Need Graph Neural Networks for Traffic Forecasting?
المؤلفون: Liu, Xu, Liang, Yuxuan, Huang, Chao, Hu, Hengchang, Cao, Yushi, Hooi, Bryan, Zimmermann, Roger
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Social and Information Networks
الوصف: Spatio-temporal graph neural networks (STGNN) have become the most popular solution to traffic forecasting. While successful, they rely on the message passing scheme of GNNs to establish spatial dependencies between nodes, and thus inevitably inherit GNNs' notorious inefficiency. Given these facts, in this paper, we propose an embarrassingly simple yet remarkably effective spatio-temporal learning approach, entitled SimST. Specifically, SimST approximates the efficacies of GNNs by two spatial learning techniques, which respectively model local and global spatial correlations. Moreover, SimST can be used alongside various temporal models and involves a tailored training strategy. We conduct experiments on five traffic benchmarks to assess the capability of SimST in terms of efficiency and effectiveness. Empirical results show that SimST improves the prediction throughput by up to 39 times compared to more sophisticated STGNNs while attaining comparable performance, which indicates that GNNs are not the only option for spatial modeling in traffic forecasting.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2301.12603
رقم الأكسشن: edsarx.2301.12603
قاعدة البيانات: arXiv