Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Ranking Before Serving: Low-Latency LLM Serving via Pairwise Learning-to-Rank

About

Efficient scheduling of large language model (LLM) inference tasks is critical for achieving low latency and high throughput, a challenge that is becoming increasingly acute with the rise of reasoning-capable LLMs whose generation lengths are highly variable. Traditional strategies like First Come, First-Serve (FCFS) often suffer from Head-of-Line (HOL) blocking, where long-running tasks delay shorter ones queued behind them. In this paper, we introduce PARS, a prompt-aware LLM task scheduler that mitigates HOL blocking by approximating shortest-job-first (SJF) scheduling through pairwise ranking with a margin ranking loss. PARS effectively predicts response-length-based task ordering directly from prompts, thereby optimizing scheduling decisions with minimal overhead. In addition, it integrates seamlessly with vLLM, a state-of-the-art LLM serving system, for the research community. Extensive experiments across multiple LLM models and real-world inference use cases, including chat, math, and code generation, demonstrate that PARS significantly reduces latency by up to 15.7x compared to the vLLM default scheduler. Cross-model evaluations demonstrate that our design generalizes effectively, allowing effective scheduling across diverse LLMs without requiring model-specific retraining.

Yiheng Tao, Yihe Zhang, Matthew Dearing, Xin Wang, Yuping Fan, Michael E. Papka, Zhiling Lan• 2025

Related benchmarks

TaskDatasetResultRank
LLM Inference SchedulingAlpaca
P90 Per-token Latency (s/token)0.18
20
LLM Inference SchedulingLMSYS-Chat-1M
Average Per-token Latency (s/token)0.1864
16
LLM Request SchedulingLMSYS (test)
Average Latency (s)0.1103
12
LLM Request SchedulingLongBench (test)
Average Latency52.5263
12
LLM ServingMATH-plus
p90 Latency (ms/token)185.1
12
LLM ServingAlpaca
Average Latency (ms/token)124.9
12
LLM ServingMATH-plus
Average Latency (ms/token)115.7
12
LLM ServingEvol-CodeAlpaca
p90 Latency (ms/token)279.3
12
LLM ServingEvol-CodeAlpaca
Average Latency (ms/token)171.6
12
Ranking PredictionAlpaca Llama
Kendall's Tau0.75
4
Showing 10 of 21 rows

Other info

Follow for update