GAN-Enhanced Nonlinear Fusion Model for Stock Price Prediction

Yingcheng Xu,Yunfeng Zhang,Peide Liu,Qiuyue Zhang,Yuqi Zuo
DOI: https://doi.org/10.1007/s44196-023-00394-4
IF: 2.259
2024-01-20
International Journal of Computational Intelligence Systems
Abstract:Stock price prediction is a significant field of finance research for both academics and practitioners. Numerous studies have proved that the stock movement can be fully reflect various internal features of stock price including non-stationary behavior, high persistence in the conditional variance. The fusion of time-series prediction model such as Auto-Regressive Integrated Moving Average (ARIMA) and neural network is an availability but difficult approach for stock price prediction. Although the orientation has been studied through some methods in different research, there are still difficulties with the poor capture ability of time-series features and insufficient effectiveness of integrating temporal feature and frequency domain information. In this paper, we propose a Generative Adversarial Network (GAN) framework with the Convolution Neural Networks (CNN) as the discriminator and a hybrid model as the generator for forecasting the stock price. The hybrid model includes Attention-based Convolution Neural Networks (ACNN), Long Short-Term Memory (LSTM), and ARIMA model. Moreover, this proposed framework uses the Generative Adversarial patten and Attention Mechanism to achieve effective analysis and feature extraction for stock price movement. The extensive experiments in different history periods of dataset demonstrate an improvement in forecasting of stock price using our model as compared to the baseline models.
computer science, artificial intelligence, interdisciplinary applications
What problem does this paper attempt to address?
The paper is primarily dedicated to addressing the problem of stock price prediction, particularly in handling the nonlinear relationships, non-stationary behavior, and long-term dependencies in time series data. The authors propose a new framework based on Generative Adversarial Networks (GAN), called Time Series Generative Adversarial Network (TSGAN), aimed at improving the accuracy of stock price predictions. The specific issues addressed in the paper include: 1. **Difficulty of existing models in capturing time series features**: Traditional statistical methods such as the Autoregressive Integrated Moving Average (ARIMA) model are somewhat effective in handling noise but have limitations in capturing nonlinear characteristics. 2. **Insufficient utilization of frequency domain information**: Existing research has not sufficiently integrated time features with frequency domain information. 3. **Market characteristics of different historical periods**: Stock price data reflect different market characteristics and behaviors at different times, requiring prediction models to adapt to these changes. 4. **Complex nonlinear relationships**: The nonlinear relationships between political policies, economic changes, social events, and investor behavior lead to delays in the prediction task. To address the above challenges, the paper presents the following contributions: - Proposes a new TSGAN model that combines the ARIMA model, Convolutional Neural Network (CNN)-Long Short-Term Memory (LSTM) hybrid model, and CNN as the generator and discriminator, effectively enhancing the feature representation capability of stock price data from different historical periods. - Introduces the combination of LSTM and CNN into the GAN framework as the generator for the first time, to address the long-term dependency issues of LSTM in adversarial training. - Experimental results show that the proposed nonlinear fusion model significantly outperforms single models and nonlinear ensemble methods, demonstrating stronger robustness and practicality in stock price prediction. In summary, this paper aims to improve the accuracy of stock price prediction by introducing a novel Time Series Generative Adversarial Network framework, particularly addressing the issues that existing methods struggle to handle.