NVIDIA Kumo Forecast combina representaciones de series temporales preentrenadas con recuperación de patrones históricos y interpretabilidad
huggingface ·
NVIDIA ha presentado Kumo Forecast, un modelo de pronóstico de series temporales multivariadas anteriormente conocido como NV-Tesseract Forecasting. Construido sobre la arquitectura base Transformer MOMENT-1-large con aproximadamente 340 millones de parámetros, el modelo combina representaciones preentrenadas con una cabeza de pronóstico entrenable. Cuenta con un modo único de Representación y Recuperación Consciente del Dominio (DARR) que fusiona predicciones directas con patrones recuperados de datos históricos mediante un enfoque de vecinos más cercanos. La biblioteca ofrece una API centrada en DataFrames y soporta horizontes de pronóstico configurables, incluida la extensión autoregresiva para predicciones más largas. Una innovación clave es su marco de interpretabidad agnóstico al modelo, que proporciona explicaciones específicas por horizonte, como atribuciones de rezagos y diagnósticos de flujo semántico, sin alterar la arquitectura central. Kumo Forecast está disponible bajo la licencia Apache 2.0 e incluye demostraciones para mantenimiento predictivo e integración de plataformas. Los desarrolladores pueden acceder al código y los pesos del modelo a través de Hugging Face para comenzar a construir o ajustar finamente el sistema en sus propios conjuntos de datos.
huggingface