针对深度学习集群资源异构性,提出高效调度方案提升利用率与训练速度。
Resource Heterogeneity-Aware and Utilization-Enhanced Scheduling for Deep Learning Clusters
- 基于任务级异构感知优化框架,跨时空维度动态调度作业。
- 相比先进调度器Gavel,整体训练时间提速1.2倍,资源利用率提升45%。
- 支持多副本并行训练,适合大规模分布式深度学习场景。
当前深度学习模型在配备GPU、TPU等加速器的集群上训练时,调度系统或缺乏细粒度异构感知,或导致资源严重闲置。为此,我们提出一种新型任务级异构感知调度器Hadar,基于优化框架以提升资源利用率。Hadar利用深度学习作业在异构集群中的性能特征,将任务级性能异构性建模为优化问题,并在空间与时间维度上做出调度决策。其采用原对偶框架与对偶子程序求解优化问题并指导调度设计。基于代表性DL训练工作负载的追踪仿真表明,与当前最先进的异构感知调度器Gavel相比,Hadar使总训练时长加速1.20倍。进一步地,我们将Hadar增强为HadarE,通过将每个作业分叉为多个副本,实现作业在不同节点上异构GPU上的并发训练,以增强资源利用率。HadarE在物理深度学习集群上进行了广泛评估,相比Hadar和Gavel,在Amazon AWS(提升45%)和实验室集群上分别实现了50%(或80%)的总时长缩减,同时生成的模型推理质量始终优于Hadar。
原文摘要 · Abstract (English)
Scheduling deep learning (DL) models to train on powerful clusters with accelerators like GPUs and TPUs, presently falls short, either lacking fine-grained heterogeneity awareness or leaving resources substantially under-utilized. To fill this gap, we propose a novel design of a task-level heterogeneity-aware scheduler, Hadar, based on an optimization framework that can boost resource utilization. Hadar leverages the performance traits of DL jobs on a heterogeneous DL cluster, characterizes the task-level performance heterogeneity in the optimization problem, and makes scheduling decisions across both spatial and temporal dimensions. It involves the primal-dual framework employing a dual subroutine, to solve the optimization problem and guide the scheduling design. Our trace-driven simulation with representative DL model training workloads demonstrates that Hadar accelerates the total time duration by 1.20x when compared with its state-of-the-art heterogeneity-aware counterpart, Gavel. Further, our Hadar scheduler is enhanced to HadarE by forking each job into multiple copies to let a job train concurrently on heterogeneous GPUs resided on separate available nodes (i.e., machines or servers) for resource utilization enhancement. HadarE is evaluated extensively on physical DL clusters for comparison with Hadar and Gavel. With substantial enhancement in cluster resource utilization (by 1.45x), HadarE exhibits considerable speed-ups in DL model training, reducing the total time duration by 50% (or 80%) on an Amazon's AWS (or our lab) cluster, while producing trained DL models with consistently better inference quality than those trained by Hadar.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。