中文
相关论文

相关论文: FlashPrefill: Instantaneous Pattern Discovery and …

200 篇论文

The integration of long-context capabilities with visual understanding unlocks unprecedented potential for Vision Language Models (VLMs). However, the quadratic attention complexity during the pre-filling phase remains a significant…

计算机视觉与模式识别 · 计算机科学 2025-05-26 Yucheng Li , Huiqiang Jiang , Chengruidong Zhang , Qianhui Wu , Xufang Luo , Surin Ahn , Amir H. Abdi , Dongsheng Li , Jianfeng Gao , Yuqing Yang , Lili Qiu

An efficient attention implementation is essential for large models due to its quadratic time complexity. Fortunately, attention commonly exhibits sparsity, i.e., many values in the attention map are near zero, allowing for the omission of…

机器学习 · 计算机科学 2025-11-20 Jintao Zhang , Chendong Xiang , Haofeng Huang , Jia Wei , Haocheng Xi , Jun Zhu , Jianfei Chen

Transformers are slow and memory-hungry on long sequences, since the time and memory complexity of self-attention are quadratic in sequence length. Approximate attention methods have attempted to address this problem by trading off model…

机器学习 · 计算机科学 2022-06-24 Tri Dao , Daniel Y. Fu , Stefano Ermon , Atri Rudra , Christopher Ré

The ever-increasing sizes of large language models necessitate distributed solutions for fast inference that exploit multi-dimensional parallelism, where computational loads are split across various accelerators such as GPU clusters.…

人工智能 · 计算机科学 2024-12-12 Qingyuan Li , Bo Zhang , Liang Ye , Yifan Zhang , Wei Wu , Yerui Sun , Lin Ma , Yuchen Xie

Long-Context Transformer Models (LCTMs) are vital for real-world applications but suffer high computational costs due to attention's quadratic complexity. Block-sparse attention mitigates this by focusing computation on critical regions,…

计算与语言 · 计算机科学 2025-03-21 Ruyi Xu , Guangxuan Xiao , Haofeng Huang , Junxian Guo , Song Han

While Transformers have shown remarkable success in natural language processing, their attention mechanism's large memory requirements have limited their ability to handle longer contexts. Prior approaches, such as recurrent memory or…

计算与语言 · 计算机科学 2023-11-21 Amirkeivan Mohtashami , Martin Jaggi

Long-context inference in decoder-only language models is costly because long prompts are processed during Prefill, cached at every layer, and repeatedly attended to during autoregressive Decode. We introduce \emph{Shallow Prefill, dEEp…

人工智能 · 计算机科学 2026-05-08 Jungsuk Oh , Hyeseo Jeon , Hyunjune Ji , Kyongmin Kong , Jay-Yoon Lee

The evolution of large language models (LLMs) towards applications with ultra-long contexts faces challenges posed by the high computational and memory costs of the Transformer architecture. While existing sparse and linear attention…

Diffusion Language Models (DLMs) enable globally coherent, bidirectional, and controllable text generation, offering advantages over traditional autoregressive LLMs, while scaling to ultra-long sequences remains costly. Many existing…

计算机视觉与模式识别 · 计算机科学 2026-05-20 Wenhu Zhang , Yiming Wu , Huanyu Wang , Yaoyang Liu , Huanzhang Dou , Senqiao Yang , Sitong Wu , Hanbin Zhao , Jiaya Jia

The quadratic computational complexity of standard attention mechanisms presents a severe scalability bottleneck for LLMs in long-context scenarios. While hybrid attention mechanisms combining Full Attention (FA) and Sparse Attention (SA)…

机器学习 · 计算机科学 2026-04-10 Quantong Qiu , Zhiyi Hong , Yi Yang , Haitian Wang , Kebin Liu , Qingqing Dang , Juntao Li , Min Zhang

Long-context LLMs increasingly rely on extended, reusable prefill prompts for agents and domain Q&A, pushing attention and KV-cache to become the dominant decode-time bottlenecks. While sparse attention reduces computation and transfer…

机器学习 · 计算机科学 2026-04-13 Chuxu Song , Zhencan Peng , Jiuqi Wei , Chuanhui Yang

Diffusion Large Language Models (dLLMs) deliver strong long-context processing capability in a non-autoregressive decoding paradigm. However, the considerable computational cost of bidirectional full attention limits the inference…

计算与语言 · 计算机科学 2026-02-03 Lingkun Long , Yushi Huang , Shihao Bai , Ruihao Gong , Jun Zhang , Ao Zhou , Jianlei Yang

Self-attention based Transformer has demonstrated the state-of-the-art performances in a number of natural language processing tasks. Self-attention is able to model long-term dependencies, but it may suffer from the extraction of…

计算与语言 · 计算机科学 2019-12-30 Guangxiang Zhao , Junyang Lin , Zhiyuan Zhang , Xuancheng Ren , Qi Su , Xu Sun

The computational challenges of Large Language Model (LLM) inference remain a significant barrier to their widespread deployment, especially as prompt lengths continue to increase. Due to the quadratic complexity of the attention…

Effective attention modules have played a crucial role in the success of Transformer-based large language models (LLMs), but the quadratic time and memory complexities of these attention modules also pose a challenge when processing long…

分布式、并行与集群计算 · 计算机科学 2024-06-07 Ao Sun , Weilin Zhao , Xu Han , Cheng Yang , Zhiyuan Liu , Chuan Shi , Maosong Sun

Accommodating long sequences efficiently in autoregressive Transformers, especially within an extended context window, poses significant challenges due to the quadratic computational complexity and substantial KV memory requirements…

计算与语言 · 计算机科学 2024-06-25 Chao Lou , Zixia Jia , Zilong Zheng , Kewei Tu

Deploying Large Language Models (LLMs) on edge devices remains challenging due to their quadratically increasing computations with the sequence length. Existing studies for dynamic attention pruning are designed for hardware with massively…

人工智能 · 计算机科学 2025-07-29 Jiawen Qi , Chang Gao , Zhaochun Ren , Qinyu Chen

The quadratic computational complexity of the standard attention mechanism constitutes a fundamental bottleneck for large language models in long-context inference. While existing KV cache compression methods alleviate memory pressure, they…

计算与语言 · 计算机科学 2026-05-06 Jinyu Guo , Zhihan Zhang , Jiehui Xie , Md. Tamim Iqbal , Dongshen Han , Lik-Hang Lee , Sung-Ho Bae , Jie Zou , Yang Yang , Chaoning Zhang

Transformers have demonstrated great success in numerous domains including natural language processing and bioinformatics. This success stems from the use of the attention mechanism by these models in order to represent and propagate…

机器学习 · 计算机科学 2025-02-10 Nathaniel Tomczak , Sanmukh Kuppannagari

Effectively processing long contexts is a critical challenge for language models. While standard Transformers are limited by quadratic complexity and poor length extrapolation, alternative architectures like sliding window attention and…

计算与语言 · 计算机科学 2026-05-01 Jiaqi Leng , Xiang Hu , Junxiong Wang , Jianguo Li , Wei Wu , Yucheng Lu