Affiliation:
1. Institute for Quantum Information & State Key Laboratory of High Performance Computing, College of Computer Science and Technology, National University of Defense Technology, Changsha 410073, China
2. Institute of Software, School of Computer Science, Peking University, Beijing 100871, China
Abstract
As the Information Age brings people an amount of data, research on data prediction has been widely concerned. Time series data, a sequence of data points collected over an interval of time, are very common in many areas such as weather forecasting, stock markets, and so on. Thus, research on time series data prediction is of great significance. Traditional prediction methods are usually based on correlations between data points, but correlations do not always reflect the relationship exactly within the data. In this paper, we propose the LiNGAM Weight Adjust–LSTM (LWA-LSTM) algorithm, which combines a causality discovery algorithm, feature weight adjustment, and a deep neural network for time series data prediction. Several stocks in the Chinese stock market were selected and the algorithm was used to predict the stock price. Comparing the prediction effect of the model with that of the LSTM model alone, the results show that the LWA-LSTM model can find the stable relationship between the data better and has fewer prediction errors.
Subject
Fluid Flow and Transfer Processes,Computer Science Applications,Process Chemistry and Technology,General Engineering,Instrumentation,General Materials Science