ARTICLE DETAIL

资讯详情

深耕网站视觉设计与运营推广的一线实战洞察。

从论文到代码:Chronos-Bolt-Mini零样本预测原理与实现详解

从论文到代码:Chronos-Bolt-Mini零样本预测原理与实现详解 从论文到代码Chronos-Bolt-Mini零样本预测原理与实现详解【免费下载链接】chronos-bolt-mini项目地址: https://ai.gitcode.com/hf_mirrors/autogluon/chronos-bolt-miniChronos-Bolt-Mini是一款基于T5编码器-解码器架构的时间序列预测模型属于Chronos-Bolt模型家族的轻量级版本。它通过创新的时间序列分块处理技术和直接多步预测方法实现了高精度、高效率的零样本时间序列预测能力在无需针对特定数据集进行训练的情况下即可提供可靠的预测结果。 Chronos-Bolt-Mini核心优势解析Chronos-Bolt-Mini作为时间序列基础模型的突破性成果具备三大核心优势极致高效的推理性能相比原始Chronos模型Chronos-Bolt-Mini实现了250倍速度提升和20倍内存效率优化。在处理1024个时间序列上下文长度512预测步长64的场景中其推理速度远超同类模型完美平衡了性能与资源消耗。卓越的零样本预测精度在27个数据集的综合评测中Chronos-Bolt-Mini的零样本预测性能超越了多种传统统计模型和深度学习模型。通过加权分位数损失WQL和平均绝对标度误差MASE评估该模型在未接触过测试数据的情况下仍能提供高精度预测。轻量级架构设计作为仅有2100万参数的轻量级模型Chronos-Bolt-Mini基于t5-efficient-mini架构构建既保证了预测性能又显著降低了部署门槛可在普通CPU环境下高效运行。 零样本预测原理深度解析创新的时间序列分块机制Chronos-Bolt-Mini采用时间序列分块技术将历史数据分割为包含多个观测值的补丁patches。具体而言模型使用16的输入补丁大小和16的补丁步幅config.json中input_patch_size16input_patch_stride16这种设计使模型能够捕捉时间序列中的局部模式和长期依赖关系。T5编码器-解码器架构应用模型基于T5架构将时间序列预测转化为序列到序列的生成任务编码器处理分块后的时间序列上下文默认上下文长度2048提取关键特征表示解码器直接生成多步未来预测采用直接多步预测策略而非传统的递归预测方式概率预测能力Chronos-Bolt-Mini支持生成概率预测通过预测多个分位数config.json中quantiles包含0.1至0.9来量化预测不确定性为决策提供更全面的信息支持。 简单三步实现零样本预测1️⃣ 环境准备安装AutoGluon库即可快速开始pip install autogluon2️⃣ 模型加载与训练使用以下代码加载Chronos-Bolt-Mini并完成模型初始化from autogluon.timeseries import TimeSeriesPredictor, TimeSeriesDataFrame # 加载示例时间序列数据 df TimeSeriesDataFrame(https://autogluon.s3.amazonaws.com/datasets/timeseries/m4_hourly/train.csv) # 初始化预测器指定预测长度和模型 predictor TimeSeriesPredictor(prediction_length48).fit( df, hyperparameters{ Chronos: {model_path: autogluon/chronos-bolt-mini}, }, )3️⃣ 执行预测调用predict方法即可获得零样本预测结果predictions predictor.predict(df) 模型配置参数详解Chronos-Bolt-Mini的核心配置参数定义在config.json中关键参数包括context_length: 2048 - 模型可处理的历史数据长度prediction_length: 64 - 默认预测步长d_model: 384 - 模型隐藏层维度num_heads: 8 - 注意力头数量num_layers: 4 - 编码器/解码器层数这些参数共同决定了模型的容量和性能可根据具体应用场景进行调整。 进阶应用指南模型微调对于特定领域的时间序列数据可通过微调进一步提升Chronos-Bolt-Mini的预测性能。详细微调方法请参考官方教程。协变量支持Chronos-Bolt-Mini支持纳入外部协变量如温度、节假日等进行预测增强模型对外部因素的建模能力。SageMaker部署模型可轻松部署到Amazon SageMakerfrom sagemaker.jumpstart.model import JumpStartModel model JumpStartModel( model_idautogluon-forecasting-chronos-bolt-base, instance_typeml.c5.2xlarge, ) predictor model.deploy() 总结Chronos-Bolt-Mini通过创新的分块技术和高效的T5架构重新定义了时间序列预测的性能标准。其零样本预测能力消除了传统模型对大量标注数据的依赖2100万参数的轻量级设计使其能够在各种计算环境中高效部署。无论是科研探索还是工业应用Chronos-Bolt-Mini都为时间序列预测任务提供了强大而灵活的解决方案。如需获取更多技术细节请参考项目相关论文article{ansari2024chronos, title{Chronos: Learning the Language of Time Series}, author{Ansari, Abdul Fatir and Stella, Lorenzo and Turkmen, Caner and Zhang, Xiyuan, and Mercado, Pedro and Shen, Huibin and Shchur, Oleksandr and Rangapuram, Syama Syndar and Pineda Arango, Sebastian and Kapoor, Shubham and Zschiegner, Jasper and Maddix, Danielle C. and Mahoney, Michael W. and Torkkola, Kari and Gordon Wilson, Andrew and Bohlke-Schneider, Michael and Wang, Yuyang}, journal{Transactions on Machine Learning Research}, issn{2835-8856}, year{2024}, url{https://openreview.net/forum?idgerNCVqqtR} }该项目采用Apache-2.0开源许可欢迎社区贡献和应用。【免费下载链接】chronos-bolt-mini项目地址: https://ai.gitcode.com/hf_mirrors/autogluon/chronos-bolt-mini创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表