Stock and market index prediction using Informer network

Yuze Lu,Hailong Zhang,Qiwen Guo
2023-05-22
Abstract:Applications of deep learning in financial market prediction has attracted huge attention from investors and researchers. In particular, intra-day prediction at the minute scale, the dramatically fluctuating volume and stock prices within short time periods have posed a great challenge for the convergence of networks result. Informer is a more novel network, improved on Transformer with smaller computational complexity, longer prediction length and global time stamp features. We have designed three experiments to compare Informer with the commonly used networks LSTM, Transformer and BERT on 1-minute and 5-minute frequencies for four different stocks/ market indices. The prediction results are measured by three evaluation criteria: MAE, RMSE and MAPE. Informer has obtained best performance among all the networks on every dataset. Network without the global time stamp mechanism has significantly lower prediction effect compared to the complete Informer; it is evident that this mechanism grants the time series to the characteristics and substantially improves the prediction accuracy of the networks. Finally, transfer learning capability experiment is conducted, Informer also achieves a good performance. Informer has good robustness and improved performance in market prediction, which can be exactly adapted to real trading.
Statistical Finance,Artificial Intelligence
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve This paper aims to address several key issues in stock market prediction, particularly in short-term fluctuation prediction on a minute-level time scale. Specifically: 1. **Limitations of Existing Models**: - Although LSTM models perform well in stock price prediction, their serial structure leads to high memory requirements and a complex backpropagation process, making the results difficult to converge. - While Transformer models outperform LSTM in handling long sequence data, they still face issues in computational complexity, memory consumption, and long-sequence output speed. 2. **Introduction of the Informer Model**: - Informer is an improved model based on Transformer, addressing the high time complexity, memory bottleneck, and low decoding efficiency issues in long-sequence prediction with Transformer. - Informer introduces a global timestamp mechanism, enabling the model to better identify active trading periods (such as opening and closing times), thereby improving prediction accuracy. With these improvements, Informer performs excellently in minute-level stock price prediction tasks, enabling real-time stock price prediction and reducing financial risks.