HPC Cluster Task Prediction Based on Multimodal Temporal Networks with Hierarchical Attention Mechanism
In recent years, the increasing adoption of High-Performance Computing (HPC) clusters in scientific research and engineering has exposed challenges such as resource imbalance, node idleness, and overload, which hinder scheduling efficiency. Accurate multidimensional task prediction remains a key bottleneck. To address this, we propose a hybrid prediction model that integrates Informer, Long Short-Term Memory (LSTM), and Graph Neural Networks (GNN), enhanced by a hierarchical attention mechanism combining multi-head self-attention and cross-attention. The model captures both long- and short-term temporal dependencies and deep semantic relationships across features. Built on a multitask learning framework, it predicts task execution time, CPU usage, memory, and storage demands with high accuracy. Experiments show prediction accuracies of 89.9%, 87.9%, 86.3%, and 84.3% on these metrics, surpassing baselines like Transformer-XL. The results demonstrate that our approach effectively models complex HPC workload dynamics, offering robust support for intelligent cluster scheduling and holding strong theoretical and practical significance.
Authors
- Xuemei Bai (ORCID: https://orcid.org/0009-0005-7718-2185)
- Jing-Wen Zhou (ORCID: https://orcid.org/0009-0002-4727-6867)
- Zhijun Wang (ORCID: https://orcid.org/0000-0003-4281-1074)
Institutions
- Changchun University of Science and Technology (CN)
- Changchun Normal University (CN)
Publication Details
- Journal
- Computers
- Published
- 2025-08-18
- DOI
- https://doi.org/10.3390/computers14080335
- Citations
- 2
- Primary Topic
- Distributed and Parallel Computing Systems
- Type
- article
- Field-Weighted Citation Impact
- 2.97