NVIDIA Kumo Forecast kombiniert vortrainierte Zeitreihendarstellungen mit historischer Musterrückgabe und Interpretierbarkeit
huggingface ·
NVIDIA hat Kumo Forecast vorgestellt, ein multivariates Zeitreihen-Prognosemodell, das zuvor als NV-Tesseract Forecasting bekannt war. Das Modell basiert auf dem MOMENT-1-large-Transformer-Backbone mit etwa 340 Millionen Parametern und kombiniert vortrainierte Darstellungen mit einem trainierbaren Prognosekopf. Es verfügt über einen einzigartigen Domain-Aware Representation and Retrieval (DARR)-Modus, der direkte Vorhersagen mit Mustern aus historischen Daten verbindet, die mittels eines Nearest-Neighbor-Ansatzes abgerufen werden. Die Bibliothek bietet eine DataFrame-first-API und unterstützt konfigurierbare Prognosehorizonte, einschließlich einer autoregressiven Erweiterung für längere Vorhersagen. Eine Schlüsselinnovation ist ihr modellagnostisches Interpretierbarkeitsframework, das horizonspezifische Erklärungen wie Lag-Zuschreibungen und semantische Flussdiagnosen bereitstellt, ohne die Kernarchitektur zu verändern. Kumo Forecast ist unter der Apache 2.0-Lizenz verfügbar und umfasst Demos zur prädiktiven Wartung und Plattformintegration. Entwickler können über Hugging Face auf den Modellcode und die Gewichte zugreifen, um das System auf ihren eigenen Datensätzen zu entwickeln oder feinabzustimmen.
huggingface