An Integrated Machine Learning Framework for Stock Price Prediction

Quanzhi Bi,Hongfei Yan,Chong Chen,Qi Su
DOI: https://doi.org/10.1007/978-3-030-56725-5_8
2020-01-01
Abstract:Predicting the future price of financial assets has always been an important research topic in the field of quantitative finance. This paper attempts to use the latest artificial intelligence technologies to design and implement a framework for financial asset price prediction. The framework we use is divided into three modules: Feature Engineering, Regressor, and Hyper Optimizer. The Feature Engineering module extract multiple features using technical indicators, FinBERT, FFT, ARIMA, stacked auto-encoder, PCA and XGBoost. The Regressor module consists of a generative adversarial network, where the generator is Seq2Seq and the discriminator is GRU. The HyperOptimizer module will tune the parameter in GAN using the Bayesian optimization algorithm. Finally, we conducted numerical experiments on our framework, which shows that the framework implemented in this paper performs better than the benchmark method.
What problem does this paper attempt to address?