Multi-factor Based Stock Price Prediction Using Hybrid Neural Networks with Attention Mechanism

被引:17
作者
Li, Chen [1 ]
Zhang, Xu [1 ]
Qaosar, Mahboob [1 ]
Ahmed, Saleh [1 ]
Alam, Kazi Md Rokibul [2 ]
Morimoto, Yasuhiko [1 ]
机构
[1] Hiroshima Univ, Grad Sch Engn, Hiroshima, Japan
[2] Khulna Univ Engn & Technol, Dept Comp Sci & Engn, Khulna, Bangladesh
来源
IEEE 17TH INT CONF ON DEPENDABLE, AUTONOM AND SECURE COMP / IEEE 17TH INT CONF ON PERVAS INTELLIGENCE AND COMP / IEEE 5TH INT CONF ON CLOUD AND BIG DATA COMP / IEEE 4TH CYBER SCIENCE AND TECHNOLOGY CONGRESS (DASC/PICOM/CBDCOM/CYBERSCITECH) | 2019年
关键词
stock prediction; multi-factor; CNN; LSTM; attention mechanism;
D O I
10.1109/DASC/PiCom/CBDCom/CyberSciTech.2019.00176
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The prediction of time series data, such as stock prices, is difficult since there exist many factors that affect the prediction model. Also, the influence of different factors on a stock price may be linear or nonlinear. The generation of good models for stock prices challenge the researchers in recent years. Long Short-Term Memory (LSTM) is a variation of Recurrent Neural Network (RNN), which can capture temporal sequence and have gained great success on time series prediction. Also, Convolutional Neural Network (CNN) is superior for extracting features from multi-dimensional sequences. In this paper, we propose a CNN-LSTM hybrid neural network with multiple factors to predict stock prices. Moreover, we add an attention mechanism to improve the scalability and the accuracy of the CNN-LSTM model. In the experiments, we compare our proposed model with different approaches in two real stock datasets. The results confirm the efficiency and scalability of our proposed method.
引用
收藏
页码:961 / 966
页数:6
相关论文
共 26 条
  • [1] [Anonymous], 2017, DEEP STOCK REPRESENT
  • [2] Predicting Deep Zero-Shot Convolutional Neural Networks using Textual Descriptions
    Ba, Jimmy Lei
    Swersky, Kevin
    Fidler, Sanja
    Salakhutdinov, Ruslan
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 4247 - 4255
  • [3] Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
  • [4] A deep learning framework for financial time series using stacked autoencoders and long-short term memory
    Bao, Wei
    Yue, Jun
    Rao, Yulei
    [J]. PLOS ONE, 2017, 12 (07):
  • [5] LEARNING LONG-TERM DEPENDENCIES WITH GRADIENT DESCENT IS DIFFICULT
    BENGIO, Y
    SIMARD, P
    FRASCONI, P
    [J]. IEEE TRANSACTIONS ON NEURAL NETWORKS, 1994, 5 (02): : 157 - 166
  • [6] Box G.E.P., 1970, Time Series Analysis, Forecasting and Control, V65, P1509, DOI 10.1080/01621459.1970.10481180
  • [7] Attentive Collaborative Filtering: Multimedia Recommendation with Item- and Component-Level Attention
    Chen, Jingyuan
    Zhang, Hanwang
    He, Xiangnan
    Nie, Liqiang
    Liu, Wei
    Chua, Tat-Seng
    [J]. SIGIR'17: PROCEEDINGS OF THE 40TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL, 2017, : 335 - 344
  • [8] Chen JF, 2016, INT C CLOUD COMP BIG, P87, DOI [10.1109/CCBD.2016.51, 10.1109/CCBD.2016.027]
  • [9] Chen K, 2015, PROCEEDINGS 2015 IEEE INTERNATIONAL CONFERENCE ON BIG DATA, P2823, DOI 10.1109/BigData.2015.7364089
  • [10] Cho K., 2014, P SSST8 8 WORKSH SYN, P103, DOI 10.3115/v1/w14-4012