English
Related papers

Related papers: BitDecoding: Unlocking Tensor Cores for Long-Conte…

200 papers

With the growing demand for long-context LLMs across a wide range of applications, the key-value (KV) cache has become a critical bottleneck for both latency and memory usage. Recently, KV-cache offloading has emerged as a promising…

Machine Learning · Computer Science 2026-05-18 Andrey Bocharnikov , Ivan Ermakov , Denis Kuznedelev , Vyacheslav Zhdanovskiy , Yegor Yershov

The linear memory growth of the KV cache poses a significant bottleneck for LLM inference in long-context tasks. Existing static compression methods often fail to preserve globally important information. Although recent dynamic retrieval…

Computation and Language · Computer Science 2026-04-21 Zhiyuan Shi , Qibo Qiu , Feng Xue , Zhonglin Jiang , Li Yu , Jian Jiang , Xiaofei He , Wenxiao Wang

The rapid expansion of context window sizes in Large Language Models~(LLMs) has enabled them to tackle increasingly complex tasks involving lengthy documents. However, this progress comes at the cost of a substantial increase in memory…

Computation and Language · Computer Science 2025-08-05 Da Ma , Lu Chen , Situo Zhang , Yuxun Miao , Su Zhu , Zhi Chen , Hongshen Xu , Hanqi Li , Shuai Fan , Lei Pan , Kai Yu

The Key-Value (KV) cache of Large Language Models (LLMs) is prefix-based, making it highly inefficient for processing contexts retrieved in arbitrary order. Position-Independent Caching (PIC) has been proposed to enable KV reuse without…

Machine Learning · Computer Science 2026-02-03 Shiju Zhao , Junhao Hu , Jiaqi Zheng , Guihai Chen

Large language models (LLMs) face low hardware efficiency during decoding, especially for long-context reasoning tasks. This paper introduces Step-3, a 321B-parameter VLM with hardware-aware model-system co-design optimized for minimizing…

Machine Learning · Computer Science 2025-07-28 StepFun , : , Bin Wang , Bojun Wang , Changyi Wan , Guanzhe Huang , Hanpeng Hu , Haonan Jia , Hao Nie , Mingliang Li , Nuo Chen , Siyu Chen , Song Yuan , Wuxun Xie , Xiaoniu Song , Xing Chen , Xingping Yang , Xuelin Zhang , Yanbo Yu , Yaoyu Wang , Yibo Zhu , Yimin Jiang , Yu Zhou , Yuanwei Lu , Houyi Li , Jingcheng Hu , Ka Man Lo , Ailin Huang , Binxing Jiao , Bo Li , Boyu Chen , Changxin Miao , Chang Lou , Chen Hu , Chen Xu , Chenfeng Yu , Chengyuan Yao , Daokuan Lv , Dapeng Shi , Deshan Sun , Ding Huang , Dingyuan Hu , Dongqing Pang , Enle Liu , Fajie Zhang , Fanqi Wan , Gulin Yan , Han Zhang , Han Zhou , Hanghao Wu , Hangyu Guo , Hanqi Chen , Hanshan Zhang , Hao Wu , Haocheng Zhang , Haolong Yan , Haoran Lv , Haoran Wei , Hebin Zhou , Heng Wang , Heng Wang , Hongxin Li , Hongyu Zhou , Hongyuan Wang , Huiyong Guo , Jia Wang , Jiahao Gong , Jialing Xie , Jian Zhou , Jianjian Sun , Jiaoren Wu , Jiaran Zhang , Jiayu Liu , Jie Cheng , Jie Luo , Jie Yan , Jie Yang , Jieyi Hou , Jinguang Zhang , Jinlan Cao , Jisheng Yin , Junfeng Liu , Junhao Huang , Junzhe Lin , Kaijun Tan , Kaixiang Li , Kang An , Kangheng Lin , Kenkun Liu , Lei Yang , Liang Zhao , Liangyu Chen , Lieyu Shi , Liguo Tan , Lin Lin , Lin Zhang , Lina Chen , Liwen Huang , Liying Shi , Longlong Gu , Mei Chen , Mengqiang Ren , Ming Li , Mingzhe Chen , Na Wang , Nan Wu , Qi Han , Qian Zhao , Qiang Zhang , Qianni Liu , Qiaohui Chen , Qiling Wu , Qinglin He , Qinyuan Tan , Qiufeng Wang , Qiuping Wu , Qiuyan Liang , Quan Sun , Rui Li , Ruihang Miao , Ruosi Wan , Ruyan Guo , Shangwu Zhong , Shaoliang Pang , Shengjie Fan , Shijie Shang , Shilei Jiang , Shiliang Yang , Shiming Hao , Shuli Gao , Siming Huang , Siqi Liu , Tiancheng Cao , Tianhao Cheng , Tianhao Peng , Wang You , Wei Ji , Wen Sun , Wenjin Deng , Wenqing He , Wenzhen Zheng , Xi Chen , Xiangwen Kong , Xianzhen Luo , Xiaobo Yang , Xiaojia Liu , Xiaoxiao Ren , Xin Han , Xin Li , Xin Wu , Xu Zhao , Yanan Wei , Yang Li , Yangguang Li , Yangshijie Xu , Yanming Xu , Yaqiang Shi , Yeqing Shen , Yi Yang , Yifei Yang , Yifeng Gong , Yihan Chen , Yijing Yang , Yinmin Zhang , Yizhuang Zhou , Yuanhao Ding , Yuantao Fan , Yuanzhen Yang , Yuchu Luo , Yue Peng , Yufan Lu , Yuhang Deng , Yuhe Yin , Yujie Liu , Yukun Chen , Yuling Zhao , Yun Mou , Yunlong Li , Yunzhou Ju , Yusheng Li , Yuxiang Yang , Yuxiang Zhang , Yuyang Chen , Zejia Weng , Zhe Xie , Zheng Ge , Zheng Gong , Zhenyi Lu , Zhewei Huang , Zhichao Chang , Zhiguo Huang , Zhirui Wang , Zidong Yang , Zili Wang , Ziqi Wang , Zixin Zhang , Binxing Jiao , Daxin Jiang , Heung-Yeung Shum , Xiangyu Zhang

Vision-language models (VLMs) show remarkable performance in multimodal tasks. However, excessively long multimodal inputs lead to oversized Key-Value (KV) caches, resulting in significant memory consumption and I/O bottlenecks. Previous KV…

Computation and Language · Computer Science 2025-01-28 Zunhai Su , Wang Shen , Linge Li , Zhe Chen , Hanyu Wei , Huangqi Yu , Kehong Yuan

Deploying long-context large language models (LLMs) is essential but poses significant computational and memory challenges. Caching all Key and Value (KV) states across all attention heads consumes substantial memory. Existing KV cache…

Computation and Language · Computer Science 2024-10-15 Guangxuan Xiao , Jiaming Tang , Jingwei Zuo , Junxian Guo , Shang Yang , Haotian Tang , Yao Fu , Song Han

Generative large language models (LLMs) have revolutionized multiple domains. Modern LLMs predominantly rely on an autoregressive decoding strategy, which generates output tokens sequentially and employs a key-value cache (KV cache) to…

Cryptography and Security · Computer Science 2026-02-13 Ye Yu , Yifan Zhou , Yi Chen , Pedro Soto , Wenjie Xiong , Meng Li

Autoregressive large language models (LLMs) have made remarkable progress in various natural language generation tasks. However, they incur high computation cost and latency resulting from the autoregressive token-by-token generation. To…

Computation and Language · Computer Science 2023-07-07 Luciano Del Corro , Allie Del Giorno , Sahaj Agarwal , Bin Yu , Ahmed Awadallah , Subhabrata Mukherjee

The deployment of large language models (LLMs) is often hindered by the extensive memory requirements of the Key-Value (KV) cache, especially as context lengths increase. Existing approaches to reduce the KV cache size involve either…

Computation and Language · Computer Science 2024-11-05 Alessio Devoto , Yu Zhao , Simone Scardapane , Pasquale Minervini

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

Recent large vision-language models (LVLMs) demonstrate remarkable capabilities in processing extended multi-modal sequences, yet the resulting key-value (KV) cache expansion creates a critical memory bottleneck that fundamentally limits…

Computer Vision and Pattern Recognition · Computer Science 2026-01-29 Xuyang Liu , Xiyan Gui , Yuchao Zhang , Linfeng Zhang

Edge acceleration for large language models is crucial for their widespread application; however, achieving fast attention inference and efficient decoding on resource-constrained edge accelerators remains challenging. This paper presents…

Hardware Architecture · Computer Science 2026-01-19 Junming Zhang , Qinyan Zhang , Huajun Sun , Feiyang Gao , Sheng Hu , Rui Nie , Xiangshui Miao

Multimodal Large Language Models face severe challenges in computational efficiency and memory consumption due to the substantial expansion of the visual KV cache when processing long visual contexts. Existing KV cache compression methods…

Machine Learning · Computer Science 2026-05-07 Sihao Liu , YuFan Xiong , Zhonghua Jiang , Zhaode Wang , chengfei lv Shengyu Zhang

The Key-Value (KV) cache is integral to efficient autoregressive inference in large language models (LLMs), yet its unbounded growth in stateful multi-turn scenarios presents major challenges. This paper examines the interplay between KV…

Machine Learning · Computer Science 2025-11-10 Pratik Poudel

KV-cache memory is a major bottleneck in real-world LLM serving, where systems must simultaneously support latency-sensitive small-batch requests and high-throughput concurrent workloads. Although many KV-cache compression methods improve…

Large language models (LLMs) are increasingly applied in long-context scenarios such as multi-turn conversations. However, long contexts pose significant challenges for inference efficiency, including high memory overhead from Key-Value…

Computation and Language · Computer Science 2025-12-18 Kuan Lu , Shuhang Lin , Sai Wu , Yichen Yao , Junhan Yang , Huan Li , Wei Chu , Xu Yinghui , Yuan Qi , Gang Chen

The efficiency of Large Language Model~(LLM) inference is often constrained by substantial memory bandwidth and capacity demands. Existing techniques, such as pruning, quantization, and mixture of experts/depth, reduce memory capacity…

Hardware Architecture · Computer Science 2025-04-23 Rui Xie , Asad Ul Haq , Linsen Ma , Yunhua Fang , Zirak Burzin Engineer , Liu Liu , Tong Zhang

Quantization is essential for efficient large language model (LLM) inference, yet the dequantization step-converting low-bit weights back to high-precision for matrix multiplication has become a critical bottleneck on modern AI…

Machine Learning · Statistics 2026-05-15 Lingchao Zheng , Yuwei Fan , Jun Li , Chengqiu Hu , Qichen Liao , Junyi Fan , Rui Shi , Fangzheng Miao

Fault-tolerant quantum computing demands decoders that are fast, accurate, and adaptable to circuit structure and realistic noise. While machine learning (ML) decoders have demonstrated impressive performance for quantum memory, their use…

Quantum Physics · Physics 2025-09-16 J. Pablo Bonilla Ataides , Andi Gu , Susanne F. Yelin , Mikhail D. Lukin