Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
skip to main content
10.1145/3669754.3669825acmotherconferencesArticle/Chapter ViewAbstractPublication PagesiccaiConference Proceedingsconference-collections
research-article
Open access

A Clustering-based Multi-Task Learning Method using Graph Attention Network for Short-term Traffic Forecasting

Published: 30 August 2024 Publication History

Abstract

In modern intelligent transportation systems, accurate short-term traffic forecasting is pivotal for managers and travelers. While recent spatio-temporal traffic forecasting models excel in short-term predictions, data from different traffic node inherently possesses diverse characteristics, yet existing models tend to neglect the disparities between these nodes and treat neighboring nodes with different data in a uniform manner, overlooking their unique attributes, which can result in imprecise prediction results. To address these concerns, this paper introduces, the Clustering-based Multi-Task Learning Method using Graph Attention Network (CMTGAT). This model adopts a multi-task learning framework, enhanced with time-series clustering for nuanced pattern recognition and improved prediction accuracy in diverse traffic scenarios. Experimental results demonstrate that the CMTGAT model outperforms existing methods, almost doubling in forecasting precision. The integration of the time series clustering and the graph attention mechanism helps to reduce the MAPE approximately by 6%.

References

[1]
Anton Agafonov and Alexander Yumaganov. 2020. Spatio-Temporal Graph Convolutional Networks for Short-Term Traffic Forecasting. 2020 International Conference on Information Technology and Nanotechnology (ITNT) (2020), 1–6. https://api.semanticscholar.org/CorpusID:226970303
[2]
Christopher M Bishop. 1995. Neural networks for pattern recognition. Oxford university press.
[3]
George Box. 2013. Box and Jenkins: time series analysis, forecasting and control. In A Very British Affair: Six Britons and the Development of Time Series Analysis During the 20th Century. Springer, 161–215.
[4]
Leo Breiman. 2001. Random forests. Machine learning 45 (2001), 5–32.
[5]
Shaked Brody, Uri Alon, and Eran Yahav. 2021. How attentive are graph attention networks?arXiv preprint arXiv:2105.14491 (2021).
[6]
Junyoung Chung, Caglar Gulcehre, KyungHyun Cho, and Yoshua Bengio. 2014. Empirical evaluation of gated recurrent neural networks on sequence modeling. arXiv preprint arXiv:1412.3555 (2014).
[7]
Corinna Cortes and Vladimir Vapnik. 1995. Support-vector networks. Machine learning 20 (1995), 273–297.
[8]
Jeffrey Donahue, Lisa Anne Hendricks, Sergio Guadarrama, Marcus Rohrbach, Subhashini Venugopalan, Kate Saenko, and Trevor Darrell. 2015. Long-term recurrent convolutional networks for visual recognition and description. In Proceedings of the IEEE conference on computer vision and pattern recognition. 2625–2634.
[9]
Harris Drucker, Christopher J Burges, Linda Kaufman, Alex Smola, and Vladimir Vapnik. 1996. Support vector regression machines. Advances in neural information processing systems 9 (1996).
[10]
Martin Ester, Hans-Peter Kriegel, Jörg Sander, Xiaowei Xu, 1996. A density-based algorithm for discovering clusters in large spatial databases with noise. In kdd, Vol. 96. 226–231.
[11]
Sepp Hochreiter and Jürgen Schmidhuber. 1997. Long short-term memory. Neural computation 9, 8 (1997), 1735–1780.
[12]
Alex Krizhevsky, Ilya Sutskever, and Geoffrey E Hinton. 2012. Imagenet classification with deep convolutional neural networks. Advances in neural information processing systems 25 (2012).
[13]
Yaguang Li, Rose Yu, Cyrus Shahabi, and Yan Liu. 2017. Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. arXiv preprint arXiv:1707.01926 (2017).
[14]
Zahra Mahdavi and Maryam Khademi. 2012. Prediction of Oil Production with: Data Mining, Neuro-Fuzzy and Linear Regression. International Journal of Computer Theory and Engineering 4, 3 (2012), 446.
[15]
Franco Manessi, Alessandro Rozza, and Mario Manzo. 2020. Dynamic graph convolutional networks. Pattern Recognition 97 (2020), 107000.
[16]
Farzin Owramipur, Parinaz Eskandarian, and Faezeh Sadat Mozneb. 2013. Football result prediction with Bayesian network in Spanish League-Barcelona team. International Journal of Computer Theory and Engineering 5, 5 (2013), 812.
[17]
David E Rumelhart, Geoffrey E Hinton, and Ronald J Williams. 1986. Learning representations by back-propagating errors. nature 323, 6088 (1986), 533–536.
[18]
Ilya Sutskever, Oriol Vinyals, and Quoc V Le. 2014. Sequence to sequence learning with neural networks. Advances in neural information processing systems 27 (2014).
[19]
Petar Velickovic, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Lio, Yoshua Bengio, 2017. Graph attention networks. stat 1050, 20 (2017), 10–48550.
[20]
Zonghan Wu, Shirui Pan, Guodong Long, Jing Jiang, Xiaojun Chang, and Chengqi Zhang. 2020. Connecting the dots: Multivariate time series forecasting with graph neural networks. In Proceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining. 753–763.
[21]
Yuhanis Yusof and Zuriani Mustaffa. 2011. Dengue outbreak prediction: A least squares support vector machines approach. International Journal of Computer Theory and Engineering 3, 4 (2011), 489.
[22]
Min-Ling Zhang and Zhi-Hua Zhou. 2013. A review on multi-label learning algorithms. IEEE transactions on knowledge and data engineering 26, 8 (2013), 1819–1837.

Index Terms

  1. A Clustering-based Multi-Task Learning Method using Graph Attention Network for Short-term Traffic Forecasting

    Recommendations

    Comments

    Information & Contributors

    Information

    Published In

    cover image ACM Other conferences
    ICCAI '24: Proceedings of the 2024 10th International Conference on Computing and Artificial Intelligence
    April 2024
    491 pages
    ISBN:9798400717055
    DOI:10.1145/3669754
    Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

    Publisher

    Association for Computing Machinery

    New York, NY, United States

    Publication History

    Published: 30 August 2024

    Permissions

    Request permissions for this article.

    Check for updates

    Author Tags

    1. Graph self-attention network
    2. Spatio-temporal graph
    3. Traffic forecasting
    4. multi-task learning
    5. time-series clustering

    Qualifiers

    • Research-article
    • Research
    • Refereed limited

    Conference

    ICCAI 2024

    Contributors

    Other Metrics

    Bibliometrics & Citations

    Bibliometrics

    Article Metrics

    • 0
      Total Citations
    • 84
      Total Downloads
    • Downloads (Last 12 months)84
    • Downloads (Last 6 weeks)29
    Reflects downloads up to 25 Dec 2024

    Other Metrics

    Citations

    View Options

    View options

    PDF

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format.

    HTML Format

    Login options

    Media

    Figures

    Other

    Tables

    Share

    Share

    Share this Publication link

    Share on social media