Stock Price Prediction with Heavy-Tailed Distribution Time-Series Generation Based on WGAN-BiLSTM

Ming Kang
DOI: https://doi.org/10.1007/s10614-024-10639-9
IF: 1.741
2024-06-14
Computational Economics
Abstract:Accurate stock price prediction is essential for investing in the financial market. Aiming at the problem's accuracy, the prediction model is restricted due to the lack of sufficient data samples for the stock price of newly listed companies. This paper proposes a novel approach to improve the generalization of Bidirectional Long Short-Term Memory Network (BiLSTM) with Wasserstein Generative Adversarial Network (WGAN) on stock time-series data augmentation. WGAN is used to learn the distribution rules of the real data, and the two-sample KS test and the area between the log–log plots are adopted to evaluate the extent of heavy-tailed distribution of the generated samples similar to that of real-world data. BiLSTM is employed to extract and predict data's forward and reverse time information. With the stock price data, the prediction results of different algorithm models without and after augmented data are compared. Experimental results show that the accuracy of BiLSTM enhanced by WGAN is significantly improved, and the proposed WGAN-BiLSTM model is suitable for stock price prediction in few-shot settings.
economics,mathematics, interdisciplinary applications,management
What problem does this paper attempt to address?