English
Related papers

Related papers: PrefillShare: A Shared Prefill Module for KV Reuse…

200 papers

Diffusion Language Models (DLMs) have been seen as a promising competitor for autoregressive language models. However, diffusion language models have long been constrained by slow inference. A core challenge is that their non-autoregressive…

Computation and Language · Computer Science 2025-05-22 Xinyin Ma , Runpeng Yu , Gongfan Fang , Xinchao Wang

With the rapid evolution of Large Language Models (LLMs), multi-round workflows, such as autonomous agents and iterative retrieval, have become increasingly prevalent. However, this raises hurdles for serving LLMs under prefill-decode (PD)…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-17 Wenhao He , Youhe Jiang , Penghao Zhao , Quanqing Xu , Eiko Yoneki , Bin Cui , Fangcheng Fu

Existing large language model (LLM) serving systems fall into two categories: 1) a unified system where prefill phase and decode phase are co-located on the same GPU, sharing the unified computational resource and storage, and 2) a…

Computation and Language · Computer Science 2025-04-29 Ke Hong , Lufang Chen , Zhong Wang , Xiuhong Li , Qiuli Mao , Jianping Ma , Chao Xiong , Guanyu Wu , Buhe Han , Guohao Dai , Yun Liang , Yu Wang

Two widely adopted techniques for LLM inference serving systems today are hybrid batching and disaggregated serving. A hybrid batch combines prefill and decode tokens of different requests in the same batch to improve resource utilization…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-21 Amna Masood , Pratishtha Gaur , Nuwan Jayasena

Large language models (LLMs) are often used for infilling tasks, which involve predicting or generating missing information in a given text. These tasks typically require multiple interactions with similar context. To reduce the computation…

Computation and Language · Computer Science 2025-05-30 Tianyu Guo , Hande Dong , Yichong Leng , Feng Liu , Cheater Lin , Nong Xiao , Xianwei Zhang

Recent advances in Post-Training Quantization (PTQ) techniques have significantly increased demand for serving quantized large language models (LLMs), enabling higher throughput and substantially reduced memory usage with minimal accuracy…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-16 Kyungmin Bin , Seungbeom Choi , Jimyoung Son , Jieun Choi , Daseul Bae , Daehyeon Baek , Kihyo Moon , Minsung Jang , Hyojung Lee

In Text-to-SQL tasks, existing LLM-based methods often include extensive database schemas in prompts, leading to long context lengths and increased prefilling latency. While user queries typically focus on recurrent table sets-offering an…

Computation and Language · Computer Science 2026-01-14 Jinbo Su , Yuxuan Hu , Cuiping Li , Hong Chen , Jia Li , Lintao Ma , Jing Zhang

Transformer-based large language models (LLMs) demonstrate impressive performance across various natural language processing tasks. Serving LLM inference for generating long contents, however, poses a challenge due to the enormous memory…

Machine Learning · Computer Science 2024-07-01 Wonbeom Lee , Jungi Lee , Junghwan Seo , Jaewoong Sim

Modern LLM serving systems must sustain high throughput while meeting strict latency SLOs across two distinct inference phases: compute-intensive prefill and memory-bound decode phases. Existing approaches either (1) aggregate both phases…

Machine Learning · Computer Science 2025-11-10 Lei Gao , Chaoyi Jiang , Hossein Entezari Zarch , Daniel Wong , Murali Annavaram

Large Language Models (LLMs) are increasingly deployed in multi-agent systems, where effective inter-model communication is crucial. Existing communication protocols either rely on natural language, incurring high inference costs and…

Machine Learning · Computer Science 2026-02-24 Xiangyu Shi , Marco Chiesa , Gerald Q. Maguire , Dejan Kostic

Modern large language model (LLM) systems increasingly rely on multi-turn pipelines that are composed of multiple task-specific adapters, yet existing serving frameworks remain inefficient, incurring substantial recomputation overhead when…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-12-23 Allison Li , Kristjan Greenewald , Thomas Parnell , Navid Azizan

Prefix-sharing among multiple prompts presents opportunities to combine the operations of the shared prefix, while attention computation in the decode stage, which becomes a critical bottleneck with increasing context lengths, is a…

Large Language Models (LLMs) use key-value (KV) cache to reduce redundant computation in autoregressive generation. However, the KV cache size increases linearly during generation, leading to excessive memory usage, especially for long…

Computation and Language · Computer Science 2025-03-04 Jian Yuan , Ziwei He , Haoli Bai , Jingwen Leng , Bo Jiang

Large multimodal models (LMMs) typically employ an encoding module to transform multimodal data inputs into embeddings, which are then fed to language models for further processing. However, efficiently serving LMMs remains highly…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-30 Tianyu Guo , Tianming Xu , Xianjie Chen , Junru Chen , Nong Xiao , Xianwei Zhang

Long-context Large Language Models (LLMs) face significant memory bottlenecks during inference due to the linear growth of key-value (KV) cache with sequence length. While individual optimization techniques like KV cache quantization,…

Machine Learning · Computer Science 2025-12-02 Sai Gokhale , Devleena Das , Rajeev Patwari , Ashish Sirasao , Elliott Delaye

Prefix caching is a key latency optimization for autoregressive LLM serving, yet existing systems assume dense per-token key/value reuse. State-space models change the structure of the problem: a recurrent layer can resume from a single…

Machine Learning · Computer Science 2026-05-08 Mikhail Shirokikh , Sergey Nikolenko

Serving Large Language Models (LLMs) is a GPU-intensive task where traditional autoscalers fall short, particularly for modern Prefill-Decode (P/D) disaggregated architectures. This architectural shift, while powerful, introduces…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-28 Rongzhi Li , Ruogu Du , Zefang Chu , Sida Zhao , Chunlei Han , Zuocheng Shi , Yiwen Shao , Huanle Han , Long Huang , Zherui Liu , Shufan Liu

Efficient inference with Large Language Models (LLMs) increasingly relies on Key-Value (KV) caches to store previously computed key and value vectors at each layer. These caches are essential to minimize redundant computation during…

Hardware Architecture · Computer Science 2026-04-08 Oteo Mamo , Olga Kogiou , Hyunjin Yi , Weikuan Yu

Besides typical generative applications, like ChatGPT, GitHub Copilot, and Cursor, we observe an emerging trend that LLMs are increasingly used in traditional discriminative tasks, such as recommendation, credit verification, and data…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-05-13 Kuntai Du , Bowen Wang , Chen Zhang , Yiming Cheng , Qing Lan , Hejian Sang , Yihua Cheng , Jiayi Yao , Xiaoxuan Liu , Yifan Qiao , Ion Stoica , Junchen Jiang

Prefill-Decode (P/D) disaggregation has emerged as a widely adopted optimization strategy for Large Language Model (LLM) inference. However, there currently exists no well-established methodology for determining the optimal number of P/D…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-03-06 Luchang Li , Dongfang Li , Bozhao Gong , Yu Zhang