Hierarchical Multi-Scale Gaussian Transformer for Stock Movement Prediction

Hierarchical Multi-Scale Gaussian Transformer for Stock Movement Prediction

Qianggang Ding, Sifan Wu, Hao Sun, Jiadong Guo, Jian Guo

Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence
Special Track on AI in FinTech. Pages 4640-4646. https://doi.org/10.24963/ijcai.2020/640

Predicting the price movement of finance securities like stocks is an important but challenging task, due to the uncertainty of financial markets. In this paper, we propose a novel approach based on the Transformer to tackle the stock movement prediction task. Furthermore, we present several enhancements for the proposed basic Transformer. Firstly, we propose a Multi-Scale Gaussian Prior to enhance the locality of Transformer. Secondly, we develop an Orthogonal Regularization to avoid learning redundant heads in the multi-head self-attention mechanism. Thirdly, we design a Trading Gap Splitter for Transformer to learn hierarchical features of high-frequency finance data. Compared with other popular recurrent neural networks such as LSTM, the proposed method has the advantage to mine extremely long-term dependencies from financial time series. Experimental results show our proposed models outperform several competitive methods in stock price prediction tasks for the NASDAQ exchange market and the China A-shares market.
Keywords:
Foundation for AI in FinTech: Deep learning and representation for FinTech
Foundation for AI in FinTech: Modeling financial structure and hierarchy
AI for trading: AI for algorithmic trading
AI for trading: AI for novel financial mechanisms
AI for trading: AI for novel financial models
AI for trading: AI for predictive trading
AI for trading: General