IPA: Inference Pipeline Adaptation to Achieve High Accuracy and Cost-Efficiency

Abstract

Efficiently optimizing multi-model inference pipelines for fast, accurate, and cost-effective inference is a crucial challenge in machine learning production systems, given their tight end-to-end latency requirements. To simplify the exploration of the vast and intricate trade-off space of latency, accuracy, and cost in inference pipelines, providers frequently opt to consider one of them. However, the challenge lies in reconciling latency, accuracy, and cost trade-offs. To address this challenge and propose a solution to efficiently manage model variants in inference pipelines, we present IPA, an online deep learning Inference Pipeline Adaptation system that efficiently leverages model variants for each deep learning task. Model variants are different versions of pre-trained models for the same deep learning task with variations in resource requirements, latency, and accuracy. IPA dynamically configures batch size, replication, and model variants to optimize accuracy, minimize costs, and meet user-defined latency Service Level Agreements (SLAs) using Integer Programming. It supports multi-objective settings for achieving different trade-offs between accuracy and cost objectives while remaining adaptable to varying workloads and dynamic traffic patterns. Navigating a wider variety of configurations allows \namex{} to achieve better trade-offs between cost and accuracy objectives compared to existing methods. Extensive experiments in a Kubernetes implementation with five real-world inference pipelines demonstrate that IPA improves end-to-end accuracy by up to 21% with a minimal cost increase. The code and data for replications are available at https://github.com/reconfigurable-ml-pipeline/ipa.

References in corpus (7)

• Deep Learning Recommendation Model for Personalization and Recommendation Systems
• Cloudburst: Stateful Functions-as-a-Service
• AI Chains: Transparent and Controllable Human-AI Interaction by Chaining Large Language Model Prompts
• Reconciling High Accuracy, Cost-Efficiency, and Low Latency of Inference Serving Systems
• Rafiki: Machine Learning as an Analytics Service System
• XRBench: An Extended Reality (XR) Machine Learning Benchmark Suite for the Metaverse
• Desiderata for next generation of ML model serving

Cited by in corpus (1)

• Adaptive Configuration Selection for Multi-Model Inference Pipelines in Edge Computing