skip to main content
research-article

Graph-Based Stock Recommendation by Time-Aware Relational Attention Network

Authors Info & Claims
Published:20 July 2021Publication History
Skip Abstract Section

Abstract

The stock market investors aim at maximizing their investment returns. Stock recommendation task is to recommend stocks with higher return ratios for the investors. Most stock prediction methods study the historical sequence patterns to predict stock trend or price in the near future. In fact, the future price of a stock is correlated not only with its historical price, but also with other stocks. In this article, we take into account the relationships between stocks (corporations) by stock relation graph. Furthermore, we propose a Time-aware Relational Attention Network (TRAN) for graph-based stock recommendation according to return ratio ranking. In TRAN, the time-aware relational attention mechanism is designed to capture time-varying correlation strengths between stocks by the interaction of historical sequences and stock description documents. With the dynamic strengths, the nodes of the stock relation graph aggregate the features of neighbor stock nodes by graph convolution operation. For a given group of stocks, the proposed TRAN model can output the ranking results of stocks according to their return ratios. The experimental results on several real-world datasets demonstrate the effectiveness of our TRAN for stock recommendation.

References

  1. F. Feng, X. He, X. Wang, C. Luo, Y. Liu, and T. S. Chua. 2019. Temporal relational ranking for stock prediction. ACM Transactions on Information Systems 37, 2 (2019), 27:1–27:30. DOI:http://dx.doi.org/10.1145/3309547 Google ScholarGoogle ScholarDigital LibraryDigital Library
  2. S. Deng, N. Zhang, W. Zhang, J. Chen, J. Z. Pan, and H. Chen. 2019. Knowledge-driven stock trend prediction and explanation via temporal convolutional network. In Proceedings of the 27th World Wide Web Conference. 678–685. DOI:http://dx.doi.org/10.1145/3308560.3317701 Google ScholarGoogle ScholarDigital LibraryDigital Library
  3. Josif Grabocka, Nicolas Schilling, and Lars Schmidt-Thieme. 2016. Latent time-series motifs. ACM Transactions on Knowledge Discovery from Data 11, 1 (2016), 1–20. DOI:http://dx.doi.org/10.1145/2940329 Google ScholarGoogle ScholarDigital LibraryDigital Library
  4. Mu-Yen Chen. 2014. A high-order fuzzy time series forecasting model for internet stock trading. Future Generation Computer Systems 37 (2014), 461–467. http://dx.doi.org/10.1016/j.future.2013.09.025 Google ScholarGoogle ScholarDigital LibraryDigital Library
  5. Jason Lines, Sarah Taylor, and Anthony Bagnall. 2018. Time series classification with HIVE-COTE: The hierarchical vote collective of transformation-based ensembles. ACM Transactions on Knowledge Discovery from Data 12, 5 (2018), 52:1–52:35. http://dx.doi.org/10.1145/3182382 Google ScholarGoogle ScholarDigital LibraryDigital Library
  6. Sina Sajadmanesh, Sogol Bazargani, Jiawei Zhang, and Hamid R Rabiee. 2019. Continuous-time relationship prediction in dynamic heterogeneous information networks. ACM Transactions on Knowledge Discovery from Data 13, 4 (2019), 1–31. DOI:http://dx.doi.org/10.1145/3333028 Google ScholarGoogle ScholarDigital LibraryDigital Library
  7. Hung-Hsuan Chen. 2018. Behavior2Vec: Generating distributed representations of users’ behaviors on products for recommender systems. ACM Transactions on Knowledge Discovery from Data 12, 4 (2018), 1–20. DOI:http://dx.doi.org/10.1145/3184454 Google ScholarGoogle ScholarDigital LibraryDigital Library
  8. Yu Lei and Wenjie Li. 2019. Interactive recommendation with user-specific deep reinforcement learning. ACM Transactions on Knowledge Discovery from Data 13, 6 (2019), 1–15. DOI:http://dx.doi.org/10.1145/3359554 Google ScholarGoogle ScholarDigital LibraryDigital Library
  9. Huiting Liu, Huimin Liu, Qiang Ji, Peng Zhao, and Xindong Wu. 2020. Collaborative deep recommendation with global and local item correlations. Neurocomputing 385 (2020), 278–291. DOI:http://dx.doi.org/10.1016/j.neucom.2019.12.087Google ScholarGoogle ScholarDigital LibraryDigital Library
  10. L. Zhang, C. Aggarwal, and G. J. Qi. 2017. Stock price prediction via discovering multi-frequency trading patterns. In Proceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2141–2149. DOI:http://dx.doi.org/10.1145/3097983.3098117 Google ScholarGoogle ScholarDigital LibraryDigital Library
  11. Qikai Liu, Xiang Cheng, Sen Su, and Shuguang Zhu. 2018. Hierarchical complementary attention network for predicting stock price movements with news. In Proceedings of the 27th ACM International Conference on Information and Knowledge Management. 1603–1606. DOI:http://dx.doi.org/10.1145/3269206.3269286 Google ScholarGoogle ScholarDigital LibraryDigital Library
  12. Y. Xu and S. B. Cohen. 2018. Stock movement prediction from tweets and historical prices. In Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics. 1970–1979. DOI:http://dx.doi.org/10.18653/v1/P18-1183Google ScholarGoogle Scholar
  13. Y. Chen, Z. Wei, and X. Huang. 2018. Incorporating corporation relationship via graph convolutional neural networks for stock price prediction. In Proceedings of the ACM International Conference on Information and Knowledge Management. 1655–1658. DOI:http://dx.doi.org/10.1145/3269206.3269269 Google ScholarGoogle ScholarDigital LibraryDigital Library
  14. Burton G. Malkiel and Eugene F. Fama. 1970. Efficient capital markets: A review of theory and empirical work. The Journal of Finance 25, 2 (1970), 383–417. DOI:http://dx.doi.org/10.2307/2325486Google ScholarGoogle ScholarCross RefCross Ref
  15. Guang Liu and Xiaojie Wang. 2018. A numerical-based attention method for stock market prediction with dual information. IEEE Access 7 (2018), 7357–7367. DOI:http://dx.doi.org/10.1109/ACCESS.2018.2886367Google ScholarGoogle ScholarCross RefCross Ref
  16. P. K. Agarwal, R. B. Avraham, H. Kaplan, and M. Sharir. 2014. Computing the discrete Fréchet distance in subquadratic time. SIAM Journal on Computing 43, 2 (2014), 429–449. DOI:http://dx.doi.org/10.1137/130920526Google ScholarGoogle ScholarCross RefCross Ref
  17. Eugene F. Fama. 1965. The behavior of stock-market prices. The Journal of Business 38, 1 (1965), 34–105. DOI:http://dx.doi.org/10.1086/294743Google ScholarGoogle ScholarCross RefCross Ref
  18. Xiao Ding, Yue Zhang, Ting Liu, and Junwen Duan. 2015. Deep learning for event-driven stock prediction. In Proceedings of the 24th International Joint Conference on Artificial Intelligence. 2327–2333. Google ScholarGoogle ScholarDigital LibraryDigital Library
  19. H. Xie, J. Bian, M. Wang, and H. Qiao. 2014. Is technical analysis informative in UK stock market? Evidence from decomposition-based vector autoregressive (DVAR) model. Journal of Systems Science and Complexity 27, 1 (2014), 144–156. DOI:http://dx.doi.org/10.1007/s11424-014-3280-9Google ScholarGoogle ScholarCross RefCross Ref
  20. R. Bisoi and P. K. Dash. 2014. A hybrid evolutionary dynamic neural network for stock market trend analysis and prediction using unscented Kalman filter. Applied Soft Computing 19 (2014), 41–56. DOI:http://dx.doi.org/10.1016/j.asoc.2014.01.039Google ScholarGoogle ScholarCross RefCross Ref
  21. Chenglin Xiao, Weili Xia, and Jijiao Jiang. 2020. Stock price forecast based on combined model of ARI-MA-LS-SVM. Neural Computing and Applications 32, 10 (2020), 5379–5388. DOI:http://dx.doi.org/10.1007/s00521-019-04698-5Google ScholarGoogle ScholarCross RefCross Ref
  22. Zhe Lin. 2018. Modelling and forecasting the stock market volatility of SSE composite index using GARCH models. Future Generation Computer Systems 79 (2018), 960–972. DOI:http://dx.doi.org/10.1016/j.future.2017.08.033 Google ScholarGoogle ScholarDigital LibraryDigital Library
  23. D. M. Nelson, A. C. Pereira, and R. A. de Oliveira. 2017. Stock market’s price movement prediction with LSTM neural networks. In Proceedings of the International Joint Conference on Neural Networks. 1419–1426. DOI:http://dx.doi.org/10.1109/IJCNN.2017.7966019Google ScholarGoogle ScholarCross RefCross Ref
  24. J. Wang, T. Sun, B. Liu, Y. Cao, and H. Zhu. 2019. CLVSA: A convolutional LSTM based variational sequence-to-sequence model with attention for predicting trends of financial markets. In Proceedings of the 28th International Joint Conference on Artificial Intelligence. 3705–3711. DOI:http://dx.doi.org/10.24963/ijcai.2019/514 Google ScholarGoogle ScholarDigital LibraryDigital Library
  25. Sepp Hochreiter and Jürgen Schmidhuber. 1997. Long short-term memory. Neural Computation 9, 8 (1997), 1735–1780. DOI:http://dx.doi.org/10.1162/neco.1997.9.8.1735 Google ScholarGoogle ScholarDigital LibraryDigital Library
  26. X. Ding, Y. Zhang, T. Liu, and J. Duan. 2015. Deep learning for event-driven stock prediction. In Proceedings of the 24th International Joint Conference on Artificial Intelligence. 2327–2333. Google ScholarGoogle ScholarDigital LibraryDigital Library
  27. Yanshuang Li, Xintian Zhuang, Jian Wang, and Weiping Zhang. 2020. Analysis of the impact of Sino-US trade friction on China’s stock market based on complex networks. The North American Journal of Economics and Finance 52 (2020), 101185. DOI:http://dx.doi.org/10.1016/j.najef.2020.101185Google ScholarGoogle ScholarCross RefCross Ref
  28. Yuqiang Fang, Ruili Wang, Bin Dai, and Xindong Wu. 2015. Graph-based learning via auto-grouped sparse regularization and kernelized extension. IEEE Transactions on Knowledge and Data Engineering 27, 1 (2015), 142–154. DOI:http://dx.doi.org/10.1109/TKDE.2014.2312322Google ScholarGoogle ScholarCross RefCross Ref
  29. T. N. Kipf and M. Welling. 2017. Semi-supervised classification with graph convolutional networks. In Proceedings of the International Conference on Learning Representations. 1–14.Google ScholarGoogle Scholar
  30. Xiaoting Ying, Cong Xu, Jianliang Gao, Jianxin Wang, and Zhao Li. 2020. Time-aware graph relational attention network for stock recommendation. In Proceedings of the 29th ACM International Conference on Information and Knowledge Management. 2281–2284. Google ScholarGoogle ScholarDigital LibraryDigital Library
  31. Jiexia Ye, Juanjuan Zhao, Kejiang Ye, and Chengzhong Xu. 2020. Multi-view graph convolutional networks for relationship-driven stock prediction. arXiv:2005.04955. Retrieved from https://arxiv.org/abs/2005.04955.Google ScholarGoogle Scholar
  32. Thomas Fischer and Christopher Krauss. 2018. Deep learning with long short-term memory networks for financial market predictions. European Journal of Operational Research 270, 2 (2018), 654–669. DOI:http://dx.doi.org/10.1016/j.ejor.2017.11.054Google ScholarGoogle ScholarCross RefCross Ref
  33. C. Chen, L. Zhao, J. Bian, C. Xing, and T. Y. Liu. 2019. Investment behaviors can tell what inside: Exploring stock intrinsic properties for stock trend prediction. In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2376–2384. DOI:http://dx.doi.org/10.1145/3292500.3330663 Google ScholarGoogle ScholarDigital LibraryDigital Library
  34. Tanawat Chiewhawan and Peerapon Vateekul. 2020. Stock return prediction using dual-stage attention model with stock relation inference. In Proceedings of the 12th Asian Conference on Intelligent Information and Database Systems. 492–503. DOI:http://dx.doi.org/10.1007/978-3-030-41964-6_42Google ScholarGoogle ScholarCross RefCross Ref
  35. Wayne Xin Zhao, Jinpeng Wang, Yulan He, Ji-Rong Wen, Edward Y Chang, and Xiaoming Li. 2016. Mining product adopter information from online reviews for improving product recommendation. ACM Transactions on Knowledge Discovery from Data 10, 3 (2016), 1–23. DOI:http://dx.doi.org/10.1145/2842629 Google ScholarGoogle ScholarDigital LibraryDigital Library
  36. D. M. Blei, A. Y. Ng, and M. I. Jordan. 2003. Latent dirichlet allocation. Journal of Machine Learning Research 3, Jan (2003), 993–1022. Google ScholarGoogle ScholarDigital LibraryDigital Library
  37. Ting Hua, Chang-Tien Lu, Jaegul Choo, and Chandan K. Reddy. 2020. Probabilistic topic modeling for comparative analysis of document collections. ACM Transactions on Knowledge Discovery from Data 14, 2 (2020), 1–27. DOI:http://dx.doi.org/10.1145/3369873 Google ScholarGoogle ScholarDigital LibraryDigital Library
  38. Wei Bao, Jun Yue, and Yulei Rao. 2017. A deep learning framework for financial time series using stacked autoencoders and long-short term memory. PloS One 12, 7 (2017), e0180944. DOI:http://dx.doi.org/10.1371/journal.pone.0180944Google ScholarGoogle ScholarCross RefCross Ref
  39. Petar Veličković, Guillem Cucurull, Arantxa Casanova, Adriana Romero, Pietro Lio, and Yoshua Bengio. 2018. Graph attention networks. In Proceedings of the 6th International Conference on Learning Representations.Google ScholarGoogle Scholar
  40. Q. Le and T. Mikolov. 2014. Distributed representations of sentences and documents. In Proceedings of the International Conference on Machine Learning. 1188–1196. Google ScholarGoogle ScholarDigital LibraryDigital Library
  41. M. Dixon, D. Klabjan, and J. H. Bang. 2017. Classification-based financial markets prediction using deep neural networks. Algorithmic Finance 6, 3-4 (2017), 67–77. DOI:http://dx.doi.org/10.3233/AF-170176Google ScholarGoogle ScholarCross RefCross Ref
  42. G. W. Schwert. 2002. Stock volatility in the new millennium: How wacky is nasdaq?Journal of Monetary Economics 49, 1 (2002), 3–26. DOI:http://dx.doi.org/10.3233/AF-170176Google ScholarGoogle Scholar

Index Terms

  1. Graph-Based Stock Recommendation by Time-Aware Relational Attention Network

          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

          Full Access

          • Published in

            cover image ACM Transactions on Knowledge Discovery from Data
            ACM Transactions on Knowledge Discovery from Data  Volume 16, Issue 1
            February 2022
            475 pages
            ISSN:1556-4681
            EISSN:1556-472X
            DOI:10.1145/3472794
            Issue’s Table of Contents

            Copyright © 2021 Association for Computing Machinery.

            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: 20 July 2021
            • Accepted: 1 February 2021
            • Revised: 1 November 2020
            • Received: 1 July 2020
            Published in tkdd Volume 16, Issue 1

            Permissions

            Request permissions about this article.

            Request Permissions

            Check for updates

            Qualifiers

            • research-article
            • Refereed

          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