ALLOT: Budgeted Hybrid-Memory Routing for Knowledge Updates in LLMs
Organizations: Baidu Inc.
Abstract
For large language models (LLMs), parametric adaptation is costly when retrieval already suffices. We introduce ALLOT, a hybrid-memory routing framework that separates learned write priority from a hard parametric budget. A memory-aware router combines frozen text representations, retrieval confidence, and relation metadata; a single ranking supports multiple write budgets while preserving all facts in external memory. On CounterFact with Qwen3-4B, ALLOT reaches 0.760 accuracy at a 20% parametric-write budget and recovers 78.4% of the budget-matched oracle gain, with 80% fewer parametric writes than dual-writing every fact. At this budget, jointly adding retrieval and relation features to text improves normalized oracle gain by 6.2 percentage points. Complementary Qwen3-0.6B shared-store results achieve dual-write-level accuracy with 6-14.5% parametric writes, and cross-benchmark transfer retains approximately 88% of in-domain gain. These results support allocating adaptation capacity according to its incremental value rather than treating every factual update as an equally valuable training target.
Figures & tables
| Method | 5% | 10% | 15% | 20% | 30% |
|---|---|---|---|---|---|
| Oracle | 0.718 | 0.734 | 0.751 | 0.768 | 0.801 |
| ALLOT (D) | 0.713 | 0.730 | 0.746 | 0.760 | 0.788 |
| Uplift | 0.712 | 0.726 | 0.734 | 0.747 | 0.761 |
| Logistic | 0.714 | 0.724 | 0.736 | 0.746 | 0.764 |
| GBT | 0.713 | 0.723 | 0.733 | 0.744 | 0.759 |
| Random | 0.708 | 0.716 | 0.724 | 0.732 | 0.748 |
| Features | 5% | 10% | 15% | 20% | 30% |
|---|---|---|---|---|---|
| A: text | 52.2 | 57.4 | 62.7 | 72.2 | 68.5 |
| B: text + ret. | 64.1 | 63.5 | 66.8 | 69.1 | 68.5 |
| C: text + rel. | 40.3 | 51.3 | 62.7 | 62.9 | 64.3 |
| D: all | 52.2 | 75.7 | 79.3 | 78.4 | 74.8 |
| Ret. heuristic | 28.4 | 26.9 | 12.9 | 4.3 | 1.3 |
| Data | Policy | Acc. | Para. | MB (est.) |
|---|---|---|---|---|
| zsRE | Retrieval | .624 | 0% | 2 |
| Both always | .644 | 100% | 553 | |
| ALLOT | .646 | 6% | 37 | |
| CF | Retrieval | .516 | 0% | 2 |
| Both always | .585 | 100% | 553 | |
| ALLOT | .585 | 14.5% | 63 |
Memory-Augmented Language Models
Memory-augmented language models combine persistent or temporary memory with neural computation to retain, retrieve, and use information beyond the active context.
See all 292 papers