mlsanityData to 7 Oct 2026

Efficient LLM Scheduling by Learning to Rank

arXiv 2408.15792 ↗First posted 28 Aug 2024cs.LGcited by 83

On the map: Efficient LLM inference and serving → Serving systems and request scheduling

Abstract (arXiv)

In Large Language Model (LLM) inference, the output length of an LLM request is typically regarded as not known a priori. Consequently, most LLM serving systems employ a simple First-come-first-serve (FCFS) scheduling strategy, leading to Head-Of-Line (HOL) blocking and reduced throughput and service quality. In this paper, we reexamine this assumption -- we show that, although predicting the exact generation length of each request is infeasible, it is possible to predict the relative ranks of output lengths in a batch of requests, using learning to rank. The ranking information offers valuable guidance for scheduling requests. Building on this insight, we develop a novel scheduler for LLM inference and serving that can approximate the shortest-job-first (SJF) schedule better than existing approaches. We integrate this scheduler with the state-of-the-art LLM serving system and show significant performance improvement in several important applications: 2.8x lower latency in chatbot serving and 6.5x higher throughput in synthetic data generation. Our code is available at https://github.com/hao-ai-lab/vllm-ltr.git

Key facts. Cited by 83 papers in the map (data through 7 Oct 2026). Placed in Serving systems and request scheduling, one of 12 directions in Efficient LLM inference and serving. Builds on 23 of its 50 references that are in the map. First posted on arXiv 28 Aug 2024.

Builds on

23 of 50 references.

Closest ancestors · 7

Same direction, newest first

  1. 2024Andes: Defining and Enhancing Quality-of-Experience in LLM-Based Text Streaming Services25 Apr
  2. 2024Efficient Interactive LLM Serving with Proxy Model-based Sequence Length Prediction12 Apr
  3. 2024Taming Throughput-Latency Tradeoff in LLM Inference with Sarathi-Serve4 Mar
  4. 2024DistServe: Disaggregating Prefill and Decoding for Goodput-optimized Large Language Model Serving18 Jan
  5. 2023Fairness in Serving Large Language Models31 Dec
  6. 2023Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline22 May
  7. 2023Fast Distributed Inference Serving for Large Language Models10 May

Other references · 16

  1. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  2. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
  3. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
  4. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
  5. 2024Length-Controlled AlpacaEval: A Simple Way to Debias Automatic Evaluatorscited by 865
  6. 2023AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedbackcited by 738
  7. 2023Splitwise: Efficient generative LLM inference using phase splittingcited by 536
  8. 2023LMSYS-Chat-1M: A Large-Scale Real-World LLM Conversation Datasetcited by 449
  9. 2023A Long Way to Go: Investigating Length Correlations in RLHFcited by 285
  10. 2018Learning to Optimize Tensor Programscited by 138
  11. 2024Inference without Interference: Disaggregate LLM Inference for Mixed Downstream Workloadscited by 120
  12. 2024DynamoLLM: Designing LLM Inference Clusters for Performance and Energy Efficiencycited by 106
4 more
  1. 2023S3: Increasing GPU Utilization during Generative Inference for Higher Throughputcited by 81
  2. 2024D'ej`aVu: KV-cache Streaming for Fast, Fault-tolerant Generative LLM Servingcited by 48
  3. 2021NeuralNDCG: Direct Optimisation of a Ranking Metric via Differentiable Relaxation of Sortingcited by 39
  4. 2024Enabling Efficient Batch Serving for LMaaS via Generation Length Predictioncited by 20

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 9; 2025: 29; 2026: 45.

83 papers.

Sort:

By area:

  1. 2024Efficiently Scaling LLM Reasoning with Certaindexcited by 56
  2. 2025StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generationcited by 56
  3. 2025OmniRouter: Budget and Performance Controllable Multi-LLM Routingcited by 35
  4. 2024Don't Stop Me Now: Embedding Based Scheduling for LLMscited by 34
  5. 2024BatchLLM: Optimizing Large Batched LLM Inference with Global Prefix Sharing and Throughput-oriented Token Batchingcited by 31
  6. 2025Seer: Online Context Learning for Fast Synchronous LLM Reinforcement Learningcited by 21
  7. 2024BlendServe: Optimizing Offline Inference for Auto-regressive Large Models with Resource-aware Batchingcited by 19
  8. 2025Online Scheduling for LLM Inference with KV Cache Constraintscited by 17
  9. 2025JITServe: SLO-aware LLM Serving with Imprecise Request Informationcited by 17
  10. 2025Optimizing LLM Inference: Fluid-Guided Online Scheduling with Memory Constraintscited by 16
  11. 2025Taming the Titans: A Survey of Efficient LLM Inference Servingcited by 16
  12. 2025semi-PD: Towards Efficient LLM Serving via Phase-Wise Disaggregated Computation and Unified Storagecited by 15
38 more
  1. 2025TETRIS: Optimal Draft Token Selection for Batch Speculative Decodingcited by 14
  2. 2024No Request Left Behind: Tackling Heterogeneity in Long-Context LLM Inference with Medhacited by 12
  3. 2024Fast Inference for Augmented Large Language Modelscited by 10
  4. 2025HyGen: Efficient LLM Serving via Elastic Online-Offline Request Co-locationcited by 10
  5. 2024Chameleon: Adaptive Caching and Scheduling for Many-Adapter LLM Inference Environmentscited by 9
  6. 2024Multi-Bin Batching for Increasing LLM Inference Throughputcited by 9
  7. 2025On Evaluating Performance of LLM Inference Serving Systemscited by 9
  8. 2026Predicting LLM Output Length via Entropy-Guided Representationscited by 9
  9. 2025Adaptively Robust LLM Inference Optimization under Prediction Uncertaintycited by 8
  10. 2025LLM Serving Optimization with Variable Prefill and Decode Lengthscited by 7
  11. 2026Scheduling LLM Inference with Uncertainty-Aware Output Length Predictionscited by 7
  12. 2025Apt-Serve: Adaptive Request Scheduling on Hybrid Cache for Scalable LLM Inference Servingcited by 6
  13. 2025Scorpio: Serving Right Requests at the Right Time for Heterogeneous SLOs in LLM Inferencecited by 6
  14. 2026OrbitFlow: SLO-Aware Long-Context LLM Serving with Fine-Grained KV Cache Reconfigurationcited by 6
  15. 2026SAGA: Workflow-Atomic Scheduling for AI Agent Inference on GPU Clusterscited by 5
  16. 2026TRIAGE: Evaluating Prospective Metacognitive Control in LLMs under Resource Constraintscited by 5
  17. 2025Queueing, Predictions, and LLMs: Challenges and Open Problemscited by 4
  18. 2026Chimera: Latency- and Performance-Aware Multi-agent Serving for Heterogeneous LLMscited by 4
  19. 2025Evaluating the Efficacy of LLM-Based Reasoning for Multiobjective HPC Job Schedulingcited by 3
  20. 2026FlowPrefill: Decoupling Preemption from Prefill Scheduling Granularity to Mitigate Head-of-Line Blocking in LLM Servingcited by 3
  21. 2026Robust Length Prediction: A Perspective from Heavy-Tailed Prompt-Conditioned Distributionscited by 3
  22. 2025Semi-Clairvoyant Scheduling of Speculative Decoding Requests to Minimize LLM Inference Latencycited by 2
  23. 2025Ranking Before Serving: Low-Latency LLM Serving via Pairwise Learning-to-Rankcited by 2
  24. 2025Justitia: Fair and Efficient Scheduling of Task-parallel LLM Agents with Selective Pamperingcited by 2
  25. 2026Queue Length Regret Bounds for Contextual Queueing Banditscited by 2
  26. 2026SuperInfer: SLO-Aware Rotary Scheduling and Memory Management for LLM Inference on Superchipscited by 2
  27. 2026Predictive Scheduling for Efficient Inference-Time Reasoning in Large Language Modelscited by 2
  28. 2026Adapter-Augmented Bandits for Online Multi-Constrained Multi-Modal Inference Schedulingcited by 2
  29. 2026Agentic AI Systems Should Be Designed as Marginal Token Allocatorscited by 2
  30. 2026Observation, Not Prediction: Conversation-Level Disaggregated Scheduling for Agentic Servingcited by 2
  31. 2024TimelyLLM: Segmented LLM Serving System for Time-sensitive Robotic Applicationscited by 1
  32. 2025Semantic Scheduling for LLM Inferencecited by 1
  33. 2025Efficient Serving of LLM Applications with Probabilistic Demand Modelingcited by 1
  34. 2025STAR: Decode-Phase Rescheduling for LLM Inferencecited by 1
  35. 2025CascadeInfer: Length-Aware Scheduling of LLM Serving with Low Latency and Load Balancingcited by 1
  36. 2026PackInfer: Compute- and I/O-Efficient Attention for Batched LLM Inferencecited by 1
  37. 2026Multi-Layer Scheduling for MoE-Based LLM Reasoningcited by 1
  38. 2026RLHFless: Serverless Computing for Efficient RLHFcited by 1

50 most cited of 83 All 83 (JSON)

Similar, not cited

Close in wording, no citation either way

  1. 2025Optimal Scheduling Algorithms for LLM Inference: Theory and Practice
  2. 2025Throughput-Optimal Scheduling Algorithms for LLM Inference and AI Agents
  3. 2025ELIS: Efficient LLM Iterative Scheduling System with Response Length Predictor
  4. 2025Efficient LLM Serving on Hybrid Real-time and Best-effort Requests
  5. 2026EWSJF: An Adaptive Scheduler with Hybrid Partitioning for Mixed-Workload LLM Inference
  6. 2024ScaleLLM: A Resource-Frugal LLM Serving Framework by Optimizing End-to-End Efficiency
  7. 2026Cascade: Exploiting SLO-Aware latency budget for fair and high goodput LLM inference serving
  8. 2024LLM Inference Unveiled: Survey and Roofline Model Insights