Romil Bhardwaj

EECS Department, University of California, Berkeley

Technical Report No. UCB/EECS-2023-273

December 14, 2023

http://www2.eecs.berkeley.edu/Pubs/TechRpts/2023/EECS-2023-273.pdf

The increasing computational demands of Machine Learning (ML) models, coupled with a slowdown in hardware advancements, have led to a significant compute supply-demand gap. This gap is evident in the rising costs and limited availability of resources needed for training complex ML models like GPT-4. These challenges hinder the progress and accessibility of ML.

This dissertation aims to bridge the compute supply-demand gap by improving resource efficiency of ML. We introduce Ekya, Cilantro, and ESCHER, three new systems and methods for improving resource efficiency at different layer in the ML stack. Ekya, at the ML application layer, implements a Thief Scheduling algorithm and a Microprofiler to intelligently redistribute resources between inference and retraining tasks, thereby making continuous learning four times more resource-efficient. Cilantro, in the cluster management layer, utilizes online learning to develop dynamic resource-performance models, enabling performance-aware resource allocation in multi-tenant environments. At the orchestration layer, ESCHER introduces ephemeral resources, allowing ML applications to specify custom scheduling requirements without overhauling the underlying cluster manager. This unique approach provides applications with the flexibility to adapt to evolving needs while maintaining simplicity in system design. Together, these systems represent a comprehensive approach to mitigating the compute supply-demand gap, contributing sustainable and efficient resource management techniques.

Advisors: Ion Stoica and Prabal Dutta

---

BibTeX citation:

@phdthesis{Bhardwaj:EECS-2023-273,
    Author= {Bhardwaj, Romil},
    Title= {Efficient Resource Management for Machine Learning},
    School= {EECS Department, University of California, Berkeley},
    Year= {2023},
    Month= {Dec},
    Url= {http://www2.eecs.berkeley.edu/Pubs/TechRpts/2023/EECS-2023-273.html},
    Number= {UCB/EECS-2023-273},
    Abstract= {The increasing computational demands of Machine Learning (ML) models, coupled with a slowdown in hardware advancements, have led to a significant compute supply-demand gap. This gap is evident in the rising costs and limited availability of resources needed for training complex ML models like GPT-4. These challenges hinder the progress and accessibility of ML.

This dissertation aims to bridge the compute supply-demand gap by improving resource efficiency of ML. We introduce Ekya, Cilantro, and ESCHER, three new systems and methods for improving resource efficiency at different layer in the ML stack. Ekya, at the ML application layer, implements a Thief Scheduling algorithm and a Microprofiler to intelligently redistribute resources between inference and retraining tasks, thereby making continuous learning four times more resource-efficient. Cilantro, in the cluster management layer, utilizes online learning to develop dynamic resource-performance models, enabling performance-aware resource allocation in multi-tenant environments. At the orchestration layer, ESCHER  introduces ephemeral resources, allowing ML applications to specify custom scheduling requirements without overhauling the underlying cluster manager. This unique approach provides applications with the flexibility to adapt to evolving needs while maintaining simplicity in system design. Together, these systems represent a comprehensive approach to mitigating the compute supply-demand gap, contributing sustainable and efficient resource management techniques.},
}

EndNote citation:

%0 Thesis
%A Bhardwaj, Romil 
%T Efficient Resource Management for Machine Learning
%I EECS Department, University of California, Berkeley
%D 2023
%8 December 14
%@ UCB/EECS-2023-273
%U http://www2.eecs.berkeley.edu/Pubs/TechRpts/2023/EECS-2023-273.html
%F Bhardwaj:EECS-2023-273