GraphCNNpred: A stock market indices prediction using a Graph based deep learning system

Yuhui Jin
2024-07-17
Abstract:The application of deep learning techniques for predicting stock market prices is a prominent and widely researched topic in the field of data science. To effectively predict market trends, it is essential to utilize a diversified dataset. In this paper, we give a graph neural network based convolutional neural network (CNN) model, that can be applied on diverse source of data, in the attempt to extract features to predict the trends of indices of \text{S}\&\text{P} 500, NASDAQ, DJI, NYSE, and RUSSEL. The experiments show that the associated models improve the performance of prediction in all indices over the baseline algorithms by about $4\% \text{ to } 15\%$, in terms of F-measure. A trading simulation is generated from predictions and gained a Sharpe ratio of over 3.
Computational Finance,Machine Learning
What problem does this paper attempt to address?
The paper attempts to address the issue of improving prediction accuracy in stock market index forecasting. Specifically, the paper proposes a method based on Graph Convolutional Neural Networks (Graph CNN), aiming to extract features from diverse data sources to predict the index trends of major stock markets such as S&P 500, NASDAQ, DJI, NYSE, and RUSSEL. Experimental results show that compared to baseline algorithms, the model improves prediction performance by approximately 4% to 15% across all indices, particularly excelling in the F-measure metric. Additionally, trading simulations based on this model achieved a Sharpe ratio of over 3. The paper combines Graph Neural Networks (GNN) and Convolutional Neural Networks (CNN) to address the challenge of traditional methods in simultaneously handling time series features and relational features. This approach not only respects the time series characteristics of the data but also fully utilizes all available features, thereby enhancing the accuracy and robustness of the predictions.