用预测优化大模型推理队列调度,提升系统性能
Queueing, Predictions, and LLMs: Challenges and Open Problems
- 将预测服务时间融入队列调度决策
- 大模型推理具动态内存与可抢占特性,影响调度效果
- 适合研究系统调度与排队理论的学者参考
队列系统为机器学习预测(如预计服务时间)的应用提供了广阔机会,以提升系统性能。尽管已有研究探索了基于预测服务时间的队列,目标多为缩短任务在系统中的停留时间,但如何有效利用预测仍存在诸多开放问题。本文综述相关工作,强调预测的有效性,并提出关于队列性能的新问题。随后聚焦于大型语言模型(LLM)系统的实际调度场景,该场景因推理请求具有可变的推理时长、受键值缓存(KV store)内存限制的动态内存占用,以及多种预占策略而带来新的挑战。文章介绍LLM调度的关键背景,提出新模型与未解问题,主张将排队论洞察应用于提升LLM系统调度效率。
原文摘要 · Abstract (English)
Queueing systems present many opportunities for applying machine-learning predictions, such as estimated service times, to improve system performance. This integration raises numerous open questions about how predictions can be effectively leveraged to improve scheduling decisions. Recent studies explore queues with predicted service times, typically aiming to minimize job time in the system. We review these works, highlight the effectiveness of predictions, and present open questions on queue performance. We then move to consider an important practical example of using predictions in scheduling, namely Large Language Model (LLM) systems, which presents novel scheduling challenges and highlights the potential for predictions to improve performance. In particular, we consider LLMs performing inference. Inference requests (jobs) in LLM systems are inherently complex; they have variable inference times, dynamic memory footprints that are constrained by key-value (KV) store memory limitations, and multiple possible preemption approaches that affect performance differently. We provide background on the important aspects of scheduling in LLM systems, and introduce new models and open problems that arise from them. We argue that there are significant opportunities for applying insights and analysis from queueing theory to scheduling in LLM systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。