Yumei Shi

dblp:340/5957 · DBLP profile ↗
← Back
5ranked-venue papers
0as first author
5since 2021 · last 2026
0009-0006-3312-348XORCID · reported

Domains — the database's venue-derived domains; a paper can count in several

Systems, architecture and hardware · 5 · 5 since 2021
YearPublicationVenuePosition
2026 Graph-Based Batch Job Load Balancing Scheduling for Multi-Dimensional Resources in Heterogeneous GPU Clusters
abstract
GPU clusters serve as a cornerstone of high-performance computing and support a wide range of batch jobs with complex resource demands. However, the diverse requirements of batch jobs and resource heterogeneity present significant challenges to efficient scheduling. Existing approaches either rely on static rules or overlook the interdependencies among virtual machines introduced by resource heterogeneity, making it difficult to address the diverse resource demands and dynamic load balancing. In this paper, we propose a novel scheduling model based on Graph Neural Networks (GNNs) and Double Deep Q-Networks (DDQNs), termed GNN-DDQN, for batch job load balancing and scheduling of multi-dimensional resources (e.g. GPU, CPU and memory) in heterogeneous clusters. We propose a system model that integrates batch job resource requests, multi-dimensional resource configurations, and a multi-objective optimization framework. The scheduling problem is formulated as a Markov Decision Process based on this model. A GNN is employed to effectively capture the interdependencies among virtual machines, while a DDQN optimizes scheduling decisions using a dynamic target network update mechanism. Extensive experiments are conducted using two real-world Alibaba cluster traces. Results demonstrate the effectiveness and generalization capabilities of the proposed scheduling model. Compared to baseline methods, the results confirm its superiority in load balancing, job latency, fairness, and time efficiency.
Yumei Shi, Shiping Chen 0002, Guangshun Yao, Shengxiang Wang
IEEE Trans. Computers2
2025 AdaGap: An adaptive gap-aware resource allocation strategy for GPU sharing in heterogeneous clusters
Shiping Chen 0004, Yumei Shi, Guangshun Yao
Future Gener. Comput. Syst.3
2024 GPARS: Graph predictive algorithm for efficient resource scheduling in heterogeneous GPU clusters
Shiping Chen 0002, Yumei Shi
Future Gener. Comput. Syst.3
2024 MSHGN: Multi-scenario adaptive hierarchical spatial graph convolution network for GPU utilization prediction in heterogeneous GPU clusters
Shiping Chen 0002, Yumei Shi
J. Parallel Distributed Comput.4
2024 Utilization-prediction-aware energy optimization approach for heterogeneous GPU clusters
Shiping Chen 0002, Yumei Shi
J. Supercomput.3