T0, un modelo de 256M de pesos abiertos redefine la predicción de series temporales
The Forecasting Company lanzó T0, un modelo base pequeño pero potente que produce pronósticos probabilísticos para todo, desde la demanda eléctrica hasta los reembolsos de medicamentos.
The Forecasting Company ha lanzado T0, un modelo base de pesos abiertos para la predicción de series temporales que, con solo 256 millones de parámetros, iguala o supera a modelos especializados más grandes en una variedad de benchmarks. Tres variantes —t0-alpha, t0-beta y t0-gamma— están disponibles públicamente en Hugging Face y se pueden descargar sin ninguna autenticación 1.
A diferencia de muchos modelos de IA generativa que producen texto o imágenes, T0 está diseñado para una tarea muy diferente: predecir qué viene después en una secuencia de números. Produce pronósticos probabilísticos, lo que significa que genera un rango de resultados probables en lugar de una única suposición. Esto lo hace útil para todo, desde la demanda de electricidad hasta las ventas minoristas.
Cómo funciona
T0 toma datos históricos como entrada —llamados "contexto"— y también puede aceptar covariables, que son señales adicionales que el modelo conoce de antemano. Las características del calendario, promociones planeadas, pronósticos del tiempo e información similar se pueden alimentar como covariables futuras, ayudando al modelo a hacer predicciones más inteligentes 1.
El modelo admite varias opciones de preprocesamiento que mejoran la precisión. Estas incluyen recortar valores faltantes al inicio de una serie, promediar la predicción con una versión reflejada de la serie negada, recortar predicciones negativas a cero para datos que no pueden ser negativos, y ajustar a números enteros las series de valores enteros 1.
Los usuarios pueden ejecutar T0 en Apple Silicon a través del entorno de ejecución MLX, y el código está disponible en GitHub bajo una licencia abierta 1.
Pruebas en el mundo real
Los desarrolladores demostraron T0 en varios conjuntos de datos públicos. Se utilizó para pronosticar la demanda nacional de electricidad de Francia usando datos de Enedis, y los reembolsos mensuales de medicamentos del sistema nacional de seguro de salud francés. Un ejemplo más complejo involucró precios europeos de electricidad anticipados en el día siguiente a través de cuatro zonas de puja, donde el modelo utilizó carga, producción de energía renovable, calendarios festivos y datos meteorológicos como covariables 1.
El arranque rápido requiere solo unas pocas líneas de código. Los usuarios instalan el paquete, cargan el modelo, pasan un contexto de observaciones pasadas y solicitan predicciones para un horizonte elegido, y reciben pronósticos de cuantil que muestran el rango de resultados probables 1.
Por qué importa el tamaño
Con 256 millones de parámetros, T0 es notablemente compacto en comparación con los grandes modelos de lenguaje que pueden tener cientos de miles de millones. Esta menor huella significa que se ejecuta más rápido y en hardware más económico mientras sigue ofreciendo resultados competitivos. El modelo fue diseñado desde cero como un modelo base de predicción, no adaptado de un modelo de lenguaje 1.
El lanzamiento ocurre cuando la comunidad de IA explora cada vez más cómo las técnicas de modelos base pueden aplicarse más allá del texto y las imágenes. T0 demuestra que los mismos principios —entrenar con datos diversos, usar mecanismos de atención y ofrecer salidas probabilísticas— pueden funcionar de manera efectiva para tareas de predicción numérica también.
Los interesados pueden encontrar el código y los pesos del modelo en la página de GitHub del proyecto 1.
Los datos clave
- T0 es un modelo de predicción de series temporales de 256M de parámetros y pesos abiertos de The Forecasting Company
- Tres variantes —t0-alpha, t0-beta y t0-gamma— se pueden descargar gratuitamente sin autenticación
- El modelo produce pronósticos probabilísticos y puede usar covariables conocidas del futuro, como el clima y los datos del calendario
- Se probó en la demanda eléctrica francesa, los reembolsos de medicamentos y los precios europeos de electricidad anticipados
- T0 admite el entorno de ejecución MLX para Apple Silicon e incluye opciones como el promedio simétrico y el ajuste a enteros