State Space Models are Strong Text Rerankers
Xu, Zhichao, Yan, Jinghua, Gupta, Ashim, Srikumar, Vivek
–arXiv.org Artificial Intelligence
Transformers dominate NLP and IR; but their inference inefficiencies and challenges in extrapolating to longer contexts have sparked interest in alternative model architectures. Among these, state space models (SSMs) like Mamba offer promising advantages, particularly $O(1)$ time complexity in inference. Despite their potential, SSMs' effectiveness at text reranking -- a task requiring fine-grained query-document interaction and long-context understanding -- remains underexplored. This study benchmarks SSM-based architectures (specifically, Mamba-1 and Mamba-2) against transformer-based models across various scales, architectures, and pre-training objectives, focusing on performance and efficiency in text reranking tasks. We find that (1) Mamba architectures achieve competitive text ranking performance, comparable to transformer-based models of similar size; (2) they are less efficient in training and inference compared to transformers with flash attention; and (3) Mamba-2 outperforms Mamba-1 in both performance and efficiency. These results underscore the potential of state space models as a transformer alternative and highlight areas for improvement in future IR applications.
arXiv.org Artificial Intelligence
Dec-18-2024
- Country:
- North America > United States
- Utah (0.04)
- New York > New York County
- New York City (0.04)
- Minnesota > Hennepin County
- Minneapolis (0.14)
- Europe
- Italy (0.04)
- United Kingdom > England
- Cambridgeshire > Cambridge (0.04)
- Ireland > Leinster
- County Dublin > Dublin (0.04)
- Greece > Central Macedonia
- Thessaloniki (0.04)
- Asia
- Singapore (0.04)
- Middle East > Jordan (0.04)
- North America > United States
- Genre:
- Research Report > New Finding (0.87)
- Technology: