arrow
Return

Cornucopia: Elastic GPU cluster management for mixed batch-interactive distributed training

delete2026-07-01
delete0
PRE
AI
H
Huang, Zimeng
李
李石坚 (Shijian Li)
F
Fan, Zihao
B
BoJiang(江波) (Bo Jiang) *
T
Tian Guo *
DOI:10.1016/j.jpdc.2026.105334delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
The ability to quickly train deep learning (DL) models plays an important role in accelerating the model design process. DL practitioners often need to submit trial-and-error jobs when designing new models or debugging existing models; once the model architectures and the hyperparameters are set, practitioners will then train those models to convergence for inference deployment. Therefore, these trial-and-error jobs often need to be interactive as DL practitioners are actively working on the model design. However, existing GPU cluster managers often assume that training jobs are batch-oriented, leading to very long queueing time for interactive training jobs. To cater to a mix of interactive and batch training workloads, optimizing for both queueing time and job completion time (JCT) simultaneously, we have designed a new cluster manager called Cornucopia. We leverage a new feature called elastic training, which allows a job to be partially preempted to release GPU resources to a higher-priority job to start execution. Specifically, Cornucopia consists of three main components: a scheduler that determines the job queue based on priorities, a profiler that assesses the resource acceleration ratios of jobs, and an allocator that makes decisions regarding resource allocation and preemption. Via large-scale trace-driven simulations, we show that Cornucopia reduces the queuing time of interactive jobs by up to 90% while ensuring JCT compared to several recent GPU schedulers.
Keywords:
Distributed training
GPU cluster
Elastic training
Job scheduling

Journal

Journal of Parallel and Distributed Computing cover
Journal of Parallel and Distributed Computing
IF:
4
Papers:
3.8K
Citations:
4.8K

Organization

W
worcester polytechnic institute
Scholars:
49
Papers: 23
Citations: 0
S
Shanghai Jiao Tong University
Scholars:
5.0K
Papers: 1.5K
Citations: 0
Cited Papers

Cited Papers

No cited papers available