Abstract
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
