English

Understand and Accelerate Memory Processing Pipeline for Disaggregated LLM Inference

Distributed, Parallel, and Cluster Computing 2026-05-12 v2 Artificial Intelligence

Abstract

Modern large language models (LLMs) increasingly depends on efficient long-context processing and generation mechanisms, including sparse attention, retrieval-augmented generation (RAG), and compressed contextual memory, to support complex reasoning. We show that these optimizations can be unified into a four-step memory processing pipeline: Prepare Memory, Compute Relevancy, Retrieval, and Apply to Inference. Through systematic profiling, we identify a 22%-97% memory processing overhead in LLM inference and strong heterogeneity in its computational characteristics. Motivated by this insight, we argue that \textbf{heterogeneous systems} are well-suited to accelerate memory processing and thus end-to-end inference. We demonstrate this approach on a GPU-FPGA system by offloading sparse, irregular, and memory-bounded operations to FPGAs while retaining compute-intensive operations on GPUs. Evaluated on an AMD MI210 GPU and an Alveo U55C FPGA, our system is up to 2.2×2.2\times faster and achieves up to 4.7×4.7\times less energy across multiple LLM inference optimizations than the GPU baseline (similar results hold on NVIDIA A100). These results establish heterogeneous systems as a practical direction for efficient LLM memory processing and inform future heterogeneous hardware design.

Keywords

Cite

@article{arxiv.2603.29002,
  title  = {Understand and Accelerate Memory Processing Pipeline for Disaggregated LLM Inference},
  author = {Zifan He and Rui Ma and Yizhou Sun and Jason Cong},
  journal= {arXiv preprint arXiv:2603.29002},
  year   = {2026}
}

Comments

Accepted by ICML 2026

R2 v1 2026-07-01T11:45:00.833Z