SCOOT: SLO-Oriented Performance Tuning for LLM Inference Engines

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cheng, Ke, Wang, Zhi, Hu, Wen, Yang, Tiannuo, Li, Jianguo, Zhang, Sheng
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912238785265664
author Cheng, Ke
Wang, Zhi
Hu, Wen
Yang, Tiannuo
Li, Jianguo
Zhang, Sheng
author_facet Cheng, Ke
Wang, Zhi
Hu, Wen
Yang, Tiannuo
Li, Jianguo
Zhang, Sheng
contents As large language models (LLMs) are gaining increasing popularity across a wide range of web applications, it is of great importance to optimize service-level objectives (SLOs) for LLM inference services to enhance user satisfaction and improve the competitiveness of cloud vendors. In this paper, we observe that adjusting the parameters of LLM inference engines can improve service performance, and the optimal parameter configurations of different services are different. Therefore, we propose SCOOT, an automatic performance tuning system to optimize SLOs for each LLM inference service by tuning the parameters of the inference engine. SCOOT jointly exploits single-objective and multiple-objective Bayesian optimization (BO) techniques to handle various optimization objectives via exploration and exploitation. Moreover, SCOOT prunes the search space with known constraints and adopts a random forest to learn hidden constraints during the tuning process to mitigate invalid exploration. To improve the tuning efficiency, SCOOT utilizes the parallel suggestion to accelerate the tuning process. Extensive experiments demonstrate that SCOOT considerably outperforms existing tuning techniques in SLO optimization while greatly improving the tuning efficiency. Moreover, SCOOT is universally applicable to various LLM inference engines including vLLM and TensorRT-LLM. Currently, SCOOT has already been implemented in the production environment at Ant Group.
format Preprint
id arxiv_https___arxiv_org_abs_2408_04323
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle SCOOT: SLO-Oriented Performance Tuning for LLM Inference Engines
Cheng, Ke
Wang, Zhi
Hu, Wen
Yang, Tiannuo
Li, Jianguo
Zhang, Sheng
Distributed, Parallel, and Cluster Computing
As large language models (LLMs) are gaining increasing popularity across a wide range of web applications, it is of great importance to optimize service-level objectives (SLOs) for LLM inference services to enhance user satisfaction and improve the competitiveness of cloud vendors. In this paper, we observe that adjusting the parameters of LLM inference engines can improve service performance, and the optimal parameter configurations of different services are different. Therefore, we propose SCOOT, an automatic performance tuning system to optimize SLOs for each LLM inference service by tuning the parameters of the inference engine. SCOOT jointly exploits single-objective and multiple-objective Bayesian optimization (BO) techniques to handle various optimization objectives via exploration and exploitation. Moreover, SCOOT prunes the search space with known constraints and adopts a random forest to learn hidden constraints during the tuning process to mitigate invalid exploration. To improve the tuning efficiency, SCOOT utilizes the parallel suggestion to accelerate the tuning process. Extensive experiments demonstrate that SCOOT considerably outperforms existing tuning techniques in SLO optimization while greatly improving the tuning efficiency. Moreover, SCOOT is universally applicable to various LLM inference engines including vLLM and TensorRT-LLM. Currently, SCOOT has already been implemented in the production environment at Ant Group.
title SCOOT: SLO-Oriented Performance Tuning for LLM Inference Engines
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2408.04323