LUMIA Lab pairs 410M base with 6.9B memory, matches Pythia-12B
TL;DR
- Pairing Pythia-410M with a 6.9B pretrained memory scores 37.34 average across 17 benchmarks, edging Pythia-12B's 37.24 with 39% fewer total parameters.
- Attaching 1.7B domain-specific memories to Qwen3 base models from 0.6B to 14B improves average scores by more than 9 points at every scale.
- The team pretrained memory modules on 300B tokens using a distributed Faiss indexing pipeline with sparse, batch-wise loading of kNN distributions.
A paper out of Shanghai Jiao Tong's LUMIA Lab stuck with me this week because it points at a genuinely different way to spend a parameter budget. The team behind Memory Decoder at Scale pairs a small 410M-parameter Pythia base with a separately pretrained 6.9B parametric memory module, and reports that the combination scores 37.34 on average across 17 benchmarks, edging Pythia-12B's 37.24 while using 39% fewer total parameters.
The angle worth pausing on is not the leaderboard win, it is what it says about scaling. The default assumption has been to grow the base model. This work argues that if you split long-term memory out into its own module and pretrain it on 300B tokens, you get a better parameter-performance tradeoff than pouring the same budget into a bigger backbone. To make that practical at 6.9B, the authors had to build a distributed Faiss indexing and retrieval pipeline with sparse, batch-wise loading of kNN distributions, because a standard Faiss pipeline was infeasible at that data scale.
The domain-adaptation result is the one operators should read twice. On Qwen3 base models from 0.6B up to 14B, attaching a 1.7B domain-specific memory improved the average score across three domains by more than 9 points at every scale. The project page reports 47 out of 51 general-memory task-scale combinations improved and 15 out of 15 domain evaluations improved across all Qwen3 scales, which is a broader coverage claim than a single headline number.
The honest caveats are that the comparison anchors on Pythia and Qwen3 base checkpoints rather than modern instruction-tuned frontier systems, and the write-up does not put a number on end-to-end inference latency once the distributed retrieval is running, which is what actually determines serving cost. What the abstract also does not give you is the compute bill for pretraining a 6.9B memory on 300B tokens, or whether the same trick survives RLHF and instruction tuning.
Still, the direction is the interesting part. If pairing small base models with much larger pretrained memory modules holds up beyond base-model benchmarks, the cost calculus for anyone deploying domain-specific inference gets meaningfully more forgiving, and vector-index infrastructure stops being a retrieval afterthought and starts looking like part of the model itself.
Originally reported by huggingface.co
Read the original article →Original headline: Tsinghua's Memory Decoder at Scale Pairs 410M Base With 6.9B External Memory, Matches Pythia-12B