A hybrid model combining variational mode decomposition and an attention-GRU network for stock price index forecasting

被引:21
作者
Niu, Hongli [1 ]
Xu, Kunliang [1 ]
机构
[1] Univ Sci & Technol Beijing, Sch Econ & Management, Beijing 100083, Peoples R China
关键词
variational mode decomposition; Gated Recurrent Units; attention mechanism; forecasting; stock price; SHORT-TERM LOAD; NEURAL-NETWORK;
D O I
10.3934/mbe.2020367
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
In this paper we introduce a new hybrid model based on variational mode decomposition (VMD) and Gated Recurrent Units (GRU) network improved by attention mechanism to enhance the accuracy of stock price indices forecasting. In the process of establishing the model, VMD is made a use to decompose the primary series into some almost orthogonal subsequences. The attention mechanism is introduced into GRU to assign different weights to the input elements in advance so that better predictive results can be achieved for each component. In empirical experiment, London FTSE Index (FTSE) and Nasdaq Index (IXIC) are adopted to examine the performance of VMD-AttGRU model. Empirical results report that the developed hybrid model outperforms the single models and indeed raises the accuracy of stock price indices forecasting. In addition, the introduction of attention mechanism can increase the level predictive accuracy but decrease the correctness of direction forecasting.
引用
收藏
页码:7151 / 7166
页数:16
相关论文
共 30 条
[1]  
[Anonymous], ARXIV150804025, DOI DOI 10.18653/V1/2021.FNDINGSEMNLP.101
[2]   Distributed optimization and statistical learning via the alternating direction method of multipliers [J].
Boyd S. ;
Parikh N. ;
Chu E. ;
Peleato B. ;
Eckstein J. .
Foundations and Trends in Machine Learning, 2010, 3 (01) :1-122
[3]   Financial time series forecasting model based on CEEMDAN and LSTM [J].
Cao, Jian ;
Li, Zhi ;
Li, Jian .
PHYSICA A-STATISTICAL MECHANICS AND ITS APPLICATIONS, 2019, 519 :127-139
[4]   Exploring the attention mechanism in LSTM-based Hong Kong stock price movement prediction [J].
Chen, Shun ;
Ge, Lei .
QUANTITATIVE FINANCE, 2019, 19 (09) :1507-1515
[5]   Recurrent neural network for combined economic and emission dispatch [J].
Deng, Ting ;
He, Xing ;
Zeng, Zhigang .
APPLIED INTELLIGENCE, 2018, 48 (08) :2180-2198
[6]   Neural Mechanisms of Selective Visual Attention [J].
Moore, Tirin ;
Zirnsak, Marc .
ANNUAL REVIEW OF PSYCHOLOGY, VOL 68, 2017, 68 :47-72
[7]   Variational Mode Decomposition [J].
Dragomiretskiy, Konstantin ;
Zosso, Dominique .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2014, 62 (03) :531-544
[8]  
Graves A, 2012, STUD COMPUT INTELL, V385, P1, DOI [10.1007/978-3-642-24797-2, 10.1162/neco.1997.9.1.1]
[9]  
JIN HH, 2019, ENERGY, V189
[10]   Transductive LSTM for time-series prediction: An application to weather forecasting [J].
Karevan, Zahra ;
Suykens, Johan A. K. .
NEURAL NETWORKS, 2020, 125 :1-9