← 最新论文
💰 quantitative finance

From Index to Equity: Pre-Training Transformers for Stock Return Prediction

本文表明,尽管LSTM和XGBoost基准模型在实际应用中实现了更高的平均日收益率,但在多伦多证券交易所指数上预训练Transformer模型仍能显著优于这些基准,提升其对个股回报方向和数值预测的能力。

原作者: Marie Soehl Coolsaet, Roberto Gallardo, Zhen Gao

发布于 2026-05-26
📖 1 分钟阅读☕ 轻松阅读

原作者: Marie Soehl Coolsaet, Roberto Gallardo, Zhen Gao

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下,你正在尝试预测某个特定、微小街区的天气。那里极其混乱:一栋房子可能阳光明媚,而隔壁却在遭遇冰雹。现在,想象一下先尝试预测整个国家的天气。宏观图景要平滑得多,也更容易理解,因为各个街区的剧烈波动被平均化了。

本文讲述的是利用一个超级聪明的计算机大脑(称为Transformer)来预测股票价格,作者发现,在要求它预测“街区级”天气之前,最好先让它学习“国家级”天气。

以下是他们研究历程的分解:

1. 问题:股市是混乱的

预测单只股票明天是涨是跌,就像试图猜测飓风中被吹起的一片树叶的确切路径。市场杂乱无章,瞬息万变,且充满噪音。传统的计算机模型(如LSTMXGBoost)曾试图解决这一问题,但它们往往举步维艰,因为它们在每次面对新股票时都要从头开始学习。

2. 解决方案:“学徒”策略(预训练)

作者决定尝试一种名为迁移学习的新教学方法。这就像培训一位主厨:

  • 步骤 1(预训练): 不是立即教主厨烹饪某道特定的怪异菜肴,而是先让他们在一锅巨大且平滑的汤(TSX 市场指数)上练习。这锅汤代表了整个股票市场。它比单只股票要少一些混乱。主厨在没有被某种怪异食材分心的情况下,学会了烹饪的通用规则(风味如何相互作用、热量如何运作)。
  • 步骤 2(微调): 一旦主厨理解了这锅大汤的基本原理,你就交给他一份特定的、棘手的食谱(单只股票)。因为他们已经掌握了基础,所以学习特定食谱的速度和质量都远胜于从零开始。

3. 工具:Transformer 与老牌劲旅

该团队使用了一种名为Transformer的现代人工智能架构(与驱动 ChatGPT 等工具的技术相同)。他们将其与更古老、可靠的工具进行了比较:

  • LSTM: 一种擅长记忆序列的旧式人工智能。
  • XGBoost: 一种强大的统计工具,通过构建多棵决策树来寻找模式。
  • 集成模型(Ensemble): 一个“委员会”,它平均了上述所有模型的预测结果。

4. 他们的发现(结果)

实验有两个主要目标:猜测方向(涨或跌)和猜测确切数值(涨多少)。

  • “方向”游戏(分类):

    • 当 Transformer 试图从头学习单只股票时,它感到困惑且表现不佳。
    • 当他们使用“学徒策略”(先在大型市场指数上进行预训练)时,Transformer 在猜测股票涨跌方面有了显著提升。
    • 然而,老牌XGBoost模型在单纯猜对“涨或跌”方面仍然是最好的。
  • “确切数值”游戏(回归):

    • 在尝试预测确切的回报率百分比时,微调后的 Transformer微调后的 LSTM击败了 XGBoost 模型。它们在计算具体数值方面更加准确。
    • 但这里有个转折: 尽管 Transformer 在计算数值方面更准确,但在模拟交易测试中,集成模型(委员会)实际上赚到了最多的钱。
    • 为什么? Transformer 非常谨慎,预测的数值非常接近零(保守的赌注),而 XGBoost/集成模型则愿意押注波动更大的股票。在股市中,有时过于“安全”意味着你会错失大赚的机会。

5. 现实世界的工具

研究人员并没有止步于理论。他们构建了一个实时交易应用程序(使用名为 Streamlit 的工具)。

  • 你可以输入股票代码(如"TD"或"RY")。
  • 该应用程序获取最新数据,将其输入他们的模型,并告诉你哪 5 只股票看起来涨幅最大,哪 5 只看起来会崩盘。
  • 这就像拥有一个为你个人投资组合提供每日天气预报的天气应用程序。

核心结论

论文总结道,虽然没有模型是完美的(股市仍然极难预测),但先让 AI 学习宏观市场图景有助于它更好地理解微观图景

Transformer 模型是理解股票走势细微差别的有力竞争者,但在本次特定测试中,老牌方法(如 XGBoost)和“委员会”方法在创造最大利润方面仍居首位。作者计划在未来让 Transformer 变得更大、更聪明,也许是通过教它阅读全球新闻和经济报告,而不仅仅是过去的价格图表。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →