PreprintCited by 40Topic ModelingNatural Language Processing TechniquesParallel Computing and Optimization Techniques

The DOI matches its DataCite record; 33 of the 33 references it cites were found in a registry or an index; 13 of the 19 sources its claims cite were checked for support.

ChunkAttention: Efficient Self-Attention with Prefix-Aware KV Cache and Two-Phase Partition

Open access
Ye, Lu, Tao, Ze, Huang, Yong, Li, Yang·2024·arXiv

Abstract

Self-attention is an essential component of large language models (LLM) but a significant source of inference latency for long sequences. In multi-tenant LLM serving scenarios, the compute and memory operation cost of self-attention can be optimized by using the probability that multiple LLM requests have shared system prompts in prefixes. In this paper, we introduce ChunkAttention, a prefix-aware self-attention module that can detect matching prompt prefixes across multiple requests and share their key/value tensors in memory at runtime to improve the memory utilization of KV cache. This is achieved by breaking monolithic key/value tensors into smaller chunks and structuring them into the auxiliary prefix tree. Consequently, on top of the prefix-tree based KV cache, we design an efficient self-attention kernel, where a two-phase partition algorithm is implemented to improve the data locality during self-attention computation in the presence of shared system prompts. Experiments show that ChunkAttention can speed up the self-attention kernel by 3.2-4.8$\times$ compared to the state-of-the-art implementation, with the length of the system prompt ranging from 1024 to 4096.

Overview

AI snapshot

Written from the paper’s full text.

## TL;DR

ChunkAttention is a self-attention module for LLM inference that exploits shared system prompts by storing KV cache in a prefix tree and using a two-phase partition kernel. On A100 (80G), it reports 3.2–4.8× speedups over the state of the art when shared system prompts are 1024–4096 tokens long.

## Key findings

  • In microkernel experiments on NVIDIA A100 (80G) with FP16, ChunkAttn outperformed PagedAttn* by 2.8–3.2× when shared prefix length \(n_s\) ranged from 1024 to 4096; for example, at \(n_p=1024, n_s=1024\), latency dropped from 154.46 µs to 56.00 µs.
  • In end-to-end tests on OpenLlama2 7B with ChunkLlama, token rate improved by 3.2×, 2.7×, and 2.2× at \(n_c = 256, 512,\) and 1024 for one setting, and by 4.8×, 4.1×, 3.2×, and 2.4× for shared prefix lengths 512, 1024, 2048, and 4096 respectively.
  • Table 4 reports that with \(n_p=1024, n_s=1024, n_c=512\), peak KV cache memory fell from 14.79 GB to 3.28 GB; with \(n_p=2048, n_s=2048, n_c=512\), it fell from 21.09 GB to 3.40 GB.
  • The paper states that without shared system prompts, ChunkLlama shows no performance regression versus vLLM, and can achieve comparable throughput to the SOTA PagedAttention kernel.

## Methods

  • The paper proposes Prefix-Aware KV Cache (PAKV), organizing KV cache as a prefix tree of chunked context tokens so matching prompt prefixes can share key/value tensors in memory at runtime.
  • It adds Two-Phase Partition (TPP) for decoding: a chunk-first phase over shared chunks and a sequence-first phase over per-sequence chunks, using online softmax and specialized CUDA kernels.
  • Evaluation includes both self-attention microkernel tests and end-to-end GPT-style serving tests on NVIDIA A100 (80G) with CUDA 11.8; microkernel settings include FP16, head dimension 128, 32 heads, and chunk size 64.

## Limitations

  • The paper states that the shared system prompt must appear at the beginning of the sequence for KV sharing; if developers place it elsewhere, PAKV cannot save memory even when many tokens are common.
  • The paper states that ChunkAttention’s best performance depends on low-level CUDA implementations and is tuned for common configurations such as 128 head dimension and hardware like NVIDIA A100, GeForce RTX 4090, and Intel Xeon CPU; other configurations require case-by-case tuning.
  • Evident from the text: experiments are reported only on A100 (80G) and OpenLlama2 7B / Llama 7B-style settings, so broader generalization to other models and hardware is not directly shown in the paper.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

DataCite · 10.48550/arxiv.2402.15220

Cites

0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Verification

Claims the paper makes

13 of 19 cited sources checked · 6 supported · 0 flagged · accuracy not yet measured

  • Premisereasoning
    § 1 Introduction

    The Kwon et al. proposal has limitations: predefined system prompts are static and inflexible in frequent refreshes for large-scale deployments.

    “The proposal has limitations: i) predefined system prompts are static and inflexible in frequent refreshes for largescale deployments since both application developers and the service provider are involved in the operation loop”
    Quote in the paperSubstantial claim1 of 1 sources checked
    • cites Kwon et al. (2023) source

      The “Shared prefix” section says vLLM reserves physical blocks for predefined shared prefixes, with their KV caches stored in advance by the service provider. It does not say these prefixes are inflexible to refresh at scale or that developers and the provider must both participate in an update loop. (LLM, conf=0.83)

      Experimental — accuracy not yet measured
  • Premisereasoning
    § 1 Introduction

    The Kwon et al. proposal has memory waste in case of long system prompts and low hit rate.

    “ii) there is memory waste in case of long system prompts and low hit rate”
    Quote in the paperSubstantial claim1 of 1 sources checked
    • cites Kwon et al. (2023) source

      The “Shared prefix” section (page unavailable) says vLLM reserves physical blocks for predefined cached prefixes, so the proposal does allocate memory to them. It does not discuss low hit rates or say that long system prompts cause memory waste under that condition; the waste described in §3.1 concerns earlier systems instead. (LLM, conf=0.75)

      Experimental — accuracy not yet measured
  • Findingcitation
    § 1 Introduction

    System prompts in LLM-based applications can be long (more than 1K tokens).

    “we reveal that system prompts can be long (§2.1), providing opportunities for optimizing selfattention”
    • Also stated in: § 1 Introduction · plain
      “gyudoza (2023) shows that system prompts of various LLM-based applications have more than 1K tokens.”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites gyudoza (2023)

      Couldn’t be matched to a paper in Crossref, so it can’t be checked.

  • Findingreasoning
    § Abstract

    ChunkAttention can detect matching prompt prefixes across multiple requests and share their key/value tensors in memory at runtime to improve KV cache memory utilization.

    “we introduce ChunkAttention, a prefix-aware self-attention module that can detect matching prompt prefixes across multiple requests and share their key/value tensors in memory at runtime to improve the memory utilization of KV cache.”
    Quote in the paperSubstantial claim
  • Findingown data
    § Abstract

    ChunkAttention speeds up the self-attention kernel by 3.2-4.8× compared to the state-of-the-art implementation, for system prompt lengths from 1024 to 4096.

    “Experiments show that ChunkAttention can speed up the self-attention kernel by 3.2-4.8 × compared to the state-of-the-art implementation, with the length of the system prompt ranging from 1024 to 4096.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one. 26
  • Probably not a substantial claim. 17
  • It may state two separate claims as one.

    Premisecitation
    § 1 Introduction

    The only related work is Kwon et al. (2023), which reserves memory for key/value tensors of predefined system prompts.

    “To our knowledge, the only related work is a proposal by Kwon et al. (2023), in which the service provider reserves memory for key/value tensors of a set of predefined system prompts from application developers.”
    Quote in the paperSubstantial claim1 of 1 sources checked
    • cites Kwon et al. (2023) source

      The “Shared prefix” section says the LLM service provider can reserve physical blocks for predefined shared prefixes and store their KV cache in advance. It does not identify those prefixes specifically as system prompts supplied by application developers. The sentence’s “only related work” framing is the citing authors’ own assessment, not a claim established by this source. (LLM, conf=0.84)

      Experimental — accuracy not yet measured
  • Probably not a substantial claim. The paper’s conclusions may not depend on it.

    Premisecitation
    § 5 Related Work

    The most relevant work on optimizing the memory utilization of KV cache is PagedAttention in vLLM.

    “The most relevant work on optimizing the memory utilization of KV cache is PagedAttention in vLLM (Kwon et al., 2023).”
    Quote in the paperNot a substantial claim1 of 1 sources checked
    • cites Kwon et al., 2023 source
  • It may state two separate claims as one.

    Premisecitation
    § 5 Related Work

    Only a proposal on service providers to pre-configure shared prompts is mentioned in vLLM, and it is not implemented in vLLM releases up to 0.2.7.

    “However, only a proposal on service providers to pre-configure shared prompts is mentioned, and it is not implemented in vLLM releases (up to 0.2.7).”
    Quote in the paperNot a substantial claim1 of 1 sources checked
    • cites Kwon et al., 2023 source

      The “Shared prefix” section describes how a service provider can predefine shared prefixes and reserve KV-cache blocks for them. Section 6.4 also reports vLLM performance tests using shared prefixes, so the paper does more than merely mention a proposal. Our copy does not discuss vLLM releases through 0.2.7 or establish whether those releases implemented the feature. (LLM, conf=0.87)

      Experimental — accuracy not yet measured
  • It may state two separate claims as one.

    Premisecitation
    § 5 Related Work

    Partition strategies in ChunkAttention are built on online softmax and inspired by FlashAttention.

    “Partition strategies in ChunkAttention are built on online softmax (Milakov and Gimelshein, 2018) and inspired by FlashAttention (Dao et al., 2022; Dao, 2023), which adopted the same algorithm.”
    Quote in the paperNot a substantial claim3 of 3 sources checked
    • cites Milakov and Gimelshein, 2018 source
    • cites Dao et al., 2022 source
    • cites Dao, 2023 source
  • It may state two separate claims as one.

    Findingcitation
    § 1 Introduction

    The self-attention module has low arithmetic intensity and high latency.

    “The selfattention module has low arithmetic intensity (Williams et al., 2009) and high latency.”
    Quote in the paperSubstantial claim1 of 1 sources checked
    • cites Williams et al., 2009 source

      The citation is attached to the claim that self-attention has low arithmetic intensity, not to the separate statement about latency. Section 3, “The Roofline Model,” defines intensity metrics, and Section 6 demonstrates the model on other kernels. Searches for self-attention and transformer-related terms found no passage assessing the arithmetic intensity of a self-attention module. (LLM, conf=0.75)

      Experimental — accuracy not yet measured
  • Probably not a substantial claim. The paper’s conclusions may not depend on it. It may state two separate claims as one.

    Premisecitation
    § 5 Related Work

    FlashAttention-2 altered tiling strategies and additionally doubled the speed.

    “FlashAttention-2 altered tiling strategies and additionally doubled the speed.”
    Quote in the paperNot a substantial claim1 of 1 sources checked
    • cites Dao, 2023 source
  • Probably not a substantial claim. The paper’s conclusions may not depend on it.

    Premisecitation
    § 5 Related Work

    PagedAttention introduces the paging technique in OS to solve the problem of memory waste caused by dynamic and unknown sequence lengths during decoding.

    “It introduces the paging technique in OS to solve the problem of memory waste caused by dynamic and unknown sequence lengths during decoding.”
    Quote in the paperNot a substantial claim1 of 1 sources checked
    • cites Kwon et al., 2023 source
  • Probably not a substantial claim. The paper’s conclusions may not depend on it. It may state two separate claims as one.

    Premisecitation
    § 5 Related Work

    FlashAttention accelerated self-attention by 2-4× while cutting memory operations by 10-20×.

    “FlashAttention thoroughly researched and implemented various tiling techniques, accelerating selfattention by 2-4 × while cutting memory operations by 10-20 × .”
    Quote in the paperNot a substantial claim1 of 1 sources checked
    • cites Dao et al., 2022 source

      Appendix B.5 (p. 22) says FlashAttention is 2–4× faster than standard attention. The Introduction (p. 2) reports up to 9× fewer HBM accesses, not 10–20× fewer memory operations; the up-to-20× figure in §4.3 (p. 10) concerns memory footprint instead. (LLM, conf=0.88)

      Experimental — accuracy not yet measured
  • Probably not a substantial claim. It may be too vague to check against evidence. The paper’s conclusions may not depend on it.

    Premisecitation
    § 7 Limitations

    Fine-tuning is another promising way to infuse domain knowledge into LLMs.

    “In addition to using system prompts, fine-tuning is another promising way to infuse domain knowledge into LLMs (Houlsby et al., 2019; Hu et al., 2023).”
    Quote in the paperNot a substantial claim1 of 2 sources checked
    • cites Houlsby et al., 2019 source

      The Introduction (p. 1) describes fine-tuning a pretrained network for a downstream task and says it often performs better than feature-based transfer. Related Work (p. 8) also reports strong results from fine-tuning language models. These passages support fine-tuning as a promising NLP transfer method, but they do not address infusing domain knowledge into LLMs or compare fine-tuning with system prompts. (LLM, conf=0.83)

      Experimental — accuracy not yet measured
    • cites Hu et al., 2023

      Couldn’t be matched to a paper in Crossref, so it can’t be checked.

  • Probably not a substantial claim. The paper’s conclusions may not depend on it. It may state two separate claims as one.

    Findingown data
    § 1 Introduction

    For a ChatGPT-like online chatbot, the system prompt can be as long as 1766 tokens with only 6 plugins activated, and all requests sent to the chatbot share one single system prompt.

    “For a ChatGPT-like online chatbot, the system prompt can be as long as 1766 tokens with only 6 plugins activated, and all requests sent to the chatbot share one single system prompt (Appendix A).”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper may give no reason why it holds.

    Findingnone given
    § Abstract

    Self-attention is a significant source of inference latency for long sequences in LLMs.

    “Self-attention is an essential component of large language models (LLM) but a significant source of inference latency for long sequences.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper may give no reason why it holds.

    Findingnone given
    § 1 Introduction

    Using FP16, the KV cache for each token in GPT3(175B) requires 4.5MB of memory.

    “For instance, using FP16, the KV cache for each token in GPT3(175B) requires 4.5MB of memory.”
    Quote in the paperNot a substantial claim
  • It may state two separate claims as one.

    Findingown data
    § 1 Introduction

    ChunkAttention can be significantly faster as the length of shared system prompts grows and has no performance degradation without shared system prompts, compared to existing highly optimized implementations.

    “Our experiments show that ChunkAttention can be significantly faster as the length of shared system prompts grows and has no performance degradation without shared system prompts, compared to existing highly optimized implementations.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Findingown data
    § 4.1 Microkernel Evaluation

    As decoding proceeds, sequences start to diverge, and the performance gain of ChunkAttn gradually decreases.

    “As the decoding proceeds, sequences start to diverge, and the performance gain of ChunkAttn gradually decreases, as shown in Figure 3.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper may give no reason why it holds. It may state two separate claims as one.

    Findingnone given
    § 2.1 Shared System Prompt

    The system prompt is shared between multiple requests and can be very long.

    “The system prompt is shared between multiple requests and can be very long. This can be observed in various LLM-based applications, from online chatbots to offline experiments.”
    Quote in the paperNot a substantial claim
  • It may state two separate claims as one.

    Findingown data
    § 4.1 Microkernel Evaluation

    Naive is 6.6× and 2.1× slower than ChunkAttn and PagedAttn*, respectively, at n_s=4096.

    “The Naive is 6.6 × and 2.1 × slower than ChunkAttn and PagedAttn*, respectively ( n s =4096).”
    Quote in the paperSubstantial claim
    • The authors state this as an assumption: Iteration-based batching is enabled to form batches for ChunkAttention's kernel to run efficiently.
  • It may state two separate claims as one.

    Interpretationreasoning
    § 3.2 Two-phase Partition (TPP)

    Without the partial attention results generated by the chunk-first phase, the sequence-first phase needs to load shared chunks from RAM b times, which adds significant MOPs.

    “However, without the partial attention results generated by the chunk-first phase, it needs to load shared chunks from RAM b times, which adds significant MOPs.”
    Quote in the paperSubstantial claim
    • The authors state this as an assumption: Iteration-based batching is enabled to form batches for ChunkAttention's kernel to run efficiently.
  • Probably not a substantial claim. The paper’s conclusions may not depend on it. The paper may give no reason why it holds. It may state two separate claims as one.

    Interpretationnone given
    § 3.3 Further Optimizations

    On CPU devices, the overhead of serializing is insignificant, and reduction can be implemented using spin locks.

    “However, on CPU devices, the overhead of serializing is insignificant, and reduction can be implemented using spin locks.”
    Quote in the paperNot a substantial claim
    • The authors state this as an assumption: Iteration-based batching is enabled to form batches for ChunkAttention's kernel to run efficiently.
  • It may state two separate claims as one.

    Interpretationreasoning
    § 7 Limitations

    When application developers do not put the system prompt at the beginning, KV caches of the entire sequences are different, and PAKV cannot save memory in this case, although they have a large number of tokens in common.

    “As a result, when application developers do not put the system prompt at the beginning for performance concerns after evaluations or unintentional mistakes, KV caches of the entire sequences are different, and PAKV cannot save memory in this case, although they have a large number of tokens in common.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 3.1 Prefix Aware KV Cache (PAKV)

    In memory-limited inference scenarios, sharing common prefixes helps increase batch size and thus improve throughput.

    “In memory-limited inference scenarios, this helps increase the batch size and thus improve throughput.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim.

    Interpretationreasoning
    § 3.1 Prefix Aware KV Cache (PAKV)

    The KV cache should be made prefix-aware by organizing the KV cache of all sequences under decoding into a prefix tree.

    “Given the property, we argue that the KV cache should be made prefix-aware, which is to organize the KV cache of all sequences under decoding into a prefix tree.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 1 Introduction

    ChunkAttention's KV cache only stores key/value tensors of sequences currently in decoding and has zero memory waste.

    “The KV cache only stores key/value tensors of sequences currently in decoding and has zero memory waste.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Study design and conductown data
    § 6 Conclusion

    ChunkAttention was evaluated in various configurations and at different levels, proving its feasibility and that side effects can be managed.

    “We evaluate ChunkAttention in various configurations and at different levels, proving its feasibility and the side effects can be managed.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 3.1 Prefix Aware KV Cache (PAKV)

    When multiple sequences share common prefix tokens, key/value tensors are the same and thus can be shared in memory.

    “When multiple sequences share common prefix tokens, key/value tensors are the same and thus can be shared in memory.”
    Quote in the paperNot a substantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 1 Introduction

    KV cache restricts batch size and system throughput.

    “KV cache additionally restricts the batch size and system throughput.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 3.3 Further Optimizations

    ChunkAttention manages context generation overhead by latency hiding and lazy context copy.

    “ChunkAttention manages the overhead in two ways: i) latency hiding. The context generation step on CPU can be overlapped with other kernels on GPU prior to self-attention. ii) lazy context copy.”
    Quote in the paperNot a substantial claim
    • The authors state this as an assumption: Iteration-based batching is enabled to form batches for ChunkAttention's kernel to run efficiently.
  • Probably not a substantial claim. The paper may give no reason why it holds. It may state two separate claims as one.

    Premisenone given
    § 5 Related Work

    FlashAttention is inflexible regarding noncontiguous memory or variable sequence lengths, making it more suitable for model training than inference.

    “However, FlashAttention is inflexible regarding noncontiguous memory or variable sequence lengths, making it more suitable for model training than inference.”
    Quote in the paperNot a substantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 5 Related Work

    ChunkAttention handles variable sequence lengths during decoding and batches attention operations of several sequences to reduce memory operations.

    “ChunkAttention handles variable sequence lengths during decoding and batches attention operations of several sequences to reduce memory operations.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 4.1 Microkernel Evaluation

    The improvement of ChunkAttn over PagedAttn* is lower because PagedAttn* benefits from physically shared KV cache memory, and only TPP makes a difference.

    “The improvement of ChunkAttn over PagedAttn* is lower since PagedAttn* benefits from physically shared KV cache memory, and only TPP makes a difference here.”
    Quote in the paperSubstantial claim
  • It may state two separate claims as one.

    Interpretationreasoning
    § 1 Introduction

    ChunkAttention's KV cache is a prefix tree built with chunked context tokens and key/value tensors, making it prefix-aware and able to dynamically detect and remove redundancy at runtime without human involvement.

    “KV cache in ChunkAttention is a prefix tree built with chunked context tokens and key/value tensors. Thus, the KV cache is prefixaware and can dynamically detect and remove redundancy at runtime without human involvement.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim. The paper’s conclusions may not depend on it.

    Interpretationreasoning
    § 7 Limitations

    It is not cost-efficient for each application to fine-tune models and deploy private instances.

    “It is not cost-efficient for each application to finetune models and deploy private instances.”
    Quote in the paperNot a substantial claim
    • Rests on an assumption: LLMs are typically pre-trained and centrally hosted for multiple applications to share due to high training and deployment cost.
  • Probably not a substantial claim. The paper may give no reason why it holds. It may state two separate claims as one.

    Interpretationnone given
    § 5 Related Work

    ChunkAttention uses a prefix tree to manage memory and aims to discover redundancy in KV cache across user requests at runtime automatically.

    “Our solution, which differs from the paging one, uses the prefix tree to manage memory and aims to discover redundancy in KV cache across user requests at runtime automatically.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper’s conclusions may not depend on it. The paper may give no reason why it holds.

    Premisenone given
    § 5 Related Work

    There is little gain when the query token count is always one during decoding.

    “There is little gain when the query token count is always one during decoding.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper’s conclusions may not depend on it. It may state two separate claims as one.

    Study design and conductreasoning
    § 4.1 Microkernel Evaluation

    PagedAttn* simulates the KV cache sharing scenario and helps observe the performance of PagedAttn's CUDA kernel.

    “However, its paging design enables us to manually create a fixed page table, mapping virtually non-shared memory to the same physical memory. It simulates the KV cache sharing scenario and helps us observe the performance of PagedAttn's CUDA kernel, which is denoted as PagedAttn*.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim. The paper may give no reason why it holds.

    Interpretationnone given
    § 3.1 Prefix Aware KV Cache (PAKV)

    A key property of the prefix tree is that sequences covered by each chunk in the prefix tree are contiguous in the sequence index dimension.

    “A key property of the prefix tree is that sequences covered by each chunk in the prefix tree are contiguous in the sequence index dimension.”
    Quote in the paperNot a substantial claim
35 of 35
  • The authors state this as an assumption.

    Iteration-based batching is enabled to form batches for ChunkAttention's kernel to run efficiently.

    “The ChunkAttention in this paper assumes that iteration-based batching is enabled to form batches for its kernel to run efficiently.”
  • The paper relies on this without support.

    LLMs are typically pre-trained and centrally hosted for multiple applications to share due to high training and deployment cost.

    “Due to the high training and deployment cost, LLMs are typically pre-trained and centrally hosted for multiple applications to share.”
2 of 2

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • vLLM and ChunkedAttention explore some simple reuse cases (e.g., system prompt sharing) but do not cover multi-level tree-structured sharing or LRU caching.

    “vLLM [23] and ChunkedAttention [58] explore some simple reuse cases (e.g., system prompt sharing) but do not cover multi-level tree-structured sharing or LRU caching.”

    From SGLang: Efficient Execution of Structured Language Model Programs (2023) · cites this paper as [58]

    ChunkAttention does study shared system prompts, but the claim that it does not cover multi-level tree-structured sharing conflicts with §3.1, which describes shared KV-cache chunks in a prefix tree with parent–child relationships. The Introduction also says the cache stores tensors only for sequences currently decoding; this is consistent with not retaining them through LRU caching, but does not rescue the tree-sharing claim. (LLM, conf=0.87)

    Checked by a reasoning model — may be wrong.

    Experimental — accuracy not yet measured
  • HydraGen, FlashInfer, and ChunkedAttention focus on CUDA kernel optimizations and do not include the concept of an LRU cache.

    “HydraGen [18], FlashInfer [59], and ChunkedAttention [58] focus on CUDA kernel optimizations and do not include the concept of an LRU cache.”

    From SGLang: Efficient Execution of Structured Language Model Programs (2023) · cites this paper as [58]

    The Abstract and Conclusion describe an optimized self-attention kernel, supporting the kernel-optimization part of the claim. But they also present prefix-aware KV-cache management as a central contribution, so describing ChunkAttention as focused on kernel optimization alone is incomplete. Section 3.1 describes a pool allocator with used and free chunk lists, not an LRU policy; the copy does not expressly state that LRU is absent. (LLM, conf=0.82)

    Experimental — accuracy not yet measured
  • A significant portion of input prefixes in LLM inference are reusable.

    “Opportunities for Reusing KV Caches: A significant portion of input prefixes in LLM inference are reusable [66], presenting an opportunity to reduce redundant computation through KV cache reuse.”
  • In ChatGPT-like applications, a single system prompt can exceed 1,700 tokens when multiple plugins are activated, and all requests share this same prompt.

    “In ChatGPTlike applications, a single system prompt can exceed 1,700 tokens when multiple plugins are activated [66], and all requests share this same prompt.”
  • Many prompts in LLM inference share common prefixes.

    “Since many prompts share common prefixes, prefix sharing has been proposed to reuse shared prefix KV caches during inference.”

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

  • ChunkAttention said to have zero memory waste, but elsewhere memory loss is bounded not zero

    Minor

    One passage says the KV cache has zero memory waste, while another states that some memory space for alignment is unused and the memory loss is bounded by a nonzero expression. These claims cannot both be true as written.

1 of 1

Citations

Cites

The publisher hasn’t deposited this paper’s reference list.

Research works printed but not linked: 1

  1. Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. 2019. Language models are unsupervised multitask learners. OpenAI blog , 1(8):9.

Cited by

Sorted by citations

Metadata

Authors
Ye, L., Tao, Z., Huang, Y., & Li, Y.
Affiliations
Microsoft (United States) (US)
Venue
arXiv
Publisher
arXiv
Published
2024
Pages
11608-11620
ISSN
0736-587X (ISSN-L)
OpenAlex
W4402671542
Cited by
34
References
0
Citations by year
2026: 21 · 2025: 11 · 2024: 2
FWCI
10.02
Type
conference paper
Language
en
Open access
Locations
  • Proceedings of the conference - Association for Computational Linguistics. Meeting (publishedVersion, cc-by)
  • arXiv (Cornell University) (submittedVersion, cc-by)
  • arXiv (Cornell University) (cc-by)
Topics
  • Physical Sciences › Computer Science › Artificial Intelligence › Topic Modeling
  • Physical Sciences › Computer Science › Artificial Intelligence › Natural Language Processing Techniques
  • Physical Sciences › Computer Science › Hardware and Architecture › Parallel Computing and Optimization Techniques
Keywords
key-value cache, prefix tree, data locality, inference latency, memory efficiency, large language models

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

A language model drew the claims from the paper on 2026-10-01, under version 3 of the claim definition (model deepseek-v4.1-flash, prompt extract-v3).

A second model labelled each claim on 2026-10-01 (model jev-1.13-20260917, definition revision gold-guideline-v0.2).

Cited sources are checked with a language model’s help (gpt-6-sol).

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review