论文标题

关系驱动的股票移动预测的多画卷积网络

Multi-Graph Convolutional Network for Relationship-Driven Stock Movement Prediction

论文作者

Ye, Jiexia, Zhao, Juanjuan, Ye, Kejiang, Xu, Chengzhong

论文摘要

由于金融市场的动荡性质,股票价格变动预测通常被认为是一项非常具有挑战性的任务。以前的工作通常主要根据其自己的信息来预测股票价格,从而忽略了相关股票之间的交叉效应。但是,众所周知,单个股票价格与其他股票的价格相关。为了考虑交叉效应,我们提出了一个称为多GCGRU的深度学习框架,该框架包括图形卷积网络(GCN)和门控复发单元(GRU)来预测库存运动。具体而言,我们首先根据金融领域知识将股票之间的多个关系编码为图表,并利用GCN根据这些预定义的图提取交叉效应。为了进一步摆脱先验知识,我们探讨了数据自动学到的自适应关系。 GCN产生的互相关功能与历史记录相连,然后进食GRU,以建模股票价格的时间依赖性。在中国市场上的两个股票指数的实验表明,我们的模型表现优于其他基线。请注意,我们的模型可以纳入包含专家知识的更有效的股票关系以及学习数据驱动的关系。

Stock price movement prediction is commonly accepted as a very challenging task due to the volatile nature of financial markets. Previous works typically predict the stock price mainly based on its own information, neglecting the cross effect among involved stocks. However, it is well known that an individual stock price is correlated with prices of other stocks in complex ways. To take the cross effect into consideration, we propose a deep learning framework, called Multi-GCGRU, which comprises graph convolutional network (GCN) and gated recurrent unit (GRU) to predict stock movement. Specifically, we first encode multiple relationships among stocks into graphs based on financial domain knowledge and utilize GCN to extract the cross effect based on these pre-defined graphs. To further get rid of prior knowledge, we explore an adaptive relationship learned by data automatically. The cross-correlation features produced by GCN are concatenated with historical records and then fed into GRU to model the temporal dependency of stock prices. Experiments on two stock indexes in China market show that our model outperforms other baselines. Note that our model is rather feasible to incorporate more effective stock relationships containing expert knowledge, as well as learn data-driven relationship.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源