跳到正文
arXiv:cs.LG· Daijin Hu·· 3 小时前

JointServe:在显存受限 GPU 上服务多模型智能体

JointServe: Serving Multi-Model Agents on Memory-Constrained GPUs

AI 导读

JointServe 是一个面向显存受限 GPU 的多模型智能体服务系统,将权重分片与 KV 块视为竞争同一内存预算的驻留对象,通过调用意图预取、部分 k* 驻留和抢占式传输调度来提前准备模型。

正文

View PDF HTML (experimental)

Abstract:LLM agents are evolving from a single model calling tools to planners that orchestrate multiple expert models whose capabilities live in distinct weights. This shift breaks a hidden assumption of today's serving systems: that the working set of models fits in GPU memory. On the memory-constrained GPUs typical of private and edge deployments, naive whole-model swapping serializes weight transfer with computation and, without host-backed KV restoration, discards session state at a switch. We present JointServe, a serving system that treats weight shards and KV blocks as residency objects competing for one unified memory budget. JointServe exploits three structural properties of agent workloads (directed calls, gapped execution, and dual-state residency) to prepare models before they are called: an intent signal parsed from the caller's own output stream triggers prefetch during generation, partial k* residency reduces switching to incremental layers, and a preemptive transfer scheduler protects live-session KV with a guaranteed bandwidth share. A carve-out control step adjusts the weight/KV allocation as workload demands change. Evaluations show that JointServe reduces session completion time by 21.7% over the strongest whole-model swap baseline and stays usable in a crash regime where every whole-model swap baseline fails to finish (a 46-fold gap). On replayed SWE-bench, tau-bench, and OSWorld traces, it cuts session completion time by up to 49% and lifts SLO attainment from 2.8-24.5% to 76.8-100%.
Comments: 18 pages, 12 figures, 1 table
Subjects: Distributed, Parallel, and Cluster Computing (cs.DC); Machine Learning (cs.LG)
Cite as: arXiv:2610.19822 [cs.DC]
  (or arXiv:2610.19822v1 [cs.DC] for this version)
  https://doi.org/10.48550/arXiv.2610.19822

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Daijin Hu [view email]
[v1] Wed, 14 Oct 2026 08:59:35 UTC (3,470 KB)

来源:arXiv:cs.LG · arxiv.org