简介

TimesFM是Google Research推出的预训练时间序列基础模型,专门用于时间序列预测。简单说,你给它一段历史数据,它能告诉你未来会怎样——销售预测、流量预估、股价走势,都行。

最新2.5版本把参数量从500M砍到200M,但上下文长度反而从2048飙升到16K,预测范围最高支持1000步。更骚的是,这东西已经整合进Google BigQuery,直接当云服务用。

核心功能

超长上下文理解

16K的上下文窗口意味着你能喂给它更长的历史数据。以前2048个时间点就到头了,现在能吃下16倍的数据量。对于周期性强的数据(比如有年周期规律的销售数据),预测准确度会更高。

连续分位数预测

不只是给个点预测,TimesFM能输出从10%到90%的完整分位数分布。相当于告诉你:最可能的结果是多少,最好情况能有多好,最坏情况会多糟。做风险评估的人应该会喜欢这个功能。

轻量化设计

200M参数的模型比上一代500M小了一半多,效果没掉。显存占用更低,推理更快,RTX 3080就能跑起来。

BigQuery原生集成

不想折腾本地部署的话,可以直接在Google BigQuery里调用TimesFM模型,SQL语句就能做预测。云端算力帮你跑,省事。

价格方案

方案价格适合场景
开源版本免费本地部署,完全自主控制
BigQuery版本按查询量计费企业级应用,无需运维

开源版成本:

  • GPU需求:RTX 3080以上即可(10GB显存够用)
  • 云端租用:约$0.5-1/小时(按需GPU实例)
  • 自己有硬件的话,完全免费

BigQuery版定价:

  • 按处理数据量计费,约$5/TB
  • 预测API调用另计费
  • 不想折腾基础设施的团队可以考虑

👉 TimesFM GitHub仓库(开源免费) 👉 BigQuery TimesFM文档(云端付费)

优缺点对比

优点缺点
开源免费,商用友好对小数据集可能过拟合
16K上下文长度,行业领先需要一定技术门槛部署
分位数预测,风险量化实用文档相对简单,踩坑要自己摸索
参数轻量,消费级硬件可跑BigQuery集成仅限GCP用户
Google出品,持续更新不支持多变量时间序列(2.5版本)

适合人群

适合:

  • 需要做销售/流量/库存预测的电商运营
  • 金融量化分析师,需要快速验证预测模型
  • 数据科学家,想省去模型训练时间
  • 物联网场景,预测设备状态或能耗

不适合:

  • 完全没有编程基础的小白
  • 数据量特别小的场景(少于100个时间点)
  • 需要多变量联合预测的复杂场景

使用教程

环境准备

# 克隆仓库
git clone https://github.com/google-research/timesfm.git
cd timesfm

# 创建虚拟环境
uv venv
source .venv/bin/activate

# 安装依赖(PyTorch版本)
uv pip install -e .[torch]

加载模型

import torch
import timesfm

# 加载预训练模型
model = timesfm.TimesFM_2p5_200M_torch.from_pretrained(
    "google/timesfm-2.5-200m-pytorch"
)

# 配置预测参数
model.compile(
    timesfm.ForecastConfig(
        max_context=1024,      # 最大上下文长度
        max_horizon=256,       # 最大预测步数
        normalize_inputs=True,
        use_continuous_quantile_head=True,
    )
)

执行预测

import numpy as np

# 准备输入数据(两个示例序列)
inputs = [
    np.linspace(0, 1, 100),      # 线性趋势
    np.sin(np.linspace(0, 20, 67)),  # 正弦波
]

# 预测未来12步
point_forecast, quantile_forecast = model.forecast(
    horizon=12,
    inputs=inputs,
)

print(f"点预测: {point_forecast.shape}")      # (2, 12)
print(f"分位数预测: {quantile_forecast.shape}")  # (2, 12, 10)

不需要训练,直接推理。

使用建议

TimesFM开源免费,没有"购买"一说,但可以给几个参考:

个人开发者或小团队:直接用开源版本,租个按需GPU实例跑,成本可控。RTX 3080/4080级别的显卡完全够用。

企业用户:评估一下自己的数据量和预测频率。偶尔用几次的话,BigQuery按量付费更划算;高频使用的话,自建服务成本更低。

与竞品对比:

  • vs Prophet:TimesFM更准,但Prophet更简单
  • vs ARIMA:TimesFM不需要手动调参,开箱即用
  • vs 深度学习自建:省去大量训练时间和算力成本

时间序列预测这事儿,以前需要专业建模师搞几周,现在TimesFM几分钟就出结果。Google用海量时间序列数据预训练出来的模型,泛化能力确实强。

对于有预测需求的运营、分析师、开发者来说,这是一个值得投入时间学习的工具。开源免费,BigQuery云端可选,灵活性够用。


相关链接: