跳到正文
Hugging Face Blog·· 1 小时前精选AI 评分64

英伟达发布开源时间序列预测模型 Kumo Forecast

NVIDIA Kumo Forecast Combines Pretrained Time-Series Representations with Historical Pattern Retrieval and Interpretability

AI 导读

英伟达发布多变量时间序列预测模型 Kumo Forecast,结合了预训练 Transformer 主干与域感知表示检索机制。该模型基于 MOMENT-1-large 主干,包含约 3.4 亿参数的冻结基础编码器与可训练预测头,支持直接预测与历史模式检索增强模式,并提供模型无关的可解释性框架,采用 Apache 2.0 协议开源。

推荐理由

原文介绍了该模型的双重预测机制、架构组成及开源协议,想在时间序列预测中引入检索增强与可解释性的开发者可参考其实现方式。

来源:Hugging Face Blog · huggingface.co