English
Related papers

Related papers: Scaling Linear Attention with Sparse State Expansi…

200 papers

Large language models have shown strong reasoning capabilities through chain-structured methods such as Chain-of-Thought. Recent studies optimize thought structures by generating parallel or tree-like structures, switching between long and…

Computation and Language · Computer Science 2025-10-30 Jinghan Zhang , Fengran Mo , Tharindu Cyril Weerasooriya , Xinyue Ye , Dongjie Wang , Yanjie Fu , Kunpeng Liu

In recent years, Large Language Models (LLMs) have demonstrated remarkable capabilities across a wide array of text-centric tasks. However, their `large' scale introduces significant computational and storage challenges, particularly in…

Computation and Language · Computer Science 2024-07-03 Chaoran Zhang , Lixin Zou , Dan Luo , Min Tang , Xiangyang Luo , Zihao Li , Chenliang Li

In recent years, the success of large language models (LLMs) has driven the exploration of scaling laws in recommender systems. However, models that demonstrate scaling laws are actually challenging to deploy in industrial settings for…

Information Retrieval · Computer Science 2026-01-27 Weijiang Lai , Beihong Jin , Di Zhang , Siru Chen , Jiongyan Zhang , Yuhang Gou , Jian Dong , Xingxing Wang

Deep learning-based speech enhancement (SE) methods often face significant computational challenges when needing to meet low-latency requirements because of the increased number of frames to be processed. This paper introduces the SlowFast…

Audio and Speech Processing · Electrical Eng. & Systems 2025-01-07 Longbiao Cheng , Ashutosh Pandey , Buye Xu , Tobi Delbruck , Vamsi Krishna Ithapu , Shih-Chii Liu

Linear attention offers a computationally efficient yet expressive alternative to softmax attention. However, recent empirical results indicate that the hidden state of trained linear attention models often exhibits a low-rank structure,…

Machine Learning · Computer Science 2026-02-13 Philipp Nazari , T. Konstantin Rusch

Sparsity-aware training is an effective approach for transforming large language models (LLMs) into hardware-friendly sparse patterns, thereby reducing latency and memory consumption during inference. In this paper, we propose Continuous…

Machine Learning · Computer Science 2025-10-01 Weiyu Huang , Yuezhou Hu , Jun Zhu , Jianfei Chen

Various forms of sparse attention have been explored to mitigate the quadratic computational and memory cost of the attention mechanism in transformers. We study sparse transformers not through a lens of efficiency but rather in terms of…

Machine Learning · Computer Science 2025-06-19 Parikshit Ram , Kenneth L. Clarkson , Tim Klinger , Shashanka Ubaru , Alexander G. Gray

Sparse autoencoders (SAEs) are a promising unsupervised approach for identifying causally relevant and interpretable linear features in a language model's (LM) activations. To be useful for downstream tasks, SAEs need to decompose LM…

Machine Learning · Computer Science 2024-08-02 Senthooran Rajamanoharan , Tom Lieberum , Nicolas Sonnerat , Arthur Conmy , Vikrant Varma , János Kramár , Neel Nanda

Efficient state space models (SSMs), such as linear recurrent neural networks and linear attention variants, offer computational advantages over Transformers but struggle with tasks requiring long-range in-context retrieval-like text…

Computation and Language · Computer Science 2025-02-25 Sam Blouir , Jimmy T. H. Smith , Antonios Anastasopoulos , Amarda Shehu

Long-context inference in large language models is bottlenecked by the quadratic cost of full attention. Existing efficient alternatives often rely either on native sparse training or on heuristic token eviction, creating an undesirable…

Computation and Language · Computer Science 2026-05-19 Yanke Zhou , Yiduo Li , Hanlin Tang , Maohua Li , Kan Liu , Lan Tao , Lin Qu , Yuan Yao , Xiaoxing Ma

Sparse autoencoders (SAEs) are a popular method for interpreting concepts represented in large language model (LLM) activations. However, there is a lack of evidence regarding the validity of their interpretations due to the lack of a…

Machine Learning · Computer Science 2025-02-25 Subhash Kantamneni , Joshua Engels , Senthooran Rajamanoharan , Max Tegmark , Neel Nanda

Large Language Models (LLMs) have demonstrated success across many benchmarks. However, they still exhibit limitations in long-context scenarios, primarily due to their short effective context length, quadratic computational complexity, and…

Computation and Language · Computer Science 2025-09-26 Manlai Liang , Mandi Liu , Jiangzhou Ji , Huaijun Li , Haobo Yang , Yaohan He , Jinlong Li

Transformer architecture has been very successful long runner in the field of Deep Learning (DL) and Large Language Models (LLM) because of its powerful attention-based learning and parallel-natured architecture. As the models grow gigantic…

Machine Learning · Computer Science 2026-01-21 Phani Kumar , Nyshadham , Jyothendra Varma , Polisetty V R K , Aditya Rathore

A key advantage of Recurrent Neural Networks (RNNs) over Transformers is their linear computational and space complexity enables faster training and inference for long sequences. However, RNNs are fundamentally unable to randomly access…

Computation and Language · Computer Science 2025-11-04 Xiang Hu , Jiaqi Leng , Jun Zhao , Kewei Tu , Wei Wu

Effectively learning from sequential data is a longstanding goal of Artificial Intelligence, especially in the case of long sequences. From the dawn of Machine Learning, several researchers have pursued algorithms and architectures capable…

Machine Learning · Computer Science 2025-08-19 Matteo Tiezzi , Michele Casoni , Alessandro Betti , Marco Gori , Stefano Melacci

Large language models (LLMs) have demonstrated remarkable proficiency in a wide range of natural language processing applications. However, the high energy and latency overhead induced by the KV cache limits the edge deployment, especially…

Performance · Computer Science 2025-12-09 Zizhuo Fu , Xiaotian Guo , Wenxuan Zeng , Shuzhang Zhong , Yadong Zhang , Peiyu Chen , Runsheng Wang , Le Ye , Meng Li

Large reasoning models achieve strong performance through test-time scaling, but this incurs substantial computational overhead due to long decoding from short prompts. While sparse attention can reduce latency and memory usage, existing…

Computation and Language · Computer Science 2026-04-29 Lijie Yang , Zhihao Zhang , Arti Jain , Shijie Cao , Baihong Yuan , Yiwei Chen , Zhihao Jia , Ravi Netravali

State-based sequence models like RWKV-7 offer a compelling alternative to Transformer architectures, achieving linear complexity while demonstrating greater expressive power in short-context scenarios and enabling state tracking beyond the…

Machine Learning · Computer Science 2025-04-14 Liu Xiao , Li Zhiyuan , Lin Yueyu

Large language models (LLMs) enhanced with retrieval-augmented generation (RAG) have introduced a new paradigm for web search. However, the limited context awareness of LLMs degrades their performance on RAG tasks. Existing methods to…

Computation and Language · Computer Science 2024-10-08 Tao Tan , Yining Qian , Ang Lv , Hongzhan Lin , Songhao Wu , Yongbo Wang , Feng Wang , Jingtong Wu , Xin Lu , Rui Yan

Processing long contexts has become a critical capability for modern large language models (LLMs). However, serving long-context LLMs comes with significant inference costs due to the high memory overhead of the key-value (KV) cache.…

Machine Learning · Computer Science 2025-03-04 Qihui Zhou , Peiqi Yin , Pengfei Zuo , James Cheng