OpenLTM

GitHub
539 54 中等 1 次阅读 今天MIT开发框架
AI 解读 由 AI 自动生成,仅供参考

OpenLTM 是一个专注于大型时间序列模型(LTM)的开源代码库,旨在为开发者提供从预训练到评估的完整流水线。它主要解决了时间序列领域缺乏统一开发框架和高质量预训练模型的痛点,让研究人员能够更轻松地构建、复现并应用如 Timer、Moirai、GPT4TS 等前沿模型,从而高效处理零样本预测、特征提取等多样化任务。

该项目特别适合人工智能研究人员、数据科学家以及希望深入探索时间序列大模型的开发者使用。无论是想要微调现有模型,还是从头开始预训练新架构,OpenLTM 都提供了丰富的脚本和数据集支持。其核心亮点在于集成了多种基于 Transformer 的可扩展骨干网络,并发布了包括 260B 参数在内的多个大规模预训练权重,显著降低了进入该领域的门槛。此外,项目还持续更新社区贡献的实现方案,促进了学术成果的快速落地与共享,是探索时间序列基础模型不可或缺的工具平台。

使用场景

某新能源发电集团的数据科学团队正致力于构建覆盖全国数千个风电场的超短期功率预测系统,以应对电网调度对精度的严苛要求。

没有 OpenLTM 时

  • 模型重复造轮子:团队需从零搭建 Transformer 架构并编写预训练代码,耗时数月才能复现如 Timer 或 Moirai 等前沿算法,研发周期严重滞后。
  • 冷启动数据匮乏:面对新并网的风电场,因缺乏历史数据无法训练专用模型,传统方法在零样本(Zero-shot)场景下预测误差极高。
  • 基线对比困难:业界大模型众多(如 GPT4TS、Time-LLM),但代码分散且标准不一,难以在同一框架下公平评估哪种模型最适合当前业务。
  • 微调门槛高昂:现有开源模型多依赖特定环境或缺乏友好的微调脚本,工程师需花费大量精力处理数据对齐和权重加载问题。

使用 OpenLTM 后

  • 开箱即用加速研发:直接调用 OpenLTM 集成的 Timer-XL 或 Moment 等预训练模型及标准化流水线,将新算法验证周期从数月缩短至数天。
  • 泛化能力解决冷启动:利用大规模预训练带来的强大泛化性,对新风场实现零样本预测,显著提升了无历史数据场景下的调度可靠性。
  • 统一评估高效选型:依托统一的开发与评估管道,快速横向对比多种大模型表现,迅速锁定最适合风电波动特性的最优架构。
  • 便捷微调落地业务:通过提供的 PyTorch 检查点加载脚本和微调指南,轻松将通用大模型适配到私有气象数据上,大幅降低工程部署难度。

OpenLTM 通过提供标准化的大时间序列模型开发流水线与预训练资源,让企业能以最低成本将前沿 AI 技术转化为实际的业务预测力。

运行环境要求

操作系统
  • 未说明
GPU
  • 非必需(推理可用 CPU)
  • 微调推荐 RTX 4090
  • 预训练推荐 A100
内存

未说明

依赖
notes该工具支持多种运行场景:CPU 可用于推理,RTX 4090 适用于模型适配(微调),A100 适用于大规模预训练。需手动下载数据集(如 UTSD 或 ERA5)并放置于 ./dataset 目录。提供 260B 时间点数预训练的 Timer-XL 检查点可供下载使用。
python3.11
requirements.txt 中定义的依赖(具体列表未在 README 中展示)
OpenLTM hero image

快速开始

OpenLTM

OpenLTM 是一个开源代码库,旨在提供一套用于开发和评估大型时间序列模型的流水线。

对于深度时间序列模型,我们推荐 Time-Series-Library 以及这篇全面的 综述

您可以在 Large-Time-Series-ModelHuggingFace 中试用我们开箱即用的时间序列基础模型。

更新

:triangular_flag_on_post: 新闻 (2025.05) 我们发布了 260B 参数量的预训练模型 Timer 的 PyTorch 版本。该版本等同于 HuggingFace 模型,但在微调方面更加友好。此 notebook 展示了如何加载和使用该检查点。

:triangular_flag_on_post: 新闻 (2025.04) 非常感谢 frndtls 实现了 TTMs 及其他 LLM4TS 方法。

:triangular_flag_on_post: 新闻 (2024.12) 非常感谢 khairulislam 实现了 GPT4TS

:triangular_flag_on_post: 新闻 (2024.10) 我们收录了几款大型时间序列模型,发布了预训练代码,并提供了相关脚本。

什么是 LTM?

LTM (Large Time-Series Model,又称时间序列基础模型,TSFM)是一种基于可扩展骨干网络(如 Transformer)并通过大规模预训练构建的深度时间序列模型,可用于处理多种时间序列数据(例如零样本预测)以及下游任务(例如时间序列的通用特征提取)。更多信息请参阅以下教程:[CN][EN]

模型清单

使用方法

  1. 安装 Python 3.11。为方便起见,请执行以下命令。
pip install -r requirements.txt
  1. 将下载的数据放入 ./dataset 文件夹中。此处提供一份 数据集概览
  • 对于单变量预训练(如果您使用的是预训练检查点,则可跳过此步骤):

    • UTSD 包含 10 亿个时间点,用于大规模预训练(numpy 格式):[下载]
    • ERA5 系列(40 年跨度,数千种变量)适用于特定领域的模型:[下载]
  • 对于监督训练或模型适配

  1. 从头开始训练,或使用我们位于 ./checkpoints 文件夹中的预训练检查点。
  1. 我们在 ./scripts/ 文件夹下提供了一些监督训练、预训练和适配脚本:
# 监督训练
# (a) 一对一预测
bash ./scripts/supervised/forecast/moirai_ecl.sh
# (b) 一对多(滚动)预测
bash ./scripts/supervised/rolling_forecast/timer_xl_ecl.sh

# 大规模预训练
# (a) 在 UTSD 上预训练
bash ./scripts/pretrain/timer_xl_utsd.sh
# (b) 在 ERA5 上预训练
bash ./scripts/pretrain/timer_xl_era5.sh

# 模型适配
# (a) 全样本微调
bash ./scripts/adaptation/full_shot/timer_xl_etth1.sh
# (b) 少样本微调
bash ./scripts/adaptation/few_shot/timer_xl_etth1.sh
  1. 开发您的大型时间序列模型。
  • 将模型文件添加到 ./models 文件夹中。您可以参考 ./models/timer_xl.py
  • 将新添加的模型纳入 ./exp/exp_basic.py 中的 Exp_Basic.model_dict
  • ./scripts 文件夹下创建相应的脚本。

推荐资源

在此我们列出一些时间序列大模型的资源,它们支持开箱即用(例如零样本预测):

[!NOTE] 与其它模态的基础模型相比,时间序列大模型目前规模仍然较小。例如,推理时使用 CPU 即可,适配阶段可以使用 RTX 4090 显卡,而预训练则建议使用 A100 显卡。

引用

如果您觉得本仓库对您有帮助,请引用我们的论文。

@inproceedings{liutimer,
  title={Timer: Generative Pre-trained Transformers Are Large Time Series Models},
  author={Liu, Yong and Zhang, Haoran and Li, Chenyu and Huang, Xiangdong and Wang, Jianmin and Long, Mingsheng},
  booktitle={Forty-first International Conference on Machine Learning}
}

@article{liu2024timer,
  title={Timer-XL: Long-Context Transformers for Unified Time Series Forecasting},
  author={Liu, Yong and Qin, Guo and Huang, Xiangdong and Wang, Jianmin and Long, Mingsheng},
  journal={arXiv preprint arXiv:2410.04803},
  year={2024}
}

@article{liu2025sundial,
  title={Sundial: A Family of Highly Capable Time Series Foundation Models},
  author={Liu, Yong and Qin, Guo and Shi, Zhiyuan and Chen, Zhi and Yang, Caiyin and Huang, Xiangdong and Wang, Jianmin and Long, Mingsheng},
  journal={arXiv preprint arXiv:2502.00816},
  year={2025}
}

致谢

我们非常感谢以下 GitHub 仓库提供的宝贵代码和努力:

贡献者

如果您有任何问题或希望使用这些代码,请随时联系:

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

160k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

109.2k|★★☆☆☆|今天
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架