English
Related papers

Related papers: Priming: Hybrid State Space Models From Pre-traine…

200 papers

Recent research has shown that combining Mamba with Transformer architecture, which has selective state space and quadratic self-attention mechanism, outperforms using Mamba or Transformer architecture alone in language modeling tasks. The…

Computation and Language · Computer Science 2024-07-23 Jingze Shi , Ting Xie , Bingheng Wu , Chunjun Zheng , Kai Wang

State Space Models (SSMs) have emerged as a promising alternative to Transformers for long-context sequence modeling, offering linear $O(N)$ computational complexity compared to the Transformer's quadratic $O(N^2)$ scaling. This paper…

Machine Learning · Computer Science 2026-01-06 Abidemi Koledoye , Chinemerem Unachukwu , Gold Nwobu , Hasin Rana

Should LLM reasoning live in a separate module, or within a single model's forward pass and representational space? We study dual-architecture latent reasoning, where a fluent Base exchanges latent messages with a Coprocessor, and test two…

Long-sequence modeling faces a fundamental trade-off between the efficiency of compressive fixed-size memory in RNN-like models and the fidelity of lossless growing memory in attention-based Transformers. Inspired by the Multi-Store Model…

Computation and Language · Computer Science 2025-12-18 Yunhao Fang , Weihao Yu , Shu Zhong , Qinghao Ye , Xuehan Xiong , Lai Wei

State Space Models (SSMs), particularly the Mamba architecture, have recently emerged as powerful alternatives to Transformers for sequence modeling, offering linear computational complexity while achieving competitive performance. Yet,…

Computer Vision and Pattern Recognition · Computer Science 2025-11-18 Mohamed A. Mabrok , Yalda Zafari

With the growing demand for deploying large language models (LLMs) across diverse applications, improving their inference efficiency is crucial for sustainable and democratized access. However, retraining LLMs to meet new user-specific…

Machine Learning · Computer Science 2026-01-21 Mingyu Yang , Mehdi Rezagholizadeh , Guihong Li , Vikram Appia , Emad Barsoum

The hybrid analog/digital architecture that connects a limited number of RF chains to multiple antennas through phase shifters could effectively address the energy consumption issues in massive multiple-input multiple-output (MIMO) systems.…

Information Theory · Computer Science 2025-01-07 Yifan Guo

State space models (SSMs), such as Mamba, have emerged as an efficient alternative to transformers for long-context sequence modeling. However, despite their growing adoption, SSMs lack the interpretability tools that have been crucial for…

Computation and Language · Computer Science 2025-02-26 Hugo Pitorro , Marcos Treviso

State space models (SSMs) have demonstrated state-of-the-art sequence modeling performance in some modalities, but underperform attention in language modeling. Moreover, despite scaling nearly linearly in sequence length instead of…

Machine Learning · Computer Science 2023-05-02 Daniel Y. Fu , Tri Dao , Khaled K. Saab , Armin W. Thomas , Atri Rudra , Christopher Ré

In this technical report, we present Zamba, a novel 7B SSM-transformer hybrid model which achieves competitive performance against leading open-weight models at a comparable scale. Zamba is trained on 1T tokens from openly available…

Machine Learning · Computer Science 2024-05-28 Paolo Glorioso , Quentin Anthony , Yury Tokpanov , James Whittington , Jonathan Pilault , Adam Ibrahim , Beren Millidge

Applying pre-trained models to assist point cloud understanding has recently become a mainstream paradigm in 3D perception. However, existing application strategies are straightforward, utilizing only the final output of the pre-trained…

Computer Vision and Pattern Recognition · Computer Science 2025-05-28 Yaohua Zha , Yanzi Wang , Hang Guo , Jinpeng Wang , Tao Dai , Bin Chen , Zhihao Ouyang , Xue Yuerong , Ke Chen , Shu-Tao Xia

Linear State Space Models (SSMs) offer remarkable performance gains in efficient sequence modeling, with constant inference-time computation and memory complexity. Recent advances, such as Mamba, further enhance SSMs with input-dependent…

Machine Learning · Computer Science 2025-06-24 Zheng Zhan , Liliang Ren , Shuohang Wang , Liyuan Liu , Yang Liu , Yeyun Gong , Yanzhi Wang , Yelong Shen

State space models (SSMs) have emerged as a powerful framework for modelling long-range dependencies in sequence data. Unlike traditional recurrent neural networks (RNNs) and convolutional neural networks (CNNs), SSMs offer a structured and…

Machine Learning · Computer Science 2024-10-07 Siddhanth Bhat

State Space Models (SSMs), as key components of Mamaba, have gained increasing attention for vision models recently, thanks to their efficient long sequence modeling capability. Given the computational cost of deploying SSMs on…

Computer Vision and Pattern Recognition · Computer Science 2025-01-24 Yinglong Li , Xiaoyu Liu , Jiacheng Li , Ruikang Xu , Yinda Chen , Zhiwei Xiong

State Space Models (SSMs) have emerged as an efficient alternative to the transformer architecture. Recent studies show that SSMs can match or surpass Transformers on code understanding tasks, such as code retrieval, when trained under…

Artificial Intelligence · Computer Science 2026-02-09 Jiali Wu , Abhinav Anand , Shweta Verma , Mira Mezini

Speculative decoding accelerates autoregressive inference by drafting candidate tokens with a fast model and verifying them in parallel with the target. Self-speculative methods avoid the need for an external drafter but have been studied…

Computation and Language · Computer Science 2026-05-05 Hector Borobia , Elies Seguí-Mas , Guillermina Tormo-Carbó

Transformers have demonstrated strong performance across a wide range of sequence modeling tasks, but their quadratic attention complexity limits scalability to long sequences. Linear models such as Mamba and sliding-window attention (SWA)…

Machine Learning · Computer Science 2025-09-03 Aref Jafari , Yuhe Fan , Benyamin Jamialahmadi , Parsa Farinneya , Boxing Chen , Marzieh S. Tahaei

Transformers have become one of the foundational architectures in point cloud analysis tasks due to their excellent global modeling ability. However, the attention mechanism has quadratic complexity, making the design of a linear complexity…

Computer Vision and Pattern Recognition · Computer Science 2024-11-26 Dingkang Liang , Xin Zhou , Wei Xu , Xingkui Zhu , Zhikang Zou , Xiaoqing Ye , Xiao Tan , Xiang Bai

Transformers with linear attention (i.e., linear transformers) and state-space models have recently been suggested as a viable linear-time alternative to transformers with softmax attention. However, these models still underperform…

Machine Learning · Computer Science 2025-01-16 Songlin Yang , Bailin Wang , Yu Zhang , Yikang Shen , Yoon Kim

The attention mechanism in Transformers is an important primitive for accurate and scalable sequence modeling. Its quadratic-compute and linear-memory complexity however remain significant bottlenecks. Linear attention and state-space…

Machine Learning · Computer Science 2026-03-03 Han Guo , Songlin Yang , Tarushii Goel , Eric P. Xing , Tri Dao , Yoon Kim