News-driven stock market index prediction based on trellis network and sentiment attention mechanism

Abstract

Applying deep learning, especially time series neural network, to stock market prediction, has become one of the important applications in the quantitative finance field. However, due to the multi-correlation and volatility of the stock market, how to timely and accurately predict it has become a challenging issue. In order to cope with this challenge, a news-driven stock market index prediction model based on TrellisNet and a sentiment attention mechanism (SA-TrellisNet) is proposed. A sentiment analysis model based on CNN and LSTM is presented to obtain the sentiment index of massive news crawled from authoritative financial websites. Furthermore, a sentiment attention mechanism is designed for data fusion of stock data and news sentiment index as the input of the simple and efficient TrellisNet network for model training and prediction. The performance of our model is systematically evaluated using seven major international stock market indices including S&P500, NYSE, DJI, NASDAQ, FTSE 100, Nikkei 225 and SSE, and comparative experiments demonstrate that SA-TrellisNet is competitive to the other state-of-the-art methods in predicting stock market indices.