arXiv:2409.08282q-fin.STcs.CE2024-09被引 26

融合长短期股票关系与改进GRU,提升股市趋势预测准确率

LSR-IGRU: Stock Trend Prediction Based on Long Short-Term Relationships and Improved GRU

  • 构建长短期股票关系矩阵,引入行业与隔夜价格信息
  • 改进GRU输入机制,显著提升趋势预测准确率
  • 在中美多数据集验证有效,适合量化交易系统应用

股票价格预测是金融领域的重要挑战。近年来,随着深度学习和图神经网络的发展,研究开始关注股票间的相互关系。然而,现有方法多聚焦于短期动态关系,并直接融合关系与时间信息,忽视了市场中股票复杂的非线性动态特征及潜在的高阶交互关系。为此,本文提出一种基于长短期股票关系与改进GRU输入的股价趋势预测模型LSR-IGRU。首先,构建股票间的长短期关系矩阵:首次引入二级行业信息捕捉长期关系,利用隔夜价格信息建立短期关系;其次,改进GRU每一步的输入,使模型更有效地融合时间与长短期关系信息,显著提升趋势预测精度;最后,在中国与美国多个股票市场数据集上进行广泛实验,验证了该模型优于当前主流基线方法。此外,将该模型应用于某金融公司的算法交易系统,实现了显著更高的累计投资组合收益。代码已开源:https://github.com/ZP1481616577/Baselines_LSR-IGRU。

原文摘要 · Abstract (English)

Stock price prediction is a challenging problem in the field of finance and receives widespread attention. In recent years, with the rapid development of technologies such as deep learning and graph neural networks, more research methods have begun to focus on exploring the interrelationships between stocks. However, existing methods mostly focus on the short-term dynamic relationships of stocks and directly integrating relationship information with temporal information. They often overlook the complex nonlinear dynamic characteristics and potential higher-order interaction relationships among stocks in the stock market. Therefore, we propose a stock price trend prediction model named LSR-IGRU in this paper, which is based on long short-term stock relationships and an improved GRU input. Firstly, we construct a long short-term relationship matrix between stocks, where secondary industry information is employed for the first time to capture long-term relationships of stocks, and overnight price information is utilized to establish short-term relationships. Next, we improve the inputs of the GRU model at each step, enabling the model to more effectively integrate temporal information and long short-term relationship information, thereby significantly improving the accuracy of predicting stock trend changes. Finally, through extensive experiments on multiple datasets from stock markets in China and the United States, we validate the superiority of the proposed LSR-IGRU model over the current state-of-the-art baseline models. We also apply the proposed model to the algorithmic trading system of a financial company, achieving significantly higher cumulative portfolio returns compared to other baseline methods. Our sources are released at https://github.com/ZP1481616577/Baselines_LSR-IGRU.

股票预测GRU关系建模量化交易

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。