NVIDIA Kumo Forecast unisce rappresentazioni preaddestrate delle serie temporali con recupero di pattern storici e interpretabilità
huggingface ·
NVIDIA ha presentato Kumo Forecast, un modello di previsione delle serie temporali multivariate precedentemente noto come NV-Tesseract Forecasting. Costruito sul backbone Transformer MOMENT-1-large con circa 340 milioni di parametri, il modello combina rappresentazioni pre-addestrate con una testa di previsione addestrabile. Dispone di una modalità unica chiamata Domain-Aware Representation and Retrieval (DARR), che fonde le previsioni dirette con i modelli recuperati dai dati storici mediante un approccio nearest-neighbor. La libreria offre un'API orientata ai DataFrame e supporta orizzonti di previsione configurabili, inclusa l'estensione autoregressiva per previsioni più lunghe. Un'innovazione chiave è il suo framework di interpretabilità agnostico rispetto al modello, che fornisce spiegazioni specifiche per ogni orizzonte, come attribuzioni dei lag e diagnostica del flusso semantico, senza alterare l'architettura di base. Kumo Forecast è disponibile con licenza Apache 2.0 e include demo per la manutenzione predittiva e l'integrazione di piattaforma. Gli sviluppatori possono accedere al codice e ai pesi del modello tramite Hugging Face per iniziare a costruire o affinare il sistema sui propri dataset.
huggingface