Article

Machine Learning Inference on Serverless Platforms Using Model Decomposition

Adrien Gallego, Uraz Odyurt, Yi Cheng, Yuandou Wang, Zhiming Zhao

Abstract

Serverless offers a scalable and cost-effective service model for users to run applications without focusing on underlying infrastructure or physical servers. While the Serverless architecture is not designed to address the unique challenges posed by resource-intensive workloads, e.g., Machine Learning (ML) tasks, it is highly scalable. Due to the limitations of Serverless function deployment and resource provisioning, the combination of ML and Serverless is a complex undertaking. We tackle this problem through decomposition of large ML models into smaller sub-models, referred to as slices. We set up ML inference tasks using these slices as a Serverless workflow, i.e., sequence of functions. Our experimental evaluations are performed on the Serverless offering by AWS for demonstration purposes, considering an open-source format for ML model representation, Open Neural Network Exchange. Achieved results portray that our decomposition method enables the execution of ML inference tasks on Serverless, regardless of the model size, benefiting from the high scalability of this architecture while lowering the strain on computing resources, such as required run-time memory.

Cite as » BibTeX download badge

Metadata

Type:
Article
Year:
2024
Venue:
International Conference on Utility and Cloud Computing (UCC '23)
DOI:
10.1145/3603166.3632535

Links

Licence

Creative Commons Attribution (CC BY) licence Artefacts shared as PDF are licenced under CC BY 4.0.