Grid-Oriented Probabilistic Load Forecasting for AI Data Centers

Authors: Qingyi Yao, Ding Lin, Tianqiao Zhao, Meng Yue, Jianhui Wang

Submitted to arXiv on 06 Nov. 2026
Abstract

Probabilistic load forecasts for AI data centers need to represent both uncertainty under changing workloads and temporal dependence to support grid operations. This paper develops an open, grid-oriented benchmark using measured GPU power from a 448-GPU AI cluster. Forecasts are evaluated through four stages: probabilistic prediction, online uncertainty calibration, temporal scenario generation, and horizon-wide reserve sizing. Following a change in cluster utilization, a quantile LightGBM forecast achieves only 63.4% coverage for a nominal 90% interval, while online regime-conditioned calibration increases coverage to 87.3%. Even with well-calibrated marginal forecasts, independent sampling produces unrealistic ramp distributions and excessive reserve procurement. Reconstructing temporal dependence from historical forecast errors reduces the discrepancy in ramp statistics and lowers realized reserve cost by 7.4-10.6% across three forecasting tasks. These findings show that marginal forecast quality alone is insufficient for operational decisions and motivate joint evaluation of calibration, temporal structure, and decision outcomes.

The papers around it, by date and closeness

Higher is closer to this paper; left is older. Lines join papers close to each other: groups of lines are sub-topics. Select a point to see why it is there. Back to the summary of this paper

Closest papers Prior work it builds on Papers that cite it

Papers around this one

Ranked by how close their title and abstract are to this paper’s. Select some to read them together.

Closest papers

  1. Close Another approach to the problem
    Probabilistic Load Forecasting via Point Forecast Feature Integration
    Qicheng Chang, Yishen Wang, Xiao Lu et al. · 2019 · Machine Learning · cited 10 times · AI summary available

    Probabilistic Load Forecasting via Point Forecast Feature Integration predicts short-term power-system load probabilistically using point forecasts as an input feature.

    Abstract

    Short-term load forecasting is a critical element of power systems energy management systems. In recent years, probabilistic load forecasting (PLF) has gained increased attention for its ability to provide uncertainty information that helps to improve the reliability and economics of system operation performances. This paper proposes a two-stage probabilistic load forecasting framework by integrating point forecast as a key probabilistic forecasting feature into PLF. In the first stage, all related features are utilized to train a point forecast model and also obtain the feature importance. In the second stage the forecasting model is trained, taking into consideration point forecast feature…

  2. Close Broader context
    Benchmarking CPU, GPU, and ML Power-Flow Solvers from 14 to 70,000 Buses
    Benjamin Donnot, Sami Tazi, Eva Boguslawski et al. · 2026 · Distributed, Parallel, and Cluster Computing · AI summary available

    Benchmarking CPU, GPU, and ML Power-Flow Solvers compares computational methods for power-flow calculations rather than forecasting AI-cluster demand.

    In short

    Optimized parallel CPU and batched GPU physics solvers match or exceed tested GNN power-flow surrogates in throughput while delivering substantially higher accuracy.

  3. Close Broader context
    DxPU: Large Scale Disaggregated GPU Pools in the Datacenter
    Bowen He, Xiao Zheng, Yuan Chen et al. · 2023 · Distributed, Parallel, and Cluster Computing · cited 18 times · AI summary available

    DxPU studies pooling and allocating datacenter GPUs, addressing GPU infrastructure rather than probabilistic power forecasting.

    Abstract

    The rapid adoption of AI and convenience offered by cloud services have resulted in the growing demands for GPUs in the cloud. Generally, GPUs are physically attached to host servers as PCIe devices. However, the fixed assembly combination of host servers and GPUs is extremely inefficient in resource utilization, upgrade, and maintenance. Due to these issues, the GPU disaggregation technique has been proposed to decouple GPUs from host servers. It aggregates GPUs into a pool, and allocates GPU node(s) according to user demands. However, existing GPU disaggregation systems have flaws in software-hardware compatibility, disaggregation scope, and capacity. In this paper, we present a new implem…

  4. Close Broader context
    Matching Workloads to Blackwell: Power-Resolved Benchmarking of NVIDIA B200 and B300 across HPC and AI Workloads
    Mohsen Zand, Trung Dac Nguyen, H. Birali Runesha · 2026 · Distributed, Parallel, and Cluster Computing · AI summary available

    Matching Workloads to Blackwell measures GPU power and performance across AI and HPC workloads, rather than forecasting cluster load for reserve decisions.

    In short

    Across AI and HPC benchmarks, B300 does not uniformly outperform B200: advantages depend on workload, while kernel granularity, double-precision capability and software maturity shape substantial regressions.

  5. Related Broader context
    Electricity Demand Forecasting with Hybrid Statistical and Machine Learning Algorithms: Case Study of Ukraine
    Tatiana Gonzalez Grandon, Johannes Schwenzer, Thomas Steens et al. · 2023 · Machine Learning · preprint, not peer reviewed · cited 6 times · AI summary available

    Electricity Demand Forecasting with Hybrid Statistical and Machine Learning Algorithms forecasts national Ukrainian electricity demand from hourly consumption data.

    Abstract

    This article presents a novel hybrid approach using statistics and machine learning to forecast the national demand of electricity. As investment and operation of future energy systems require long-term electricity demand forecasts with hourly resolution, our mathematical model fills a gap in energy forecasting. The proposed methodology was constructed using hourly data from Ukraine's electricity consumption ranging from 2013 to 2020. To this end, we analysed the underlying structure of the hourly, daily and yearly time series of electricity consumption. The long-term yearly trend is evaluated using macroeconomic regression analysis. The mid-term model integrates temperature and calendar reg…

  6. Related Broader context
    Not All GPUs Are Created Equal: Characterizing Variability in Large-Scale, Accelerator-Rich Systems
    Prasoon Sinha, Akhil Guliani, Rutwik Jain et al. · 2022 · Distributed, Parallel, and Cluster Computing · cited 47 times · Not summarized yet

    Not All GPUs Are Created Equal characterizes variability in accelerator-rich systems, while this paper forecasts aggregate AI-cluster power and grid reserves.

    Abstract

    Scientists are increasingly exploring and utilizing the massive parallelism of general-purpose accelerators such as GPUs for scientific breakthroughs. As a result, datacenters, hyperscalers, national computing centers, and supercomputers have procured hardware to support this evolving application paradigm. These systems contain hundreds to tens of thousands of accelerators, enabling peta- and exa-scale levels of compute for scientific workloads. Recent work demonstrated that power management (PM) can impact application performance in CPU-based HPC systems, even when machines have the same architecture and SKU (stock keeping unit). This variation occurs due to manufacturing variability and th…

  7. Related Another approach to the problem
    TimeGPT in Load Forecasting: A Large Time Series Model Perspective
    Wenlong Liao, Fernando Porte-Agel, Jiannong Fang et al. · 2024 · Machine Learning · cited 92 times · Not summarized yet

    TimeGPT in Load Forecasting considers a pretrained time-series transformer for load forecasting, whereas this paper evaluates probabilistic forecasts and reserve decisions for AI-cluster power.

    Abstract

    Machine learning models have made significant progress in load forecasting, but their forecast accuracy is limited in cases where historical load data is scarce. Inspired by the outstanding performance of large language models (LLMs) in computer vision and natural language processing, this paper aims to discuss the potential of large time series models in load forecasting with scarce historical data. Specifically, the large time series model is constructed as a time series generative pre-trained transformer (TimeGPT), which is trained on massive and diverse time series datasets consisting of 100 billion data points (e.g., finance, transportation, banking, web traffic, weather, energy, health…

  8. Related Broader context
    A hybrid-model approach for reducing the performance gap in building energy forecasting
    Xia Chen, Tong Guo, Martin Kriegel et al. · 2022 · Advanced Engineering Informatics 52 (2022): 101627 · cited 39 times · AI summary available

    A hybrid-model approach for reducing the performance gap in building energy forecasting combines physics-based and machine-learning models to predict building energy use.

    Abstract

    The performance gap between predicted and actual energy consumption in the building domain remains an unsolved problem in practice. The gap exists differently in both current mainstream methods: the first-principles model and the machine learning (ML) model. Inspired by the concept of time-series decomposition to identify different uncertainties, we proposed a hybrid-model approach by combining both methods to minimize this gap: 1. Use the first-principles method as an encoding tool to convert the building static features and predictable patterns in time-series simulation results; 2. The ML method combines the results as extra inputs with historical records simultaneously, trains the model t…

How are these papers found?

We compare the title and abstract of this paper with those of every paper summarized on this site, and with the papers Semantic Scholar recommends, those it cites and those citing it (for a much-cited paper, its most recent citations and, from OpenAlex, the most cited ones). Among the prior work and the follow-ups, the papers that use each other in depth, a method it uses and the much-cited follow-ups come a little higher. The levels (very close, close, related) come from the similarity of the texts, not from a judgment of quality; * marks a paper without an abstract, compared by its title. The sentence under a paper is written by an AI from the two abstracts: check it in the papers. A paper marked not relevant by three readers leaves the list.

Updated on 9 November 2026.