arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.05693q-fin.ST
收藏

基于DRAGAN和特征匹配的股票市场预测

Stock market forecasting using DRAGAN and feature matching

Fateme Shahabi Nejad, Mohammad Mehdi Ebadzadeh

AI总结:

针对股票价格预测中GANs训练不稳定、模式崩溃及相关性捕捉难题,提出基于DRAGAN和特征匹配的框架,通过窗口化和判别器输入条件限制提升性能,实验优于LSTM、基础GANs及WGAN-GP。

摘要:

近年来,应用机器学习方法预测股票价格一直是备受关注的研究课题之一。尽管该领域基于生成对抗网络(GANs)的研究报道较少,但其结果颇具前景。GANs是在不同领域成功应用的强大生成模型,但存在训练不稳定和模式崩溃等固有挑战。此外,一个主要问题是捕捉股票价格中的相关性。因此,我们的挑战分为两大类:捕捉相关性以及GANs的固有问题。本文提出了一种基于DRAGAN和特征匹配的新型股票价格预测框架,该框架可提高训练稳定性并缓解模式崩溃。我们采用窗口化方法,使生成器获取时间相关性;同时,通过对判别器输入进行条件限制,捕捉时间相关性以及价格与特征之间的相关性。对多只股票数据的实验结果表明,我们提出的方法优于作为基准方法的长短期记忆(LSTM),以及作为GANs不同变体的基础GANs和WGAN-GP。

英文摘要:

Applying machine learning methods to forecast stock prices has been one of the research topics of interest in recent years. Almost few studies have been reported based on generative adversarial networks (GANs) in this area, but their results are promising. GANs are powerful generative models successfully applied in different areas but suffer from inherent challenges such as training instability and mode collapse. Also, a primary concern is capturing correlations in stock prices. Therefore, our challenges fall into two main categories: capturing correlations and inherent problems of GANs. In this paper, we have introduced a novel framework based on DRAGAN and feature matching for stock price forecasting, which improves training stability and alleviates mode collapse. We have employed windowing to acquire temporal correlations by the generator. Also, we have exploited conditioning on discriminator inputs to capture temporal correlations and correlations between prices and features. Experimental results on data from several stocks indicate that our proposed method outperformed long short-term memory (LSTM) as a baseline method, also basic GANs and WGAN-GP as two different variants of GANs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑