A Case for Task Sampling based Learning for Cluster Job Scheduling
Akshay Jajoo, Y. Charlie Hu, Xiaojun Lin, Nan Deng
Abstract
The ability to accurately estimate job runtime properties allows a scheduler to effectively schedule jobs. State-of-the-art online cluster job schedulers use history-based learning, which uses past job execution information to estimate the runtime properties of newly arrived jobs. However, with fast-paced development in cluster technology (in both hardware and software) and changing user inputs, job runtime properties can change over time, which lead to inaccurate predictions. In this paper, we explore the potential and limitation of real-time learning of job runtime properties, by proactively sampling and scheduling a small fraction of the tasks of each job. Such a task-sampling-based approach exploits the similarity among runtime properties of the tasks of the same job and is inherently immune to changing job behavior. Our study focuses on two key questions in comparing task-sampling-based learning (learning in space) and history-based learning (learning in time): (1) Can learning in space be more accurate than learning in time? (2) If so, can delaying scheduling the remaining tasks of a job till the completion of sampled tasks be more than compensated by the improved accuracy and result in improved job performance? Our analytical and experimental analysis of 3 production traces with different skew and job distribution shows that learning in space can be substantially more accurate. Our simulation and testbed evaluation on Azure of the two learning approaches anchored in a generic job scheduler using 3 production cluster job traces shows that despite its online overhead, learning in space reduces the average Job Completion Time (JCT) by 1.28x, 1.56x, and 1.32x compared to the prior-art history-based predictor.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers3
- TapFinger: Task Placement and Fine-Grained Resource Allocation for Edge Machine LearningYihong Li, Tianyu Zeng, Xiaoxi Zhang, Jingpu Duan et al.INFOCOM 2023 · 39 citations
- Serverless Cold Starts and Where to Find ThemArtjom Joosen, Ahmed Hassan, Martin Asenov, Rajkarn Singh et al.EuroSys 2025 · 30 citations
- When will my ML Job finish? Toward providing Completion Time Estimates through Predictability-Centric SchedulingAbdullah Bin Faisal, Noah Martin, Hafiz Mohsin Bashir, Swaminathan Lamelas et al.OSDI 2024 · 6 citations
Related papers
- Runtime Variation in Big Data AnalyticsYiwen Zhu, Rathijit Sen, Robert Horton, John Mark AgostaSIGMOD 2023 · 5 citations
- SchedInspector: A Batch Job Scheduling Inspector Using Reinforcement LearningDi Zhang, Dong Dai, Bing XieHPDC 2022 · 26 citations
- Cilantro: Performance-Aware Resource Allocation for General Objectives via Online FeedbackRomil Bhardwaj, Kirthevasan Kandasamy, Asim Biswal, Wenshuo Guo et al.OSDI 2023 · 41 citations
- A Dual-Agent Scheduler for Distributed Deep Learning Jobs on Public Cloud via Reinforcement LearningMingzhe Xing, Hangyu Mao, Shenglin Yin, Lichen Pan et al.KDD 2023 · 9 citations
- GREEN: Carbon-efficient Resource Scheduling for Machine Learning ClustersKaiqiang Xu, Decang Sun, Han Tian, Junxue Zhang et al.NSDI 2025 · 23 citations
