arXiv · 2511.05560
Sample-Efficient Language Modeling with Linear Attention and Lightweight Enhancements
Abstract
We study architectural and optimization techniques for sample-efficient language modeling under the constraints of the BabyLM 2025 shared task. Our model, BLaLM, replaces self-attention with a linear-time mLSTM token mixer and explores lightweight enhancements, including short convolutions, sliding window attention with dynamic modulation, and Hedgehog feature maps. To support training in low-resource settings, we curate a high-quality corpus emphasizing readability and pedagogical structure. Experiments across both STRICT and STRICT-SMALL tracks show that (1) linear attention combined with sliding window attention consistently improves zero-shot performance, and (2) the Muon optimizer stabilizes convergence and reduces perplexity over AdamW. These results highlight effective strategies for efficient language modeling without relying on scale.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Patrick Haller, Jonas Golde, Alan Akbik. 2025-11-04. Sample-Efficient Language Modeling with Linear Attention and Lightweight Enhancements. https://arxiv.org/abs/2511.05560
Cite the original work for its findings. Save a collection to share your selection of sources.