《计算机应用》唯一官方网站 ›› 2022, Vol. 42 ›› Issue (7): 2265-2273.DOI: 10.11772/j.issn.1001-9081.2021081487

• 前沿与综合应用 • 上一篇    

基于多重注意力机制的图神经网络股市波动预测方法

李晓寒1(), 王俊1, 贾华丁1, 萧刘2   

  1. 1.西南财经大学 经济信息工程学院,成都 611130
    2.四川久远银海软件股份有限公司 住房金融行业部,成都 610063
  • 收稿日期:2021-08-19 修回日期:2021-11-30 接受日期:2021-12-03 发布日期:2022-01-07 出版日期:2022-07-10
  • 通讯作者: 李晓寒
  • 作者简介:王俊(1987—),男,山东青岛人,副教授,博士,CCF会员,主要研究方向:金融科技、金融智能
    贾华丁(1956—),男,四川成都人,教授,博士,CCF高级会员,主要研究方向:机器学习、算法交易、扩频序列设计
    萧刘(1986—),男,四川成都人,高级工程师,主要研究方向:机器学习、大数据、社会保障经济与金融。
  • 基金资助:
    四川省科技计划项目(2020JDJQ0061)

Stock market volatility prediction method based on graph neural network with multi-attention mechanism

Xiaohan LI1(), Jun WANG1, Huading JIA1, Liu XIAO2   

  1. 1.School of Economic Information Engineering,Southwestern University of Finance and Economics,Chengdu Sichuan 611130,China
    2.Housing Finance Industry Department,Sichuan Jiuyuan Yinhai Software Company Limited,Chengdu Sichuan 610063,China
  • Received:2021-08-19 Revised:2021-11-30 Accepted:2021-12-03 Online:2022-01-07 Published:2022-07-10
  • Contact: Xiaohan LI
  • About author:WANG Jun, born in 1987, Ph. D., associate professor. His research interests include financial technology, financial intelligence.
    JIA Huading, born in 1956, Ph. D., professor. His research interests include machine learning, algorithmic trading, spread spectrum sequence design.
    XIAO Liu, born in 1986, senior engineer. His research interests include machine learning, big data,social security, economics and finance.
  • Supported by:
    Science and Technology Program of Sichuan Province(2020JDJQ0061)

摘要:

股票市场是金融市场关键组成部分,因此对股票市场波动的研究对合理化控制金融市场风险、提高投资收益提供了重要支持,一直以来都是学术界和相关业界的关注焦点,然而,股票市场会受到各种因素的影响。面对股票市场中多源化、异构化的信息,如何高效挖掘、融合股票市场的多源异构数据具有挑战性。为了充分解释不同信息及信息间相互作用对于股票市场价格波动的影响,提出一种基于多重注意力机制的图神经网络来预测股票市场的价格波动。首先,引入关系维度构建股票市场交易数据和新闻文本的异构子图,并利用多重注意力机制实现图数据的融合;其次,通过图神经网络门控循环单元(GRU)进行图分类,在此基础上完成对股票市场中上证综合指数、沪深300指数、深证成份指数这三个重要指数波动的预测。实验结果表明,从异构信息特性角度,相较于股票市场交易数据,股市新闻信息对于股票价格影响存在滞后性;从异构信息融合角度,所提方法与支持向量机(SVM)、随机森林、多核k-means (MKKM)聚类等算法相比,预测准确率分别提升了17.88个百分点、30.00个百分点和38.00个百分点,并进行了模型交易策略的量化投资模拟。

关键词: 股市预测, 多重注意力机制, 图神经网络, 股市新闻, 图数据

Abstract:

Stock market is an essential element of financial market, therefore, the study on volatility of stock market plays a significant role in taking effective control of financial market risks and improving returns on investment. For this reason, it has attracted widespread attention from both academic circle and related industries. However, there are multiple influencing factors for stock market. Facing the multi-source and heterogeneous information in stock market, it is challenging to find how to mine and fuse multi-source and heterogeneous data of stock market efficiently. To fully explain the influence of different information and information interaction on the price changes in stock market, a graph neural network based on multi-attention mechanism was proposed to predict price fluctuation in stock market. First of all, the relationship dimension was introduced to construct heterogeneous subgraphs for the transaction data and news text of stock market, and multi-attention mechanism was adopted for fusion of the graph data. Then, the graph neural network Gated Recurrent Unit (GRU) was applied to perform graph classification. On this basis, prediction was made for the volatility of three important indexes: Shanghai Composite Index, Shanghai and Shenzhen 300 Index, Shenzhen Component Index. Experimental results show that from the perspective of heterogeneous information characteristics, compared with the transaction data of stock market, the news information of stock market has the lagged influence on stock volatility; from the perspective of heterogeneous information fusion, compared with algorithms such as Support Vector Machine (SVM), Random Forest (RF) and Multiple Kernel k-Means (MKKM) clustering, the proposed method has the prediction accuracy improved by 17.88 percentage points, 30.00 percentage points and 38.00 percentage points respectively; at the same time, the quantitative investment simulation was performed according to the model trading strategy.

Key words: stock market prediction, multi-attention mechanism, graph neural network, stock market news, graph data

中图分类号: