Stock Return Prediction Using Dual-Stage Attention Model with Stock Relation Inference
- 327 Downloads
Deep learning models have become widely accessible for stock prediction tasks. However, most of the research in this area focuses on only a single stock or an index and often formulates the problem to optimize only on the accuracy. Our paper proposed a more profit-oriented framework by formulating the problem into multiple stock returns prediction as well as introducing a relation inference for stock ranking. This setup can diversify investment and eventually enhance trading profits while maintaining the regression accuracy. Moreover, it is become more challenging to process multiple time-series features simultaneously because of the great variety of available information in the financial market. We mitigate this with the state-of-the-art model for time-series forecasting, the Dual-stage attention recurrent neural networks (DA-RNN), and train them with the shared-parameter model setting. The attention layer within DA-RNN helps the model captures the relevance insight among the features. We conducted experiments on major 64 target stocks from the SET market with RMSE, mean reciprocal rank, and annualized profit returns as evaluation metrics. The results show that our proposed model framework (DA-RANK) can predict multiple stock returns in ranking order and able to produce a desirable improvement in profitability over other baseline models.
KeywordsDeep learning Ranking-aware loss function Long Short-term memory model Dual-stage attention Stock prediction Stock ranking
Our work is partly supported on the research budget by Capital Market Research Institute (CMRI), The Stock Exchange of Thailand (SET), during Capital Market Research Innovation contest 2019. and permission to use the SET SMART dataset for academic purposes from the Financial Laboratory, Chulalongkorn Business School and Asst. Prof. Tanakorn Likitapiwat.
- 3.Qin, Y., Song, D., Chen, H., Cheng, W., Jiang, G., Cottrell, G.: A dual-stage attention-based recurrent neural network for time series prediction. arXiv e-prints (2017)Google Scholar
- 8.Chen, K., Zhou, Y., Dai, F.: A LSTM-based method for stock returns prediction: a case study of china stock market. In: 2015 IEEE International Conference on Big Data (Big Data), pp. 2823–2824. IEEE (2015)Google Scholar
- 9.Nelson, D.M., Pereira, A.C., de Oliveira, R.A.: Stock market’s price movement prediction with LSTM neural networks. In: 2017 International Joint Conference on Neural Networks (IJCNN), pp. 1419–1426. IEEE (2017)Google Scholar
- 12.Hollis, T., Viscardi, A., Yi, S.E.: A comparison of LSTMs and attention mechanisms for forecasting financial time series. CoRR abs/1812.07699 (2018)Google Scholar
- 13.Guo, T., Lin, T.: Multi-variable LSTM neural network for autoregressive exogenous model. arXiv preprint arXiv:1806.06384 (2018)
- 14.Akita, R., Yoshihara, A., Matsubara, T., Uehara, K.: Deep learning for stock prediction using numerical and textual information. In: 2016 IEEE/ACIS 15th International Conference on Computer and Information Science (ICIS), 26–29 June 2016, pp. 1–6 (2016)Google Scholar
- 15.Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. arXiv preprint arXiv:1502.03167 (2015)
- 16.Oncharoen, P., Vateekul, P.: Deep learning using risk-reward function for stock market prediction. Paper Presented at the 2018 2nd International Conference on Computer Science and Artificial Intelligence, Shenzhen, China (2018)Google Scholar