skip to main content
10.1145/3573942.3574019acmotherconferencesArticle/Chapter ViewAbstractPublication PagesaiprConference Proceedingsconference-collections
research-article

Attention-based BiLSTM model for stock price prediction

Published: 16 May 2023 Publication History

Abstract

Abstract: Stock price prediction is a hot issue in the field of quantitative finance. Investors hope to discover the objective laws of stock price fluctuations from historical data, optimize their investment strategies, and avoid risks to obtain better investment returns. With the development of deep learning technology, neural networks have shown good forecasting effects in task of time series data forecasting. Aiming at the temporal correlation of stock data, a bidirectional LSTM network stock price prediction model fused with attention mechanism is proposed. The experimental results show that the bidirectional LSTM network can effectively learn the correlation between data when performing the prediction task, and the attention module helps the model to better capture the key information in the stock data. Compared with other prediction networks, the model has higher prediction accuracy and lower prediction error, and achieves the best prediction performance on different datasets, which can provide help for stock price prediction.

References

[1]
Hochreiter S, Schmidhuber J . Long Short-Term Memory[J]. Neural Computation, 1997, 9(8):1735-1780.
[2]
Huang Z, Wei X, Kai Y. Bidirectional LSTM-CRF Models for Sequence Tagging[J]. Computer Science, 2015.
[3]
Yao Jinhai. Study on Stock Index Prediction Based on ARIMA and Information Granular SVR Combination [J]. Operations Research and Management Science,2022,31(05):214-220.
[4]
Wang F.Wang X Y.Chen S.Machine Learning in Economics Research:Review and Prospective[J]. The Journal of Quantitative & Technical Economics. 2020,37(04):146-164.
[5]
Engle R F, Gonzalo R J. The Spline-GARCH Model for Low-frequency Volatility and Its Global Macroeconomic Causes [J].The Review of Financial Studies,2008,21(3).
[6]
L. Di Persio and O. Honchar. Artificial neural networks architectures for stock price prediction:Comparisons and applications. International journal of circuits, systems and signal processing,10(2016):403–413, 2016.
[7]
Krauss,X.A.Do, and N.Huck. Deep neural networks, gradient-boosted trees, random forests: Statistical arbitrage on the s&p 500. European Journal of Operational Research,259(2):689–702, 2017.
[8]
J.Patel, S.Shah,P.Thakkar, and K.Kotecha. Predicting stock and stock price index movement using trend deterministic data preparation and machine learning techniques. Expert Systems with Applications, 42(1):259–268, 2015.
[9]
N. I. Sapankevych and R. Sankar. Time series prediction using support vector machines: A survey. IEEE Computational Intelligence Magazine, 4(2):24–38, 2009.
[10]
Xu G X, Yang Z J. Research for Construction and Application of PCA-GA-SVM Model. The Journal of Quantitative & Technical Economics. 2011,28(02):135-147.
[11]
Ahmed N K, Atiya A F, Gayar N E, An Empirical Comparison of Machine Learning Models for Time Series Forecasting[J]. Econometric Reviews, 2010, 29(5-6):594-621.
[12]
CHEN J, LIU D X, WU D S. Stock Index Forecasting Method Based on Feature Selection and LSTM Model[J]. Compuyer Engineering and Applications, 2019, 55(6): 108-112.
[13]
PENG Y, LIU Y H, ZHANG R F, Modeling and Analysis of Stock Price Forecast Based on LSTM[J]. Computer Engineering and Applications, 2019,55(11):209-212.
[14]
T. Fischer and C. Krauss. Deep learning with long short-term memory networks for financial market predictions. European Journal of Operational Research, 270(2):654–669, 2018.
[15]
S. Siami-Namini, N. Tavakoli, and A. Siami Namin. A comparison of arima and lstm in forecasting time series. In 2018 17th IEEE International Conference on Machine Learning and Applications (ICMLA), pages 1394–1401, 2018.

Recommendations

Comments

Information & Contributors

Information

Published In

cover image ACM Other conferences
AIPR '22: Proceedings of the 2022 5th International Conference on Artificial Intelligence and Pattern Recognition
September 2022
1221 pages
ISBN:9781450396899
DOI:10.1145/3573942
Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

Publisher

Association for Computing Machinery

New York, NY, United States

Publication History

Published: 16 May 2023

Permissions

Request permissions for this article.

Check for updates

Author Tags

  1. Attention Mechanism
  2. BiLSTM Network
  3. Deep Learning
  4. Stock Price Prediction

Qualifiers

  • Research-article
  • Research
  • Refereed limited

Conference

AIPR 2022

Contributors

Other Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

  • 0
    Total Citations
  • 89
    Total Downloads
  • Downloads (Last 12 months)38
  • Downloads (Last 6 weeks)7
Reflects downloads up to 01 Mar 2025

Other Metrics

Citations

View Options

Login options

View options

PDF

View or Download as a PDF file.

PDF

eReader

View online with eReader.

eReader

HTML Format

View this article in HTML Format.

HTML Format

Figures

Tables

Media

Share

Share

Share this Publication link

Share on social media