零代码上手Kairos-50M：3分钟实现跨领域时间序列预测的Python实例

【免费下载链接】Kairos_50m项目地址: https://ai.gitcode.com/hf_mirrors/mldi-lab/Kairos_50m

Kairos-50M是一款仅需5000万参数的时间序列基础模型，专为跨领域零样本预测设计。它采用动态补丁 tokenizer、混合大小编码和动态旋转位置嵌入（DRoPE）技术，能高效处理不同信息密度的异构时间序列数据，让普通用户也能轻松实现专业级预测分析。

🚀 为什么选择Kairos-50M？三大核心优势

🔄 动态自适应架构

Kairos-50M的混合大小编码器可根据局部信息密度自动选择tokenization粒度，配合动态旋转位置嵌入（DRoPE）技术，能为不同频谱特征的时间序列数据生成定制化位置编码，解决传统模型在异构数据上的适配难题。

📊 开箱即用的零样本预测

无需任何领域微调，模型即可在电力负荷、股票价格、气象数据等多个领域展现优异性能。这得益于其在包含3000多亿时间点的PreSTS语料库上的预训练，使其具备强大的跨场景泛化能力。

⚡ 高效轻量设计

仅5000万参数的体量却能超越许多大型模型，特别适合资源有限的环境。配置文件config.json显示，模型采用8头注意力机制和6层Transformer结构，在保证性能的同时大幅降低计算需求。

🔍 模型核心参数速览

📋 3分钟快速上手步骤

1️⃣ 环境准备

确保已安装transformers库（5.12.1及以上版本）：

pip install transformers --upgrade

2️⃣ 获取模型

通过Git克隆完整项目：

git clone https://gitcode.com/hf_mirrors/mldi-lab/Kairos_50m cd Kairos_50m

3️⃣ 零代码预测实现

使用以下Python代码即可完成时间序列预测：

import torch from tsfm.model.kairos import AutoModel # 加载模型 model = AutoModel.from_pretrained( "mldi-lab/Kairos_50m", trust_remote_code=True ) # 准备输入数据（示例为随机生成，实际应用中替换为你的时间序列数据） batch_size, context_length = 1, 2048 seqs = torch.randn(batch_size, context_length) # 执行预测 forecast = model( past_target=seqs.clone().detach().float(), prediction_length=96, # 预测未来96个时间步 generation=True, preserve_positivity=True, # 确保预测结果非负 average_with_flipped_input=True # 使用数据增强提升稳定性 ) # 获取预测结果 prediction = forecast["prediction_outputs"] print(f"预测结果形状: {prediction.shape}") # 输出应为 (1, 96)

💡 实用技巧与最佳实践

数据预处理建议

• 输入数据需保持二维结构（批量大小×时间序列长度）
• 建议对数据进行标准化处理（均值为0，标准差为1）
• 长序列数据可分块输入，每块不超过2048个时间点

参数调优指南

• prediction_length：根据实际需求调整预测步长，建议不超过输入序列长度的50%
• preserve_positivity：对非负数据（如销售额、温度）设为True
• average_with_flipped_input：时序波动较大时启用，可降低预测方差

📚 进阶学习资源

• 官方技术论文：Kairos: Toward Adaptive and Parameter-Efficient Time Series Foundation Models
• 模型家族：除50M版本外，还有轻量级的Kairos-23M和Kairos-10M可供选择

🤝 引用与致谢

如果使用Kairos-50M进行研究，请引用以下论文：

@article{feng2025kairos, title={Kairos: Toward Adaptive and Parameter-Efficient Time Series Foundation Models}, author={Feng, Kun and Lan, Shaocheng and Fang, Yuchen and He, Wenchao and Ma, Lintao and Lu, Xingyu and Ren, Kan}, journal={arXiv preprint arXiv:2509.25826v2}, year={2025} }

本项目采用Apache 2.0开源许可，详情参见LICENSE文件。

【免费下载链接】Kairos_50m项目地址: https://ai.gitcode.com/hf_mirrors/mldi-lab/Kairos_50m