skip to main content
10.1145/3512576.3512594acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicitConference Proceedingsconference-collections
research-article

Traffic Matrix Prediction with Attention-based Recurrent Neural Network

Authors Info & Claims
Published:11 April 2022Publication History

ABSTRACT

Traffic matrix (TM) shows the traffic volume of a network. Therefore, TM prediction is of great significance for network management. Attention mechanism has been successful in many sub-domains of machine learning, such as computer vision and natural language processing, and it performs particularly well on time series data. In this work, we first introduce attention mechanisms into the traffic matrix prediction field by proposing an attention-based deep learning model for traffic matrix prediction. This model is composed of two parts, encoder and decoder. We use a recurrent neural network (RNN) architecture as the encoder and our decoder has an attention layer and a linear layer. Attention mechanism allows the model to have better memory ability, so the model can concentrate on those important data regardless of distance. We also reduce the time consumption of our model using GPU-based parallel acceleration. Finally, we evaluate the effectiveness of our model on a real world TM dataset, and the results show our implementations on the proposed model perform better than the baseline models.

References

  1. A. Azzouni and G. Pujolle. 2017. A Long Short-Term Memory Recurrent Neural Network Framework for Network Traffic Matrix Prediction. arXiv (2017).Google ScholarGoogle Scholar
  2. D. Bahdanau, K. Cho, and Y. Bengio. 2014. Neural Machine Translation by Jointly Learning to Align and Translate. Computer Science (2014).Google ScholarGoogle Scholar
  3. M. Barabas, G. Boanea, A. B. Rus, V. Dobrota, and J. Domingo-Pascual. 2011. Evaluation of network traffic prediction based on neural networks with multi-task learning and multiresolution decomposition. In 2011 IEEE 7th International Conference on Intelligent Computer Communication and Processing.Google ScholarGoogle Scholar
  4. K. Cho, B Van Merrienboer, C. Gulcehre, D. Ba Hdanau, F. Bougares, H. Schwenk, and Y. Bengio. 2014. Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation. Computer Science (2014).Google ScholarGoogle Scholar
  5. J. Chung, C. Gulcehre, K. H. Cho, and Y. Bengio. 2014. Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling. arXiv (2014).Google ScholarGoogle Scholar
  6. H. Feng and Y. Shu. 2005. Study on network traffic prediction techniques. In International Conference on Wireless Communications.Google ScholarGoogle Scholar
  7. B. Fortz and M. Thorup. 2001. Internet traffic engineering by optimizing OSPF weights. proceedings of ieee infocom mar(2001).Google ScholarGoogle Scholar
  8. S. Hochreiter and J. Schmidhuber. 1997. Long Short-Term Memory. Neural Computation 9, 8 (1997), 1735–1780.Google ScholarGoogle ScholarDigital LibraryDigital Library
  9. Y. Kim, C. Denton, L. Hoang, and A. M. Rush. 2017. Structured Attention Networks. (2017).Google ScholarGoogle Scholar
  10. Z. Liu, Z. Wang, X. Yin, X. Shi, and Y. Tian. 2019. Traffic Matrix Prediction Based on Deep Learning for Dynamic Traffic Engineering. In 2019 IEEE Symposium on Computers and Communications (ISCC).Google ScholarGoogle Scholar
  11. N. Ramakrishnan and T. Soni. 2018. Network Traffic Prediction Using Recurrent Neural Networks. In 2018 17th IEEE International Conference on Machine Learning and Applications (ICMLA).Google ScholarGoogle Scholar
  12. A. Sang and S. Q. Li. 2000. A predictability analysis of network traffic. In Infocom Nineteenth Joint Conference of the IEEE Computer & Communications Societies IEEE.Google ScholarGoogle Scholar
  13. A. Soule, A. Lakhina, N. Taft, K. Papagiannaki, K. Salamatian, A. Nucci, M. Crovella, and C. Diot. 2005. Traffic Matrices: Balancing Measurements, Inference and Modeling. Performance evaluation review 33, 1 (2005), p.362–373.Google ScholarGoogle ScholarDigital LibraryDigital Library
  14. S. Troia, R. Alvizu, Y. Zhou, G. Maier, and A. Pattavina. 2018. Deep Learning-Based Traffic Prediction for Network Optimization. In International Conference on Transparent Optical Networks (ICTON) 2018.Google ScholarGoogle Scholar
  15. Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Lukasz Kaiser, and Illia Polosukhin. 2017. Attention Is All You Need. arXiv (2017).Google ScholarGoogle Scholar
  16. L. Wei, H. Ao, O. Liang, W. Ding, and Z. Ge. 2014. Prediction and correction of traffic matrix in an IP backbone network. IEEE (2014).Google ScholarGoogle Scholar
  17. Y. Zhang. 2011,. Abilene: American Research and Education Network dataset. http://www.cs.utexas.edu/yzhang/research/AbileneTMGoogle ScholarGoogle Scholar

Recommendations

Comments

Login options

Check if you have access through your login credentials or your institution to get full access on this article.

Sign in
  • Published in

    cover image ACM Other conferences
    ICIT '21: Proceedings of the 2021 9th International Conference on Information Technology: IoT and Smart City
    December 2021
    584 pages
    ISBN:9781450384971
    DOI:10.1145/3512576

    Copyright © 2021 ACM

    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 ACM 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: 11 April 2022

    Permissions

    Request permissions about this article.

    Request Permissions

    Check for updates

    Qualifiers

    • research-article
    • Research
    • Refereed limited
  • Article Metrics

    • Downloads (Last 12 months)40
    • Downloads (Last 6 weeks)12

    Other Metrics

PDF Format

View or Download as a PDF file.

PDF

eReader

View online with eReader.

eReader

HTML Format

View this article in HTML Format .

View HTML Format