SP-LAN: A Stock Prediction Model Based on LSTM-Attention Network.

Jingyou Sun,Dong Li,Xing Wang,Yue Kou,Peixuan Li,Yang Xie
DOI: https://doi.org/10.1007/978-981-99-1354-1_7
2022-01-01
Abstract:With the prosperity of the stock market and the rapid development of artificial intelligence, stock prediction technology has been widely concerned in recent years. Currently, most existing works only focus on stock prediction models via machine learning, ignoring some problems, such as local minima, gradient vanishing and interaction of information at different times. The stock data has characteristics like high noise and nonlinearity, which increase the difficulty of modeling. To solve these problems, we propose the Stock Prediction Model based on LSTM-Attention Network (called SP-LAN). Unlike traditional methods, SP-LAN can train the model with excellent generalization ability. SP-LAN also can make the model better learn interaction of information at different times, thus improving the prediction effectiveness of our model. In addition, we propose a novel feature selection strategy based on embedded importance to reduce the dimensions of data, which makes model train easier. Finally, we conduct experimental studies based on the daily trading dataset of SSE 50. The experimental results demonstrate the effectiveness and efficiency of our model.
What problem does this paper attempt to address?