Explainable Load Forecasting with Covariate-Informed Time Series Foundation Models

arXiv cs.LG / 5/1/2026

📰 NewsModels & Research

Key Points

  • The paper proposes an efficient method to compute SHAP (Shapley Additive Explanations) for Time Series Foundation Models (TSFMs) to improve transparency for critical energy applications like power-grid forecasting.
  • It exploits TSFMs’ flexibility in handling context length and covariates to perform efficient temporal and covariate masking, making SHAP-based explanation scalable.
  • The authors evaluate two TSFMs—Chronos-2 and TabPFN-TS—on day-ahead load forecasting for a transmission system operator, showing zero-shot predictive performance comparable to a Transformer trained on multiple years of TSO data.
  • The generated explanations are reported to match established domain knowledge, with the models appropriately leveraging weather and calendar information for load prediction.
  • Overall, the study argues that TSFMs can be both transparent and reliable tools for operational energy forecasting when paired with explainability techniques like SHAP.

Abstract

Time Series Foundation Models (TSFMs) have recently emerged as general-purpose forecasting models and show considerable potential for applications in energy systems. However, applications in critical infrastructure like power grids require transparency to ensure trust and reliability and cannot rely on pure black-box models. To enhance the transparency of TSFMs, we propose an efficient algorithm for computing Shapley Additive Explanations (SHAP) tailored to these models. The proposed approach leverages the flexibility of TSFMs with respect to input context length and provided covariates. This property enables efficient temporal and covariate masking (selectively withholding inputs), allowing for a scalable explanation of model predictions using SHAP. We evaluate two TSFMs - Chronos-2 and TabPFN-TS - on a day-ahead load forecasting task for a transmission system operator (TSO). In a zero-shot setting, both models achieve predictive performance competitive with a Transformer model trained specifically on multiple years of TSO data. The explanations obtained through our proposed approach align with established domain knowledge, particularly as the TSFMs appropriately use weather and calendar information for load prediction. Overall, we demonstrate that TSFMs can serve as transparent and reliable tools for operational energy forecasting.