A new attention-based LSTM model for closing stock price prediction

被引:4
|
作者
Lin, Yuyang [1 ]
Huang, Qi [1 ]
Zhong, Qiyin [1 ]
Li, Muyang [1 ]
Li, Yan [1 ]
Ma, Fei [1 ]
机构
[1] Xian Jiaotong Liverpool Univ, Sch Sci, Appl Math, Suzhou, Peoples R China
关键词
Deep learning; long short-term memory; attention mechanism; stock market prediction;
D O I
10.1142/S2424786322500141
中图分类号
F8 [财政、金融];
学科分类号
0202 ;
摘要
Financial time-series prediction has been a demanding and popular subject in many fields. Latest progress in the deep learning technique, especially the deep neural network, shows great potentials in accomplishing this difficult task. This study explores the possible neural networks to improve the accuracy of the financial time-series prediction, while the main focus is to predict the closing price for next trading day. In this paper, we propose a new attention-based LSTM model (AT-LSTM) by combining the Long Short-Term Memory (LSTM) networks with the attention mechanism. Six stock markets indices with four features were used as the input to the model. We evaluate the model performance in terms of MSE, RMSE and MAE. The results for these three metrics are 0.4537, 0.6736 and 0.4858, respectively. The results suggest that our model is skillful in capturing financial time series, and the predictions are robust and stable. Furthermore, we compared our results with the previous work. As a result, our proposed AT-LSTM exhibits a significant performance improvement and outperforms other methods.
引用
收藏
页数:17
相关论文
共 50 条
  • [1] An Attention-Based LSTM Model for Stock Price Trend Prediction Using Limit Order Books
    Li, Yunhao
    Li, Liuliu
    Zhao, Xudong
    Ma, Tianyi
    Zou, Ying
    Chen, Ming
    [J]. 5TH ANNUAL INTERNATIONAL CONFERENCE ON INFORMATION SYSTEM AND ARTIFICIAL INTELLIGENCE (ISAI2020), 2020, 1575
  • [2] Attention-Based Event Relevance Model for Stock Price Movement Prediction
    Liu, Jian
    Chen, Yubo
    Liu, Kang
    Zhao, Jun
    [J]. KNOWLEDGE GRAPH AND SEMANTIC COMPUTING: LANGUAGE, KNOWLEDGE, AND INTELLIGENCE, CCKS 2017, 2017, 784 : 37 - 49
  • [3] Stock closing price prediction based on sentiment analysis and LSTM
    Zhigang Jin
    Yang Yang
    Yuhong Liu
    [J]. Neural Computing and Applications, 2020, 32 : 9713 - 9729
  • [4] Stock closing price prediction based on sentiment analysis and LSTM
    Jin, Zhigang
    Yang, Yang
    Liu, Yuhong
    [J]. NEURAL COMPUTING & APPLICATIONS, 2020, 32 (13): : 9713 - 9729
  • [5] Applied attention-based LSTM neural networks in stock prediction
    Cheng, Li-Chen
    Huang, Yu-Hsiang
    Wu, Mu-En
    [J]. 2018 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2018, : 4716 - 4718
  • [6] Multi-scale local cues and hierarchical attention-based LSTM for stock price trend prediction
    Teng, Xiao
    Zhang, Xiang
    Luo, Zhigang
    [J]. NEUROCOMPUTING, 2022, 505 : 92 - 100
  • [7] Stock price prediction based on LSTM and LightGBM hybrid model
    Tian, Liwei
    Feng, Li
    Yang, Lei
    Guo, Yuankai
    [J]. JOURNAL OF SUPERCOMPUTING, 2022, 78 (09): : 11768 - 11793
  • [8] Stock price prediction based on LSTM and LightGBM hybrid model
    Liwei Tian
    Li Feng
    Lei Yang
    Yuankai Guo
    [J]. The Journal of Supercomputing, 2022, 78 : 11768 - 11793
  • [9] A NOVEL MODEL FOR STOCK CLOSING PRICE PREDICTION USING CNN-ATTENTION-GRU-ATTENTION
    Lu, Wenjie
    Li, Jiazheng
    Wang, Jingyang
    Wu, Shaowen
    [J]. ECONOMIC COMPUTATION AND ECONOMIC CYBERNETICS STUDIES AND RESEARCH, 2022, 56 (03): : 251 - 264
  • [10] An Improved Attention-based Bidirectional LSTM Model for Cyanobacterial Bloom Prediction
    Jianjun Ni
    Ruping Liu
    Guangyi Tang
    Yingjuan Xie
    [J]. International Journal of Control, Automation and Systems, 2022, 20 : 3445 - 3455