Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2511.00315 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911244226658304 |
|---|---|
| author | Xiong, Lee Tkachenko, Maksim Effendi, Johanes Cai, Ting |
| author_facet | Xiong, Lee Tkachenko, Maksim Effendi, Johanes Cai, Ting |
| contents | We propose Factorization Memory, an efficient recurrent neural network (RNN) architecture that achieves performance comparable to Transformer models on short-context language modeling tasks while also demonstrating superior generalization in long-context scenarios. Our model builds upon Mamba-2, enabling Factorization Memory to exploit parallel computations during training while preserving constant computational and memory complexity during inference. To further optimize model efficiency and representational capacity, we develop a sparse formulation of Factorization Memory that updates only a subset of recurrent states at each step while preserving the strong performance of its dense counterpart. To our knowledge, this represents the first RNN architecture that successfully combines sparse memory activation with competitive performance across both short and long-context settings. This work provides a systematic empirical analysis of Factorization Memory in comparison to Transformer and Mamba-2 architectures. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2511_00315 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Language Modeling With Factorization Memory Xiong, Lee Tkachenko, Maksim Effendi, Johanes Cai, Ting Computation and Language Artificial Intelligence We propose Factorization Memory, an efficient recurrent neural network (RNN) architecture that achieves performance comparable to Transformer models on short-context language modeling tasks while also demonstrating superior generalization in long-context scenarios. Our model builds upon Mamba-2, enabling Factorization Memory to exploit parallel computations during training while preserving constant computational and memory complexity during inference. To further optimize model efficiency and representational capacity, we develop a sparse formulation of Factorization Memory that updates only a subset of recurrent states at each step while preserving the strong performance of its dense counterpart. To our knowledge, this represents the first RNN architecture that successfully combines sparse memory activation with competitive performance across both short and long-context settings. This work provides a systematic empirical analysis of Factorization Memory in comparison to Transformer and Mamba-2 architectures. |
| title | Language Modeling With Factorization Memory |
| topic | Computation and Language Artificial Intelligence |
| url | https://arxiv.org/abs/2511.00315 |