Computer Science ›› 2019, Vol. 46 ›› Issue (10): 84-89.doi: 10.11896/jsjkx.180901771

• Big Data & Data Science • Previous Articles     Next Articles

Stock Recommendation System Based on Deep Bidirectional LSTM

ZENG An1, NIE Wen-jun2   

  1. (School of Computer,Guangdong University of Technology,Guangzhou 510006,China)1
    (Guangdong Key Laboratory of Big Data Analysis and Processing,Guangzhou 510006,China)2
  • Received:2018-09-19 Revised:2019-02-21 Online:2019-10-15 Published:2019-10-21

Abstract: With the diversity of applications scenarios and rapid growth of data,the stock prediction models based on classical statistical methods are unable to meet the requirements for high prediction accuracy.But traditional stock reco-mmendation models either never consider the time factor or just consider the unidirectional relationship over time.However,existing stock recommendation models based on deep learning rarely consider the time factor.This paper proposed a deep bidirectional LSTM model for stock prediction,which makes full use of context relationship in the forward direction and backward direction of time series.The problem of vanishing gradient and exploding gradient are solved by introducing LSTM when dealing with long-term sequence.The proposed model can learn information which has long-term dependence on time.At the same time,dropout strategy is introduced to prevent over-fitting caused by deep network model and speed up the training.Experiments on S&P500 dataset show that the neural network prediction model based on the deep bidirectional LSTM outperforms the existing prediction models,the error is about 5% lower,and the coefficient of determination (r2) is increased by 10%.

Key words: Bidirectional long short-term memory, Deep recurrent neural networks, Recommendation system, Stock forecast

CLC Number: 

  • TP181
[1]DEGIANNAKIS S,FILIS G,HASSANI H.Forecasting global stock market implied volatility indices[J].Journal of Empirical Finance,2018,46:111-129.
[2]ZHANG W Y,ZHANG S X,ZHANG S,et al.A multi-factor and high-order stock forecast model based on Type-2 FTS using cuckoo search and self-adaptive harmony search[J].Neurocomputing,2017,240:13-24.
[3]NARENDRA BABU C,REDDY B E.Prediction of selected Indian stock using a partitioning-interpolation based ARIMA-GARCH model[J].Applied Computing and Informatics,2015,11(2):130-143.
[4]ROSAS-ROMERO R,DÍAZ-TORRES A,ETCHEVERRY G. Forecasting of stock return prices with sparse representation of financial time series over redundant dictionaries[J].Expert Systems With Applications,2016,57:37-48.
[5]CHEN Y T,TAO L,LI H B.The Prediction of Dow Jones Cbn China 600 Inde Based on “Most Updated” GM(1,1) Model and Grey Dynamic Neural Network[C]//2010 Third International Conference on Information and Computing.Wuxi:IEEE,2010:56-59.
[6]HUANG K Y,JANE C J.A hybrid model for stock market forecasting and portfolio selection based on ARX,grey system and RS theories[J].Expert Systems With Applications,2008,36(3):5387-5392.
[7]WANG D X,LIU X,WANG M D.A DT-SVM Strategy for Stock Futures Prediction with Big Data[C]//Computational Science and Engineering.Sydney,NSW,Australia:IEEE,2013:1005-1012.
[8]CHEN Y J,HAO Y T.A feature weighted support vector machine and K-nearest neighbor algorithm for stock market indices prediction[J].Expert Systems with Applications,2017,80:340-355.
[9]WANG P G.Research on Stock Price Prediction Based on BP Wavelet Neural Network with Mexico Hat Wavelet Basis[C]//Proceedings of 1st International Conference on Education,Economics and Management Research.Advances in Social Science,Education and Humanities Research (ICEEMR 2017) .Singapore Management University,2017,95:4.
[10]NAYAK S C,MISRA B B,BEHERA H S.Artificial chemical reaction optimization of neural networks for efficient prediction of stock market indices[J].Ain Shams Engineering Journal,2015,8(3):371-390.
[11]CHATZIS S P,SIAKOULIS V,PETROPOULOS A,et al. Forecasting stock market crisis events using deep and statistical machine learning techniques[J].Expert Systems with Applications,2018,112:353-371.
[12]GILES C L,KUHN G M,WILLIAMS R J.Dynamic recurrent neural networks:Theory and applications [J].IEEE Trans.Neural Networks,1994,15:153-156.
[13]SCHUSTER M,PALIWAL K K.Bidirectional Recurrent Neural Networks [J].IEEE Transactions on Signal Processing,1997,45:11.
[14]HOCHREITER S,SCHMIDHUBER J.Long Short-term Me-mory [J].Neural computation,1997,9(8):1735-1780.
[15]GRAVES A,SCHMIDHUBER J.Framewise Phoneme Classification with Bidirectional LSTM and Other Neural Network Architectures [J].Neural Networks,2005,18(5):602-610.
[16]SUN R Q.Research on the price trend forecasting model of us stock index based on LSTM neural network[D].Capital University of Economids and Business,2015.
[17]FISCHER T,KRAUSS C.Deep learning with long?short-term memory networks for financial market predictions[J].European Journal of Operational Research,2018,270(2):654-669.
[18]SRIVASTAVA N,HINTON G,KRIZHEVSKY A.Dropout:A Simple Way to Prevent Neural Networks from Overtting [J].Journal of Machine Learning Research,2014(15):1929-1958.
[19]HOCHREITER S,SCHMIDHUBER J.Long short-term memory[J].Neural computation,1997,9(8).
[20]HIRANSHA M,GOPALAKRISHNAN E A,VIJAY KRISHNA Menon,et al.NSE Stock Market Prediction Using Deep-Learning Models[J].Procedia Computer Science,2018,132:1351-1362.
[21]YETIS Y,KAPLAN H,JAMSHIDI M.Stock market prediction by using artificial neural network[C]//World Automation Congress (WAC) .Waikoloa,HI,USA:IEEE,2014:718-722.
[22]REN J,WANG J H,WANG C M,et al.Stock Index Forecast Based on Regularized LSTM Model[J].Computer Application and Software,2018,35(4):44-48,108.
[1] QIN Qi-qi, ZHANG Yue-qin, WANG Run-ze, ZHANG Ze-hua. Hierarchical Granulation Recommendation Method Based on Knowledge Graph [J]. Computer Science, 2022, 49(8): 64-69.
[2] FANG Yi-qiu, ZHANG Zhen-kun, GE Jun-wei. Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning [J]. Computer Science, 2022, 49(8): 70-77.
[3] SHUAI Jian-bo, WANG Jin-ce, HUANG Fei-hu, PENG Jian. Click-Through Rate Prediction Model Based on Neural Architecture Search [J]. Computer Science, 2022, 49(7): 10-17.
[4] QI Xiu-xiu, WANG Jia-hao, LI Wen-xiong, ZHOU Fan. Fusion Algorithm for Matrix Completion Prediction Based on Probabilistic Meta-learning [J]. Computer Science, 2022, 49(7): 18-24.
[5] CAI Xiao-juan, TAN Wen-an. Improved Collaborative Filtering Algorithm Combining Similarity and Trust [J]. Computer Science, 2022, 49(6A): 238-241.
[6] HE Yi-chen, MAO Yi-jun, XIE Xian-fen, GU Wan-rong. Matrix Transformation and Factorization Based on Graph Partitioning by Vertex Separator for Recommendation [J]. Computer Science, 2022, 49(6A): 272-279.
[7] XIONG Zhong-min, SHU Gui-wen, GUO Huai-yu. Graph Neural Network Recommendation Model Integrating User Preferences [J]. Computer Science, 2022, 49(6): 165-171.
[8] HONG Zhi-li, LAI Jun, CAO Lei, CHEN Xi-liang, XU Zhi-xiong. Study on Intelligent Recommendation Method of Dueling Network Reinforcement Learning Based on Regret Exploration [J]. Computer Science, 2022, 49(6): 149-157.
[9] YU Ai-xin, FENG Xiu-fang, SUN Jing-yu. Social Trust Recommendation Algorithm Combining Item Similarity [J]. Computer Science, 2022, 49(5): 144-151.
[10] CHEN Jin-peng, HU Ha-lei, ZHANG Fan, CAO Yuan, SUN Peng-fei. Convolutional Sequential Recommendation with Temporal Feature and User Preference [J]. Computer Science, 2022, 49(1): 115-120.
[11] ZHAN Wan-jiang, HONG Zhi-lin, FANG Lu-ping, WU Zhe-fu, LYU Yue-hua. Collaborative Filtering Recommendation Algorithm Based on Adversarial Learning [J]. Computer Science, 2021, 48(7): 172-177.
[12] YU Sheng, LI Bin, SUN Xiao-bing, BO Li-li, ZHOU Cheng. Approach for Knowledge-driven Similar Bug Report Recommendation [J]. Computer Science, 2021, 48(5): 91-98.
[13] DING Ling, XIANG Yang. Chinese Event Detection with Hierarchical and Multi-granularity Semantic Fusion [J]. Computer Science, 2021, 48(5): 202-208.
[14] XIAO Shi-tao, SHAO Ying-xia, SONG Wei-ping, CUI Bin. Hybrid Score Function for Collaborative Filtering Recommendation [J]. Computer Science, 2021, 48(3): 113-118.
[15] HAO Zhi-feng, LIAO Xiang-cai, WEN Wen, CAI Rui-chu. Collaborative Filtering Recommendation Algorithm Based on Multi-context Information [J]. Computer Science, 2021, 48(3): 168-173.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!