Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
skip to main content
10.1145/3511808.3557089acmconferencesArticle/Chapter ViewAbstractPublication PagescikmConference Proceedingsconference-collections
research-article

Temporal and Heterogeneous Graph Neural Network for Financial Time Series Prediction

Published: 17 October 2022 Publication History

Abstract

The price movement prediction of stock market has been a classical yet challenging problem, with the attention of both economists and computer scientists. In recent years, graph neural network has significantly improved the prediction performance by employing deep learning on company relations. However, existing relation graphs are usually constructed by handcraft human labeling or nature language processing, which are suffering from heavy resource requirement and low accuracy. Besides, they cannot effectively response to the dynamic changes in relation graphs. Therefore, in this paper, we propose a temporal and heterogeneous graph neural network-based (THGNN) approach to learn the dynamic relations among price movements in financial time series. In particular, we first generate the company relation graph for each trading day according to their historic price. Then we leverage a transformer encoder to encode the price movement information into temporal representations. Afterward, we propose a heterogeneous graph attention network to jointly optimize the embeddings of the financial time series data by transformer encoder and infer the probability of target movements. Finally, we conduct extensive experiments on the stock market in the United States and China. The results demonstrate the effectiveness and superior performance of our proposed methods compared with state-of-the-art baselines. Moreover, we also deploy the proposed THGNN in a real-world quantitative algorithm trading system, the accumulated portfolio return obtained by our method significantly outperforms other baselines.

References

[1]
Klaus Adam, J. Nicolini, and A. Marcet. 2008. Stock Market Volatility and Learning. IO: Theory (2008).
[2]
Michel Ballings, D. V. Poel, Nathalie Hespeels, and Ruben Gryp. 2015. Evaluating multiple classifiers for stock price direction prediction. Expert Syst. Appl. 42 (2015), 7046--7056.
[3]
Chi Chen, Li Zhao, Jiang Bian, Chunxiao Xing, and Tie-Yan Liu. 2019. Investment Behaviors Can Tell What Inside: Exploring Stock Intrinsic Properties for Stock Trend Prediction. Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (2019).
[4]
Yingmei Chen, Zhongyu Wei, and Xuanjing Huang. 2018. Incorporating Corporation Relationship via Graph Convolutional Neural Networks for Stock Price Prediction. Proceedings of the 27th ACM International Conference on Information and Knowledge Management (2018).
[5]
Dawei Cheng, Zhibin Niu, and Liqing Zhang. 2020. Delinquent events prediction in temporal networked-guarantee loans. IEEE Transactions on Neural Networks and Learning Systems (2020).
[6]
Dawei Cheng, Xiaoyang Wang, Ying Zhang, and Liqing Zhang. 2020. Graph Neural Network for Fraud Detection via Spatial-temporal Attention. IEEE Transactions on Knowledge and Data Engineering (2020).
[7]
Dawei Cheng, Fangzhou Yang, Xiaoyang Wang, Y. Zhang, and Liqing Zhang. 2020. Knowledge Graph-based Event Embedding Framework for Financial Quantitative Investments. Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval (2020).
[8]
Dawei Cheng, Fangzhou Yang, Sheng Xiang, and Jin Liu. 2022. Financial time series forecasting with multi-modality graph neural network. Pattern Recognition 121 (2022), 108218. https://doi.org/10.1016/j.patcog.2021.108218
[9]
Dawei Cheng, Yiyi Zhang, Fangzhou Yang, Yi Tu, Zhibin Niu, and Liqing Zhang. 2019. A dynamic default prediction framework for networked-guarantee loans. In Proceedings of the 28th ACM International Conference on Information and Knowledge Management. 2547--2555.
[10]
Rui Cheng and Qing Li. 2021. Modeling the Momentum Spillover Effect for Stock Prediction via Attribute-Driven Graph Attention Networks. In AAAI.
[11]
Kyunghyun Cho, B. V. Merrienboer, Çaglar Gülçehre, Dzmitry Bahdanau, Fethi Bougares, Holger Schwenk, and Yoshua Bengio. 2014. Learning Phrase Representations using RNN Encoder--Decoder for Statistical Machine Translation. ArXiv abs/1406.1078 (2014).
[12]
Xiao Ding, Yue Zhang, Ting Liu, and Junwen Duan. [n.d.]. Using Structured Events to Predict Stock Price Movement: An Empirical Investigation. In Proceedings of the 2014 Conference on Empirical Methods in Natural Language Processing, 2014, October 25--29, 2014. 1415--1425.
[13]
M. Dixon, D. Klabjan, and J. Bang. 2017. Classification-based financial markets prediction using deep neural networks. Algorithmic Finance 6 (2017), 67--77.
[14]
Fuli Feng, Huimin Chen, Xiangnan He, Ji Ding, Maosong Sun, and Tat-Seng Chua. 2019. Enhancing Stock Movement Prediction with Adversarial Training. In IJCAI.
[15]
Fuli Feng, Xiangnan He, Xiang Wang, Cheng Luo, Yiqun Liu, and Tat-Seng Chua. 2019. Temporal Relational Ranking for Stock Prediction. ACM Transactions on Information Systems (TOIS) 37 (2019), 1 -- 30.
[16]
Gartheeban Ganeshapillai, John Guttag, and Andrew Lo. 2013. Learning connections in financial time series. In International Conference on Machine Learning. PMLR, 109--117.
[17]
S. Hochreiter and J. Schmidhuber. 1997. Long Short-Term Memory. Neural Computation 9 (1997), 1735--1780.
[18]
Florian Holzschuher and René Peinl. 2013. Performance of graph query languages: comparison of cypher, gremlin and native access in Neo4j. In EDBT '13.
[19]
Joel F Houston, Chen Lin, and Zhongyan Zhu. 2016. The financial implications of supply chain changes. Management Science 62, 9 (2016), 2520--2542.
[20]
Shaoxiong Ji, Shirui Pan, Erik Cambria, Pekka Marttinen, and S Yu Philip. 2021. A survey on knowledge graphs: Representation, acquisition, and applications. IEEE Transactions on Neural Networks and Learning Systems (2021).
[21]
Zhigang Jin, Ya-Chi Yang, and Yuhong Liu. 2019. Stock closing price prediction based on sentiment analysis and LSTM. Neural Computing and Applications 32 (2019), 9713--9729.
[22]
Raehyun Kim, Chan Ho So, Minbyul Jeong, Sanghoon Lee, Jinkyu Kim, and Jaewoo Kang. 2019. HATS: A Hierarchical Graph Attention Network for Stock Movement Prediction. ArXiv abs/1908.07999 (2019).
[23]
Diederik P. Kingma and Jimmy Ba. 2015. Adam: A Method for Stochastic Optimization. CoRR abs/1412.6980 (2015).
[24]
Qing Li, Jinghua Tan, Jun Wang, and HsinChun Chen. 2020. A Multimodal Event-driven LSTM Model for Stock Prediction Using Online News. IEEE Transactions on Knowledge and Data Engineering (2020), 1--1. https://doi.org/10.1109/TKDE.2020.2968894
[25]
Wei Li, Ruihan Bao, Keiko Harimoto, Deli Chen, Jingjing Xu, and Qi Su. 2020. Modeling the Stock Relation with Graph Network for Overnight Stock Movement Prediction. In IJCAI.
[26]
Zhige Li, Derek Yang, Li Zhao, Jiang Bian, Tao Qin, and Tie-Yan Liu. 2019. Individualized Indicator for All: Stock-wise Technical Indicator Optimization with Stock Embedding. Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (2019).
[27]
Xin Liang, Dawei Cheng, Fangzhou Yang, Yifeng Luo, Weining Qian, and Aoying Zhou. 2020. F-HMTC: Detecting Financial Events for Investment Decisions Based on Neural Hierarchical Multi-Label Text Classification. In IJCAI. 4490--4496.
[28]
Qikai Liu, Xiang Cheng, Sen Su, and Shuguang Zhu. [n.d.]. Hierarchical Complementary Attention Network for Predicting Stock Price Movements with News. In Proceedings of the 27th ACM International Conference on Information and Knowledge Management, CIKM 2018, Torino, Italy, October 22--26, 2018. 1603--1606.
[29]
Weiwen Liu, Yin Zhang, Jianling Wang, Yun He, James Caverlee, Patrick PK Chan, Daniel S Yeung, and Pheng-Ann Heng. 2021. Item Relationship Graph Neural Networks for E-Commerce. IEEE Transactions on Neural Networks and Learning Systems (2021).
[30]
Xiao-Yang Liu, Jingyang Rui, Jiechao Gao, Liuqing Yang, Hongyang Yang, Zhaoran Wang, Christina Dan Wang, and Guo Jian. 2021. FinRL-Meta: Data-Driven Deep Reinforcement Learning in Quantitative Finance. Data-Centric AI Workshop, NeurIPS (2021).
[31]
Simone Merello, Andrea Picasso Ratto, L. Oneto, and E. Cambria. 2019. Ensemble Application of Transfer Learning and Sample Weighting for Stock Market Prediction. 2019 International Joint Conference on Neural Networks (IJCNN) (2019), 1--8.
[32]
Daniel N. Myers and James W. McGuffee. 2015. Choosing Scrapy. Journal of Computing Sciences in Colleges.
[33]
World Federation of Exchanges database. 2021. Market capitalization of listed domestic companies. https://data.worldbank.org/indicator/CM.MKT.LCAP.CD accessed 25-July-2021.
[34]
Ashwin Paranjape, Austin R. Benson, and Jure Leskovec. 2017. Motifs in Temporal Networks. Proceedings of the Tenth ACM International Conference on Web Search and Data Mining (2017).
[35]
Ramit Sawhney, S. Agarwal, Arnav Wadhwa, and R. Shah. 2020. Deep Attentive Learning for Stock Movement Prediction From Social Media Text and Company Correlations. In EMNLP.
[36]
Ramit Sawhney, Shivam Agarwal, Arnav Wadhwa, and Rajiv Shah. 2020. Deep Attentive Learning for Stock Movement Prediction from Social Media Text and Company Correlations. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP). 8415--8426.
[37]
Y. Tu, Li Niu, Junjie Chen, Dawei Cheng, and Liqing Zhang. 2020. Learning From Web Data With Self-Organizing Memory Module. 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2020), 12843--12852.
[38]
Yi Tu, Li Niu, Weijie Zhao, Dawei Cheng, and Liqing Zhang. 2020. Image cropping with composition and saliency aware aesthetic score map. In Proceedings of the AAAI Conference on Artificial Intelligence, Vol. 34. 12104--12111.
[39]
Ashish Vaswani, Noam M. Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, and Illia Polosukhin. 2017. Attention is All you Need. ArXiv abs/1706.03762 (2017).
[40]
Yaowei Wang, Qing Li, Zhexue Huang, and Mark Junjie Li. [n.d.]. EAN: Event Attention Network for Stock Price Trend Prediction based on Sentimental Embedding. In Proceedings of the 11th ACM Conference on Web Science, WebSci 2019, Boston, MA, USA, June 30 - July 03, 2019. 311--320.
[41]
Zonghan Wu, Shirui Pan, Fengwen Chen, Guodong Long, Chengqi Zhang, and S Yu Philip. 2020. A comprehensive survey on graph neural networks. IEEE transactions on neural networks and learning systems 32, 1 (2020), 4--24.
[42]
Sheng Xiang, Dong Wen, Dawei Cheng, Ying Zhang, Lu Qin, Zhengping Qian, and Xuemin Lin. 2021. General graph generators: experiments, analyses, and improvements. The VLDB Journal (2021).
[43]
Wentao Xu, Weiqing Liu, Chang Xu, Jiang Bian, Jian Yin, and Tie-Yan Liu. 2021. REST: Relational Event-driven Stock Trend Forecasting. Proceedings of the Web Conference 2021 (2021).
[44]
Chengxuan Ying, Tianle Cai, Shengjie Luo, Shuxin Zheng, Guolin Ke, Di He, Yanming Shen, and Tie-Yan Liu. 2021. Do Transformers Really Perform Bad for Graph Representation?. In NeurIPS.
[45]
Dawei Zhou, Lecheng Zheng, Jiawei Han, and Jingrui He. 2020. A Data-Driven Graph Generative Model for Temporal Interaction Networks. Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (2020).

Cited By

View all
  • (2024)Design and Implementation of an Intelligent Financial Management System Based on Enterprise Legal SystemJournal of Organizational and End User Computing10.4018/JOEUC.35022436:1(1-25)Online publication date: 12-Aug-2024
  • (2024)Graph Time-series Modeling in Deep Learning: A SurveyACM Transactions on Knowledge Discovery from Data10.1145/363853418:5(1-35)Online publication date: 28-Feb-2024
  • (2024)Neural Attributed Community Search at Billion ScaleProceedings of the ACM on Management of Data10.1145/36267381:4(1-25)Online publication date: 26-Apr-2024
  • Show More Cited By

Index Terms

  1. Temporal and Heterogeneous Graph Neural Network for Financial Time Series Prediction

    Recommendations

    Comments

    Information & Contributors

    Information

    Published In

    cover image ACM Conferences
    CIKM '22: Proceedings of the 31st ACM International Conference on Information & Knowledge Management
    October 2022
    5274 pages
    ISBN:9781450392365
    DOI:10.1145/3511808
    • General Chairs:
    • Mohammad Al Hasan,
    • Li Xiong
    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]

    Sponsors

    Publisher

    Association for Computing Machinery

    New York, NY, United States

    Publication History

    Published: 17 October 2022

    Permissions

    Request permissions for this article.

    Check for updates

    Author Tags

    1. financial time series
    2. graph neural network
    3. heterogeneous graph
    4. stock movement prediction

    Qualifiers

    • Research-article

    Funding Sources

    • Australian Research Council
    • National Natural Science Foundation of China

    Conference

    CIKM '22
    Sponsor:

    Acceptance Rates

    CIKM '22 Paper Acceptance Rate 621 of 2,257 submissions, 28%;
    Overall Acceptance Rate 1,861 of 8,427 submissions, 22%

    Upcoming Conference

    Contributors

    Other Metrics

    Bibliometrics & Citations

    Bibliometrics

    Article Metrics

    • Downloads (Last 12 months)533
    • Downloads (Last 6 weeks)47
    Reflects downloads up to 30 Aug 2024

    Other Metrics

    Citations

    Cited By

    View all
    • (2024)Design and Implementation of an Intelligent Financial Management System Based on Enterprise Legal SystemJournal of Organizational and End User Computing10.4018/JOEUC.35022436:1(1-25)Online publication date: 12-Aug-2024
    • (2024)Graph Time-series Modeling in Deep Learning: A SurveyACM Transactions on Knowledge Discovery from Data10.1145/363853418:5(1-35)Online publication date: 28-Feb-2024
    • (2024)Neural Attributed Community Search at Billion ScaleProceedings of the ACM on Management of Data10.1145/36267381:4(1-25)Online publication date: 26-Apr-2024
    • (2024)Heterogeneous Dual-Dynamic Attention Network for Modeling Mutual Interplay of StocksIEEE Transactions on Artificial Intelligence10.1109/TAI.2024.33742695:7(3595-3606)Online publication date: Jul-2024
    • (2024)Multi-Relational Graph Diffusion Neural Network with Parallel Retention for Stock Trends ClassificationICASSP 2024 - 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)10.1109/ICASSP48485.2024.10447394(6545-6549)Online publication date: 14-Apr-2024
    • (2024)Curriculum learning empowered reinforcement learning for graph-based portfolio management: Performance optimization and comprehensive analysisNeural Networks10.1016/j.neunet.2024.106537179(106537)Online publication date: Nov-2024
    • (2024)Towards a better negative sampling strategy for dynamic graphsNeural Networks10.1016/j.neunet.2024.106175(106175)Online publication date: Feb-2024
    • (2024)Predicting stock market trends with self-supervised learningNeurocomputing10.1016/j.neucom.2023.127033568:COnline publication date: 14-Mar-2024
    • (2024)Artificial intelligence techniques in financial trading: A systematic literature reviewJournal of King Saud University - Computer and Information Sciences10.1016/j.jksuci.2024.10201536:3(102015)Online publication date: Mar-2024
    • (2024)VGC-GANExpert Systems with Applications: An International Journal10.1016/j.eswa.2023.121204236:COnline publication date: 1-Feb-2024
    • Show More Cited By

    View Options

    Get Access

    Login options

    View options

    PDF

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    Media

    Figures

    Other

    Tables

    Share

    Share

    Share this Publication link

    Share on social media