Exploring the attention mechanism in LSTM-based Hong Kong stock price movement prediction

Abstract

State-of-the-art methods using attention mechanism in Recurrent Neural Networks have shown exceptional performance targeting sequential predictions and classifications. We explore the attention mechanism in Long–Short-Term Memory (LSTM) network based stock price movement prediction. Our proposed model significantly enhances the LSTM prediction performance in the Hong Kong stock market. The attention LSTM (AttLSTM) model is compared with the LSTM model in Hong Kong stock movement prediction. Further parameter tuning results also demonstrate the effectiveness of the attention mechanism in LSTM-based prediction method.