Forecasting stock prices with a feature fusion LSTM-CNN model using different representations of the same data

被引:176
作者
Kim, Taewook [1 ]
Kim, Ha Young [1 ,2 ]
机构
[1] Ajou Univ, Dept Financial Engn, Suwon, South Korea
[2] Ajou Univ, Dept Data Sci, Suwon, South Korea
来源
PLOS ONE | 2019年 / 14卷 / 02期
关键词
TECHNICAL ANALYSIS; NEURAL-NETWORKS; RETURNS; PREDICTION; VOLATILITY; SYSTEM; VOLUME;
D O I
10.1371/journal.pone.0212320
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Forecasting stock prices plays an important role in setting a trading strategy or determining the appropriate timing for buying or selling a stock. We propose a model, called the feature fusion long short-term memory-convolutional neural network (LSTM-CNN) model, that combines features learned from different representations of the same data, namely, stock time series and stock chart images, to predict stock prices. The proposed model is composed of LSTM and a CNN, which are utilized for extracting temporal features and image features. We measure the performance of the proposed model relative to those of single models (CNN and LSTM) using SPDR S&P 500 ETF data. Our feature fusion LSTM-CNN model outperforms the single models in predicting stock prices. In addition, we discover that a candlestick chart is the most appropriate stock chart image to use to forecast stock prices. Thus, this study shows that prediction error can be efficiently reduced by using a combination of temporal and image features from the same data rather than using these features separately.
引用
收藏
页数:23
相关论文
共 50 条
[1]   Stock Price Prediction Using the ARIMA Model [J].
Adebiyi, Ayodele A. ;
Adewumi, Aderemi O. ;
Ayo, Charles K. .
2014 UKSIM-AMSS 16TH INTERNATIONAL CONFERENCE ON COMPUTER MODELLING AND SIMULATION (UKSIM), 2014, :106-112
[2]  
[Anonymous], 2017, DEEP STOCK REPRESENT
[3]  
[Anonymous], P 3 INT C LEARNING R
[4]  
[Anonymous], PROC CVPR IEEE
[5]  
[Anonymous], INCEPTION V4 INCEPTI
[6]  
[Anonymous], WIDER DEEPER REVISIT
[7]  
[Anonymous], 2017, COMMUN ACM, DOI DOI 10.1145/3065386
[8]  
[Anonymous], J FINANC QUANT ANAL
[9]  
[Anonymous], P IEEE INT C COMP VI
[10]  
[Anonymous], P 2006 IEEE C EV COM