IDEAS home Printed from https://ideas.repec.org/a/wsi/ijfexx/v09y2022i03ns2424786322500141.html
   My bibliography  Save this article

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

Author

Listed:
  • Yuyang Lin

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

  • Qi Huang

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

  • Qiyin Zhong

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

  • Muyang Li

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

  • Yan Li

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

  • Fei Ma

    (Applied Mathematics, School of Sciences, Xi’an Jiaotong-Liverpool University, Suzhou, P. R. China)

Abstract

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.

Suggested Citation

  • Yuyang Lin & Qi Huang & Qiyin Zhong & Muyang Li & Yan Li & Fei Ma, 2022. "A new attention-based LSTM model for closing stock price prediction," International Journal of Financial Engineering (IJFE), World Scientific Publishing Co. Pte. Ltd., vol. 9(03), pages 1-17, September.
    • Handle: RePEc:wsi:ijfexx:v:09:y:2022:i:03:n:s2424786322500141
    DOI: 10.1142/S2424786322500141
    as

    Download full text from publisher

    File URL: http://www.worldscientific.com/doi/abs/10.1142/S2424786322500141
    Download Restriction: Access to full text is restricted to subscribers
    File URL: https://libkey.io/10.1142/S2424786322500141?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:wsi:ijfexx:v:09:y:2022:i:03:n:s2424786322500141. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Tai Tone Lim (email available below). General contact details of provider: http://www.worldscientific.com/worldscinet/ijfe .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.