Accurately forecasting GPU workloads is essential for AI infrastructure, enabling efficient scheduling, resource allocation, and power management. Modern workloads are highly volatile, multiple periodicity, and heterogeneous, making them challenging for traditional predictors. We propose PRISM, a primitive-based compositional forecasting framework combining dictionary-driven temporal decomposition with adaptive spectral refinement. This dual representation extracts stable, interpretable workload signatures across diverse GPU jobs. Evaluated on large-scale production traces, PRISM achieves state-of-the-art results. It significantly reduces burst-phase errors, providing a robust, architecture-aware foundation for dynamic resource management in GPU-powered AI platforms.
@article{arxiv.2603.25378,
title = {PRISM: Dynamic Primitive-Based Forecasting for Large-Scale GPU Cluster Workloads},
author = {Xin Wu and Fei Teng and Xingwang Li and Bin Zheng and Qiang Duan},
journal= {arXiv preprint arXiv:2603.25378},
year = {2026}
}