NVIDIA Kumo Forecast объединяет предварительно обученные представления временных рядов с извлечением исторических паттернов и интерпретируемостью
huggingface ·
Компания NVIDIA представила Kumo Forecast — модель для многомерного прогнозирования временных рядов, ранее известную как NV-Tesseract Forecasting. Модель построена на базе трансформерного ядра MOMENT-1-large с примерно 340 миллионами параметров и сочетает предварительно обученные представления с обучаемой головой для прогнозирования. Она оснащена уникальным режимом Domain-Aware Representation and Retrieval (DARR), который объединяет прямые прогнозы с паттернами, извлеченными из исторических данных с помощью подхода ближайших соседей. Библиотека предлагает API, ориентированный на DataFrame, и поддерживает настраиваемые горизонты прогнозирования, включая авторегрессионное расширение для более длительных предсказаний. Ключевым нововведением является архитектура-независимая система интерпретируемости, предоставляющая объяснения, специфичные для горизонта прогнозирования, такие как атрибуции лагов и диагностика семантического потока, без изменения основной архитектуры. Kumo Forecast доступна под лицензией Apache 2.0 и включает демонстрации для предиктивного обслуживания и интеграции платформ. Разработчики могут получить доступ к коду модели и весам через Hugging Face, чтобы начать создание или дообучение системы на собственных наборах данных.
huggingface