Research on Stock Index Prediction Driven by Multi-source Heterogeneous Data Fusion

doi:10.3778/j.issn.1002-8331.2012-0498

Abstract

Abstract:

With the wide application of modern information technology, capital market investors can obtain more timely and valuable information, and they are more susceptible to the influence of financial forums and professional investment websites. It has become a hot topic in this field to predict stock index by integrating multi-source heterogeneous data of capital market. A Long Short-Term Memory（LSTM） model based on multi-source heterogeneous data is proposed to predict the trend of stock indexes by quantifying three sources of data, including capital market transaction data, technical index data and investor sentiment. At the same time, a Convolutional Neural Network（CNN） sentiment analysis model is proposed to extract deep emotion features, and the investor sentiment feature model is constructed. Experimental results using “Shanghai 50 Index” data show that the prediction accuracy of LSTM model is better than the traditional model, and the increase of data sources also makes a great contribution to the improvement of model accuracy, which verifies the feasibility and effectiveness of this method.

Key words: stock forecast, transaction data, emotional analysis, Long Short-Term Memory（LSTM）, convolutional neural network

摘要：

现代信息技术的广泛应用使得资本市场投资者能够获得更及时、更有价值的信息，也更容易受到金融论坛、专业投资网站的影响。融合资本市场的多源异构数据对股票指数进行预测成为该领域的研究热点。提出了一种基于多源异构数据的长短期神经网络（Long Short-Term Memory，LSTM）模型，通过对融合资本市场交易数据、技术指标数据、投资者情绪三种源数据的量化来预测股票指数的走势。提出了一种可以提取深度情感特征的卷积神经网络（Convolutional Neural Networks，CNN）情感分析模型，构建了投资者情绪特征模型。利用“上证50指数”数据进行实验，结果显示：LSTM模型的预测准确率比传统模型更为优秀，数据源的增加也对模型准确率的提升有较大贡献，验证了该方法的可行性和有效性。

关键词: 股票预测, 交易数据, 情感分析, 长短期神经网络, 卷积神经网络

GENG Lixiao, LIU Lisha, LI Hengyu. Research on Stock Index Prediction Driven by Multi-source Heterogeneous Data Fusion[J]. Computer Engineering and Applications, 2021, 57(20): 142-149.

耿立校，刘丽莎，李恒昱. 多源异构数据融合驱动的股票指数预测研究[J]. 计算机工程与应用, 2021, 57(20): 142-149.

References

[1] FAMA E F.Efficient capital markets：II[J].The Journal of Finance，1991，46（5）：1575-1617.
[2] RECHENTHIN M，STREET W N.Using conditional probability to identify trends in intra-day high-frequency equity pricing[J].Physica A-Statistical Mechanics and Its Applications，2013，392（24）：6169-6188.
[3] 孔翔宇，毕秀春，张曙光.财经新闻与股市预测——基于数据挖掘技术的实证分析[J].数理统计与管理，2016，35（2）：215-224.
KONG X Y，BI X C，ZHANG S G.Financial news and prediction for stock market：an empirical analysis based on data mining techniques[J].Journal of Applied Statistics and Management，2016，35（2）：215-224.
[4] 陈卫华，徐国祥.基于深度学习和股票论坛数据的股市波动率预测精度研究[J].管理世界，2018，34（1）：180-181.
CHEN W H，XU G X.Research on prediction accuracy of stock market volatility based on deep learning and stock forum data[J].Management World，2008，34（1）：180-181.
[5] CHAN W S.Stock price reaction to news and no-news：drift and reversal after headlines[J].Journal of Financial Economics，2003，70（2）：223-260.
[6] VEGA C.Stock price reaction to public and private information[J].Journal of Financial Economics，2006，82（1）：103-133.
[7] CHEUNG Y W，NG L K.Stock price dynamics and firm size：an empirical investigation[J].The Journal of Finance，1992，47（5）：1985-1997.
[8] BOLLEN J，MAO H N，ZENG X J.Twitter mood predicts the stock market[J].Journal of Computational Science，2011，2（1）：1-8.
[9] LI Q，WANG T J，LI P，et al.The effect of news and public mood on stock movements[J].Information Sciences，2014，278：826-840.
[10] NGUYEN T H，SHIRAI K，VELCIN J.Sentiment analysis on social media for stock movement prediction[J].Expert Systems with Applications，2015，42（24）：9603-9611.
[11] NASSIRTOUSSI A K，AGHABOZORGI S，TEH Y W，et al.Text mining for market prediction：a systematic review[J].Expert Systems with Applications，2014，41（16）：7653-7670.
[12] GROSS-KLUSSMANN A，KONIG S，EBNER M.Buzzwords build momentum：global financial Twitter sentiment and the aggregate stock market[J].Expert Systems with Applications，2019，136：171-186.
[13] LIU Y，ZENG Q G，MERE J O，et al.Anticipating stock market of the renowned companies：a knowledge graph approach[J].Complexity，2019.
[14] XUN J Y，GUO B A.Twitter as customer’s eWOM：an empirical study on their impact on firm financial performance[J].Internet Research，2017，27（5）：1014-1038.
[15] 王燕，郭元凯.改进的XGBoost模型在股票预测中的应用[J].计算机工程与应用，2019，55（20）：202-207.
WANG Y，GUO Y K.Application of improved XGBoost model in stock forecasting[J].Computer Engineering and Applications，2019，55（20）：202-207.
[16] KRAUS M，FEUERRIEGEL S.Decision support from financial disclosures with deep neural networks and transfer learning[J].Decision Support Systems，2017，104：38-48.
[17] 赵红蕊，薛雷.基于LSTM-CNN-CBAM模型的股票预测研究[J].计算机工程与应用，2021，57（3）：203-207.
ZHAO H R，XUE L.Research on stock forecasting based on LSTM-CNN-CBAM model[J].Computer Engineering and Applications，2021，57（3）：203-207.
[18] HOCHREITER S，SCHMIDHUBER J.Long short-term memory[J].Neural Computation，1997，9（8）：1735-1780.