PUBLICATIONS

  1. Beyond Prediction: Tail-Aware Scheduling for LLM Inference
    Yueying Li, Yuanfan Chen, Jiayang Chen, Esha Choukse, Haoran Qiu, G. Edward Suh, Rodrigo Fonseca, Ziv Scully, Udit Gupta
    ICML, 2026
    Abstract

    LLM serving exhibits extreme length variability, making size-based scheduling difficult in practice. Recent LLM schedulers approximate SJF/SRPT using predicted decode lengths or rank and primarily report mean-centric metrics (e.g., TTFT/TBT). We show these prediction-driven policies can be fragile under distribution shifts, bursty arrivals, and GPU memory pressure, and still offer limited control over tail latency (P90–P99) that dominates user experience—even with perfect decode-length knowledge. We introduce a distribution-aware, prediction-free scheduling framework that replaces explicit length prediction with soft, γ-parameterized priority boosting driven by lightweight statistical signals. Our design co-optimizes scheduling with cache-aware preemption to account for memory-coupled decode dynamics that vary across workload mixes. Evaluated on Azure production traces, our method achieves a P99 TTLT up to 35–50% lower than SRPT with perfect length prediction and a TTFT 34–47% lower across various workloads, including reasoning-heavy and chat-heavy tasks, demonstrating a robust alternative for tail-latency optimization in online LLM serving.

  2. Making Sense of DPU Performance for Cloud Data Processing
    Jiasheng Hu, Chihan Cui, Yuanfan Chen, Philip A. Bernstein, Jialin Li, Qizhen Zhang
    SoCC (17th ACM Symposium on Cloud Computing), 2026
    Abstract

    Data processing units (DPUs, SoC-based SmartNICs) are emerging data center hardware that provide opportunities to address cloud data processing challenges. We developed a DPU benchmarking framework that encompasses a suite of data processing tasks from primitive compute, memory, and I/O operations and hardware-accelerated tasks to macro-level cloud database modules and a full-fledged, lightweight DBMS.

OPEN SOURCE

  • Found and fixed a silent output-corruption bug in the AITER fp8 MoE path, where padded expert weights were left uninitialized. PR #55251
  • Per-tier cache-hit metrics so operators can see where KV-cache lookups hit, miss, or exit early. PR #4734
  • Prometheus counters that aggregate those hits, misses, and early exits per model. PR #4962
  • Attribution of delivered keys to L1/L2 per object group on retrieve, exported as counters. PR #5842
  • Co-developed lookup skipping for prefixes vLLM already caches; on replayed production traffic it cut L2 prefetch loads by 56% and p50 TTFT by 16–25%. PR #5383
  • Proposed a prefill attention config for Gemma-4 head-512 layers on MI325X. PR #5649
  • Validated the maintainers' follow-up end to end in vLLM. PR #5650
  • Same-GPU weight sharing across data-parallel replicas. PR #1124
  • A placement planner for multi-stage speech serving. PR #2134

ACADEMIC SERVICE

Invited ReviewerIEEE Journal on Selected Areas in Information Theory (JSAIT)Special Issue on Energy and Data Efficiency in AI, 2026