Efficient Extractive Summarization with MAMBA-Transformer Hybrids for Low-Resource Scenarios
Organizations: Independent Researcher · University of Memphis
Abstract
Extractive summarization of long documents is bottlenecked by quadratic complexity, often forcing truncation and limiting deployment in resource-constrained settings. We introduce the first Mamba-Transformer hybrid for extractive summarization, combining the semantic strength of pre-trained transformers with the linear-time processing of state space models. Leveraging Mamba's ability to process full documents without truncation, our approach preserves context while maintaining strong summarization quality. The architecture includes: (1) a transformer encoder for sentence-level semantics, (2) a Mamba state space model to capture inter-sentence dependencies efficiently, and (3) a linear classifier for sentence relevance prediction. Across news, argumentative, and scientific domains under low-resource conditions, our method achieves: (1) large gains over BERTSUM and MATCHSUM, including +0.23 ROUGE-1 on ArXiv and statistically significant improvements on all datasets (p < 0.001); (2) consistent advantages across domains, strongest on the longest documents; (3) robust performance with limited training data; and (4) 24-27% faster inference on news summarization (CNN/DailyMail). We introduce the first hybrid Transformer-state space architecture for summarization, showing significant ROUGE improvements in low-resource scenarios.
Figures & tables
| Dataset | Avg doc length (tokens) | Avg sents/doc | Positive (%) |
|---|---|---|---|
| CNN/DailyMail | 781 | 29.4 | 18.2 |
| DebateSum | 643 | 24.1 | 22.7 |
| ArXiv | 1,247 | 41.8 | 16.9 |
| Model | ROUGE-1 | ROUGE-2 | ROUGE-L | Time (s/sample) |
|---|---|---|---|---|
| CNN/DailyMail | ||||
| MAMBA-BERT | 0.61 | 0.51 | 0.53 | 13.7 |
| BERTSUM | 0.44 | 0.28 | 0.33 | 18.0 |
| MATCHSUM | 0.40 | 0.18 | 0.27 | 18.6 |
| DebateSum | ||||
| MAMBA-BERT | 0.57 | 0.54 | 0.56 | 0.65 |