Astraea通过全局调度降低LLM代理的端到端延迟
Astraea: A State-Aware Scheduling Engine for LLM-Powered Agents
- 基于请求状态与预测的分层调度,动态分类任务类型
- 平均任务完成时间减少25.5%,在高负载下仍稳定表现
- 适配多模型规模,适合需要低延迟的智能代理系统
大型语言模型(LLMs)正被越来越多地用作智能代理。其多阶段工作流在本地计算与外部网络服务(如Web API)调用间交替进行,导致执行模式与现有推理系统(如vLLM)的调度粒度不匹配。现有系统通常聚焦于局部段优化,无法最小化完整代理工作流的端到端延迟,即整个请求生命周期的全局任务完成时间(JCT)。为此,我们提出Astraea,一种将优化从局部段转向全局请求生命周期的服务引擎。Astraea采用状态感知的分层调度算法,融合请求的历史状态与未来预测,动态根据任务的输入/输出密集或计算密集特性分类,并使用改进的HRRN策略平衡效率与公平性。此外,Astraea还实现自适应KV缓存管理器,在I/O等待期间根据系统内存压力智能处理代理状态。大量实验表明,Astraea相比基线方法平均降低25.5%的JCT,且在不同模型规模下均表现出强鲁棒性与稳定性。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are increasingly being deployed as intelligent agents. Their multi-stage workflows, which alternate between local computation and calls to external network services like Web APIs, introduce a mismatch in their execution pattern and the scheduling granularity of existing inference systems such as vLLM. Existing systems typically focus on per-segment optimization which prevents them from minimizing the end-to-end latency of the complete agentic workflow, i.e., the global Job Completion Time (JCT) over the entire request lifecycle. To address this limitation, we propose Astraea, a service engine designed to shift the optimization from local segments to the global request lifecycle. Astraea employs a state-aware, hierarchical scheduling algorithm that integrates a request's historical state with future predictions. It dynamically classifies requests by their I/O and compute intensive nature and uses an enhanced HRRN policy to balance efficiency and fairness. Astraea also implements an adaptive KV cache manager that intelligently handles the agent state during I/O waits based on the system memory pressure. Extensive experiments show that Astraea reduces average JCT by up to 25.5\% compared to baseline methods. Moreover, our approach demonstrates strong robustness and stability under high load across various model scales.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。