Quota-Based Aging for Prediction-Guided Scheduling in LLM Serving
Analyze & explore related research in Conceptio
Connect this record directly into your research workflow or AI agent. Explore citations across open repositories, query full text, and export structured proof bundles.
Quota-Based Aging for Prediction-Guided Scheduling in LLM Serving Hanzhi Zhang, Anxuan Che, Hang Qi, Qiyue Li, Jingjin Wu
arXiv:2610.17795v1 [cs.DC] 13 Oct 2026
Guangdong Prov./Zhuhai Key Lab of IRADS, Beijing Normal-Hong Kong Baptist University, Guangdong, P. R. China Email: t330005063@mail.bnbu.edu.cn; t330018003@mail.bnbu.edu.cn; t330202706@mail.bnbu.edu.cn; t330005026@mail.bnbu.edu.cn; jj.wu@ieee.org Abstract—Large language model (LLM) serving systems use token-level continuous batching to improve resource utilization, but heterogeneous response lengths complicate request scheduling. Prediction-guided short-first scheduling reduces average latency but can severely delay long requests, while unrestricted aging may over-promote delayed long requests under congestion. We propose Quota-Based Aging (QBA), a lightweight policy that combines predicted short-first admission with quota-bounded, debt-enforced promotion of aged predicted-long requests. We also use a reduced M/G/𝑛 queueing surrogate, calibrated from simulator traces, to explain first-order mean-latency trends. QBA is evaluated using a serving-aware event-driven simulator that models decode-priority batching, chunked prefill, token-budget constraints, and KV-cache-safe admission. We compare QBA with FCFS, predicted short-first, and aggressive aging under asymmetric prediction errors on synthetic and production-derived workloads. The evaluation considers queueing delay, throughput, utilization, and class-conditional tail performance. Results show that QBA provides a controllable middle ground between aggressive short-first prioritization and unrestricted aging, substantially reducing long-request tail latency while retaining much of the mean-latency benefit of prediction-guided scheduling. Index Terms—large language model serving, request scheduling, Quality of Service, aging mechanism, queueing theory
I. Introduction Large Language Models (LLMs) support interactive services such as dialogue, coding, and content generation [1], [2]. Once deployed as online services, they must process dynamically arriving requests under stringent latency requirements. Modern serving engines improve resource utilization through tokenlevel continuous batching [3], [4]. However, this execution model couples scheduling to heterogeneous prompt and response lengths, competing prefill and decode stages, a finite token budget, and KV-cache capacity. Scheduling therefore becomes a core performance-control mechanism: it affects not only average responsiveness, but also whether particular request classes experience extreme delay. Accordingly, we use Quality of Service (QoS) here to refer to the joint control of mean latency and class-conditional tail latency. Length prediction provides a natural signal for exploiting request heterogeneity. Prioritizing requests predicted to be short can reduce mean delay, consistent with classical shortjob-oriented scheduling [5], [6], while recent LLM serving work has explored latency- and efficiency-oriented execution mechanisms [7], [8]. However, predicted short-first (PSF) admission can repeatedly defer predicted-long requests and inflate
their tail latency, especially under congestion. Generic aging can mitigate starvation [9], but unrestricted promotion may place too many delayed long requests ahead of short ones and erase the mean-delay benefit that motivated predictionguided scheduling. Accordingly, we ask the following research question: How can an LLM serving scheduler use coarse binary response-length predictions to retain the meandelay benefit of short-first admission while controlling long-request tail inflation, without aging-based promotions overwhelming the short-first path under token-level serving constraints? To answer this question, we propose Quota-Based Aging (QBA), a QoS-aware bounded-aging scheduler for predictionguided LLM serving. QBA prioritizes predicted-short requests while allowing delayed predicted-long requests to become eligible for promotion through a quantile-based waiting threshold. Rather than promoting every aged predicted-long request, QBA earns one promotion opportunity after every 𝑘 successful short admissions. This opportunity is recorded as promotion debt and is cleared only when an aged predicted-long request is actually admitted. By separating aging eligibility from promotion frequency, QBA provides bounded but enforceable protection for delayed long requests, with 𝑘 serving as an interpretable control over the mean–tail latency trade-off. To characterize this trade-off, we further pair QBA with a reduced M/G/𝑛 queueing abstraction and a realistic eventdriven simulator. The analytical model intentionally collapses implementation details into effective service and parallelism parameters, making the main delay trends interpretable. The simulator then restores practical serving mechanisms, including chunk continuation, KV-cache-safe admission, and token-loaddependent batch service, so that the analytical observations can be checked under realistic dynamics. We additionally test the same scheduling mechanisms, without re-tuning, on production LLM inference traces to examine whether the observed trade-off transfers beyond the synthetic setting. The main contributions are: • We propose QBA, a practical QoS-aware bounded-aging policy designed for prediction-guided LLM serving. To the best of our knowledge, it is a novel policy targeting the coupled serving features above. QBA separates aging eligibility from actual admission: delayed long requests become eligible through a quantile-based waiting threshold, while promotion is bounded by a quota cycle and enforced
through promotion debt. • We develop a tractable M/G/𝑛 abstraction that explains how load, service variability, and bounded promotion affect mean delay under token-level continuous batching. By collapsing token-level execution into effective service moments and effective parallelism, the model gives a compact view of why short-first scheduling lowers mean delay, why aggressive aging can over-correct, and why QBA provides a more controlled operating point. • We validate the analytical trends using realistic eventdriven simulation, showing consistent favorable meandelay/tail-latency trade-off for QBA. The simulator restores serving details absent from the reduced model, including decode-priority batching, continuation handling, KVcache-safe admission, token-budget constraints, and tokenload-dependent batch service. • We further validate QBA on 100 non-overlapping windows of a production Microsoft Azure LLM inference trace [10]. Despite bursty non-Poisson arrivals and a promptresponse composition substantially different from the synthetic workload, the results confirm that QBA remains a stable intermediate operating point without any workloadspecific re-tuning. The implementation and frozen experimental artifacts are publicly available at [11]. 1
Workload
2
3
Binary Predictor
Scheduling Policies Aging-based Policies
Baselines
AA Poisson arrivals
predicted short
prompt/response heterogeneity
6
Metrics
FCFS
PSF
aggressive promotion
5
Evaluation
4
quota k + promotion debt
---
LLM Serving Engine
High-fidelity simulator
mean waiting tail latency fairness-efficiency trade-off
Reduced M/G/n analysis
Sym
Meaning
Sym
Meaning
𝜆, 𝜌 𝑃, 𝐷, 𝑆 𝑌𝑖 , 𝑌ˆ𝑖 𝜖𝑆𝐿/𝐿𝑆 𝑘
arrival rate / load prompt / decode / eff. svc. true/predicted class misclassif. rates QBA quota
𝑛 𝜏0.85 𝜋𝑆/𝐿 𝛾 𝜙𝑋 (𝜌)
effective parallelism short/long cutoff class priors aging quantile level promoted long frac.
A. LLM Serving Model We consider a single-engine LLM serving system with tokenlevel continuous batching [3], [4]. For analytical tractability, we first consider that requests arrive according to a Poisson process with rate 𝜆, a standard open-system abstraction in queueing models [6]. Request 𝑖 is represented by R 𝑖 = (𝑃𝑖 , 𝐷 𝑖 , 𝑌ˆ𝑖 ), where 𝑃𝑖 is the prompt length, 𝐷 𝑖 is the response length, and 𝑌ˆ𝑖 ∈ {𝑆, 𝐿} is a predicted short/long label. 1) Two-stage Execution: Each request evolves through a prefill stage followed by a decode stage. Prefill processes prompt tokens and builds the KV cache, while decode generates response tokens iteratively until completion [10]. Let 𝑃𝑖rem (𝑡) and 𝐷 rem 𝑖 (𝑡) denote the remaining prefill and decode workloads. A request departs when 𝑃𝑖rem (𝑡) = 0 and 𝐷 rem 𝑖 (𝑡) = 0. 2) Decode-Priority Token Batching: At batch boundary ℓ, the scheduler forms a batch under token budget 𝐵, following the batching abstraction used in modern inference-serving systems [3], [12]. Let Dℓ𝜋 and Pℓ𝜋 denote the decode and prefill requests selected by policy 𝜋. The batch must satisfy ∑︁ ∑︁ 𝑞𝑖 + 𝑢 𝑗 ≤ 𝐵, 𝑖 ∈ Dℓ𝜋
𝑗 ∈ Pℓ𝜋
QBA (Proposed)
bounded promotion
predicted long
TABLE I: Key notation.
token-level batching prefill/decode KV-safe admission
Fig. 1: Workflow of prediction-guided LLM request scheduling, including workload generation, binary response-length prediction, scheduling, token-level continuous batching, evaluation, and latency metrics. The synthetic workload follows Poisson arrivals, while the production trace (Section IV-E) retains its original bursty arrival structure.
II. System Model As summarized in Fig. 1, this section defines the LLM serving system, the reduced queueing abstraction, and the prediction labels used by the scheduler.
where 𝑞 𝑖 is the decode block allocated to request 𝑖 and 𝑢 𝑗 is the prefill chunk allocated to request 𝑗. Active decode requests are always served first, and prefill requests are admitted using the residual token budget. 3) Chunk Continuation and KV-Safe Admission: In our simulator, if a prefill request cannot be fully inserted into the current batch, it is chunked and its unfinished remainder is returned to the queue as a continuation request. Continuation requests always have the highest admission priority among prefill candidates. Let 𝑀ℓ be the live KV-cache usage at batch boundary ℓ. Admission is checked against the safe threshold 𝑀safe = 𝜂𝑀max , where 𝑀max is the hard KV capacity, and 𝜂 ∈ (0, 1) is a safety factor, reflecting the central role of KVcache memory in LLM serving [4]. A candidate prefill request is admitted only if the projected KV usage remains below 𝑀safe . 4) Batch Service Time: Let 𝑏 ℓ be the total token load of batch ℓ. Its service time is modeled as 𝑇ℓ = 𝑐 0 + 𝑎 max(0, 𝑏 ℓ − 𝑏 0 ), where 𝑐 0 is a fixed launch overhead, 𝑎 is the marginal per-token coefficient, and 𝑏 0 is a knee point obtained from profiling. The event-driven simulator preserves these serving mechanics, including decode-priority batching, continuation handling, KV-safe admission, and token-load-dependent batch service times. Thus, all scheduling policies compared in this paper operate under the same token budget, KV-cache constraint, and batch-service model.
B. Reduced M/G/𝑛 Abstraction To obtain a compact analytical model that helps to interpret the key results, we collapse the two-stage execution of each request into a single effective service time, following standard queueing practice in which complex service mechanisms are represented by effective service-time distributions [6]: 𝑆𝑖 = 𝑠0 + 𝛼𝑃𝑖 + 𝛽𝐷 𝑖 ,
(1)
where 𝑠0 is a fixed overhead and 𝛼, 𝛽 > 0 are effective prefill/decode coefficients. Hereafter, 𝑃, 𝐷, and 𝑆 denote generic prompt-length, response-length, and effective-servicetime random variables drawn from the same distribution as 𝑃𝑖 , 𝐷 𝑖 , and 𝑆𝑖 . For discrete implementation, l𝐷 m l𝑃 m 𝑖 𝑖 + 𝜏𝑑 , (2) 𝑆 𝑖 ≍ 𝑠0 + 𝜏𝑝 𝐶𝑝 𝑞 where 𝐶 𝑝 is the prefill chunk size, 𝑞 is the decode block size, and 𝜏𝑝 and 𝜏𝑑 are the effective service times of one prefill chunk and one decode block, respectively. 1) Effective Parallelism: Although the actual engine is not a classical multi-server queue, we approximate it as an M/G/𝑛 system with effective parallelism 𝑛 = ⌊min(𝐵/𝑞, 𝑛safe )⌋, where 𝐵/𝑞 is the decode concurrency implied by the token budget and 𝑛safe is the maximum safe concurrency under KV constraints [6]. Let 𝜇 = 1/E[𝑆] and 𝜌 = 𝜆E[𝑆]/𝑛. The reduced system is stable when 𝜌 < 1. 2) Moment Characterization: From (1), the first two moments are E[𝑆] = 𝑠0 + 𝛼E[𝑃] + 𝛽E[𝐷], E[𝑆 ] = 𝑠02 + 2𝑠0 𝛼E[𝑃] + 𝛽E[𝐷] 2 2 2 2
(3)
2
+ 𝛼 E[𝑃 ] + 𝛽 E[𝐷 ] + 2𝛼𝛽E[𝑃𝐷].
(4)
Let 𝐶𝑠2 = Var(𝑆)/E[𝑆] 2 . When arrivals are Poisson, 𝐶𝑎2 = 1. This abstraction collapses prefill, decode, batching, and KV admission into a single effective service time together with a static effective parallelism. It preserves first-order mean-delay trends but cannot resolve token-level dynamics such as decodepriority preemption, chunk continuation, or the time-varying batch composition. We therefore use (1)–(2) only to explain mean-level trends, while tail latencies and token-level metrics are evaluated in the simulator. C. Label Construction and Prediction Assumption To make the short/long partition well-defined, we consider that each request has an underlying response-length random variable 𝐷 𝑖 drawn from a stationary workload distribution. Rather than assuming that the predictor directly outputs an arbitrary binary label, we define the true short/long class through a workloadlevel cutoff. Let 𝜏0.85 be a fixed threshold chosen from the estimated response-length distribution, for example as a quantile corresponding to a desired short/long split. The true class of request 𝑖 is defined as ( 𝑆, 𝐷 𝑖 ≤ 𝜏0.85 𝑌𝑖 = , (5) 𝐿, 𝐷 𝑖 > 𝜏0.85
1) Binary Predictor: We assume a lightweight binary predictor that estimates the thresholded class 𝑌𝑖 rather than the exact response length. Accordingly, the scheduler observes only the predicted label 𝑌ˆ𝑖 ∈ {𝑆, 𝐿}, where 𝑆 and 𝐿 denote predictedshort and predicted-long requests. 2) Prediction Errors: The class priors are 𝜋 𝑆 = Pr(𝑌𝑖 = 𝑆) and 𝜋 𝐿 = 1−𝜋 𝑆 = Pr(𝑌𝑖 = 𝐿). We characterize prediction errors by the two misclassification rates 𝜖 𝐿𝑆 = Pr(𝑌ˆ = 𝑆 | 𝑌 = 𝐿) and 𝜖 𝑆𝐿 = Pr(𝑌ˆ = 𝐿 | 𝑌 = 𝑆), which correspond to long-to-short and short-to-long errors, respectively. Under this assumption, the predicted-class arrival rates satisfy 𝜆 𝑆 = 𝜆 (1 − 𝜖 𝑆𝐿 )𝜋 𝑆 + 𝜖 𝐿𝑆 𝜋 𝐿 , (6) 𝜆 𝐿 = 𝜆 𝜖 𝑆𝐿 𝜋 𝑆 + (1 − 𝜖 𝐿𝑆 )𝜋 𝐿 . (7) III. Scheduling Policies and Analytical Approximation This section describes the scheduling policies evaluated in the paper and their reduced analytical approximations. At each scheduling decision, active decode requests retain priority because they correspond to already-started generations. The policies below therefore mainly determine the admission order among queued prefill candidates after decode traffic and continuation requests have been handled. A. Baseline and Prediction-Guided Policies 1) FCFS Baseline: First-come-first-served (FCFS) admits queued prefill requests in arrival order, subject to the tokenbudget and KV-cache constraints defined in the system model. Let 𝑊𝑞 denote queueing delay before service and 𝑇 denote end-to-end response time. For this baseline, we use the Allen– Cunneen approximation for a multi-server queue with general service-time variability [6]: E[𝑊𝑞FCFS ] ≈
𝐶𝑎2 + 𝐶𝑠2 𝑃wait · , 2 𝑛𝜇 − 𝜆
(8)
where 𝑃wait is the Erlang-C delay probability of the corresponding M/M/𝑛 system. Hence, E[𝑇 FCFS ] ≈ E[𝑆] + E[𝑊𝑞FCFS ]. We model prediction-guided scheduling as a non-preemptive two-class priority discipline on top of the reduced M/G/𝑛 abstraction. 2) Predicted Classes: Requests are partitioned into predicted-short and predicted-long classes. Let 𝜆 𝑆 = 𝜆 Pr(𝑌ˆ = 𝑆) and 𝜆 𝐿 = 𝜆 Pr(𝑌ˆ = 𝐿), with 𝜆 𝑆 + 𝜆 𝐿 = 𝜆. Define the classconditional moments 𝑚 𝑟(𝑐) = E[𝑆𝑟 | 𝑌ˆ = 𝑐] for 𝑐 ∈ {𝑆, 𝐿} and 𝑟 = 1, 2, and the corresponding loads 𝜌 𝑐 = 𝜆 𝑐 𝑚 1(𝑐) /𝑛. 3) PSF Mean Waiting Approximation: Predicted short-first (PSF) prioritizes predicted-short requests over predicted-long requests, while preserving arrival order within each predicted class, motivated by the classical mean-delay advantage of shortjob-oriented service disciplines [5]. Exact closed forms for non-preemptive priority M/G/𝑛 queues do not yield a compact expression. We therefore use a compact priority correction that preserves the FCFS congestion term while capturing the shortfirst effect [6]. Let Ω𝑛 = (1 − 𝜌)E[𝑊𝑞FCFS ]. Then the queueing
delays of the two predicted classes are approximated by Ω𝑛 , 1 − 𝜌𝑆 Ω𝑛 E[𝑊𝑞( 𝐿) ] ≈ . (1 − 𝜌 𝑆 ) (1 − 𝜌) E[𝑊𝑞(𝑆) ] ≈
(9)
denominator counts all predicted-long jobs. These fractions are obtained at several load points and interpolated over 𝜌 to form load-dependent surrogates. The corresponding effective high-priority arrival rates are
(10)
The overall mean queueing delay under predicted short-first scheduling is 𝜆𝑆 𝜆𝐿 E[𝑊𝑞PSF ] ≈ E[𝑊𝑞(𝑆) ] + E[𝑊𝑞( 𝐿) ], (11) 𝜆 𝜆 and therefore E[𝑇 PSF ] ≈ E[𝑆] + E[𝑊𝑞PSF ]. 4) Quantile-Based Aging: To mitigate starvation of delayed predicted-long requests, the simulator uses a quantile-based aging threshold rather than a moment-based threshold, consistent with recent attention to queue-aware and scheduling-aware LLM serving [13]. Let 𝑊ref denote the empirical waiting-time sample used by the aging module. The threshold is 𝜃 𝐴 = 𝑄 𝛾 (𝑊ref ), where 𝑄 𝛾 (·) is the empirical 𝛾-quantile and 𝛾 ∈ (0, 1] is a tunable parameter. A predicted-long request becomes aged once its waiting time exceeds 𝜃 𝐴. The reference sample may be taken either from the current waiting queue or from the started-request history; in our experiments we use the history-based variant. 5) Aggressive Aging (AA): Aggressive Aging (AA) promotes all currently aged predicted-long requests ahead of ordinary predicted-short requests. This mechanism serves as a natural fairness-oriented1 extension of short-first scheduling: it mitigates starvation of delayed long requests, but can overcorrect under heavy congestion because aged long requests may repeatedly occupy the short-first admission path. This limitation motivates the bounded design of QBA. 6) Quota-Based Aging (QBA): QBA combines a per-𝑘short-admissions quota with a promotion-debt state to bound aged predicted-long promotions. A short-admission counter advances only when a short prefill fully completes (not on chunk continuations); once it reaches 𝑘, one promotion debt is earned and cannot be displaced by further short admissions, but debt is cleared (and the counter reset to 0) only when the oldest eligible aged-long request is actually admitted; while debt is pending the counter stays at 𝑘, serializing promotions to at most one per quota cycle. B. Mean-Level Surrogate and Simulator Alignment 1) Mean-Level Surrogate for AA and QBA: In the analytical model, both AA and QBA are represented through effective promoted fractions estimated from simulator traces, yielding a reduced empirical closure rather than an exact priorityqueue solution [6]. For AA, let 𝜙 𝐴 (𝜌) denote the empirical fraction of predicted-long jobs whose waiting time exceeds the aging threshold at service start. For QBA, let 𝜙𝑄 (𝜌; 𝑘) = 𝑁promoted-long /𝑁predicted-long , where the numerator counts successfully promoted predicted-long jobs under the quota rule and the 1 In this paper, fairness means tail-latency fairness, the protection of delayed long requests from starvation-induced tail inflation under short-first admission, rather than a Jain-index-style per-class balance or a direct short-versus-long latency gap. Accordingly, an efficiency–fairness trade-off denotes the tension between reducing mean delay and controlling long-request tail latency.
𝜆 𝑆( 𝐴) = 𝜆 𝑆 + 𝜙 𝐴 (𝜌)𝜆 𝐿 ,
(12)
𝜆 𝑆(𝑄) = 𝜆 𝑆 + 𝜙𝑄 (𝜌; 𝑘)𝜆 𝐿 .
(13)
The promoted high-priority class then has mixed moments 𝑚˜ 𝑟(𝑆,𝑋) =
𝜆 𝑆 𝑚 𝑟(𝑆) + 𝜙 𝑋 (𝜌)𝜆 𝐿 𝑚 𝑟(𝐿) , 𝜆 𝑆 + 𝜙 𝑋 (𝜌)𝜆 𝐿
𝑋 ∈ { 𝐴, 𝑄}, 𝑟 = 1, 2.
(14) Here 𝜙 𝑋 (𝜌) denotes 𝜙 𝐴 (𝜌) for 𝑋 = 𝐴 and 𝜙𝑄 (𝜌; 𝑘) for 𝑋 = 𝑄. Substituting these mixed moments into the PSF approximation yields reduced mean-level surrogates for the aggressive-aging variant AA and the bounded policy QBA. For this closure, 𝜆 𝐿(𝑋) = 𝜆 − 𝜆 𝑆(𝑋) , 𝜌 𝑆(𝑋) = 𝜆 𝑆(𝑋) 𝑚˜ 1(𝑆,𝑋) /𝑛, and E[𝑊𝑞(𝑆,𝑋) ] and E[𝑊𝑞(𝐿,𝑋) ] are obtained by applying the same priority-correction form as in (11) with the mixed promoted class. Thus, 𝜙 𝑋 (𝜌) controls the effective high-priority population: larger promoted fractions improve delayed-long access but also increase congestion in the short-first path. E[𝑊𝑞𝑋 ] ≈
𝜆 𝑆(𝑋)
E[𝑊𝑞(𝑆,𝑋) ] +
𝜆 𝐿(𝑋) E[𝑊𝑞(𝐿,𝑋) ], 𝜆
𝑋 ∈ { 𝐴, 𝑄}. (15) The reduced closures in (15) and the simulator share the same workload family and prediction setup; the analytical layer collapses engine details for mean-level interpretation, while the simulator preserves token-level dynamics for tail-latency, TTFT, and fairness-efficiency evaluation. 𝜆
IV. Experiments A. Experimental Setup 1) Synthetic Workload Generation: We evaluate all policies on a synthetic request stream designed to mimic heterogeneous LLM serving workloads [14]. Requests arrive according to a Poisson process with arrival rate 𝜆. For each request, the prompt length is sampled from a mixture distribution: with probability 0.7 it is drawn from N (100, 302 ), and with probability 0.3 it is drawn from a Pareto distribution with shape 1.4 and scale 40, reflecting the heavy-tailed prompt-length behavior observed in production LLM workloads [14]. Both prompt and response lengths are capped at 4096 tokens. Conditioned on the prompt length 𝑃𝑖 , the response length 𝐷 𝑖 is generated from a lognormal model 𝐷 𝑖 ∼ LogNormal 𝛽0 + 𝛽1 log(1 + 𝑃𝑖 ), 𝜎 2 , with 𝛽0 = 3, 𝛽1 = 0.75, and 𝜎 = 0.3. 2) Prediction Labels: We construct short/long labels using the empirical 0.85-quantile of the generated response-length distribution. Unless otherwise stated, the predicted label matches the thresholded ground-truth label exactly. This noiseless setting isolates the scheduling mechanism; asymmetric prediction errors are evaluated separately below.