Enregistré dans:
| Auteurs principaux: | , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | https://arxiv.org/abs/2605.04074 |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866911649765523456 |
|---|---|
| author | Saleh, Mohammad AlShaikh Chawla, Sanjay Bayhan, Sertac Abu-Rub, Haitham Ghrayeb, Ali |
| author_facet | Saleh, Mohammad AlShaikh Chawla, Sanjay Bayhan, Sertac Abu-Rub, Haitham Ghrayeb, Ali |
| contents | AI data centers experience rapid fluctuations in power demand due to the heterogeneity of computational tasks that they have to support. For example, the power profile of inference and training of large language models (LLMs) is quite distinct and big divergences can result in the instability of the underlying electricity grid. In this paper we propose, to the best of our knowledge, the first physics-informed DLinear time-series model that can accurately forecast power utilization of an AI data center 5-80 minutes (short-term forecasting) into the future. The physics, based on a multi-node lumped thermal resistance-capacitance (RC) network consistent with Newton's law of cooling, is captured using newly derived time-dependent ordinary differential equations (ODE) that separately models and interlinks power consumption with the GPU compute and memory utilization and temperature. The resulting model, that we refer to as PI-DLinear, trained and evaluated on a real AI data center dataset and is not only more accurate than the state-of-the-art (SOTA) models tested, but the forecast profile respects the underlying physics under power throttling and load transient events. Relative to the SOTA transformer-based and non-transformer-based models, improvements in forecasting accuracy (averaged across all look-back and prediction windows) range from 0.782%-39.08% for MSE, 0.993%-51.82% for MAE, and 0.370%-22.28% for RMSE. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2605_04074 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | A Physics-Aware Framework for Short-Term GPU Power Forecasting of AI Data Centers Saleh, Mohammad AlShaikh Chawla, Sanjay Bayhan, Sertac Abu-Rub, Haitham Ghrayeb, Ali Machine Learning Artificial Intelligence Computational Engineering, Finance, and Science Distributed, Parallel, and Cluster Computing Emerging Technologies Operating Systems AI data centers experience rapid fluctuations in power demand due to the heterogeneity of computational tasks that they have to support. For example, the power profile of inference and training of large language models (LLMs) is quite distinct and big divergences can result in the instability of the underlying electricity grid. In this paper we propose, to the best of our knowledge, the first physics-informed DLinear time-series model that can accurately forecast power utilization of an AI data center 5-80 minutes (short-term forecasting) into the future. The physics, based on a multi-node lumped thermal resistance-capacitance (RC) network consistent with Newton's law of cooling, is captured using newly derived time-dependent ordinary differential equations (ODE) that separately models and interlinks power consumption with the GPU compute and memory utilization and temperature. The resulting model, that we refer to as PI-DLinear, trained and evaluated on a real AI data center dataset and is not only more accurate than the state-of-the-art (SOTA) models tested, but the forecast profile respects the underlying physics under power throttling and load transient events. Relative to the SOTA transformer-based and non-transformer-based models, improvements in forecasting accuracy (averaged across all look-back and prediction windows) range from 0.782%-39.08% for MSE, 0.993%-51.82% for MAE, and 0.370%-22.28% for RMSE. |
| title | A Physics-Aware Framework for Short-Term GPU Power Forecasting of AI Data Centers |
| topic | Machine Learning Artificial Intelligence Computational Engineering, Finance, and Science Distributed, Parallel, and Cluster Computing Emerging Technologies Operating Systems |
| url | https://arxiv.org/abs/2605.04074 |