English
Related papers

Related papers: FuXi-\beta: Towards a Lightweight and Fast Large-S…

200 papers

We present QUOKA: Query-oriented KV selection for efficient attention, a training-free and hardware agnostic sparse attention algorithm for accelerating transformer inference under chunked prefill. While many queries focus on a smaller…

Machine Learning · Computer Science 2026-02-10 Dalton Jones , Junyoung Park , Matthew Morse , Mingu Lee , Chris Lott , Harper Langston

Generative recommendation models can model user behavior as sequences of events and provide a shared backbone for multiple recommendation tasks. In production, however, pre-training gains do not automatically translate into downstream…

Information Retrieval · Computer Science 2026-05-25 Qiuling Xu , Ko-Jen Hsiao , Moumita Bhattacharya

Transformer-based models have achieved dominant performance in numerous NLP tasks. Despite their remarkable successes, pre-trained transformers such as BERT suffer from a computationally expensive self-attention mechanism that interacts…

Computation and Language · Computer Science 2024-06-04 Jungmin Yun , Mihyeon Kim , Youngbin Kim

The self-attention mechanism traditionally relies on the softmax operator, necessitating positional embeddings like RoPE, or position biases to account for token order. But current methods using still face length generalisation challenges.…

Machine Learning · Computer Science 2025-05-21 Shawn Tan , Songlin Yang , Aaron Courville , Rameswar Panda , Yikang Shen

Transformers have become the foundation of numerous state-of-the-art AI models across diverse domains, thanks to their powerful attention mechanism for modeling long-range dependencies. However, the quadratic scaling complexity of attention…

Hardware Architecture · Computer Science 2026-01-29 Zhenkun Fan , Zishen Wan , Che-Kai Liu , Ashwin Sanjay Lele , Win-San Khwa , Bo Zhang , Meng-Fan Chang , Arijit Raychowdhury

Large Reasoning Models (LRMs) have shown promising accuracy improvements on complex problem-solving tasks. While these models have attained high accuracy by leveraging additional computation at test time, they need to generate long…

Computation and Language · Computer Science 2025-12-16 Coleman Hooper , Sebastian Zhao , Luca Manolache , Sehoon Kim , Michael W. Mahoney , Yakun Sophia Shao , Kurt Keutzer , Amir Gholami

Attention Residuals replace standard additive residual connections with learned softmax attention over previous layer outputs, enabling selective cross-layer routing. However, standard Attention Residuals still attend over cumulative hidden…

Machine Learning · Computer Science 2026-05-20 Cheng Luo , Zefan Cai , Junjie Hu

Robustly predicting attention regions of interest for self-driving systems is crucial for driving safety but presents significant challenges due to the labor-intensive nature of obtaining large-scale attention labels and the domain gap…

Computer Vision and Pattern Recognition · Computer Science 2025-01-30 Mengshi Qi , Xiaoyang Bi , Pengfei Zhu , Huadong Ma

While deep reinforcement learning agents demonstrate high performance across domains, their internal decision processes remain difficult to interpret when evaluated only through performance metrics. In particular, it is poorly understood…

Machine Learning · Computer Science 2025-12-01 Charlotte Beylier , Hannah Selder , Arthur Fleig , Simon M. Hofmann , Nico Scherf

The computational cost of softmax-based attention in transformers limits their applicability to long-context tasks. Adaptive sparsity, of which $\alpha$-entmax attention is an example, offers a flexible data-dependent alternative, but…

Computation and Language · Computer Science 2025-06-10 Nuno Gonçalves , Marcos Treviso , André F. T. Martins

Advanced autoregressive (AR) video generation models have improved visual fidelity and interactivity, but the quadratic complexity of attention remains a primary bottleneck for efficient deployment. While existing sparse attention solutions…

Computer Vision and Pattern Recognition · Computer Science 2026-05-26 Chengtao Lv , Yumeng Shi , Yushi Huang , Ruihao Gong , Shen Ren , Wenya Wang

Online Class-Incremental (OCI) learning has sparked new approaches to expand the previously trained model knowledge from sequentially arriving data streams with new classes. Unfortunately, OCI learning can suffer from catastrophic…

Machine Learning · Computer Science 2023-03-15 Yunfeng Fan , Wenchao Xu , Haozhao Wang , Jiaqi Zhu , Junxiao Wang , Song Guo

Large language models (LLMs) encounter computational challenges during long-sequence inference, especially in the attention pre-filling phase, where the complexity grows quadratically with the prompt length. Previous efforts to mitigate…

Machine Learning · Computer Science 2025-03-03 Xunhao Lai , Jianqiao Lu , Yao Luo , Yiyuan Ma , Xun Zhou

In this paper, our focus is on enhancing steering angle prediction for autonomous driving tasks. We initiate our exploration by investigating two veins of widely adopted deep neural architectures, namely ResNets and InceptionNets. Within…

Computer Vision and Pattern Recognition · Computer Science 2024-02-02 Swetha Nadella , Pramiti Barua , Jeremy C. Hagler , David J. Lamb , Qing Tian

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

Transformers have shown strong ability to model long-term dependencies and are increasingly adopted as world models in model-based reinforcement learning (RL) under partial observability. However, unlike natural language corpora, RL…

Machine Learning · Computer Science 2025-11-11 Daniel De Dios Allegue , Jinke He , Frans A. Oliehoek

This paper introduces FANTF (Fuzzy Attention Network-Based Transformers), a novel approach that integrates fuzzy logic with existing transformer architectures to advance time series forecasting, classification, and anomaly detection tasks.…

Machine Learning · Computer Science 2025-04-02 Sanjay Chakraborty , Fredrik Heintz

Autoregressive video diffusion models enable streaming generation, opening the door to long-form synthesis, video world models, and interactive neural game engines. However, their core attention layers become a major bottleneck at inference…

Computer Vision and Pattern Recognition · Computer Science 2026-02-03 Dvir Samuel , Issar Tzachor , Matan Levy , Micahel Green , Gal Chechik , Rami Ben-Ari

Attention mechanism has been regarded as an advanced technique to capture long-range feature interactions and to boost the representation capability for convolutional neural networks. However, we found two ignored problems in current…

Computer Vision and Pattern Recognition · Computer Science 2021-08-19 Zhu Baozhou , Peter Hofstee , Jinho Lee , Zaid Al-Ars

Visual attention has been extensively studied for learning fine-grained features in both facial expression recognition (FER) and Action Unit (AU) detection. A broad range of previous research has explored how to use attention modules to…

Computer Vision and Pattern Recognition · Computer Science 2022-03-24 Xiaotian Li , Zhihua Li , Huiyuan Yang , Geran Zhao , Lijun Yin