cs.LGMar 8, 2026

InfoMamba: An Attention-Free Hybrid Mamba-Transformer Model

Authors: Youjin Wang, Jiaqiao Zhao, Rong Fu, Run Zhou, Ruizhe Zhang, Jiani Liang, Suisuai Cao, Feng Zhou

Organizations: Renmin University of China, Beijing, China · University of Macau, Macau SAR, China · Central South University, Changsha, China · Zhejiang University, Hangzhou, China

Abstract

Balancing fine-grained local modeling with long-range dependency capture under computational constraints remains a central challenge in sequence modeling. While Transformers provide strong token mixing, they suffer from quadratic complexity, whereas Mamba-style selective state-space models (SSMs) scale linearly but often struggle to capture high-rank and synchronous global interactions. We present a consistency boundary analysis that characterizes when diagonal short-memory SSMs can approximate causal attention and identifies structural gaps that remain. Motivated by this analysis, we propose InfoMamba, an attention-free hybrid architecture. InfoMamba replaces token-level self-attention with a concept bottleneck linear filtering layer that serves as a minimal-bandwidth global interface and integrates it with a selective recurrent stream through information-maximizing fusion (IMF). IMF dynamically injects global context into the SSM dynamics and encourages complementary information usage through a mutual-information-inspired objective. Extensive experiments on classification, dense prediction, and non-vision tasks show that InfoMamba consistently outperforms strong Transformer and SSM baselines, achieving competitive accuracy-efficiency trade-offs while maintaining near-linear scaling.

Explore similar work

CardsList
  1. Attention to Mamba: A Recipe for Cross-Architecture Distillation

    Apr 1, 2026Abhinav Moudgil, Ningyuan Huang, Eeshan Gunesh Dhekane +3Cross-Architecture Knowledge DistillationLanguage Model Distillation

  2. 2Mamba2Furious: Linear in Complexity, Competitive in Accuracy

    Feb 19, 2026Gabriel Mongaras, Eric C. LarsonLinear AttentionMemory-Efficient Attention

  3. Forget Attention: Importance-Aware Attention Is All You Need

    Jun 1, 2026Suhyeong Shin, Yeongwook YangSoftmax AttentionLong-Context Language Modeling