arXiv:2411.06391cs.LGcs.AI2024-11NeurIPS被引 5

用因果发现提升新闻驱动的多股预测准确率与可解释性

CausalStock: Deep End-to-end Causal Discovery for News-driven Stock Movement Prediction

  • 基于时序因果发现构建股票间影响关系图
  • 在六大数据集上超越现有基线模型,显著提升预测性能
  • 结合大模型去噪新闻编码,适合金融量化研究者使用

现有新闻驱动的多股价格走势预测方法存在两大问题:一是股票间关系常为单向(如供应商-消费者),传统相关性难以捕捉真实影响;二是新闻数据噪声大,有效信息提取困难。为此,本文提出CausalStock框架,通过滞后依赖的时序因果发现机制建模股票间动态因果关系,并采用功能因果模型封装关系以预测股价走势。同时,利用大语言模型的文本评估能力设计去噪新闻编码器,从海量新闻中提取关键信息。实验在美、中、日、英六大数据集上验证,CausalStock在新闻驱动及通用多股预测任务中均显著优于强基线模型。此外,其因果结构提供清晰可解释的预测机制。

原文摘要 · Abstract (English)

There are two issues in news-driven multi-stock movement prediction tasks that are not well solved in the existing works. On the one hand, "relation discovery" is a pivotal part when leveraging the price information of other stocks to achieve accurate stock movement prediction. Given that stock relations are often unidirectional, such as the "supplier-consumer" relationship, causal relations are more appropriate to capture the impact between stocks. On the other hand, there is substantial noise existing in the news data leading to extracting effective information with difficulty. With these two issues in mind, we propose a novel framework called CausalStock for news-driven multi-stock movement prediction, which discovers the temporal causal relations between stocks. We design a lag-dependent temporal causal discovery mechanism to model the temporal causal graph distribution. Then a Functional Causal Model is employed to encapsulate the discovered causal relations and predict the stock movements. Additionally, we propose a Denoised News Encoder by taking advantage of the excellent text evaluation ability of large language models (LLMs) to extract useful information from massive news data. The experiment results show that CausalStock outperforms the strong baselines for both news-driven multi-stock movement prediction and multi-stock movement prediction tasks on six real-world datasets collected from the US, China, Japan, and UK markets. Moreover, getting benefit from the causal relations, CausalStock could offer a clear prediction mechanism with good explainability.

因果推理股票预测新闻分析可解释性

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