GCN-based stock relations analysis for stock market prediction

PeerJ Comput Sci. 2022 Aug 11:8:e1057. doi: 10.7717/peerj-cs.1057. eCollection 2022.

Abstract

Most stock price predictive models merely rely on the target stock's historical information to forecast future prices, where the linkage effects between stocks are neglected. However, a group of prior studies has shown that the leverage of correlations between stocks could significantly improve the predictions. This article proposes a unified time-series relational multi-factor model (TRMF), which composes a self-generating relations (SGR) algorithm that can extract relational features automatically. In addition, the TRMF model integrates stock relations with other multiple dimensional features for the price prediction compared to extant works. Experimental validations are performed on the NYSE and NASDAQ data, where the model is compared with the popular methods such as attention Long Short-Term Memory network (Attn-LSTM), Support Vector Regression (SVR), and multi-factor framework (MF). Results show that compared with these extant methods, our model has a higher expected cumulative return rate and a lower risk of return volatility.

Keywords: Graph-based learning; LSTM; Multi-factor; Stock prediction; Stock relation; Time series.

Grants and funding

This work was supported by the National Natural Science Foundation of China (No. 61902349). The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.