news 2026/8/31 4:12:29

项目分享|TimesFM:谷歌开源的时间序列基础模型

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
项目分享|TimesFM:谷歌开源的时间序列基础模型

引言

时间序列预测在金融、气象、供应链等众多领域都扮演着关键角色,而构建高效、通用的预测模型一直是行业难题。由谷歌研究院(Google Research)开发的TimesFM(Time Series Foundation Model),作为一款预训练的时间序列基础模型,为这一领域带来了新的解决方案。本文将详细介绍这一开源项目的核心信息、创新优势及技术实现,助力开发者快速了解并应用该模型。

TimesFM项目概述

TimesFM是谷歌研究院推出的解码器架构时间序列基础模型,其相关研究成果已发表于2024年ICML会议(论文:《A decoder-only foundation model for time-series forecasting》)。该项目不仅拥有官方Hugging Face模型集合,还被集成到谷歌BigQuery产品中,同时谷歌研究院博客也对其技术细节进行了深度解读(注:本文介绍的开源版本非谷歌官方支持产品)。

目前,TimesFM的最新版本为2.5,而1.0和2.0版本已归档至项目的v1子目录,可通过安装timesfm==1.3.0版本加载使用。2025年9月,TimesFM 2.5正式发布,带来了多项重要更新;同年10月,该版本恢复了通过XReg实现的协变量支持,进一步提升了实用性。

核心创新与优势

相比2.0版本,TimesFM 2.5在性能与功能上实现了显著突破,其核心创新与优势主要体现在以下方面:

  1. 参数优化与效率提升:模型参数从500M缩减至200M,在保持预测能力的同时降低了计算成本,更易于部署和应用。

  2. 更长上下文支持:上下文长度从2048扩展至16k,能够处理更长的时间序列数据,捕捉更久远的历史依赖关系。

  3. 分位数预测能力:通过可选的30M分位数头,支持最长1k horizon的连续分位数预测,为不确定性估计提供了更丰富的信息。

  4. 简化使用流程:移除了对frequency指标的依赖,减少了用户配置负担,同时新增多个预测标志,提升了模型的灵活性。

  5. 生态与API升级:同步升级推理API,计划支持Flax版本(实现更快推理),并恢复协变量支持,完善文档与示例,降低使用门槛。

技术实现与部署指南

环境搭建与安装

TimesFM支持PyTorch和Flax两种后端,安装步骤如下:

  1. 克隆仓库并进入目录:
gitclone https://github.com/google-research/timesfm.gitcdtimesfm
  1. 使用uv创建虚拟环境并安装依赖(根据需求选择后端):
# 创建虚拟环境uv venv# 激活环境source.venv/bin/activate# 安装PyTorch版本uv pipinstall-e .[torch]# 或安装Flax版本uv pipinstall-e .[flax]# 如需协变量支持,安装xreg版本uv pipinstall-e .[xreg]
  1. (可选)根据硬件环境安装对应后端(PyTorch或Jax),以适配CPU、GPU、TPU或Apple Silicon等加速设备。

核心代码示例

使用TimesFM 2.5进行预测的核心流程如下:

importtorchimportnumpyasnpimporttimesfm# 设置精度torch.set_float32_matmul_precision("high")# 加载预训练模型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,# 使用分位数头force_flip_invariance=True,infer_is_positive=True,fix_quantile_crossing=True,))# 执行预测(输入为两个示例时间序列)point_forecast,quantile_forecast=model.forecast(horizon=12,# 预测长度inputs=[np.linspace(0,1,100),np.sin(np.linspace(0,20,67)),])# 输出形状:点预测(2,12),分位数预测(2,12,10)(包含均值及10th至90th分位数)

该项目及相关内容已在AladdinEdu课题广场同步发布,欢迎前往了解更多技术实现与资源。

项目地址:AladdinEdu课题广场

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/25 7:20:27

DC-DC转换器PSpice建模:项目应用全流程解析

PSpice建模实战:手把手教你打造高精度DC-DC仿真系统 你有没有经历过这样的场景? 辛辛苦苦画完电源板,上电测试却发现输出电压“抽风”——启动过冲直接干到两倍额定值;或者负载一跳变,电压就开始振荡,示波…

作者头像 李华
网站建设 2026/8/25 10:56:32

按Token计费的GPU算力平台如何控制成本?

按Token计费的GPU算力平台如何控制成本? 在AI开发日益普及的今天,越来越多开发者选择通过云平台租用GPU资源来训练和推理大模型。尤其是按Token或按时长计费的新型算力服务,看似灵活低成本,实则“暗藏玄机”——稍有不慎&#xff…

作者头像 李华
网站建设 2026/8/24 14:45:07

Altium Designer入门手册:文本标注与图形绘制技巧

Altium Designer实战精要:用文本与图形讲好电路故事在电子设计的世界里,完成电气连接只是第一步。真正让设计“活”起来的,是那些不参与导电、却承载着大量信息的非电气元素——文本标注和图形绘制。它们就像图纸上的语言与符号,把…

作者头像 李华
网站建设 2026/8/27 8:33:23

船舶离网通信方案:Meshtastic 应用指南

Meshtastic 是一款使用小型电子无线电设备发送信号的程序,其频率无需许可证,因此任何人都可以无需认证即可使用。所有这些小型无线电设备相互连接,形成一个互连设备的“网状”网络。这种相互连接的设备网络可以用来进行通信或共享其他类型的数…

作者头像 李华
网站建设 2026/8/27 8:33:24

基于JiyuTrainer微调Baichuan、Qwen等开源LLM

基于JiyuTrainer微调Baichuan、Qwen等开源LLM 在大模型落地日益迫切的今天,一个现实问题摆在开发者面前:如何让像 Baichuan、Qwen 这样的千亿参数级语言模型真正“听懂”业务需求?预训练模型虽强,但面对客服话术优化、医疗问答生成…

作者头像 李华