Forecasting stock prices with long-short term memory neural network based on attention mechanism.
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 f...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2020-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0227222&type=printable |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850043565142441984 |
|---|---|
| author | Jiayu Qiu Bin Wang Changjun Zhou |
| author_facet | Jiayu Qiu Bin Wang Changjun Zhou |
| author_sort | Jiayu Qiu |
| collection | DOAJ |
| description | 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. |
| format | Article |
| id | doaj-art-4b9ab9093ead435c96bfdc2ae4c480d7 |
| institution | DOAJ |
| issn | 1932-6203 |
| language | English |
| publishDate | 2020-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-4b9ab9093ead435c96bfdc2ae4c480d72025-08-20T02:55:12ZengPublic Library of Science (PLoS)PLoS ONE1932-62032020-01-01151e022722210.1371/journal.pone.0227222Forecasting stock prices with long-short term memory neural network based on attention mechanism.Jiayu QiuBin WangChangjun ZhouThe 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.https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0227222&type=printable |
| spellingShingle | Jiayu Qiu Bin Wang Changjun Zhou Forecasting stock prices with long-short term memory neural network based on attention mechanism. PLoS ONE |
| title | Forecasting stock prices with long-short term memory neural network based on attention mechanism. |
| title_full | Forecasting stock prices with long-short term memory neural network based on attention mechanism. |
| title_fullStr | Forecasting stock prices with long-short term memory neural network based on attention mechanism. |
| title_full_unstemmed | Forecasting stock prices with long-short term memory neural network based on attention mechanism. |
| title_short | Forecasting stock prices with long-short term memory neural network based on attention mechanism. |
| title_sort | forecasting stock prices with long short term memory neural network based on attention mechanism |
| url | https://journals.plos.org/plosone/article/file?id=10.1371/journal.pone.0227222&type=printable |
| work_keys_str_mv | AT jiayuqiu forecastingstockpriceswithlongshorttermmemoryneuralnetworkbasedonattentionmechanism AT binwang forecastingstockpriceswithlongshorttermmemoryneuralnetworkbasedonattentionmechanism AT changjunzhou forecastingstockpriceswithlongshorttermmemoryneuralnetworkbasedonattentionmechanism |