A Comparative study on the individual stock price prediction with the application of neural network models

Wenchao Lu, Wen Ge, Rong Li, Lin Yang*

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

1 Citation (Scopus)

Abstract

According to well-stablished results in the literature, the Long Short Term Memory (LSTM) model is one of learning models most widely used in stock price prediction given its characteristic feature. In this paper, we employ a novel neural network, Gated Recurrent Unit (GRU), in performing individual stock price prediction task in Chinese A-share market. As shown by the experiment results, GRU has comparable performance with LSTM and both them outperform the conventional Recurrent Neural Network (RNN) model. Further, regression analysis indicates that there may exist quadratic relationship between prediction accuracy and training data size. Thereby attempts have been made on adding nonlinear time-weight functions to substantially improve the prediction accuracy with the LSTM model.

Original languageEnglish
Title of host publicationProceedings - 2021 International Conference on Computer Engineering and Artificial Intelligence, ICCEAI 2021
EditorsPan Lin, Yong Yang
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages235-238
Number of pages4
ISBN (Electronic)9781665439602
DOIs
Publication statusPublished - Aug 2021
Event2021 International Conference on Computer Engineering and Artificial Intelligence, ICCEAI 2021 - Shanghai, China
Duration: 27 Aug 202129 Aug 2021

Publication series

NameProceedings - 2021 International Conference on Computer Engineering and Artificial Intelligence, ICCEAI 2021

Conference

Conference2021 International Conference on Computer Engineering and Artificial Intelligence, ICCEAI 2021
Country/TerritoryChina
CityShanghai
Period27/08/2129/08/21

Keywords

  • Component
  • LSTM
  • Machine learning
  • Neural network
  • Stock price prediction

Cite this