Unknown

Dataset Information

0

Forecasting stock prices with long-short term memory neural network based on attention mechanism.


ABSTRACT: The stock market is known for its extreme complexity and volatility, and people are always looking for an accurate and effective way to guide stock trading. Long short-term memory (LSTM) neural networks are developed by recurrent neural networks (RNN) and have significant application value in many fields. In addition, LSTM avoids long-term dependence issues due to its unique storage unit structure, and it helps predict financial time series. Based on LSTM and an attention mechanism, a wavelet transform is used to denoise historical stock data, extract and train its features, and establish the prediction model of a stock price. We compared the results with the other three models, including the LSTM model, the LSTM model with wavelet denoising and the gated recurrent unit(GRU) neural network model on S&P 500, DJIA, HSI datasets. Results from experiments on the S&P 500 and DJIA datasets show that the coefficient of determination of the attention-based LSTM model is both higher than 0.94, and the mean square error of our model is both lower than 0.05.

SUBMITTER: Qiu J 

PROVIDER: S-EPMC6941898 | biostudies-literature | 2020

REPOSITORIES: biostudies-literature

altmetric image

Publications

Forecasting stock prices with long-short term memory neural network based on attention mechanism.

Qiu Jiayu J   Wang Bin B   Zhou Changjun C  

PloS one 20200103 1


The stock market is known for its extreme complexity and volatility, and people are always looking for an accurate and effective way to guide stock trading. Long short-term memory (LSTM) neural networks are developed by recurrent neural networks (RNN) and have significant application value in many fields. In addition, LSTM avoids long-term dependence issues due to its unique storage unit structure, and it helps predict financial time series. Based on LSTM and an attention mechanism, a wavelet tr  ...[more]

Similar Datasets

| S-EPMC6738919 | biostudies-other
| S-EPMC9205526 | biostudies-literature
| S-EPMC5800645 | biostudies-literature
| S-EPMC3628858 | biostudies-literature
| S-EPMC9234322 | biostudies-literature
| S-EPMC6518582 | biostudies-literature
| S-EPMC8532313 | biostudies-literature
| S-EPMC6312251 | biostudies-literature
| S-EPMC4667993 | biostudies-literature
| S-EPMC9487135 | biostudies-literature