基于注意力机制和LSTM网络的股价预测
Stock Price Prediction Based on Attention Mechanism and Long Short-Term Memory Network
DOI: 10.12677/AAM.2021.1012466, PDF,  被引量   
作者: 刘 甲, 孙德山:辽宁师范大学数学学院,辽宁 大连
关键词: LSTM注意力机制股价预测Long and Short Term Memory Network Attention Mechanism Stock Price Forecast
摘要: 随着全球经济金融一体化,股票市场的交易规模不断增大,传统的计量经济学难以充分学习股票市场的非线性变化。本文运用深度学习中的长短期记忆神经网络(LSTM)作为基本模型对股票数据的开盘价进行预测,实验选用广聚能源(000096)和北方国际(000065)两组数据,之后在LSTM网络模型中引入注意力机制(Attention Mechanism)。经实验该模型预测精度有明显提升,说明AM-LSTM网络模型在股票预测领域具有一定可靠性。
Abstract: With the integration of global economy and finance, the transaction scale of stock market keeps increasing, so it is difficult for traditional econometrics to fully learn the nonlinear changes of stock market. In this paper, long and short-term memory neural network (LSTM) in deep learning is used as the basic model to predict the opening price of stock data. In the experiment, data of Guangju Energy and Norinco International are selected, and then Attention Mechanism is introduced into THE LSTM network model. Through experiments, the prediction accuracy of the model is significantly improved, indicating that the AM-LSTM network model has certain reliability in the field of stock prediction.
文章引用:刘甲, 孙德山. 基于注意力机制和LSTM网络的股价预测[J]. 应用数学进展, 2021, 10(12): 4379-4385. https://doi.org/10.12677/AAM.2021.1012466

参考文献

[1] Silver, D., et al. (2016) Mastering the Game of Go with Deep Neural Networks and Tree Search. Nature, 529, 484-489. [Google Scholar] [CrossRef] [PubMed]
[2] Dezsi, E. and Nistor, I.A. (2016) Can Deep Machine Learning Outsmart the Market? A Comparison between Econometric Modelling and Long-Short Term Memory. Romanian Economic and Business Review, 11, 54-73.
[3] Lara-Benítez, P., Carranza-García, M. snd Riquelme, J.C. (2021) An Experimental Review on Deep Learning Architectures for Time Series Forecasting. International Journal of Neural Systems, 31, Article ID: 2130001. [Google Scholar] [CrossRef
[4] Hochreiter, S. and Schmidhuber, J. (1997) Long Short-Term Memory. Neural Computation, 9, 735-780. [Google Scholar] [CrossRef] [PubMed]
[5] Long, W., Lu, Z. and Cui, L. (2018) Deep Learning-Based Feature Engineering for Stock Price Movement Prediction. Knowledge-Based Systems, 164, 163-173. [Google Scholar] [CrossRef
[6] Bahdanau, D., Cho, K. and Bengio, Y. (2014) Neural Machine Translation by Jointly Learning to Align and Translate. arXiv:1409.0473 [cs.CL]
[7] 申翔翔, 侯新文, 尹传环. 深度强化学习中状态注意力机制的研究[J]. 智能系统学报, 2020, 15(2): 317-322.
[8] Kitaev, N., Kaiser, Ł. and Levskaya, A. (2020) Reformer: The Efficient Transformer.
https://arxiv.org/pdf/2001.04451.pdf
[9] 刘翀, 杜军平. 一种基于深度LSTM和注意力机制的金融数据预测方法[J]. 计算机科学, 2020, 47(12): 125-130.