Optimal Resource Efficiency with Fairness in Heterogeneous GPU Clusters
Zizhao Mo, Huanle Xu, Wing Cheong Lau
TL;DR
The paper tackles the challenge of maximizing training throughput while enforcing fairness in heterogeneous GPU clusters. It introduces OEF, a global optimization framework that unifies resource efficiency and multiple fairness properties across cooperative and non-cooperative settings, using profiling, convex optimization, and adjacency-aware placement. The approach provides theoretical guarantees including strategy-proofness and envy-freeness, while delivering substantial practical gains, such as up to 32% throughput improvements and 17–19% reductions in job completion time in real and large-scale traces, with modest overhead. This work offers a principled, scalable solution for multi-tenant DL workloads on diverse GPU ecosystems, with clear implications for cloud providers deploying heterogeneous accelerators.
Abstract
Ensuring the highest training throughput to maximize resource efficiency, while maintaining fairness among users, is critical for deep learning (DL) training in heterogeneous GPU clusters. However, current DL schedulers provide only limited fairness properties and suboptimal training throughput, impeding tenants from effectively leveraging heterogeneous resources. The underlying design challenge stems from inherent conflicts between efficiency and fairness properties. In this paper, we introduce OEF, a new resource allocation framework specifically developed for achieving optimal resource efficiency and ensuring diverse fairness properties in heterogeneous GPU clusters. By integrating resource efficiency and fairness within a global optimization framework, OEF is capable of providing users with maximized overall efficiency, as well as various guarantees of fairness, in both cooperative and non-cooperative environments. We have implemented OEF in a cluster resource manager and conducted large-scale experiments, showing that OEF can improve the overall training throughput by up to 32% while improving fairness compared to state-of-the-art heterogeneity-aware schedulers.
