Cargando…

Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection

With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based...

Descripción completa

Detalles Bibliográficos
Autores principales: Zhou, Qihang, Zhou, Changjun, Wang, Xiao
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Public Library of Science 2022
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8815979/
https://www.ncbi.nlm.nih.gov/pubmed/35120138
http://dx.doi.org/10.1371/journal.pone.0262501
_version_ 1784645349829246976
author Zhou, Qihang
Zhou, Changjun
Wang, Xiao
author_facet Zhou, Qihang
Zhou, Changjun
Wang, Xiao
author_sort Zhou, Qihang
collection PubMed
description With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based on Feature Selection, Convolutional Neural Network and Bidirectional Gated Recurrent Unit (FS-CNN-BGRU). Feature Selection (FS) can select the data with better performance for the results as the input data after data normalization. Convolutional Neural Network (CNN) is responsible for feature extraction. It can extract the local features of the data, pay attention to more local information, and reduce the amount of calculation. The Bidirectional Gated Recurrent Unit (BGRU) can process the data with time series, so that it can have better performance for the data with time series attributes. In the experiment, we used single CNN, LSTM and GRU models and mixed models CNN-LSTM, CNN-GRU and FS-CNN-BGRU (the model used in this manuscript). The results show that the performance of the hybrid model (FS-CNN-BGRU) is better than other single models, which has a certain reference value.
format Online
Article
Text
id pubmed-8815979
institution National Center for Biotechnology Information
language English
publishDate 2022
publisher Public Library of Science
record_format MEDLINE/PubMed
spelling pubmed-88159792022-02-05 Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection Zhou, Qihang Zhou, Changjun Wang, Xiao PLoS One Research Article With the development of recent years, the field of deep learning has made great progress. Compared with the traditional machine learning algorithm, deep learning can better find the rules in the data and achieve better fitting effect. In this paper, we propose a hybrid stock forecasting model based on Feature Selection, Convolutional Neural Network and Bidirectional Gated Recurrent Unit (FS-CNN-BGRU). Feature Selection (FS) can select the data with better performance for the results as the input data after data normalization. Convolutional Neural Network (CNN) is responsible for feature extraction. It can extract the local features of the data, pay attention to more local information, and reduce the amount of calculation. The Bidirectional Gated Recurrent Unit (BGRU) can process the data with time series, so that it can have better performance for the data with time series attributes. In the experiment, we used single CNN, LSTM and GRU models and mixed models CNN-LSTM, CNN-GRU and FS-CNN-BGRU (the model used in this manuscript). The results show that the performance of the hybrid model (FS-CNN-BGRU) is better than other single models, which has a certain reference value. Public Library of Science 2022-02-04 /pmc/articles/PMC8815979/ /pubmed/35120138 http://dx.doi.org/10.1371/journal.pone.0262501 Text en © 2022 Zhou et al https://creativecommons.org/licenses/by/4.0/This is an open access article distributed under the terms of the Creative Commons Attribution License (https://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
spellingShingle Research Article
Zhou, Qihang
Zhou, Changjun
Wang, Xiao
Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title_full Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title_fullStr Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title_full_unstemmed Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title_short Stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
title_sort stock prediction based on bidirectional gated recurrent unit with convolutional neural network and feature selection
topic Research Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8815979/
https://www.ncbi.nlm.nih.gov/pubmed/35120138
http://dx.doi.org/10.1371/journal.pone.0262501
work_keys_str_mv AT zhouqihang stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection
AT zhouchangjun stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection
AT wangxiao stockpredictionbasedonbidirectionalgatedrecurrentunitwithconvolutionalneuralnetworkandfeatureselection