Toto-2.0-2.5B-FT核心功能解析:为什么它能在时间序列预测中超越90%的模型?
【免费下载链接】Toto-2.0-2.5B-FT项目地址: https://ai.gitcode.com/hf_mirrors/Datadog/Toto-2.0-2.5B-FT
Toto-2.0-2.5B-FT是Datadog推出的时间序列预测模型,基于Toto 2.0 2.5B基础模型微调而来,专注于在GIFT-Eval基准测试中展现卓越性能,是一款专为时间序列预测优化的精调模型。
🌟 模型定位与核心价值
Toto-2.0-2.5B-FT并非通用模型,而是专为基准测试打造的精调版本。它基于Toto 2.0 2.5B基础模型,在包含GIFT-Eval训练集的混合数据上进行微调,旨在探索基础模型在单一分布内基准测试中的极限性能。
⚠️ 注意:对于实际业务场景,建议使用Toto 2.0系列基础模型。基础模型未使用任何公开数据预训练,在各类评估基准中均表现出良好的泛化能力,是部署的理想选择。
📊 性能表现:超越90%模型的关键指标
在GIFT-Eval基准测试中,Toto-2.0-2.5B-FT展现出顶尖性能:
- CRPS(连续排序概率得分):0.463,排名第二
- MASE(平均绝对比例误差):0.679,排名第二
- 在综合排名中,仅次于Toto 2.0 Family-and-Friends集成模型,位列第三
这些指标表明,该模型在时间序列预测的准确性和可靠性上已超越绝大多数同类模型,尤其在复杂的观测性数据场景中表现突出。
🔧 精调配方:平衡性能与泛化的关键
Toto-2.0-2.5B-FT的精调过程经过精心设计,以避免过拟合同时最大化性能:
数据混合策略(按比例):
- GIFT-Eval预训练数据:45%
- Datadog 5分钟以上观测指标:25%
- GIFT-Eval训练集:15%
- 合成数据(TempoPFN):10%
- Datadog 10秒/60秒观测指标:各2.5%
训练参数优化:
- NorMuon和AdamW学习率较预训练降低一个数量级(分别为0.05和0.001)
- 保持与基础模型相同的架构和推理设置
🛠️ 技术架构亮点
从config.json中可以看出,模型架构经过针对性优化:
- 模型规模:2048维隐藏层,48层Transformer结构,32个注意力头
- 序列处理:32的补丁大小,支持长序列输入
- 正则化:预归一化(pre_norm)设计,残差乘法因子0.75
- 注意力机制:使用XPos位置编码,支持动态注意力偏置
这些设计使其在处理时间序列数据时既能捕捉长期依赖,又能保持计算效率。
📚 资源与引用
- 技术报告:深入了解模型原理与实验细节
- 基础模型:推荐部署的通用版本
- GitHub仓库:获取代码与实现细节
如需引用该模型,请使用以下格式:
@misc{khwaja2026toto20timeseries, title={Toto 2.0: Time Series Forecasting Enters the Scaling Era}, author={Emaad Khwaja and Chris Lettieri and Gerald Woo and Eden Belouadah and Marc Cenac and Guillaume Jarry and Enguerrand Paquin and Xunyi Zhao and Viktoriya Zhukov and Othmane Abou-Amal and Chenghao Liu and Ameet Talwalkar and David Asker}, year={2026}, eprint={2605.20119}, archivePrefix={arXiv}, primaryClass={cs.LG}, url={https://arxiv.org/abs/2605.20119}, }🔍 如何获取与使用
要开始使用Toto-2.0-2.5B-FT,可通过以下步骤克隆仓库:
git clone https://gitcode.com/hf_mirrors/Datadog/Toto-2.0-2.5B-FT该模型特别适合需要高精度时间序列预测的场景,如观测性监控、指标预测等。通过合理配置输入参数,可实现对不同时间粒度数据的精准预测。
Toto-2.0-2.5B-FT凭借其精心设计的精调策略和优化的架构,为时间序列预测领域提供了一个高性能的基准模型,展示了基础模型在特定任务上的巨大潜力。无论是学术研究还是工业应用,都能从中获得有价值的参考。
【免费下载链接】Toto-2.0-2.5B-FT项目地址: https://ai.gitcode.com/hf_mirrors/Datadog/Toto-2.0-2.5B-FT
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考