NVIDIA Kumo Forecast 将预训练时间序列表示与历史模式检索及可解释性相结合
huggingface ·
英伟达推出了Kumo Forecast,这是一种多元时间序列预测模型,此前名为NV-Tesseract Forecasting。该模型基于拥有约3.4亿参数的MOMENT-1-large Transformer主干网络构建,将预训练表示与可训练的预测头相结合。它具备独特的领域感知表示与检索(DARR)模式,通过最近邻方法将从历史数据中检索到的模式与直接预测结果相融合。该库提供以DataFrame为核心的API,并支持可配置的预测步长,包括用于更长周期预测的自回归扩展。其一项关键创新是模型无关的可解释性框架,能够在不改变核心架构的情况下提供特定于预测步长的解释,例如滞后归因和语义流诊断。Kumo Forecast采用Apache 2.0许可证发布,并包含用于预测性维护和平台集成的演示示例。开发者可通过Hugging Face获取模型代码和权重,以便在自己的数据集上开始构建或微调该系统。
huggingface