Vol. 3 No. 4 (2024)
Articles

Feature Extraction and Model Optimization of Deep Learning in Stock Market Prediction

Published 2024-07-30

Keywords

  • Deep learning; GRU; Attention; Feature extraction

How to Cite

Wei, Y., Gu, X., Feng, Z., Li, Z., & Sun, M. (2024). Feature Extraction and Model Optimization of Deep Learning in Stock Market Prediction. Journal of Computer Technology and Software, 3(4). https://doi.org/10.5281/zenodo.13622489

Abstract

This paper delves into leveraging neural networks for equity market forecasting by amalgamating gated recurrent units (GRUs) with an attention paradigm to refine the predictive model, thereby enhancing the precision of share value and market trajectory prognostications. Conventional forecasting frameworks frequently falter in encapsulating the dependencies of prolonged sequences, particularly when contending with nonlinear temporal data, culminating in diminished forecast veracity. Consequently, the architecture devised herein initially harnesses a GRU layer to preprocess the ingested temporal sequence information, discerning the dynamic alterations and latent patterns within the series. Subsequently, the attention mechanism is superimposed on the GRU's latent state output. By computing the significance rating at every temporal juncture of the hidden state, the salience of diverse epochs is dynamically recalibrated, ensuring the model focuses on the attributes most pivotal to the anticipated outcome. This fusion not only amplifies the model's acumen for enduring interdependencies but also alleviates superfluous computational overhead, accelerates the learning phase, and fortifies versatility, all while sustaining commendable predictive efficacy.

References

  1. Q. Zhan, Y. Ma, E. Gao, D. Sun, and H. Yang, "Innovations in Time Related Expression Recognition Using LSTM Networks," International Journal of Innovative Research in Computer Science & Technology, vol. 12, no. 3, pp. 120-125, 2024.
  2. Y. Yan, S. He, Z. Yu, J. Yuan, Z. Liu, and Y. Chen, "Investigation of Customized Medical Decision Algorithms Utilizing Graph Neural Networks," arXiv preprint arXiv:2405.17460, 2024.
  3. Z. Gao, Q. Wang, T. Mei, X. Cheng, Y. Zi, and H. Yang, "An Enhanced Encoder-Decoder Network Architecture for Reducing Information Loss in Image Semantic Segmentation," arXiv preprint arXiv:2406.01605, 2024.
  4. J. Wang, H. Zhang, Y. Zhong, Y. Liang, R. Ji, and Y. Cang, "Advanced Multimodal Deep Learning Architecture for Image-Text Matching," in Proc. 2024 IEEE 4th International Conference on Electronic Technology, Communication and Information (ICETCI), pp. 1185-1191, May 2024.
  5. Y. Yang, Z. Chen, Y. Yan, M. Li, and T. Gegen, "A new method of image recognition based on deep learning generates adversarial networks and integrates traditional algorithms," Journal of Computing and Electronic Information Management, vol. 13, no. 1, pp. 57-61, 2024.
  6. Y. Cheng, J. Guo, S. Long, Y. Wu, M. Sun, and R. Zhang, "Advanced Financial Fraud Detection Using GNN-CL Model," arXiv preprint arXiv:2407.06529, 2024.
  7. H. Yang, Y. Zi, H. Qin, H. Zheng, and Y. Hu, "Advancing Emotional Analysis with Large Language Models," Journal of Computer Science and Software Applications, vol. 4, no. 3, pp. 8-15, 2024.
  8. D. Sun, Y. Liang, Y. Yang, Y. Ma, Q. Zhan, and E. Gao, "Research on Optimization of Natural Language Processing Model Based on Multimodal Deep Learning," arXiv preprint arXiv:2406.08838, 2024.
  9. H. Liu, I. Li, Y. Liang, D. Sun, Y. Yang, and H. Yang, "Research on Deep Learning Model of Feature Extraction Based on Convolutional Neural Network," arXiv preprint arXiv:2406.08837, 2024.
  10. Lien Minh D, Sadeghi-niaraki A, Huy H D, et al. Deep learning approach for shortterm stock trends prediction based on two-stream gated recurrent unit network[J]. IEEE Access, IEEE, 2018, 6: 55392–55404.
  11. Friday, Ibanga Kpereobong, et al. "CAGTRADE: Predicting Stock Market Price Movement with a CNN-Attention-GRU Model." Asia-Pacific Financial Markets (2024): 1-26.
  12. Dip Das, Joy, et al. "Encoder–Decoder Based LSTM and GRU Architectures for Stocks and Cryptocurrency Prediction." Journal of Risk and Financial Management 17.5 (2024): 200.
  13. Khan, Shahzeb, and Vipin Kumar. "A novel hybrid GRU-CNN and residual bias (RB) based RB-GRU-CNN models for prediction of PTB Diagnostic ECG time series data." Biomedical Signal Processing and Control 94 (2024): 106262.
  14. Tsantekidis A, Passalis N, Tefas A,et al. Forecasting stock prices from the limit order book using convolutional neural networks[J]. Proceedings - 2017 IEEE 19th Conference on Business Informatics, 2017, 1: 7–12.
  15. Li, Y., Yan, X., Xiao, M., Wang, W., & Zhang, F. (2023, December). Investigation of creating accessibility linked data based on publicly available accessibility datasets. In Proceedings of the 2023 13th International Conference on Communication and Network Security (pp. 77-81).
  16. Anghel, A., Papandreou, N., Parnell, T., De Palma, A., & Pozidis, H. (2018). Benchmarking and optimization of gradient boosting decision tree algorithms. arXiv preprint arXiv:1809.04559.
  17. Dey, R., & Salem, F. M. (2017, August). Gate-variants of gated recurrent unit (GRU) neural networks. In 2017 IEEE 60th international midwest symposium on circuits and systems (MWSCAS) (pp. 1597-1600). IEEE.