@vintcessun: 一块能跨模型插拔的“知识硬盘”——难点不在搬记忆表,而在为目标模型配好读头。 https://arxiv.org/abs/2608.17050 论文先用源模型训练 Engram 哈希记忆,再冻结记忆与目标骨干,仅训练 reader 完成寻…
摘要
该论文提出一种通过目标侧reader适配实现跨模型记忆迁移的方法,使用Engram哈希记忆和轻量级reader,在问答任务中实现38.8分,并适用于Agent记忆的更新和审计。局限是仅测试到9B模型,缩放规律未知。
查看缓存全文
缓存时间: 2026/08/25 16:10
一块能跨模型插拔的“知识硬盘”——难点不在搬记忆表,而在为目标模型配好读头。
https://arxiv.org/abs/2608.17050
论文先用源模型训练 Engram 哈希记忆,再冻结记忆与目标骨干,仅训练 reader 完成寻址、隐藏空间对齐和指定层注入。双层四分支 reader 在受控 QA 得 38.8,跨架构迁移也有增益。对 Agent memory 很有价值:知识可更新、审计、复用,且无 RAG 上下文开销。局限是只测到 9B,缩放规律与联合训练仍未知。
Cross-Model Memory Transfer via Target-Side Reader Adaptation
Source: https://arxiv.org/html/2608.17050 Mingyuan LiAffiliation:Emails: [email protected], [email protected], [email protected], [email protected] YuAffiliation:Emails: [email protected], [email protected], [email protected], [email protected] WangAffiliation:Emails: [email protected], [email protected], [email protected], [email protected] Ji††thanks:Corresponding author.Affiliation:Emails: [email protected], [email protected], [email protected], [email protected]:ELLIS Institute FinlandAffiliation:University of TurkuAffiliation:University of Technology Sydney
Abstract
Methods for improving knowledge use in large language models typically fall into two regimes. Non-parametric retrieval offers flexible access to external knowledge, but adds retrieval latency, context overhead, and only shallow integration with the backbone. Parametric adaptation is efficient at inference time, but entangles knowledge with model weights and can be hard to update, audit, or transfer. Engram-style hashed memory occupies a middle regime: it stores learned information in an external, addressable table, yet consumes that table through a small learned reader. This raises a basic question: when such a memory is moved across backbones, what matters more, the frozen memory itself or the target-side reader? We study this question through cross-model frozen-memory extraction, in which a memory trained on a source model is frozen and attached to a different target model, with only a lightweight reader trained. Ablations show that learned memory content and correct addressing both matter, but the transferred table becomes useful only through a reader aligned to the target model. In downstream question answering tasks, a dual-layer, four-branch reader nearly closes the gap between same-model and cross-model reuse, achieving an average score of 38.8 under our controlled evaluation protocol. Moreover, when the provider reader is directly compatible with the target interface, the frozen artifact can provide substantial utility without target-side training, while optional reader adaptation yields further improvement. These results suggest that Engram can serve as a reusable external knowledge artifact, provided that the target has access to a compatible reader interface; target-side adaptation can further improve alignment when direct reader reuse is insufficient.
1Introduction
Language models can use knowledge in two familiar ways.Non-parametricmethods, such as retrieval-augmented generation (RAG)[Lewis et al., 2020]and nearest-neighbor retrieval[Khandelwal et al., 2020], keep knowledge outside the backbone and make updates relatively direct, but they pay for that flexibility with retrieval latency, longer contexts, and loose integration between retrieved evidence and generation.Parametricmethods, such as continued pretraining, fine-tuning, or learned memory layers tied to a single model[Wei et al., 2026], avoid retrieval at inference time, but they entangle knowledge with model weights. This makes updates, audits, and transfers across backbones difficult.
Engram-style hashed memory[Cheng et al., 2026]occupies an interesting middle ground. It stores learned information in an explicit, addressable external table, yet consumes that table through a small neural interface rather than by inserting raw documents into the context. This hybrid design is attractive because it combines the modularity of external memory with the efficiency of learned representations. But it also exposes a sharper question:is the memory itself reusable or transferable, or is it merely a co-adapted extension of the model that trained it?
The distinction matters: a memory table that is not portable is just another form of model-specific parameterization, not a reusable knowledge substrate. Thus, the portability requires an operational test:remove the source backbone and ask whether another model can still extract a useful signal from the frozen memory table.
We study this test throughcross-model frozen-memory transfer. Given a memory table trained with a source modelAA, we freeze the table, attach it to a different target modelBB, and train only a lightweight target-side reader for memory extraction. If the frozen memory still improves the target after the original host is removed, then it contains usable information beyond source-specific co-adaptation. If it fails, then the apparent memory benefit was likely tied to the original backbone.
Figure1illustrates the life cycle of memory reuse and transfer: source-side memory training, memory export and freezing, and target-side extraction through a new reader. To make this test precise, we separate three roles: (1)addressingdecides which entries are read, (2)memorydenotes the stored vectors, and (3)readermaps them into the model. We standardize addressing, freeze the memory, and adapt only the reader. This turns portability into a measurable question: can a target model learn to read a fixed memory artifact? We use a tokenizer-agnostic canonicalization pipeline to keep memory addresses stable across tokenizers, freeze the exported source memory, and train only a small target-side reader that projects memory values into the target residual stream.
Our central claim is:for portable hashed external memory, memory presence alone is not enough; successful reuse depends on the interface through which a target backbone addresses and consumes the stored representations.Nevertheless, memory content still matters. Transferred memory clearly outperforms permuted-key controls and reaches substantially better downstream Question Answering (QA) performance than random memory under the final evaluation protocol. But the amount of usable signal depends on how well the target model can align and integrate what it retrieves. This view explains why imperfect memory can still help with a capable reader, and why reader design can matter as much as retraining or enlarging the stored table in this regime.
We support this claim with a concise chain of evidence, spanning diverse model families and settings. First, a source–target transfer study shows that frozen memory consistently improves target models across different model families and scales, including Pythia[Biderman et al., 2023], QwenQwen Team [2025], TinyLlama[Zhang et al., 2024], PhiAbouelenin et al. [2025], LLaMA[Touvron et al., 2023], and Mistral[Jiang et al., 2023], with relative perplexity reductions of up to 15.7%. A supplementary peer-to-peer setting further confirms that transfer remains beneficial in both directions between models of similar scale. Second, ablation experiments show that transferred memory clearly outperforms the permuted-key and no-memory controls, while scratch-trained target memory remains competitive under the same target-side adaptation budget. This suggests that memory reuse is a joint property of the stored content and the target side reader that makes it usable. Third, on downstream tasks, upgrading the target side reader to a dual-layer, multi-branch architecture nearly closes the gap between same-model and cross-model reuse, with the best configuration reaching an average score of 38.8 across five QA tasks, achieving a new state-of-the-art result (SoTA) and outperforming MLP Memory[Wei et al., 2026]and selective gains on other knowledge-intensive benchmarks.
In summary, our contributions are:
- •We formulate cross-model frozen-memory reuse as an evaluation problem for external memory, separating portable memory from backbone-specific co-adaptation and measuring whether the target model can extract usable signal from a frozen artifact.
- •We instantiate this problem with a transfer protocol for Engram-style memory that standardizes addressing across tokenizers, and we propose lightweight target-side readers for memory extraction, including stronger multi-branch, multi-layer variants.
- •We provide an evidence chain showing that reader design is first-order: consistent gains across diverse model families and settings, ablations that isolate memory content and address integrity, and QA results showing that stronger target-side readers, rather than source-backbone identity alone, drive the best performance and achieve a new SoTA.
This paper is organized as follows.Section2reviews Engram architecture, related memory mechanisms, and cross-model alignment.Section3describes the cross-model frozen-memory transfer protocol, including tokenizer-agnostic addressing, target-side reader design, and training regime.Section4presents the experimental setup, research questions, results, and takeaways. Supplementary material inAppendixAprovides additional details on canonicalization, memory export, and the reader implementation.AppendixBgives training and evaluation details, including hyperparameters, datasets, and evaluation metrics.AppendixCextends RQ1 inSection4.1with exact matrix values and controls for tokenizer mismatch, peer-scale transfer, target scaling, and representational similarity.AppendixDextends RQ3 inSection4.3by diagnosing why downstream gains are task-dependent, separating source-corpus specialization, Phase-2 reader-fitting alignment, mixed-corpus behavior, and out-of-domain effects. The supplementary material concludes with a computational cost analysis (AppendixF), broader ethics and impact considerations (AppendixG), and limitations (AppendixH).
2Background and Related Work
2.1Engram Architecture
Engram[Cheng et al., 2026]augments a Transformer-MoE backbone with an external conditional memory. It retrieves staticnn-gram embeddings through deterministic hashing and injects the retrieved vector into the backbone through a learned gate. The key property for our setting is that Engram separatesaddressing,storage, andread: the input text determines which memory entry is read, the memory table stores the external vectors, and a small learned reader maps the retrieved vector into the backbone hidden space. This separation makes transfer possible: as long as another model can reproduce the same memory addresses through a shared addressing scheme, the frozen Engram table can be reused through a target-side reader.
For eachnn-gram ordern∈[2,N]n\in[2,N]and hash headk∈[1,K]k\in[1,K], a deterministic hashφn,k\varphi_{n,k}maps the canonicalizednn-gramgt,ng_{t,n}to an entry in memory table𝐄n,k\mathbf{E}_{n,k}. The retrieved embeddings are concatenated as
𝐞t,⨁n=2N⨁k=1K𝐄n,k[φn,k(gt,n)].\mathbf{e}_{t}\triangleq\bigoplusop\displaylimits_{n=2}^{N}\bigoplusop\displaylimits_{k=1}^{K}\mathbf{E}_{n,k}[\varphi_{n,k}(g_{t,n})].(1)Crucially, this lookup depends on the canonicalized input sequence rather than the model hidden state, so the memory table can be treated as an external artifact once the address space is standardized.
The retrieved vector is consumed through learned projections and a gate:
st\displaystyle s_{t}=RMSNorm(𝐡t)⊤RMSNorm(𝐤t)d,\displaystyle=\frac{\RMSNorm(\mathbf{h}_{t})^{\top}\RMSNorm(\mathbf{k}_{t})}{\sqrt{d}},αt\displaystyle\alpha_{t}=σ(sign(st)|st|),\displaystyle=\sigma\!\left(\mathrm{sign}(s_{t})\sqrt{|s_{t}|}\right),(2)𝐤t\displaystyle\mathbf{k}_{t}=𝐖K𝐞t,\displaystyle=\mathbf{W}_{K}\mathbf{e}_{t},𝐯t\displaystyle\mathbf{v}_{t}=𝐖V𝐞t.\displaystyle=\mathbf{W}_{V}\mathbf{e}_{t}.Here,𝐖K\mathbf{W}_{K},𝐖V\mathbf{W}_{V}, and the gate define the backbone-specific reader. This structural separation makes frozen-memory transfer possible: the source memory table can be frozen and moved to a target model, while only a lightweight target-side reader is trained to extract and align it.
2.2Memory Mechanisms
MoE[Shazeer et al., 2017,Fedus et al., 2022,Dai et al., 2024]expands capacity through conditional computation, but its useful signal remains embedded in the host backbone rather than exposed as a reusable memory artifact. Other memory-augmented models also miss the extraction question for one of two reasons: they either retrieve from large dynamic memory banks at inference time[Graves et al., 2014,Weston et al., 2015], or they store information in model-specific hidden spaces, as in KNN-LM[Khandelwal et al., 2020]. Retrieval-augmented generation and RETRO-style systems[Lewis et al., 2020,Borgeaud et al., 2022,Izacard et al., 2023]are portable at the document level, but the reusable object is a text datastore rather than a trained memory representation. Larger learned memory layers[Lample et al., 2019,Berges et al., 2025,Behrouz and Zhong, 2025,Wei et al., 2026]improve capacity or efficient knowledge access, but their stored representations are usually trained and consumed within the same model. Engram differs in the regime relevant here: deterministic hashing provides an explicit address space, the memory table is external and addressable, and a small learned reader mediates read by the backbone. This separation makes a frozen memory artifact well defined and allows portability to be tested as extraction rather than as additional model capacity. A broader qualitative comparison appears inTable7.
2.3Cross-Model Alignment
Cross-model transfer is often studied as representation or adapter alignment. Embedding-space work suggests that independently trained models can sometimes be connected by simple linear maps between hidden representations[Mikolov et al., 2013,Kornblith et al., 2019,Huh et al., 2024,Chen et al., 2025]. PEFT transfer methods such as Cross-LoRA[Xia et al., 2025], LoRA-X[Farhadzadeh et al., 2025], and Trans-LoRA[Wang et al., 2024]show that compact adapters or low-rank updates can sometimes be reused across heterogeneous models. These methods are useful baselines for parameter transfer, but they do not directly address frozen external-memory reuse: there is no separate memory table to address, freeze, and extract from after the source backbone is removed. In frozen-memory extraction, alignment starts earlier. The target model must first read the same memory entry as the source-intended lookup, even when their tokenizers differ. Otherwise, the same surface text may be routed to unrelated memory slots, and any downstream hidden-space mapping becomes ill-defined. Our method, therefore, treats alignment as a two-stage problem: tokenizer-agnostic canonicalization aligns the key space, and a lightweight target-side reader aligns the retrieved value space to the target residual stream.
3Methods
Our method recasts Engram as a transferable artifact. We freeze the source memory and fit only a small target-side reader. This requires a shared address space, frozen memory, and a lightweight reader. Relative to native Engram, we make three changes: we (1) freeze the exported source memory, (2) replace model-specific token-ID lookup with tokenizer-agnostic canonicalization, and (3) use a lightweight target-side reader attached by residual injection to read the knowledge from Engram memory.Figure1illustrates the overall architecture and training and adaptation procedures, andAlgorithm1summarizes the training procedure.
Figure 1:Overview of the cross-model memory transfer. A memory table trained with source modelAAis frozen and attached to target modelBB. Canonicalization keeps the address space fixed across tokenizers, and only the target-side reader (including multi-head and multi-layer variants) is trained. This setup operationalizes whether external memory remains reusable outside its source backbone.### 3.1Tokenizer-Agnostic Addressing
The first obstacle to transfer is the address mismatch. Native Engram hashes model-specific token IDs inside the host model, which is sufficient when storage and read happen under one tokenizer, but does not define a stable cross-model address space. If source and target tokenizers segment the same text differently, then identical surface content may be routed to different memory entries. To avoid this failure mode, we build the lookup space on canonicalized decoded text rather than raw token IDs. Let𝒯B\mathcal{T}_{B}be the tokenizer of target modelBB, and lets1:t=decode(𝒯B(x1:t))s_{1:t}=\mathrm{decode}(\mathcal{T}_{B}(x_{1:t}))denote the decoded target prefix up to positiontt. We normalize that prefix with a shared canonicalization function𝒫\mathcal{P}and extract the completed word sequence
(w1,…,wm(t))=words(𝒫(s1:t)),(w_{1},\ldots,w_{m(t)})=\mathrm{words}\!\left(\mathcal{P}(s_{1:t})\right),(3)wherewords(⋅)\mathrm{words}(\cdot)returns the canonicalized word-boundary units implied by the current decoded prefix. Word-boundary addressing is the default for the English-like settings above, but it degenerates for scripts without reliable whitespace segmentation. For such inputs, we replace word units with NFKC-normalized, case-folded non-whitespace Unicode character events and hash character spans using the same deterministic addressing interface.AppendixEevaluates this variant on Chinese and Japanese. In practice,𝒫\mathcal{P}applies NFKC normalization, lowercasing, accent stripping, and whitespace cleanup. Memory lookup is then performed over canonical word-boundarynn-grams rather than model-specific token IDs:
𝐞t=⨁n=2N⨁j=1K𝐄n,j[φn,j(wm(t)−n+1,…,wm(t))].\mathbf{e}_{t}=\bigoplusop\displaylimits_{n=2}^{N}\bigoplusop\displaylimits_{j=1}^{K}\mathbf{E}_{n,j}\!\left[\varphi_{n,j}(w_{m(t)-n+1},\ldots,w_{m(t)})\right].(4)At subword positions inside the same decoded word, the current token reuses the same right-edge word context until a new word boundary appears. The retrieved vector𝐞t\mathbf{e}_{t}is therefore determined by canonicalized text and the frozen memory artifact, not by target-specific token IDs. Two target models with different tokenizers can still read the same transferred entry as long as they share the same canonicalization and hash specification.
Algorithm 1Cross-Model Frozen-Memory ExtractionInput:Source memoryℰA\mathcal{E}_{A}, target modelBB, target corpus𝒟\mathcal{D}, selected injection layersℒ\mathcal{L}, branch countRR, canonicalization𝒫\mathcal{P}
Output:Target model
BBaugmented with frozen transferred memory
//// Phase 1 artifact is already trained; prepare transfer
1ExtractEngram tables
{𝐄n,j}\{\mathbf{E}_{n,j}\}and hash specification from
ℰA\mathcal{E}_{A};
2Freezeall transferred tables{𝐄n,j}\{\mathbf{E}_{n,j}\}[portable memory artifact];
//// Initialize target-side reader and hooks
3foreach*ℓ∈ℒ\ell\in\mathcal{L}*do
4Initializereader parameters
𝒲(B,ℓ)\mathcal{W}^{(B,\ell)};
5Set
𝒲(B,ℓ)←{𝐖K,ℓ,r(B),βℓ,r}r=1R,𝐖V,ℓ(B)\mathcal{W}^{(B,\ell)}\leftarrow\{\mathbf{W}_{K,\ell,r}^{(B)},\beta_{\ell,r}\}_{r=1}^{R},\mathbf{W}_{V,\ell}^{(B)};
6ifR=1R=1then
7Set
βℓ,1←0\beta_{\ell,1}\leftarrow 0;
8Registera forward hook at layer
ℓ\ellthat computes
𝐨t,ℓ\mathbf{o}_{t,\ell}from the current
𝐡t,ℓ\mathbf{h}_{t,\ell}and looked-up
𝐞t\mathbf{e}_{t}, then injects
𝐡t,ℓ←𝐡t,ℓ+𝐨t,ℓ\mathbf{h}_{t,\ell}\leftarrow\mathbf{h}_{t,\ell}+\mathbf{o}_{t,\ell};
//// Phase 2: Reader-only adaptation
9for*x1:T∈𝒟x_{1:T}\in\mathcal{D}*do
10Canonicalizeeach decoded prefix into shared word-boundary
nn-grams with
𝒫\mathcal{P};
11Look upfrozen memory vectors
𝐞1:T\mathbf{e}_{1:T}from
{𝐄n,j}\{\mathbf{E}_{n,j}\};
12Runthe frozen backbone on
x1:Tx_{1:T}with the registered hooks active;
13Update onlyreader parameters{𝒲(B,ℓ)}ℓ∈ℒ\{\mathcal{W}^{(B,\ell)}\}_{\ell\in\mathcal{L}}using the LM loss[backbone and memory frozen];
3.2Target-Side Reader
Once the address space is shared, the remaining problem is extraction: how should the target model consume a memory vector whose geometry was shaped by a different backbone? We use a shared-value reader family parameterized by branch countRR, where a largerRRincreases target-side alignment capacity without increasing memory size. For branchr∈{1,…,R}r\in\{1,\ldots,R\}at layerℓ\ell:
𝐤t,ℓ(r)=𝐖K,ℓ,r(B)𝐞t,𝐯t,ℓ=𝐖V,ℓ(B)𝐞t,\mathbf{k}_{t,\ell}^{(r)}=\mathbf{W}_{K,\ell,r}^{(B)}\mathbf{e}_{t},\qquad\mathbf{v}_{t,\ell}=\mathbf{W}_{V,\ell}^{(B)}\mathbf{e}_{t},(5)αt,ℓ(r)=σ(RMSNorm(𝐡t,ℓ)⊤RMSNorm(𝐤t,ℓ(r))dB+βℓ,r),\alpha_{t,\ell}^{(r)}=\sigma\!\left(\frac{\RMSNorm(\mathbf{h}_{t,\ell})^{\top}\RMSNorm(\mathbf{k}_{t,\ell}^{(r)})}{\sqrt{d_{B}}}+\beta_{\ell,r}\right),(6)𝐨t,ℓ=1R∑r=1Rαt,ℓ(r)⋅𝐯t,ℓ.\mathbf{o}_{t,\ell}=\frac{1}{R}\sumop\displaylimits_{r=1}^{R}\alpha_{t,\ell}^{(r)}\cdot\mathbf{v}_{t,\ell}.(7)The resulting reader output is injected through a residual connection:
𝐡t,ℓ←𝐡t,ℓ+𝐨t,ℓ.\mathbf{h}_{t,\ell}\leftarrow\mathbf{h}_{t,\ell}+\mathbf{o}_{t,\ell}.(8)The shared value projection forces all branches to operate on the same retrieved content and differ only inhowthey gate and align it. The added capacity is therefore an extraction mechanism rather than extra storage. The main3×33\times 3transfer matrix usesR=1R=1; the stronger QA reader usesR=4R=4.
Multi-layer injection.The same reader family can be instantiated at one or more target layersℒ\mathcal{L}. Single-layer injection is sufficient for the cross-architecture transfer matrix, whereas the best QA configuration uses dual-layer injection at layers 2 and 10 withR=4R=4branches.
3.3Training Regime
The training objective is the standard next-token language modeling loss on the target corpus. During target-side adaptation, both the exported source memory tables{𝐄n,j}\{\mathbf{E}_{n,j}\}and the target backbone parametersθB\theta_{B}remain frozen; only the reader parameters{𝒲(B,ℓ)}ℓ∈ℒ\{\mathcal{W}^{(B,\ell)}\}_{\ell\in\mathcal{L}}are updated. Storage is fixed, and only extraction is learned. Algorithm1summarizes the procedure concisely. The key operations are to freeze the exported source memory, canonicalize target inputs into the shared address space, and update only the target-side reader.
Deployment modes.
The two-phase protocol above describes the general case in which the target requires a model-specific reader. We additionally distinguish between two deployment modes. Indirect artifact reuse, a provider exports the frozen memory together with a reader whose residual interface is compatible with the consumer model; the consumer performs no target-side optimization. Inreader-adapted reuse, the same frozen memory is retained, but a lightweight target-specific reader is fitted on an available adaptation stream. The first mode minimizes consumer-side cost, while the second provides additional alignment capacity when direct reader reuse is insufficient. Both modes preserve the defining property of the protocol: the transferred memory table itself is never rewritten.
4Experiments and Results
In this section, we address the effect of cross-model memory transfer through 5 research questions. All experiments follow the same protocol: a source memory is trained (Phase 1), then frozen and attached to a target model where only a reader is trained (Phase 2) as illustrated inFigure1. We evaluate on intrinsic language modeling and downstream QA tasks. Intrinsic evaluations use the memory transfer architecture with a minimal reader, while QA evaluation uses a stronger multi-layer, multi-branch reader. Implementation details are inAppendixB.
4.1RQ1: Does frozen memory transfer across backbones, tokenizers, and scales?
Our first question is whether the exported memory remains useful once detached from the source backbone. We start with a3×33\times 3source–target matrix spanning three source memories (Pythia-160M[Biderman et al., 2023], Qwen3.5-0.8B, Qwen3.5-9B[Qwen Team, 2025]) and three target models (Pythia-410M, Qwen3.5-4B, TinyLlama-1.1B[Zhang et al., 2024]), all trained with the same 20M-token target-side protocol.AppendixCtherefore stress-tests the same conclusion against several alternative explanations. It reports the exact matrix values, introduces parameter-matched and retrieval-based comparisons, removes the small-to-large transfer asymmetry through peer-to-peer transfer, tests whether gains persist as the target scales, and examines whether representational similarity predicts transfer success.
(a)Full3×33\times 3source–target transfer matrix, reported as relative PPL reduction over the no-memory baseline.
(b)Target-scale intrinsic transfer: the same transferred Qwen3.5-0.8B memory improves Qwen3.5 targets from 2B to 9B.
Figure 2:The full 3x3 transfer matrix and target-scale transfer performance across different model types trained on Wikitext-103.Figure2summarizes this result together with the scaling trend. All nine cells in the main matrix inFigure2(a)show PPL reduction in models with transferred memory over the no-memory baseline, and the same transferred Qwen3.5-0.8B memory continues to help as the target grows from 2B to 9B (Figure2(b)), compared with the baseline without memory and the random memory control with a randomly initialized, untrained memory of the same architecture and size as the transferred memory.
The full matrix inFigure2(a)gives a positive answer in every one of the nine cells. Transferred memory improves over the no-memory baseline in all source–target pairs, with relative gains from 1.6% to 15.7%. The strongest gain appears in the Qwen3.5-0.8B→\toTinyLlama-1.1B case, showing that large improvements do not require the source and target to share an architecture or a tokenizer.
This is also not only an intrinsic-perplexity effect:Figure3later shows selective downstream gains for Qwen3.5 targets from 0.8B to 9B, andTable16shows that the same target family retains that pattern under two different source-corpus families. On Qwen3.5-2B specifically,Figure4shows the same asymmetry in a more focused form: transferred memory stays ahead of scratch on intrinsic PPL across all three budgets (Figure4(b)), while the downstream gains are selective rather than universal, with the clearest best-of-budget improvements appearing on BoolQ, RTE, and SciQ (Figure4(a)).
One subtle pattern is that source scaling helps, but not monotonically, for every target. Qwen3.5-0.8B improves over Pythia-160M on all three targets, yet Qwen3.5-9B does not always beat Qwen3.5-0.8B on the most dissimilar target (Pythia-410M), suggesting that transfer depends not only on what the source memory stores, but also on what the target reader can successfully extract. This underscores the importance of the reader’s extraction capability, as supplementary results inAppendixCconfirm that transfer is effective in peer-to-peer settings (SectionC.3) and even between backbones with low representational similarity (SectionC.5).
Takeaway.Frozen memory transfer remains effective across backbone families, tokenizer boundaries, and target scales, with gains in all nine main source–target pairs; the same Qwen3.5 target family also retains selective downstream gains beyond perplexity, and successful transfer ultimately depends on whether the target reader can extract the stored structure.
4.2RQ2: How much does reader design matter?
The previous experiments established that memory transfer remains effective even across model families. However, they do not explain where the transfer gains actually come from. One possibility is that most of the benefit is already contained in the transferred memory itself. An alternative explanation is that transfer quality is primarily limited by the target-side reader, i.e., the mechanism responsible for extracting and integrating information from the frozen memory.
To distinguish between these possibilities, we keep the transferred memory fixed and vary only the reader architecture. Specifically, we study two aspects of reader design: (i) memory placement, namely the layers at which memory is injected into the target backbone, and (ii) retrieval capacity, controlled through the number of reader branches. Our central finding is that reader design is a first-order factor: most of the gain comes from placing the reader at the right layers, while multiple branches further improve performance by accessing the same memory from different perspectives.
Table 1:QA accuracy (%) on Mistral-7B-v0.3 for different reader variants and baselines. Budget notation followsTable2trained on Wikipedia-2021[Izacard et al., 2023].VariantBudgetNQWebQATriviaQATruthQAHotpotQAAverageBase–20.629.357.732.121.032.1Non-parametric MethodsRAG[Lewis et al., 2020]–22.61.924.9-4.454.2-3.435.53.429.88.833.4(+3.9%)kNN[Khandelwal et al., 2020]–21.10.430.51.257.80.132.30.221.20.232.6(+1.4%)Parametric MethodsCPT–12.2-8.534.14.861.23.629.2-2.916.0-4.930.5(-5.0%)LoRA[Hu et al., 2022]–18.2-2.534.55.261.64.030.9-1.216.2-4.732.3(+0.5%)MLP Memory[Wei et al., 2026]–25.24.637.58.261.03.332.50.524.13.236.1(+12.3%)Engram Variants (sorted by Average)LLaMA Frozen{10}\{10\}-R110/021.81.229.60.360.62.933.11.122.31.433.5(+4.4%)Mistral Frozen{10}\{10\}-R110/021.91.229.70.460.83.132.90.822.41.433.5(+4.4%)Mistral{10}\{10\}-R110/2022.11.531.62.461.13.432.80.722.31.334.0(+5.8%)LLaMA{10}\{10\}-R110/2022.21.532.63.361.23.632.90.822.31.334.2(+6.6%)LLaMA{2,10}\{2,10\}-R110/2030.39.728.7-0.670.112.431.0-1.127.46.437.5(+16.7%)LLaMA{2,10}\{2,10\}-R410/2030.39.733.74.469.912.330.9-1.227.66.738.5(+19.9%)The first pattern is that a weak or frozen reader improves only a small fraction of the performance. With a single injection layer at{10}\{10\}, the frozen variants achieve only 33.5–33.5 average accuracy, a modest improvement over the 32.1 base model. Training the same single-layer reader raises performance to 34.0–34.2, indicating that target-side adaptation already contributes noticeably even before increasing reader capacity. This suggests that the transferred memory alone is insufficient; the target model must learn how to access it effectively.
The largest gain comes from reader placement. Moving from LLaMA{10}\{10\}-R1 to LLaMA{2,10}\{2,10\}-R1 increases average accuracy from 34.2 to 37.5. The improvement is broad rather than dataset-specific: NQ rises from 22.20 to 30.3, TriviaQA from 61.2 to 70.1, and HotpotQA from 22.3 to 27.4. These results suggest that dual-layer injection provides the target model with multiple opportunities to integrate retrieved memory throughout the forward pass, instead of forcing all memory interaction through a single late-layer interface.
Reader capacity explains the remaining improvement. Keeping the injection layers fixed at{2,10}\{2,10\}and increasing the number of branches fromR=1R=1toR=4R=4raises average accuracy from 37.5 to 38.5. The most visible gain appears on WebQA (28.7 to 33.7), while the strong improvements on NQ, TriviaQA, and HotpotQA are preserved. This behavior is consistent with the intended role of the multi-branch design: different branches provide complementary key-gating pathways over the same shared value representation. The branches, therefore, increase the flexibility with which the target conditions the contribution of a retrieved memory vector, rather than introducing independent stored value directions.
Reader structure is not the only capacity factor. We additionally vary the frozen-memory interface width in a controlled Pythia-160M→\rightarrowPythia-410M diagnostic. Increasing the memory width from 256 to 512 and 1024 reduces test perplexity from 22.368 to 22.157 and 21.808, respectively. However, a width-512 reader injected at two layers reaches 21.522, outperforming the wider width-1024 single-layer configuration. Thus, the default 512-dimensional interface is a design choice rather than a hard architectural bottleneck, and extraction capacity depends jointly on interface width and reader structure. Because changing the width also changes both memory and reader parameter counts, we treat this experiment as a capacity-sensitivity analysis rather than a causal estimate of memory width alone.
The same table also places reader improvements in context relative to existing approaches. The strongest transfer configuration, LLaMA{2,10}\{2,10\}-R4, achieves 38.5 average accuracy, outperforming all included baselines, including RAG, kNN-LM, CPT, LoRA, and MLP Memory. Notably, this improvement is obtained without modifying the transferred memory itself, reinforcing the conclusion that extraction quality, rather than memory storage, is the dominant factor governing transfer performance.
Takeaway.Reader design is a first-order determinant of transfer quality. Dual-layer injection recovers most of the available gain, while multi-branch gating provides an additional improvement by allowing the same retrieved value representation to be conditioned through multiple target-dependent key-gate pathways. Together, these design choices raise average QA accuracy from 34.2 to 38.5 without changing the transferred memory itself.
4.3RQ3: When does transferred memory help downstream?
Table 2:QA accuracy (%) on Mistral-7B-v0.3 for transfer verification and ablation studies. Deltas are measured against the baseline. Budget==P1/P2 denotes source-memory and target-reader training tokens (millions) trained on Wikipedia-2021.VariantBudgetNQWebQATriviaQATruthQAHotpotQAAverageBase–20.629.357.732.121.032.1Transfer VerificationMistral{2,10}\{2,10\}-R410/2030.29.534.04.770.012.330.8-1.327.76.838.5(+20.0%)LLaMA{2,10}\{2,10\}-R410/2030.39.733.74.469.912.330.9-1.227.66.738.5(+20.0%)Frozen Mistral{2,10}\{2,10\}-R410/030.59.830.31.070.813.232.20.227.76.838.3(+19.2%)Frozen LLaMA{2,10}\{2,10\}-R410/030.610.030.31.070.813.132.20.127.66.738.3(+19.2%)Controls and AblationsMem-only Mistral10/00.2-20.40.3-29.01.4-56.333.00.90.2-20.77.0(-78.1%)Mem-only LLaMA10/00.0-20.60.0-29.30.0-57.7–0.0-21.0–Permuted10/020.60.028.9-0.463.15.432.40.323.12.233.6(+4.7%)FFN R410/2019.6-1.027.3-2.058.10.532.1-0.017.6-3.430.9(-3.7%)Token Scaling (LLaMA2-7B source,{2,10}\{2,10\}-R4)Engram10/1030.29.529.90.670.412.831.2-0.927.66.737.9(+17.8%)Engram15/1530.29.632.73.470.112.430.6-1.527.86.838.3(+19.2%)Engram20/2030.39.733.94.669.912.330.7-1.427.66.738.5(+19.8%)Engram25/2530.59.933.13.870.012.330.9-1.227.76.738.4(+19.6%)Engram30/3030.610.033.94.670.412.731.1-1.027.97.038.8(+20.7%)RQ1–RQ2 established that memory can be transferred across model families and that transfer quality is largely determined by the target-side reader. The remaining question is not whether transfer works, but rather how far it can be pushed and where it begins to break down.
We study this question from two complementary perspectives. First, we use QA benchmarks to estimate the upper bound of transfer performance under increasingly strong reader configurations. Second, we evaluate whether these gains generalize beyond QA to broader knowledge-intensive tasks. Throughout this section,{a,b}\{a,b\}-Rrrdenotes memory injection at layers{a,b}\{a,b\}withrrretrieval branches, and “Frozen” indicates evaluation with the source reader without target-side adaptation.
Table2reveals a clear and surprisingly high transfer ceiling. Under the strongest configuration ({2,10}\{2,10\}-R4), same-model and cross-model transfer achieve nearly identical performance (38.5 vs. 38.5 average accuracy), achieving a ceiling once the reader is sufficiently expressive. Performance further saturates with scale: increasing training budgets from 10/10 to 30/30 improves accuracy from 37.9 to 38.8, suggesting diminishing returns in the high-38 regime.
Control results confirm that this ceiling is not driven by parameter addition alone. Memory-only variants collapse without a reader, and FFN-based substitutes fall below the base model, while permuting memory keys removes most gains. These results indicate that successful transfer critically depends on the correct retrieval and integration of frozen memory rather than additional capacity. The frozen-reader rows additionally expose a second deployment regime. When the provider reader is directly compatible with the target residual interface, the complete frozen artifact can be reused without target-side training: mean QA remains approximately 38.3 compared with 32.1 for the no-memory target. Optional target-reader fitting further raises the score to approximately 38.5. Thus, target-side adaptation is useful but is not an inherent requirement for every compatible source–target pair.
We next examine whether these gains extend beyond QA to a broader set of downstream tasks.
Figure3shows that transfer benefits are selective rather than universal. The high QA ceiling does not imply that transferred memory improves every downstream objective. We therefore evaluate whether the benefits extend beyond the five QA benchmarks to a broader set of knowledge-intensive and reasoning tasks. The strongest and most consistent gains appear on tasks for which additional factual or evidence-related information is likely to be useful. RTE improves at every evaluated target scale, with gains ranging from+0.7+0.7to+3.7+3.7accuracy points. SciQ also remains positive across scales, with improvements of up to+3.7+3.7points. OpenBookQA shows smaller but consistently positive gains, while BoolQ becomes positive for target models of 2B parameters and above. Together, these results suggest that transferred memory is most useful when the downstream decision can benefit from additional factual associations or evidence support.
By contrast, the gains are substantially weaker on tasks whose success is less directly tied to factual retrieval. RACE remains close to zero across model sizes, suggesting that the imported memory contributes little to broad reading-comprehension performance when retrieving factual associations is not the primary bottleneck.
Figure 3:Downstream task evaluation. Bars report the change in accuracy (�\Delta) produced by transferred memory relative to the no-memory baseline across six downstream tasks. Results are averaged over five seeds. Positive values indicate that transferred memory improves the target model after target-side adaptation on FineWeb-Edu (FW-9B)[Lozhkov et al.,].TruthfulQA provides the clearest negative boundary. Performance decreases at every target scale, with changes ranging from−0.2-0.2to−0.8-0.8accuracy points. Unlike the other evaluated tasks, TruthfulQA rewards calibrated truthfulness and resistance to commonly repeated misconceptions, rather than the recovery of frequently observed factual associations alone. Additional memory-derived associations can therefore be unhelpful when the task requires the model to reject plausible but misleading information. This result defines an important practical limit: transferred memory is most beneficial when the task rewards access to factual or evidence-related information, but it may be neutral or mildly harmful when success depends on calibration or resistance to memorized misconceptions.
Figure3establishes this task-dependent pattern, but does not by itself identify why it arises. The observed selectivity could reflect the information encoded in the source memory, the distribution used to fit the target-side reader during Phase 2, or a trade-off between specialization and broader distributional robustness. To separate these factors,AppendixDpresents a controlled diagnostic sequence. It first varies source-corpus specialization while matching the Phase 2 fitting distribution, then holds the source memory fixed while varying the reader-fitting corpus. It further examines whether mixed-corpus memory can broaden coverage without removing specialist gains, and finally evaluates whether the adapted reader introduces adverse effects on unrelated language-modeling distributions.
These supplementary experiments show that source-memory content and Phase 2 distribution alignment affect the magnitude of the realized gains, but do not remove the central downstream boundary. Improvements remain concentrated on tasks that can exploit factual or evidence-related associations, and do not extend uniformly to broader reading comprehension, reasoning, or truthfulness-oriented evaluation.
Takeaway.Memory transfer has a high practical ceiling: with a sufficiently expressive target-side reader, cross-model transfer nearly matches same-model reuse and saturates at approximately 38.5–38.8 average QA accuracy. This ceiling, however, is task-dependent rather than universal. Transferred memory provides the clearest gains on factual and evidence-oriented tasks, has little effect on broader reading comprehension, and can be mildly harmful on TruthfulQA, where calibrated resistance to misleading associations is important.
4.4RQ4: Does transferred memory improve answer-token support in question answering?
Our fourth question is whether transferred memory measurably improves support for the correct answer, rather than merely changing final task accuracy in a hard-to-interpret way. Byanswer-token support, we mean the log-probability that the target model assigns to a valid gold-answer continuation at the positions where that answer is generated. To test this, we compare four inference settings on the same LLaMA-2-7B→\toMistral-7B dual-layer QA system used inTable3: transferred memory, random memory, memory disabled, and an answer-position memory ablation. We compute the length-normalized gold-answer log-probability for an aliasaaas:
sc(i,a),1La∑t=1Lalogpc(yi,ta∣xi,yi,<ta),s_{c}(i,a)\triangleq\frac{1}{L_{a}}\sumop\displaylimits_{t=1}^{L_{a}}\log p_{c}\left(y^{a}_{i,t}\mid x_{i},y^{a}_{i,<t}\right),(9)wherepcp_{c}is the target model distribution under conditionc∈{trans,rand,disabled,ablated}c\in\{\mathrm{trans},\mathrm{rand},\mathrm{disabled},\mathrm{ablated}\}, andLaL_{a}is the token length of aliasaa. The probability is computed with teacher forcing: the model is given the prompt and the gold-answer prefix, and we evaluate the log-probability assigned to the next gold-answer token. For questions with multiple valid aliases𝒜i\mathcal{A}_{i}, the question-level score is:
sc(i),maxa∈𝒜isc(i,a).s_{c}(i)\triangleq\max_{a\in\mathcal{A}_{i}}s_{c}(i,a).(10)For datasets with a single annotated answer,𝒜i\mathcal{A}_{i}contains only that answer. Finally, the mean score difference over the dataset𝒟\mathcal{D}is:
�logp(c1−c2),1|𝒟|∑i∈𝒟(sc1(i)−sc2(i)),\Delta\log p(c_{1}-c_{2})\triangleq\frac{1}{|\mathcal{D}|}\sumop\displaylimits_{i\in\mathcal{D}}\left(s_{c_{1}}(i)-s_{c_{2}}(i)\right),(11)where𝒟\mathcal{D}is the evaluation set. InTable3, we report�logp(trans−rand)\Delta\log p(\mathrm{trans}-\mathrm{rand}),�logp(trans−disabled)\Delta\log p(\mathrm{trans}-\mathrm{disabled}), and�logp(trans−ablated)\Delta\log p(\mathrm{trans}-\mathrm{ablated}). The first two differences ask whether transferred memory helps more than either an untrained table or no memory at all. The third asks whether zeroing the memory channel only at answer-prediction positions reduces gold-answer likelihood. For NQ and TriviaQA, where many examples have multiple valid aliases, the analysis scores all annotated aliases and keeps the best-valid gold continuation rather than forcing the first alias in the dataset.
Table 3:QA memory-contribution trained on Wikipedia-2021 analysis for LLaMA-2-7B[Touvron et al., 2023]→\toMistral-7B-v0.3[Jiang et al., 2023]with dual-layer injection. The�logp\Delta\log pcolumns report mean per-example differences inlength-normalizedgold-answer log-probability. Positive values indicate that transferred memory makes the gold answer more likely.TaskTransferredRandomDisabledAblated�logp\Delta\log p(Trans. - Random)�logp\Delta\log p(Trans. - Disabled)�logp\Delta\log p(Trans. - Ablated)NQ25.120.118.326.2+0.035+0.032-0.013WebQA32.327.431.133.6+0.147+0.197-0.001TriviaQA72.565.558.972.5+0.047+0.072-0.000TruthQA30.832.631.730.9-0.581-0.258-0.029HotpotQA27.122.918.627.2+0.022+0.019-0.009Transferred memory beats random memory and memory-disabled evaluation on four of the five QA subsets, both in task score and in gold-answer log-probability. The cleanest gains appear on WebQA and TriviaQA. On WebQA, transferred improves mean gold-answer log-probability by+0.147+0.147over random memory and+0.197+0.197over memory-disabled inference. On TriviaQA, after fixing alias-aware gold-answer scoring, transferred also improves gold-answer log-probability by+0.047+0.047over random memory and+0.072+0.072over memory-disabled inference, aligning the logit-level analysis with the task-level F1 gains. NQ and HotpotQA show the same direction with smaller positive margins.
TruthfulQA remains the main negative case. Transferred is worse than both random memory and memory-disabled inference there, both in MC-average and in gold-answer log-probability. This matches the broader boundary condition already visible in RQ3: transferred memory is most useful on factual retrieval-style QA, but not on truthfulness-style calibration.
The ablation column sharpens the interpretation. For every task, transferred minus answer-position ablation is near zero or slightly negative rather than strongly positive. That means the main benefit is not coming from a simple direct injection of the correct answer token at the final prediction positions. Instead, the evidence points to an indirect effect: transferred memory improves the model’s overall answer-supporting state on factual QA tasks, but direct answer-token memory contribution is weak and sometimes slightly harmful.
Taken together, RQ4 supports a narrower claim than “memory use causes the gain”. The gain is real on NQ, WebQA, TriviaQA, and HotpotQA, and the corrected alias-aware analysis strengthens that claim for TriviaQA in particular. The results are therefore more consistent with improved factual support after target-side integration than with raw answer-position copying.
Takeaway.Transferred memory improves answer-token support selectively but positively: transferred memory increases gold-answer support on NQ, WebQA, TriviaQA, and HotpotQA relative to both random memory and no memory, with the corrected alias-aware TriviaQA analysis removing the main contradictory case, while the near-zero answer-position ablation effects indicate that the gain comes more from useful factual support than from direct answer-token injection.
4.5RQ5: How data-efficient is frozen-memory transfer?
We first ask whether frozen-memory transfer reduces the amount of target-side training required to reach competitive performance. Across both Pythia and Qwen model pairs, transferred memory reaches strong intrinsic performance substantially earlier than learning a new memory from scratch, indicating improved target-side data efficiency. The downstream gains are also positive, although their magnitude remains task-dependent.
Controlled scaling on Pythia.
Table4provides a controlled comparison on Pythia-160M→\rightarrowPythia-410M by tracking the same transfer setting across 5M, 20M, and 50M target-side training tokens.
Table 4:Scaling analysis for frozen-memory transfer on Pythia-160M→\rightarrowPythia-410M. Results report test perplexity as a function of target-side training tokens, averaged over 3 seeds.ConditionTrainable Params5M20M50MTransferred1.05M21.8\mathbf{21.8}21.5\mathbf{21.5}21.5†\mathbf{21.5}^{\dagger}From scratch34.6M21.921.921.821.821.621.6 †Early stopped at step 1,500 (≈\approx12M tokens); validation PPL diverges beyond this point.
Transferred memory is already competitive after only 5M target-side tokens and essentially saturates by 20M, reaching a test PPL of 21.5. In contrast, training a new target memory from scratch improves more gradually and approaches a comparable PPL only at the largest token budget. This difference is particularly notable because the transferred condition updates only 1.05M trainable parameters, compared with 34.6M for the from-scratch condition, corresponding to roughly3%3\%as many trainable parameters.
The 50M transferred entry is reported at its best early-stopped checkpoint. Although the nominal training budget is 50M tokens, validation perplexity begins to diverge after approximately 12M tokens, and the best checkpoint therefore matches rather than improves upon the 20M result. Taken together, these results suggest that Phase 2 is primarily adapting and reusing structure already learned in the source memory rather than relearning an equivalent memory representation from the target corpus alone.
Scaling to a larger target model.
We next test whether the same trend persists at a larger scale using Qwen3.5-0.8B→\rightarrowQwen3.5-2B.Figure4reports both intrinsic perplexity and downstream QA performance as the amount of target-side training data increases.
(a)Qwen3.5-2B downstream QA scaling: average zero-shot accuracy across BoolQ, RTE, and SciQ as a function of target-side training tokens.(b)Qwen3.5-2B intrinsic scaling: test perplexity versus target-side training tokens for scratch and transferred memory. Figure 4:Target-side scaling for Qwen3.5-0.8B→\rightarrowQwen3.5-2B with the source memory trained on WikiText-103. Transferred memory maintains a consistent intrinsic perplexity advantage, while downstream improvements remain positive but task-dependent.The Qwen results reinforce the Pythia scaling study. Transferred adaptation achieves lower test perplexity than scratch training across all evaluated target-side token budgets, showing that the intrinsic data-efficiency advantage is not restricted to the smaller Pythia setting. The downstream results show a similar but less uniform trend: transferred memory improves BoolQ, RTE, and SciQ by+1.5+1.5,+2.5+2.5, and+0.8+0.8points, respectively.
These results also clarify the distinction between intrinsic adaptation and downstream utility. The perplexity advantage of transfer is stable across token budgets, whereas the magnitude of downstream improvement depends on the task. Thus, transferred memory provides a consistently better target-side initialization, but the extent to which this advantage translates into downstream accuracy still depends on whether the reader can extract and route the stored signal in a task-compatible manner.
Takeaway.Frozen-memory transfer is target-data efficient: it reaches strong perplexity with substantially fewer target-side updates than learning a new memory from scratch, while downstream gains remain positive but task-dependent.
4.6RQ6: What drives the gains from memory transfer?
Having established the target-side data-efficiency advantage of transfer, we next ask what actually drives the downstream gains. In particular, we test whether the improvement can be explained simply by adding trainable target-side parameters, or whether successful transfer depends on the transferred memory content, its addressing structure, and the reader used to access it.
We study this question on LLaMA-2-7B[Touvron et al., 2023]→\rightarrowMistral-7B-v0.3[Jiang et al., 2023]using the same dual-layer QA setting that produces our strongest downstream results.Table5compares the full transferred reader against controls designed to isolate the main components of the transfer mechanism.
“Transferred (R=4R{=}4)” denotes the full dual-layer four-branch reader. “No gate” removes context-dependent gating and always injects memory values. “Affine stitch” replaces the reader with a single affine memory-to-hidden-state mapping. “Permuted keys” preserves the memory table but destroys addressing integrity through index permutation. “Random memory” replaces the learned memory table with an untrained one. “Train from scratch” learns a fresh target memory under the same downstream training setup. Finally, “FFN only” replaces memory lookup with a parameter-matched residual feed-forward module, while “No memory baseline” corresponds to the underlying Mistral-7B model without memory augmentation.
Table 5:Ablation study of frozen-memory transfer on LLaMA-2-7B→\rightarrowMistral-7B-v0.3, with the source memory trained on Wikipedia-2021. Results report held-out test perplexity and downstream QA performance.ConditionTest PPL↓\downarrow95% CINQWebQATriviaQATruthQAHotpotQAAvg↑\uparrowNo memory baseline9.99.9[9.8,10.1][9.8,10.1]20.620.629.329.357.757.732.132.121.021.032.132.1Transferred (R=4R{=}4)8.78.7[8.6,8.9][8.6,8.9]30.330.333.733.769.969.930.930.927.627.638.538.5Interface SimplificationsNo gate10.110.1[9.9,10.3][9.9,10.3]24.224.227.727.761.661.631.031.024.124.133.733.7Affine stitch9.59.5[9.4,9.7][9.4,9.7]25.025.029.229.262.862.830.430.423.923.934.334.3Content and Training ControlsPermuted keys8.78.7[8.5,8.8][8.5,8.8]18.318.334.534.557.757.730.630.621.421.432.532.5Random memory8.78.7[8.6,8.9][8.6,8.9]23.023.027.127.165.165.133.633.624.124.134.634.6Train from scratch8.18.1[8.0,8.2][8.0,8.2]29.929.933.033.070.470.430.930.927.927.938.438.4No-memory ControlFFN only (param-matchedR=4R{=}4)7.47.4[7.2,7.5][7.2,7.5]25.525.527.927.964.164.131.631.623.623.634.534.5
Parameter count alone does not explain the gains.
The full transferred reader improves average QA from 32.1 for the no-memory baseline to 38.5. More importantly, it clearly outperforms the parameter-matched FFN control, which reaches only 34.5 average accuracy. The difference cannot be attributed to intrinsic language-model fit: the FFN-only control actually obtains the lowest perplexity in the table (7.4 versus 8.7 for transfer), yet remains 4.0 points behind on average QA.
The task-level results show the same pattern. The transferred reader outperforms the parameter-matched FFN on NQ, WebQA, TriviaQA, and HotpotQA, with particularly clear margins on NQ (30.330.3versus25.525.5), TriviaQA (69.969.9versus64.164.1), and HotpotQA (27.627.6versus23.623.6). The FFN control is slightly stronger only on TruthQA. Thus, simply attaching an equally parameterized target-side residual module does not reproduce the downstream behavior of transferred memory.
The reader interface is critical.
Removing context-dependent gating reduces average QA from 38.5 to 33.7, a drop of 4.8 points. Replacing the reader with a single affine stitch similarly reduces the average to 34.3, a 4.2-point decrease. These results indicate that transfer quality depends not only on the presence of a memory table, but also on a sufficiently expressive interface for selecting, transforming, and routing retrieved information into the target model.
Addressing integrity and memory structure matter.
Permuting the memory keys while retaining the same underlying table reduces average QA from 38.5 to 32.5. This is the largest degradation among the memory-content controls and shows that preserving the association between addresses and stored values is important for downstream reuse.
Replacing the transferred table with random memory also decreases average QA to 34.6. Interestingly, random memory achieves the same test PPL as the transferred condition (8.7), yet performs substantially worse on downstream QA. This provides another example in which intrinsic perplexity alone does not predict whether a memory representation will be useful for downstream tasks.
Transfer primarily improves initialization rather than the asymptotic ceiling.
Training a fresh target memory from scratch eventually reaches an average QA score of 38.4, essentially matching the 38.5 obtained by transfer. We therefore do not interpret transfer as providing a substantially higher asymptotic performance ceiling. Instead, together with the scaling results inSection4.5, the evidence supports a more specific interpretation: the transferred memory provides a structured initialization that can reach useful target-side performance with substantially less adaptation than learning an equivalent memory representation from scratch.
This distinction also explains why perplexity and downstream QA do not rank all controls identically. For example, the parameter-matched FFN reaches the best intrinsic PPL but substantially worse QA, while scratch memory eventually matches transferred QA after sufficient target-side optimization. Successful transfer therefore depends on the interaction between memory structure, addressing, and the reader interface, rather than on parameter count or perplexity reduction alone.
Takeaway.The downstream gains cannot be explained by parameter count alone: successful transfer depends on meaningful memory addressing and a sufficiently expressive reader interface, while its main advantage over scratch training is better target-side data efficiency rather than a higher asymptotic ceiling.
5Conclusion
We study whether a learned external memory remains useful after it is detached from the backbone that trained it and attached to a different one. For Engram-style hashed memory, the answer is yes: frozen memory improves the target across tokenizers, hidden sizes, and architectures, including every cell of the main3×33\times 3transfer matrix and both directions of peer transfer. The main lesson is that portability depends not only on the frozen table but also on the target-side reader: stronger readers nearly close the same-model/cross-model gap on QA, and transferred memory clearly beats permuted-key and FFN-only controls while substantially outperforming random memory on the final QA evaluation. More broadly, cross-model frozen-memory extraction turns external-memory portability into an evaluation framework: a memory artifact should be judged by whether a new backbone can address it, align to it, and extract useful signal without retraining the stored table.
Acknowledgments
The authors wish to acknowledge CSC – IT Center for Science, Finland, for computational resources. Computational facilities were provided by the UTS eResearch High Performance Computer Cluster. Shaoxiong Ji gratefully acknowledges the support of Foundation PS through the PS Fellowship.
References
- Abouelenin et al. [2025]A. Abouelenin, A. Ashfaq, A. Atkinson, H. Awadalla, N. Bach, J. Bao, A. Benhaim, M. Cai, V. Chaudhary, C. Chen, et al.Phi-4-mini technical report: Compact yet powerful multimodal language models via mixture-of-loras.arXiv preprint arXiv:2503.01743, 2025.URLhttps://arxiv.org/pdf/2503.01743.
- Behrouz and Zhong [2025]A. Behrouz and P. Zhong.Titans: Learning to memorize at test time.arXiv preprint arXiv:2501.00663, 2025.URLhttps://arxiv.org/abs/2501.00663.
- Berant et al. [2013]J. Berant, A. Chou, R. Frostig, and P. Liang.Semantic parsing on freebase from question-answer pairs.InProceedings of the 2013 conference on empirical methods in natural language processing, pages 1533–1544, 2013.
- Berges et al. [2025]V.-P. Berges, B. Oguz, D. HAZIZA, W. tau Yih, L. Zettlemoyer, and G. Ghosh.Memory layers at scale.InForty-second International Conference on Machine Learning, 2025.URLhttps://openreview.net/forum?id=ATqGm1WyDj.
- Biderman et al. [2023]S. Biderman, H. Schoelkopf, Q. G. Anthony, H. Bradley, K. O’Brien, E. Hallahan, M. A. Khan, S. Purohit, U. S. Prashanth, E. Raff, A. Skowron, L. Sutawika, and O. van der Wal.Pythia: A suite for analyzing large language models across training and scaling.InProceedings of the 40th International Conference on Machine Learning (ICML), 2023.URLhttps://arxiv.org/abs/2304.01373.
- Borgeaud et al. [2022]S. Borgeaud, A. Mensch, J. Hoffmann, T. Cai, E. Rutherford, K. Millican, G. van den Driessche, J.-B. Lespiau, B. Damoc, A. Clark, D. de Las Casas, A. Guy, J. Menick, R. Ring, T. Hennigan, S. Huang, L. Maggiore, C. Jones, A. Cassirer, A. Brock, M. Paganini, G. Irving, O. Vinyals, S. Osindero, K. Simonyan, J. W. Rae, E. Elsen, and L. Sifre.Improving language models by retrieving from trillions of tokens.InProceedings of the 39th International Conference on Machine Learning (ICML), pages 2206–2240, 2022.URLhttps://arxiv.org/abs/2112.04426.
- Chen et al. [2025]A. Chen, J. Merullo, A. Stolfo, and E. Pavlick.Transferring linear features across language models with model stitching, 2025.URLhttps://arxiv.org/abs/2506.06609.
- Cheng et al. [2026]X. Cheng, W. Zeng, D. Dai, Q. Chen, B. Wang, Z. Xie, K. Huang, X. Yu, Z. Hao, Y. Li, H. Zhang, H. Zhang, D. Zhao, and W. Liang.Conditional memory via scalable lookup: A new axis of sparsity for large language models.arXiv preprint arXiv:2601.07372, 2026.URLhttps://arxiv.org/abs/2601.07372.
- Clark et al. [2019]C. Clark, K. Lee, M.-W. Chang, T. Kwiatkowski, M. Collins, and K. Toutanova.BoolQ: Exploring the surprising difficulty of natural yes/no questions.InProceedings of NAACL-HLT, pages 2924–2936, 2019.
- Dagan et al. [2006]I. Dagan, O. Glickman, and B. Magnini.The PASCAL recognising textual entailment challenge.In J. Quiñonero-Candela, I. Dagan, B. Magnini, and F. d’Alché Buc, editors,Machine Learning Challenges. Evaluating Predictive Uncertainty, Visual Object Classification, and Recognising Tectual Entailment, pages 177–190, Berlin, Heidelberg, 2006. Springer Berlin Heidelberg.ISBN 978-3-540-33428-6.
- Dai et al. [2024]D. Dai, C. Deng, C. Zhao, R. X. Xu, H. Gao, D. Chen, J. Li, W. Zeng, X. Yu, Y. Wu, Z. Xie, Y. K. Li, P. Huang, F. Luo, C. Ruan, Z. Sui, and W. Liang.DeepSeekMoE: Towards ultimate expert specialization in mixture-of-experts language models.arXiv preprint arXiv:2401.06066, 2024.URLhttps://arxiv.org/abs/2401.06066.
- Farhadzadeh et al. [2025]F. Farhadzadeh, D. Das, S. Borse, and F. Porikli.Lora-x: Bridging foundation models with training-free cross-model adaptation, 2025.URLhttps://arxiv.org/abs/2501.16559.
- Fedus et al. [2022]W. Fedus, B. Zoph, and N. Shazeer.Switch transformers: Scaling to trillion parameter models with simple and efficient sparsity.Journal of Machine Learning Research, 23(120):1–39, 2022.URLhttps://jmlr.org/papers/v23/21-0998.html.
- Graves et al. [2014]A. Graves, G. Wayne, and I. Danihelka.Neural turing machines, 2014.URLhttps://arxiv.org/abs/1410.5401.
- Hu et al. [2022]E. J. Hu, Y. Shen, P. Wallis, Z. Allen-Zhu, Y. Li, S. Wang, L. Wang, and W. Chen.LoRA: Low-rank adaptation of large language models.InProceedings of the 10th International Conference on Learning Representations (ICLR), 2022.URLhttps://arxiv.org/abs/2106.09685.
- Huh et al. [2024]M. Huh, B. Cheung, T. Wang, and P. Isola.The platonic representation hypothesis.InProceedings of the 41st International Conference on Machine Learning (ICML), 2024.URLhttps://arxiv.org/abs/2405.07987.
- Izacard et al. [2023]G. Izacard, P. Lewis, M. Lomeli, L. Hosseini, F. Petroni, T. Schick, J. Dwivedi-Yu, A. Joulin, S. Riedel, and E. Grave.Atlas: Few-shot learning with retrieval augmented language models.Journal of Machine Learning Research, 24(251):1–43, 2023.URLhttps://jmlr.org/papers/v24/23-0037.html.
- Jiang et al. [2023]A. Q. Jiang, A. Sablayrolles, A. Mensch, C. Bamford, D. S. Chaplot, D. de las Casas, F. Bressand, G. Lengyel, G. Lample, L. Saulnier, L. R. Lavaud, M.-A. Lachaux, P. Stock, T. L. Scao, T. Lavril, T. Wang, T. Lacroix, and W. E. Sayed.Mistral 7b, 2023.URLhttps://arxiv.org/abs/2310.06825.
- Joshi et al. [2017]M. Joshi, E. Choi, D. S. Weld, and L. Zettlemoyer.Triviaqa: A large scale distantly supervised challenge dataset for reading comprehension.InProceedings of the 55th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 1601–1611, 2017.
- Khandelwal et al. [2020]U. Khandelwal, O. Levy, D. Jurafsky, L. Zettlemoyer, and M. Lewis.Generalization through memorization: Nearest neighbor language models.InProceedings of the 8th International Conference on Learning Representations (ICLR), 2020.URLhttps://arxiv.org/abs/1911.00172.
- Kornblith et al. [2019]S. Kornblith, M. Norouzi, H. Lee, and G. E. Hinton.Similarity of neural network representations revisited.InProceedings of the 36th International Conference on Machine Learning (ICML), 2019.URLhttps://arxiv.org/abs/1905.00414.
- Kwiatkowski et al. [2019]T. Kwiatkowski, J. Palomaki, O. Redfield, M. Collins, A. Parikh, C. Alberti, D. Epstein, I. Polosukhin, J. Devlin, K. Lee, et al.Natural questions: a benchmark for question answering research.Transactions of the Association for Computational Linguistics, 7:453–466, 2019.
- Lai et al. [2017]G. Lai, Q. Xie, H. Liu, Y. Yang, and E. Hovy.RACE: Large-scale reading comprehension dataset from examinations.InProceedings of EMNLP, pages 785–794, 2017.
- Lample et al. [2019]G. Lample, A. Sablayrolles, M. Ranzato, L. Denoyer, and H. Jégou.Large memory layers with product keys.InAdvances in Neural Information Processing Systems 32 (NeurIPS), 2019.URLhttps://arxiv.org/abs/1907.05242.
- Lewis et al. [2020]P. Lewis, E. Perez, A. Piktus, F. Petroni, V. Karpukhin, N. Goyal, H. Küttler, M. Lewis, W. tau Yih, T. Rocktäschel, S. Riedel, and D. Kiela.Retrieval-augmented generation for knowledge-intensive NLP tasks.InAdvances in Neural Information Processing Systems 33 (NeurIPS), 2020.URLhttps://arxiv.org/abs/2005.11401.
- Lin et al. [2022]S. Lin, J. Hilton, and O. Evans.TruthfulQA: Measuring how models mimic human falsehoods.InProceedings of ACL, pages 3214–3252, 2022.
- [27]A. Lozhkov, L. B. Allal, L. von Werra, and T. Wolf.Fineweb-edu: the finest collection of educational content, 2024.URL https://huggingface. co/datasets/HuggingFaceFW/fineweb-edu.
- Merity et al. [2017]S. Merity, C. Xiong, J. Bradbury, and R. Socher.Pointer sentinel mixture models.InProceedings of the 5th International Conference on Learning Representations (ICLR), 2017.URLhttps://arxiv.org/abs/1609.07843.
- Mihaylov et al. [2018]T. Mihaylov, P. Clark, T. Khot, and A. Sabharwal.Can a suit of armor conduct electricity? a new dataset for open book question answering.InProceedings of EMNLP, pages 2381–2391, 2018.
- Mikolov et al. [2013]T. Mikolov, Q. V. Le, and I. Sutskever.Exploiting similarities among languages for machine translation.arXiv preprint arXiv:1309.4168, 2013.URLhttps://arxiv.org/abs/1309.4168.
- Paperno et al. [2016]D. Paperno, G. Kruszewski, A. Lazaridou, N. Q. Pham, R. Bernardi, S. Pezzelle, M. Baroni, G. Boleda, and R. Fernández.The LAMBADA dataset: Word prediction requiring a broad discourse context.InProceedings of the 54th Annual Meeting of the Association for Computational Linguistics (ACL), pages 1525–1534, 2016.doi:10.18653/v1/P16-1144.
- Qwen Team [2025]Qwen Team.Qwen3 technical report.arXiv preprint arXiv:2505.09388, 2025.
- Raffel et al. [2020]C. Raffel, N. Shazeer, A. Roberts, K. Lee, S. Narang, M. Matena, Y. Zhou, W. Li, and P. J. Liu.Exploring the limits of transfer learning with a unified text-to-text transformer.Journal of Machine Learning Research, 21(140):1–67, 2020.
- Sanjeev and Troynikov [2024]S. Sanjeev and A. Troynikov.Embedding adapters.Technical report, Chroma, May 2024.URLhttps://trychroma.com/research/embedding-adapters.Accessed on July 8, 2024.
- Shazeer et al. [2017]N. Shazeer, A. Mirhoseini, K. Maziarz, A. Davis, Q. V. Le, G. E. Hinton, and J. Dean.Outrageously large neural networks: The sparsely-gated mixture-of-experts layer.InProceedings of the 5th International Conference on Learning Representations (ICLR), 2017.URLhttps://arxiv.org/abs/1701.06538.
- Su et al. [2025]D. Su, K. Kong, Y. Lin, J. Jennings, B. Norick, M. Kliegl, M. Patwary, M. Shoeybi, and B. Catanzaro.Nemotron-cc: Transforming common crawl into a refined long-horizon pretraining dataset, 2025.URLhttps://arxiv.org/abs/2412.02595.
- Touvron et al. [2023]H. Touvron, L. Martin, K. Stone, P. Albert, A. Almahairi, Y. Babaei, N. Bashlykov, S. Batra, P. Bhargava, S. Bhosale, et al.Llama 2: Open foundation and fine-tuned chat models.arXiv preprint arXiv:2307.09288, 2023.
- Wang et al. [2024]R. Wang, S. Ghosh, D. Cox, D. Antognini, A. Oliva, R. Feris, and L. Karlinsky.Trans-LoRA: towards data-free transferable parameter efficient finetuning, 2024.URLhttps://arxiv.org/abs/2405.17258.
- Wei et al. [2026]R. Wei, J. Cao, J. Wang, J. Kai, Q. Guo, B. Zhou, and Z. Lin.Mlp memory: A retriever-pretrained memory for large language models, 2026.URLhttps://arxiv.org/abs/2508.01832.
- Welbl et al. [2017]J. Welbl, N. F. Liu, and M. Gardner.Crowdsourcing multiple choice science questions.InProceedings of the 3rd Workshop on Noisy User-generated Text, pages 94–106, 2017.
- Weston et al. [2015]J. Weston, S. Chopra, and A. Bordes.Memory networks, 2015.URLhttps://arxiv.org/abs/1410.3916.
- Xia et al. [2025]F. Xia, M. Liao, Y. Fang, D. Li, Y. Xie, W. Li, Y. Li, D. Xia, and J. Huang.Cross-lora: A data-free lora transfer framework across heterogeneous llms, 2025.URLhttps://arxiv.org/abs/2508.05232.
- Yang et al. [2018]Z. Yang, P. Qi, S. Zhang, Y. Bengio, W. Cohen, R. Salakhutdinov, and C. D. Manning.Hotpotqa: A dataset for diverse, explainable multi-hop question answering.InProceedings of the 2018 conference on empirical methods in natural language processing, pages 2369–2380, 2018.
- Zhang et al. [2024]P. Zhang, G. Zeng, T. Wang, and W. Lu.TinyLlama: An open-source small language model.arXiv preprint arXiv:2401.02385, 2024.URLhttps://arxiv.org/abs/2401.02385.
Appendix ASupplementary Formalization and Method Details
This section supplies notation and formal details forSection3.
A.1Notations
Table6summarizes the notation used throughout this paper.
Table 6:Notation used throughout this paper.SymbolMeaningA,BA,BSource model and target model, respectively.dA,dBd_{A},d_{B}Hidden dimensions of source modelAAand target modelBB, respectively.ℰA\mathcal{E}_{A}Frozen Engram memory artifact learned with source modelAA, comprising the tables{En,k}\{E_{n,k}\}.En,kE_{n,k}Memory table associated with N-gram ordernnand hash headkk, whereEn,k∈RM×dheadE_{n,k}\in\mathbb{R}^{M\times d_{\text{head}}}.NmaxN_{\max}Maximum N-gram order used for memory addressing, withn∈{2,…,Nmax}n\in\{2,\ldots,N_{\max}\}.KKNumber of independent hash heads for each N-gram order.HHTotal number of hash heads,H=(Nmax−1)KH=(N_{\max}-1)K.MMNumber of rows in each hash-head memory table.dheadd_{\text{head}}Dimensionality of the embedding row retrieved from one hash-head table.𝐞t\mathbf{e}_{t}Concatenated memory vector retrieved at token positiontt.dmemd_{\text{mem}}Dimensionality of𝐞t\mathbf{e}_{t}, wheredmem=Hdheadd_{\text{mem}}=Hd_{\text{head}}.𝒫\mathcal{P}Canonicalization function mapping a raw decoded string to its canonical form.V,V′V,V^{\prime}Raw vocabulary and canonical vocabulary, respectively.φn,k\varphi_{n,k}Deterministic hash function for N-gram ordernnand hash headkk.𝐡t,ℓ\mathbf{h}_{t,\ell}Target-backbone hidden state at token positionttand layerℓ\ell.ℒ\mathcal{L}Set of target-backbone layers at which memory readers are injected.SSNumber of memory injection sites,S=|ℒ|S=|\mathcal{L}|.RRNumber of reader branches at each injection site.𝐖K,ℓ,r(B)\mathbf{W}_{K,\ell,r}^{(B)}Branch-specific key-projection matrix for branchrrat target layerℓ\ell.𝐖V,ℓ(B)\mathbf{W}_{V,\ell}^{(B)}Value-projection matrix shared across theRRbranches at target layerℓ\ell.αt,ℓ(r)\alpha_{t,\ell}^{(r)}Context-aware scalar gate for tokentt, layerℓ\ell, and branchrr.βℓ,r\beta_{\ell,r}Learnable scalar gate bias for branchrrat injection layerℓ\ell.𝒲(B)\mathcal{W}^{(B)}Complete target-side reader parameter set, including all key and value projections, normalization parameters, and gate biases.b,Tb,TBatch size and sequence length, respectively.
A.2Memory Transfer Conditions
Let modelAAbe a source model equipped with a trained Engram memory tableℰA\mathcal{E}_{A}, and let modelBBbe a target model with hidden dimensiondBd_{B}, tokenizer𝒯B\mathcal{T}_{B}, and potentially a different architecture. We ask whether the frozen memoryℰA\mathcal{E}_{A}can provide extractable external knowledge toBBthrough a lightweight reader𝒲(B)\mathcal{W}^{(B)}, even whendB,dAd_{B}\neq d_{A}and𝒯B,𝒯A\mathcal{T}_{B}\neq\mathcal{T}_{A}.
We allow lightweight target-side reader training, and we do not assume zero-data or universal transfer across all memory architectures. Under this scope, successful extraction is meaningful only if three conditions can be met simultaneously:
- •Deterministic addressing: memory lookup indices depend only on the input token sequence through a fixed canonicalization-and-hashing pipeline. They do not depend on the backbone’s internal hidden states. Thesame textcan therefore map to thesame memory entriesacross backbones, provided the key pipeline is standardized.
- •Architectural decoupling: the memory tableℰ\mathcal{E}is extracted through learned projection matrices𝐖K\mathbf{W}_{K}and𝐖V\mathbf{W}_{V}. These projections are natural reader interfaces because they already mediate between the memory space and the backbone hidden space.
- •Gate-based robustness: even if some transferred memory entries are imperfectly aligned, the sigmoid gateαt\alpha_{t}can softly suppress their contribution. In practice, this requires the reader to learn a sufficiently negative match score and/or bias for unhelpful memory, thereby reducing interference with the hidden state.
The remainder of this section turns these conditions into a concrete decomposition. Thekey spacedetermines which memory slot is selected for a piece of text (SectionA.3); thevalue spacedetermines how the vector stored in that slot is interpreted by the target backbone (SectionA.4). Transfer, therefore, requires both address agreement and a reader that maps the retrieved vector into the target residual stream, with gating available to limit interference when alignment is incomplete (SectionA.5).
A.3Key-Space Unification
The first obstacle is address compatibility. In Engram’s original design, the hash key is computed over model-specific canonical token IDs. For example, if modelAAassigns token ID 1823 to “San” and modelBBassigns ID 47, the resulting hash indices differ even when the underlying text is the same, so the two models would read unrelated memory entries.
To remove this mismatch, we define atokenizer-agnostic canonical key pipeline. Letdecode(⋅)\text{decode}(\cdot)denote the mapping from a tokenizer’s token sequence to its surface string. For a shared decoded text prefix, both models first normalize the same surface form before hashing:
𝒫(decode(𝒯A(x1:tAtext)))=𝒫(decode(𝒯B(x1:tBtext))),\mathcal{P}\!\left(\text{decode}(\mathcal{T}_{A}(x^{\text{text}}_{1:t_{A}}))\right)=\mathcal{P}\!\left(\text{decode}(\mathcal{T}_{B}(x^{\text{text}}_{1:t_{B}}))\right),(12)wherextext1:tAx^{\text{text}}_{1:t_{A}}andxtext1:tBx^{\text{text}}_{1:t_{B}}denote source and target token prefixes that decode to the same text span, and𝒫\mathcal{P}applies NFKC normalization, lowercasing, and accent stripping. In this construction, both models map the same decoded text to the same canonical form before hashing, so they can compute identical lookup indices into the shared memory table.
Subword boundary misalignment.The remaining difficulty is that tokenizers may segment the same word into different subword sequences. Three mitigations are available, in increasing order of robustness: (i) the original NFKC-based canonicalization when vocabularies overlap heavily; (ii) word-boundary N-grams computed over decoded strings; and (iii) character- or byte-level N-gram hashing. We adopt strategy (ii) as a practical middle ground: it substantially reduces tokenizer dependence without moving all the way to byte-level modeling. Our current experiments cover GPT-NeoX (Pythia), LLaMA, Qwen, and Phi tokenizers; languages or tokenizers without clean word boundaries may require the byte-level alternative.
A.4Value-Space Alignment via Reader
Key agreement ensures that the target model reads thesamememory entry as the source model; it does not ensure that the retrieved vector is meaningful in the target residual stream. ModelAA’s memory vectors𝐞t∈Rdmem\mathbf{e}_{t}\in\mathbb{R}^{d_{\text{mem}}}were shaped by gradients flowing throughAA’s hidden geometry, which modelBBdoes not share.
Research on embedding space alignment[Mikolov et al., 2013,Kornblith et al., 2019,Sanjeev and Troynikov, 2024]suggests that representational spaces of models trained on similar data are often related by an approximately linear transformation. The Platonic Representation Hypothesis[Huh et al., 2024]offers one interpretation of this phenomenon, andChen et al. [2025]showed empirically that affine maps can transfer linear features across LLMs. This motivates the smallest plausible bridge between source memory and target backbone:
𝐖V(B):Rdmem→RdB,𝐖K(B):Rdmem→RdB.\mathbf{W}_{V}^{(B)}:\mathbb{R}^{d_{\text{mem}}}\to\mathbb{R}^{d_{B}},\qquad\mathbf{W}_{K}^{(B)}:\mathbb{R}^{d_{\text{mem}}}\to\mathbb{R}^{d_{B}}.(13)The memory tableℰA\mathcal{E}_{A}remainsfrozen; only the reader parameters𝒲(B)\mathcal{W}^{(B)}are trained. For typical configurations (dmem=512d_{\text{mem}}=512,dB=4096d_{B}=4096), the reader size is
|𝒲(B)|=2⋅dmem⋅dB+2⋅dB≈4.2M.|\mathcal{W}^{(B)}|=2\cdot d_{\text{mem}}\cdot d_{B}+2\cdot d_{B}\approx 4.2\text{M}.(14)This is well within the LoRA-scale budget and small relative to multi-billion-parameter backbones. The linear choice is deliberate: it isolates whether cross-model reuse is already feasible under a minimal alignment assumption before introducing richer readers.
A.5Gating as a Robustness Mechanism
The transfer mechanism also needs a fallback when the transferred memory is unhelpful. The context-aware gate provides exactly such a mechanism. If a memory entry is poorly aligned, for example because of a hash collision or because the source geometry does not translate cleanly to the target, the key vector𝐤t=𝐖K(B)𝐞t\mathbf{k}_{t}=\mathbf{W}_{K}^{(B)}\mathbf{e}_{t}will be geometrically dissimilar to the backbone query𝐡t\mathbf{h}_{t}. By learning a sufficiently negative match score or bias in those cases, the reader can pushαt\alpha_{t}toward zero and suppress the memory contribution. The model can therefore tolerate a non-trivial amount of noisy or only partially aligned transferred memory without forcing that signal into the residual stream. This self-suppression is analogous to a soft mask in attention, but it operates on the memory channel itself and requires no explicit confidence model.
A.6Comparison with Existing Knowledge-Augmentation Methods and Native Engram
Our source memory and target reader are trained with next-token prediction rather than task-specific labels. Consequently, a single fitted reader can be evaluated across multiple downstream tasks without being retrained for each task. This does not, however, imply universal downstream improvements. As shown in RQ5 (Section4.5), transferred memory provides stable intrinsic perplexity gains, whereas downstream improvements remain task-dependent and depend on the alignment among the memory corpus, target model, reader interface, and evaluation objective.
Table7situates frozen-memory extraction relative to retrieval-based, native-memory, and model-editing approaches. The main design distinction is that our method retains an explicit external memory while making the stored artifact portable across backbones through reader-only fitting.
Table 7:Qualitative comparison of knowledge augmentation approaches. ✓ = fully supported,∘\circ= partially supported, ✗ = not supported.PropertyKNN-LMRAGRETROMem. LayersLLM Mod.EngramOursO(1)O(1)retrieval✗✗✗✗✗✓✓Parametric (trained)✗✗✓✓✓✓✓Cross-model portable✗✓✗✗∘\circ✗✓Surgical deletion✗✓✗✗✗✓✓No context overhead✗✗∘\circ✓✓✓✓Reader-only integrationN/AN/A✗✗✗✗✓
Differences from Native Engram
Table8makes the architectural changes explicit at the formula level. Relative to native Engram, the transfer-oriented version uses shared canonical hashing, freezes the exported table, and replaces within-block integration with a target-side residual reader.
Table 8:Formula-level comparison between original Engram and our transfer-oriented reader. We replace the native within-block module with a target-side reader over frozen hashed memory.AspectOriginal EngramTransfer-oriented Engram ReaderMemory indexingLayer-specific compressed-tokennn-gram hashing:rt(l,n,j)=(⨁k=0n−1c(xt−k)⋅ak(l))modpl,n,jr_{t}^{(l,n,j)}=\Bigl(\bigoplusop\displaylimits_{k=0}^{n-1}c(x_{t-k})\cdot a_{k}^{(l)}\Bigr)\bmod p_{l,n,j}Shared canonical hashing:rt(n,j)=(H(canon_ngramt(n))⊕sj)modMr_{t}^{(n,j)}=\bigl(H(\mathrm{canon\_ngram}_{t}^{(n)})\oplus s_{j}\bigr)\bmod MMemory representationLayer-specific Engram embedding:et(l)=Concatn,j𝐄l,n,j[rt(l,n,j)]e_{t}^{(l)}=\mathrm{Concat}_{n,j}\mathbf{E}_{l,n,j}[r_{t}^{(l,n,j)}]Shared memory vector:mt=Concatj𝐓j[rt(j)](dmem=512)m_{t}=\mathrm{Concat}_{j}\mathbf{T}_{j}[r_{t}^{(j)}]\hskip 9.24994pt(d_{\mathrm{mem}}=512)Key / gatest\displaystyle s_{t}=⟨RMSNorm(Wket),RMSNorm(ht)⟩d\displaystyle=\frac{\left\langle\mathrm{RMSNorm}(W_{k}e_{t}),\mathrm{RMSNorm}(h_{t})\right\rangle}{\sqrt{d}}αt\displaystyle\alpha_{t}=σ(sign(st)|st|)\displaystyle=\sigma\!\left(\mathrm{sign}(s_{t})\sqrt{|s_{t}|}\right)kt(r)=Wk(r)mt,vt=Wvmtk_{t}^{(r)}=W_{k}^{(r)}m_{t},\hskip 9.24994ptv_{t}=W_{v}m_{t}αt(r)=σ(⟨RMSNorm(ht),RMSNorm(kt(r))⟩d+βr)\alpha_{t}^{(r)}=\sigma\!\left(\frac{\langle\mathrm{RMSNorm}(h_{t}),\mathrm{RMSNorm}(k_{t}^{(r)})\rangle}{\sqrt{d}}+\beta_{r}\right)Output updateut=αtWvet,ot=ut+ShortConv(ut)u_{t}=\alpha_{t}W_{v}e_{t},\hskip 9.24994pto_{t}=u_{t}+\mathrm{ShortConv}(u_{t})ot=1R∑r=1Rαt(r)vt,ht←ht+oto_{t}=\frac{1}{R}\sumop\displaylimits_{r=1}^{R}\alpha_{t}^{(r)}v_{t},\hskip 9.24994pth_{t}\leftarrow h_{t}+o_{t}Branch structureNative multi-branch inside backboneExplicit reader branches (RRcontrols capacity)InjectionInternal transformer block componentPost-layer hook injection (e.g., layers 2 and 10)TrainingJoint training with backboneTwo-stage: source training + frozen memory + target reader fitting
Appendix BImplementation Details
B.1Models
We evaluate transfer across six independently developed model families with different architectures, tokenizers, and parameter scales: Pythia[Biderman et al., 2023], TinyLlama[Zhang et al., 2024], Qwen3.5[Qwen Team, 2025], Phi-4-mini[Abouelenin et al., 2025], LLaMA 2[Touvron et al., 2023], and Mistral[Jiang et al., 2023]. These choices are designed to separate several sources of transfer difficulty, including model scale, architectural mismatch, tokenizer mismatch, and differences in pre-training or post-training.
The main transfer matrix uses Pythia, Qwen3.5, and TinyLlama because together they provide both same-family and cross-family transfer settings across a wide range of model sizes. The Pythia pair offers a controlled same-tokenizer comparison, while transfers to TinyLlama and Qwen3.5 introduce a tokenizer and architectural mismatch. The peer-transfer study adds Phi-4-mini to test whether memory remains portable between similarly scaled but independently trained model families, rather than only from smaller to larger models.
The corpus-matched downstream suite uses Qwen3.5 source and target models, so that source scale, target scale, and corpus alignment can be varied within a single model family while keeping the broader architecture and tokenizer family controlled. This makes it easier to attribute downstream differences to memory content, target capacity, and reader fitting, rather than to unrelated model-family-level changes.
The QA experiments inTable1and associated ablations use LLaMA 2 as the memory source and Mistral as the target backbone. This pair provides a stronger cross-family setting at the 7B scale, with comparable model capacity but independently trained backbones and distinct tokenization and representation spaces. It therefore offers a demanding test of whether the reader can recover useful factual knowledge from a frozen source memory without relying on parameter scale or within-family compatibility.
Table9lists the checkpoints, experimental roles, hidden dimensions, checkpoint types, and tokenizer families used in these experiments.
Table 9:Models and checkpoints used in the transfer experiments. “Source” denotes memory construction, and “target” denotes reader fitting for a frozen memory. Here,ddis the residual hidden size used by the memory interface.FamilyModelExperimental roleParams.ddCheckpointHugging Face identifierPythiaPythia-160MSource: main matrix160M768BaseEleutherAI/pythia-160mPythia-410MTarget: main matrix410M1024BaseEleutherAI/pythia-410mQwen3.5Qwen3.5-0.8BSource: main matrix, scalingTarget: downstream0.8B1024BaseQwen/Qwen3.5-0.8B-BaseQwen3.5-2BTarget: scaling, downstream2B2048BaseQwen/Qwen3.5-2B-BaseQwen3.5-4BSource: peer, downstreamTarget: main matrix, peer,scaling, downstream4B2560BaseQwen/Qwen3.5-4B-BaseQwen3.5-9BSource: main matrix, downstreamTarget: scaling, downstream9B4096BaseQwen/Qwen3.5-9B-BaseTinyLlamaTinyLlama-1.1BTarget: main matrix1.1B2048Chat-/instruction-tunedTinyLlama/TinyLlama-1.1B-Chat-v1.0PhiPhi-4-mini-instructSource and target: peer transfer3.8B3072Instruction-tunedmicrosoft/Phi-4-mini-instructLLaMA 2LLaMA-2-7BSource: open-domain QA, ablation7B4096Basemeta-llama/Llama-2-7b-hfMistralMistral-7B-v0.3Target: open-domain QA, ablationSource: self-transfer control7B4096Basemistralai/Mistral-7B-v0.3Pythia uses the GPT-NeoX tokenizer, and the Qwen3.5 models share the Qwen3.5 tokenizer. TinyLlama and LLaMA-2 use LLaMA-family tokenizers, Phi-4-mini-instruct uses the tokenizer distributed with the Phi checkpoint, and Mistral-7B-v0.3 uses the Mistral tokenizer. These differences in the tokenizer and hidden size require the target reader to bridge representational and, in cross-tokenizer settings, tokenization boundaries rather than simply reusing a dimension-compatible memory interface.
B.2Architecture Configuration
Memory Configuration
The source Engram memory usesNmax=3N_{\max}=3(bigram and trigram),K=4K=4hash heads per order, table sizeM=65,536M=65{,}536, and head dimensiondhead=64d_{\text{head}}=64, yieldingdmem=512d_{\text{mem}}=512and approximately 33.5M total memory parameters.
Reader Configuration
We use the reader configuration appropriate to each experimental regime rather than fitting a separate reader for each evaluation benchmark. For the experiments inTable1, we use a single-layer one-branch reader, a dual-layer one-branch reader, and a stronger dual-layer, four-branch reader.Table1seems to be about different readers. For the LLaMA-2-7B→\rightarrowMistral-7B-v0.3 transfer ablation experiments inTable5, memory is injected at layers 2 and 10, with four branch-specific key/gating paths at each site and a shared value projection. This reader is fitted with 2048-token Wikipedia-2021 sequences and is then applied zero-shot to the task prompts, without task-specific reader training. For the other experiments, we use the minimal single-layer, single-branch reader fromSection3.2. The reader is attached at the default relative depthL/3L/3, whereLLis the number of transformer layers, and the target backbone and exported memory remain frozen while only the reader is optimized. The reader is fitted with the default 512-token training sequences and is then applied zero-shot to the task prompts, without task-specific reader training. These configuration choices reflect the purpose of each experiment. The QA reader-design study inTable1intentionally varies reader depth and branch capacity to measure how interface expressivity affects extraction from frozen memory, while the ablation study inTable5uses the strongest validated reader to isolate the roles of gating, addressing, and memory content.
B.3Data, Training and Evaluation
We select the training corpora to match the purpose of each experimental regime rather than using a single corpus throughout. WikiText-103 provides a compact and reproducible language-modeling benchmark for the controlled transfer matrix, scaling, and representation analyses. FineWeb-Edu and the Nemotron-CC subsets provide substantially broader and more domain-diverse text for studying downstream transfer and corpus alignment. The five QA (NQ, WebQA, TriviaQA, TruthQA, HotpotQA) experiments use Wikipedia-2021 because their evaluation tasks emphasize factual and encyclopedic knowledge. Finally, LAMBADA, WikiText-103, and C4 are used in the distribution-shift experiments so that each fitted reader can be evaluated both on its training distribution and on corpora with different genres, structures, and provenances.
ForFigures2(b),4(b)and5andTables11and4experiments, we use WikiText-103[Merity et al., 2017]. The loader removes empty records, tokenizes each remaining text with the active model tokenizer without special tokens, concatenates the resulting token stream, and partitions it into fixed, non-overlapping sequences. The standard sequence length is 512 tokens; the standard train and test splits are used for training and final evaluation, respectively. Because the loader discards the incomplete final sequence and the evaluation data are batched with “drop-last”, the test loader contains 34 or 36 batches in the standard runs due to tokenizer-dependent token counts, rather than because the evaluation size is manually set to one of those values.
For experiments inFigure3andTables16,17and20, we use FineWeb-Edu[Lozhkov et al.,](HuggingFaceFW/fineweb-edu, “sample-10BT”), a broad educational web corpus, for source-memory and reader fitting. The domain-alignment experiments additionally use Nemotron-CC HQ-DQA[Su et al., 2025]and Nemotron-CC HQ from nvidia/Nemotron-CC-v2.1; the former is a QA-heavy STEM subset and the latter is a broader organic-web subset used as a corpus-size and structure control. The resulting reader is evaluated zero-shot on RTE[Dagan et al., 2006], BoolQ[Clark et al., 2019], openBookQA[Mihaylov et al., 2018], SciQ[Welbl et al., 2017], TruthfulQA[Lin et al., 2022], and RACE[Lai et al., 2017].
For QA tasks inTable1toTable5, it contains Natural Questions (NQ)[Kwiatkowski et al., 2019], WebQA[Berant et al., 2013], TriviaQA[Joshi et al., 2017], TruthfulQA[Lin et al., 2022], and HotpotQA[Yang et al., 2018]. The default corpus for the LLaMA-2-7B→\rightarrowMistral-7B-v0.3 experiments is the pre-tokenized December-2021 English Wikipedia[Izacard et al., 2023]release (Rubin-Wei/enwiki-dec2021-preprocessed-mistral), distributed with the MLP Memory data pipeline[Wei et al., 2026]. It provides train and test splits; the implementation maps a requested validation split to test and, when the active tokenizer differs from Mistral’s, decodes and re-tokenizes the windows before chunking them.
The out-of-domain language-modeling evaluation inSectionD.5uses LAMBADA[Paperno et al., 2016], WikiText-103[Merity et al., 2017], and the English C4 validation set[Raffel et al., 2020]. LAMBADA and WikiText-103 are considered out-of-domain only when the reader is fitted to a different corpus. LAMBADA contains narrative text requiring long-range contextual prediction, whereas WikiText-103 consists of curated encyclopedic articles; both differ substantially in genre and provenance from broad web corpora such as C4 and FineWeb-Edu.
Training protocol.The principal settings are summarized inTable10. All phases optimize the next-token language-modeling objective; no task labels are used to fit a reader.
Table 10:Principal training and evaluation settings. “LR / warm-up” reports the learning rate and number of warm-up steps. Unless otherwise noted, training uses cosine decay, AdamW with weight decay0.010.01, and gradient-norm clipping at1.01.0. Alternative-corpus experiments follow the intrinsic reader-fitting protocol with LAMBADA or English C4 replacing WikiText-103. OOD rows are evaluation-only.Regime / phaseCorpusBudget / capSeq. / batchLR / warm-upTrainable components and exceptionsPrimary training settingsRQ14.1Phase 1WikiText-10350M tokens512 / 163×10−53\times 10^{-5}1,000Source backbone, memory, and source reader are trainable. For Qwen3.5 sources, the memory learning rate is10−310^{-3}.RQ14.1Phase 2WikiText-10320M tokens512 / 163×10−53\times 10^{-5}500Only the target reader is trainable; the target backbone and transferred memory remain frozen.Figure3Phase 1FineWeb-Edu50M tokens512 / 210−310^{-3}/ 1,000Source backbone, memory, and source reader trained end-to-end; gradient checkpointingFigure3Phase 2FineWeb-Edu20M tokens512 / 43×10−53\times 10^{-5}/ 500Target reader only; target backbone and transferred memory frozen; gradient checkpointing; early stopping patience 5Table1to Table5Phase 1Wikipedia-202110–30M tokens2048 / 13×10−53\times 10^{-5}1,000The source memory and reader are trainable; the source backbone is frozen. Early stopping uses patience 3.Table1to Table5Phase 2Wikipedia-202110–30M tokens2048 / 13×10−53\times 10^{-5}500Only the target reader is trainable; the target backbone and transferred memory remain frozen. Early stopping uses patience 3.Alternative-corpus trainingLAMBADA Phase 1LAMBADA train split50M tokens512 / 163×10−53\times 10^{-5}1,000Source backbone, memory, and source reader are trainable. The finite training split is cycled as needed.LAMBADA Phase 2LAMBADA train split20M tokens512 / 163×10−53\times 10^{-5}500Only the target reader is trainable; the target backbone and transferred memory remain frozen.C4 Phase 1English C4 train stream50M tokens512 / 163×10−53\times 10^{-5}1,000Source backbone, memory, and source reader are trainable.C4 Phase 2English C4 train stream20M tokens512 / 163×10−53\times 10^{-5}500Only the target reader is trainable; the target backbone and transferred memory remain frozen.Out-of-domain evaluation without additional trainingLAMBADAlambada, test2M eval tokens512 / 16—Evaluation only; source memory, reader, and target backbone remain frozen.WikiText-103wikitext-103, test2M eval tokens512 / 16—Evaluation-only distribution-shift probe; no trainable components.C4allenai/c4, English validation5M eval tokens512 / 16—Evaluation-only streaming web-text probe; no trainable components.For the standard Pythia source configurations, Phase 1 is end-to-end; the large Qwen and LLaMA source runs freeze the source backbone when required by the memory budget. After Phase 1, the exported memory table is frozen and attached to the target model. For Phase 2, only the target-side reader is optimized.
Conditions and metrics.Across transfer experiments, we compare aBaselinetarget with no memory, aTransferredcondition with a trained frozen source memory, and aRandom memorycondition with an untrained table of the same architecture. Unless noted otherwise, all conditions use three seeds (42, 137, 2024). For the performance inTables1and2, we use F1 score as the main result, and multiple-choice average and follow the official code and checkpoint to evaluate for MLP memory. For the other downstream tasks, we report the accuracy. We report test perplexity (PPL,↓\downarrow) with 95% bootstrap confidence intervals and use pairedtt-tests on per-batch perplexities. The experiments in the appendix run on a single NVIDIA H100 NVL GPU. The experiments in the main text are conducted on the LUMI supercomputer operated by CSC – IT Center for Science, Finland, using the LUMI-G partition equipped with AMD Instinct MI250X GPUs.
Evaluation-set overlap audit.We compare all 31,807 downstream evaluation questions against the exact Phase-1 and Phase-2 training streams used in the QA experiments. Natural Questions, WebQuestions, and HotpotQA contain no exact question matches. The remaining exact matches consist of four duplicated TriviaQA rows (0.013%) and one generic TruthfulQA question without its gold answer. Partial 8-gram overlaps predominantly correspond to shared phrasing rather than direct question-answer leakage. Thus, the reader training streams use no benchmark supervision, although they are not strictly overlap-free. Since the pretraining corpora of the underlying foundation models are unavailable, we interpret the QA results as controlled comparisons rather than as a contamination-free state-of-the-art claim.
Appendix CSupplementary Evidence for RQ1
The main 3×\times3 matrix inSection4.1answers the primary question of whether a frozen memory remains useful after being moved to a different backbone. The experiments below test whether that conclusion survives several plausible alternative explanations. Specifically, we examine whether the gains depend on shared tokenization, arise only when transferring from a smaller to a larger model, disappear for stronger targets, or can be explained by similarity between source and target representations.
C.1Full Cross-Architecture Matrix
Table11reports the complete cross-architecture results for all nine source–target combinations used in the main transfer study inFigure2(a). For each source memory, we compare three conditions on each target model: the target backbone without memory augmentation (Baseline), the frozen source memory integrated through a fitted target reader (Transferred), and a size-matched randomly initialized memory evaluated under the same reader fitting protocol (Random). The transferred-memory condition improves test perplexity in every source–target pair, with relative reductions ranging from1.6%1.6\%to15.7%15.7\%. The strongest improvement occurs for Qwen3.5-0.8B→\rightarrowTinyLlama-1.1B, while the random-memory control generally improves less than the corresponding transferred memory.
Table 11:Full cross-architecture transfer matrix: test PPL (↓\downarrow) for 3 source memories×\times3 target models.Randomreplaces the learned source memory with a size-matched randomly initialized memory under the same fitting protocol. Mean±\pmstd over 3 seeds. Deltas shown next to the transferred result are relative to the no-memory baseline; negative values indicate better PPL. A displayed standard deviation of0.00.0indicates a value below0.050.05after rounding to one decimal place.TargetBaselineTransferredRandomPythia-160M sourcePythia-410M21.921.921.6±0.1\mathbf{21.6\pm 0.1}-1.6%21.9±0.021.9\pm 0.0Qwen3.5-4B10.810.810.1±0.010.1\pm 0.0-6.8%10.3±0.010.3\pm 0.0TinyLlama-1.1B10.610.69.5±0.09.5\pm 0.0-10.6%10.0±0.010.0\pm 0.0Qwen3.5-0.8B sourcePythia-410M22.922.921.4±0.021.4\pm 0.0-6.8%22.1±0.022.1\pm 0.0Qwen3.5-4B10.510.59.6±0.09.6\pm 0.0-8.9%10.2±0.010.2\pm 0.0TinyLlama-1.1B10.810.89.1±0.0\mathbf{9.1\pm 0.0}-15.7%9.3±0.09.3\pm 0.0Qwen3.5-9B sourcePythia-410M22.622.621.6±0.021.6\pm 0.0-4.3%22.3±0.022.3\pm 0.0Qwen3.5-4B10.510.59.4±0.09.4\pm 0.0-10.3%10.2±0.010.2\pm 0.0TinyLlama-1.1B10.410.49.1±0.09.1\pm 0.0-12.2%9.3±0.09.3\pm 0.0
C.2Supplement Transfer Comparisons Experiments
This section supplements the transfer comparison experiments in the same-tokenizer and cross-tokenizer transfer settings and compares against parameter-matched baselines and an external retrieval reference.
Table12isolates the simplest transfer setting, where the source and target share the Pythia family and tokenizer, and asks whether frozen memory still helps when comparing against parameter-matched alternatives. The result is still favorable to the transferred memory. It outperforms both random memory and the iso-parameter LoRA baselines, while kNN-LM remains a stronger but much more expensive non-parametric reference because it requires a retrieval datastore and nearest-neighbor search at inference time.
Table 12:Same-tokenizer transfer (Pythia-160M→\toPythia-410M). Mean±\pmstd over 3 seeds. Deltas in the PPL column are relative to the no-memory baseline; negative values indicate better PPL. LoRA and Cross-LoRA are iso-parameter baselines (1.03M parameters, same 20M token budget). KNN-LM uses a 5M-token datastore withk=1024k{=}1024retrieval at inference time.ConditionTest PPL↓\downarrow95% CIParams / CostBaseline (no memory)21.921.9[20.6,23.4][20.6,23.4]—Transfer ConditionsRandom memory21.9±0.021.9\pm 0.0-0.2%[20.6,23.3][20.6,23.3]1.05MTransferred memory21.6±0.1\mathbf{21.6\pm 0.1}-1.6%[20.2,23.0]\mathbf{[20.2,23.0]}1.05MParameter-Matched BaselinesLoRA (rank 7)[Hu et al., 2022]23.3±0.823.3\pm 0.8+6.5%[21.9,24.9][21.9,24.9]1.03MCross-LoRA[Xia et al., 2025]23.9±0.623.9\pm 0.6+9.3%[22.5,25.5][22.5,25.5]1.03MExternal Retrieval ReferenceKNN-LM[Khandelwal et al., 2020]†20.420.4-7.1%[19.3,21.5][19.3,21.5]5M-tok store†Non-parametric; requiresk=1024k{=}1024nearest-neighbor search per token at inference.Table13moves to the harder Pythia→\rightarrowTinyLlama case, where tokenizer mismatch is unavoidable, and transfer depends on the shared canonicalization pipeline. Even in this setting, transferred memory reduces PPL from 10.63 to 9.50 and clearly beats random memory. This supports the claim that the exported memory is not tied to the source tokenizer as long as the target reader is given a compatible canonical interface.
Table 13:Cross-tokenizer transfer (Pythia-160M→\toTinyLlama-1.1B). Mean±\pmstd over 3 seeds. Deltas in the PPL column are relative to the no-memory baseline; negative values indicate better PPL. Reader parameters: 2.10M.ConditionTest PPL↓\downarrow95% CIParamsBaseline (no memory)10.610.6[10.0,11.3][10.0,11.3]—Transfer ConditionsRandom memory10.0±0.010.0\pm 0.0-5.9%[9.4,10.7][9.4,10.7]2.10MTransferred memory9.5±0.0\mathbf{9.5\pm 0.0}-10.6%[8.9,10.2]\mathbf{[8.9,10.2]}2.10M
C.3Peer-to-Peer Transfer
This section tests whether transfer remains useful once the source and target are both strong mid-scale models rather than a small source feeding a much larger target.Table14shows the results of peer-to-peer transfer between Phi-4-mini[Abouelenin et al., 2025](3.8B) and Qwen3.5-4B.
Table 14:Peer-to-peer transfer between Phi-4-mini (3.8B) and Qwen3.5-4B. Source memories trained on WikiText-103 (50M tokens). Mean±\pmstd over 3 seeds. Deltas shown next to the transferred result are relative to the no-memory baseline; negative values indicate better PPL.DirectionBaselineTransferredRandomPhi-4-mini→\rightarrowQwen3.5-4B10.510.59.5±0.0\mathbf{9.5\pm 0.0}-10.1%10.2±0.010.2\pm 0.0Qwen3.5-4B→\rightarrowPhi-4-mini12.612.611.5±0.011.5\pm 0.0-9.2%12.3±0.012.3\pm 0.0The gains remain bidirectional: Phi→\rightarrowQwen improves by 10.1%, and Qwen→\rightarrowPhi improves by 9.2%. Random memory is substantially weaker in both directions, even though it uses the same target-side reader architecture and optimization budget. Thus, the observed gains cannot be attributed solely to the additional parameter capacity introduced by the reader.
C.4Target Model Scaling
Table15keeps the source fixed and scales the target from 2B to 9B parameters to check whether the transfer disappears once the target becomes stronger.
Table 15:Target model scaling: Qwen3.5-0.8B source→\toQwen3.5-{2B, 4B, 9B}. Mean±\pmstd over 3 seeds. Deltas shown next to the transferred result are relative to the no-memory baseline; negative values indicate better PPL. Baselines are eval-only (no reader training).TargetBaselineTransferredRandomQwen3.5-2B13.713.711.7±0.0\mathbf{11.7\pm 0.0}-14.1%12.6±0.012.6\pm 0.0Qwen3.5-4B10.510.59.6±0.09.6\pm 0.0-8.9%10.2±0.010.2\pm 0.0Qwen3.5-9B9.39.38.5±0.08.5\pm 0.0-8.3%9.0±0.09.0\pm 0.0The gains persist across all three target sizes. The largest relative improvement appears at 2B, but even the 9B target still benefits, which suggests that stronger backbones reduce but do not eliminate the value of the transferred memory.
C.5Analysis of Backbone Representations
We use centered kernel alignment (CKA)[Kornblith et al., 2019]to measure the similarity between hidden representations produced by different target backbones. Linear CKA compares the geometry of two representation matrices and is invariant to isotropic scaling and orthogonal transformations, making it suitable for comparing models with different hidden dimensions. We use WikiText-103 to training with sequence length 512.
For each backbone, we extract hidden states from the Transformer block located approximately one third of the way through the network,⌊L/3⌋\lfloor L/3\rfloor, whereLLis the total number of Transformer blocks. This corresponds to block 4 of 12 for Pythia-160M, block 8 of 24 for Pythia-410M, and block 7 of 22 for TinyLlama-1.1B. We process the same set of input sequences with each model, mean-pool the hidden states over non-padding tokens to obtain one representation per sequence, and compute pairwise linear CKA between the resulting representation matrices. We select the one-third-depth layer to compare intermediate representations after substantial contextual processing, while avoiding the stronger model-specific specialization that may occur near the output layer.
Figure 5:Linear CKA similarity between mean-pooled backbone representations at layer⌊L/3⌋\lfloor L/3\rfloorfor the three models shown in the matrix.Figure5provides a representation-level view of why transfer need not track backbone similarity monotonically. The two Pythia models exhibit substantially higher representational similarity, with a CKA score of0.5950.595, than either model does with TinyLlama, for which the scores are only0.0130.013and0.0070.007. Nevertheless, Pythia-160M→\rightarrowTinyLlama-1.1B is among the strongest cross-tokenizer transfer settings inTable11, reducing perplexity by10.6%10.6\%. By comparison, the more representationally similar Pythia-160M→\rightarrowPythia-410M pair improves perplexity by only1.6%1.6\%.
Representational similarity alone does not predict the magnitude of transfer gains in these experiments. Nevertheless, this result is consistent with the reader analysis inSection4.2: successful transfer depends on whether the target reader can recover and route useful information from the frozen memory, rather than requiring the source and target backbones to have closely aligned intermediate representation spaces. Also, the wording of “the main text’s claim” is bad.
Appendix DSupplementary Analysis of Corpus-Dependent Downstream Transfer for RQ3
These analyses extend RQ3 inSection4.3by separating corpus specialization, reader alignment, mixed-corpus behavior, and out-of-domain effects.
D.1Reference Downstream Pattern
Figure3reports the full six-task evaluation under FineWeb-Edu[Lozhkov et al.,]aligned Phase 2 training, so the table is not duplicated here. RTE and SciQ improve most consistently, BoolQ becomes positive for targets of 2B parameters and above, and TruthfulQA remains slightly negative at every scale. This pattern motivates the following diagnostics: corpus specialization inSectionD.2, Phase 2 reader alignment inSectionD.3, mixed-corpus training inSectionD.4, and unrelated-corpus evaluation inSectionD.5.
D.2Source-Corpus Specialization
Table16compares two source-corpus families under matched Phase 2 adaptation: a broad educational web corpus (FW) and a narrower QA-heavy corpus (Nemo HQ-DQA).
Table 16:Domain alignment ablation with corpus-matched Phase 2 readers. Positive gains are shown ingreenand negative changes inred.TargetSourceRTE↑\uparrowBoolQ↑\uparrowOpenBookQA↑\uparrowSciQ↑\uparrowTruthfulQA↑\uparrowRACE↑\uparrowQwen-0.8BFW-avg+6.1-0.6+0.7+1.8-0.1-0.2Nemo-avg+7.9+4.1∗+1.1+8.4∗-0.8+1.1∗Qwen-2BFW-avg+2.7+3.2+0.3+3.5-0.4-0.0Nemo-avg+8.7∗+14.6∗+0.5-0.7+0.0+1.1∗Qwen-4BFW-avg+1.3-0.2+0.6+0.9-0.5+0.1Nemo-avg+0.4+0.6+0.2-6.3-2.9-0.4Qwen-9BFW-avg+0.5+0.6+0.2+2.7-0.7+1.0Nemo-avg-4.6+2.1∗-0.3-3.7+0.2+1.4∗Nemo-avg>>FW-avg by≥2\geq 2percentage points.The main pattern is that domain-specialized memory helps most on smaller targets and QA-style tasks. Nemo-avg is especially strong on Qwen3.5-0.8B and Qwen3.5-2B for BoolQ, RTE, and SciQ, but the advantage is less uniform on 4B and 9B targets, showing that specialization helps most when the target still needs strong external support.
Table17controls for corpus size on Qwen3.5-2B by comparing an 8B-token STEM question-answering corpus (HQ-DQA) with a broader 26B-token general web corpus (HQ). Here, the smaller STEM question-answering corpus substantially outperforms the larger general web corpus on BoolQ and RTE and also improves RACE.
Table 17:Corpus structure vs. size on Qwen-2B: accuracy change (�%\Delta\%) under corpus-matched Phase 2 readers. Positive gains are shown ingreenand negative changes inred.Source corpusBoolQ↑\uparrowRTE↑\uparrowOBQA↑\uparrowSciQ↑\uparrowTQA↑\uparrowRACE↑\uparrowHQ-DQA (8B, STEM Q&A)+15.3±\pm0.3+8.9±\pm4.7+0.3±\pm0.2+0.9±\pm3.0+0.3±\pm0.2+1.4±\pm0.4HQ (26B, organic web)+3.0±\pm0.4+1.2±\pm2.1+0.5±\pm0.1-0.5±\pm0.0+0.2±\pm0.3-0.4±\pm0.2FW-avg (reference)+3.2+2.7+0.3+3.5-0.4-0.0
D.3Phase-2 Reader-Fitting Alignment
Table18probes the failure mode suggested in the main text: stronger source memories can look worse downstream if the target-side reader is trained on a mismatched Phase 2 objective.
Table 18:Reader alignment: downstream accuracy change (�%\Delta\%) under mismatched vs. matched Phase 2 adaptation. Positive gains are shown ingreenand negative changes inred.SourcePhase 2BoolQ�%↑\Delta\%\uparrowRTE�%↑\Delta\%\uparrowSciQ�%↑\Delta\%\uparrowMismatched Phase 2 (WikiText-103)10MWikiText-103+8.3±\pm0.5+7.7±\pm3.9-4.5±\pm0.550MWikiText-103+6.6±\pm0.4+6.1±\pm4.7-1.8±\pm0.6200MWikiText-103+0.0±\pm0.0+2.9±\pm4.7-1.4±\pm0.6Matched Phase 2 (HQ-DQA)10MHQ-DQA+14.1±\pm1.2+10.7±\pm4.0-2.3±\pm3.050MHQ-DQA+15.3±\pm0.3+8.9±\pm6.2+0.9±\pm3.7200MHQ-DQA+14.0±\pm0.8+6.6±\pm4.9+2.9±\pm4.6Under WikiText-103 Phase 2 adaptation, increasing source training from 10M to 200M tokens steadily hurts BoolQ transfer. Once Phase 2 is moved to the matched HQ-DQA distribution, however, all three source budgets recover to roughly 14–15% BoolQ gain, which shows that the apparent degradation is primarily an alignment problem at the reader rather than a problem with the frozen memory itself.
Table19gives a mechanistic view of the same effect through gate statistics.
Table 19:Gate activation statistics during BoolQ evaluation versus Phase 2 WikiText-103 training. Longer source training suppresses the gate globally; BoolQ shows a lower mean gate, while the near-closed-gate fraction rises on both distributions under mismatch.SourceGate meanFrac.<0.1<0.1tokensWikiText-103BoolQWikiText-103BoolQ10M0.60.55.6%3.6%50M0.60.59.1%2.9%200M0.50.417.5%15.3%Longer source training suppresses the gate more strongly overall, and this is visible on BoolQ prompts both in the lower mean gate and in the rise of near-closed gates. At 200M source tokens, the mean gate on BoolQ drops to 0.352 and the fraction of near-closed gates rises to 15.3%, while WikiText-103 has a slightly higher near-closed fraction at 17.5% on the same threshold metric, consistent with broad suppression under reader mismatch rather than a BoolQ-only effect.
D.4Broadening Specialist Memory through Corpus Mixing
Table20tests whether a strong specialist memory can be broadened without destroying the downstream gains that make it useful in the first place. The answer is mostly yes. A 50/50 HQ-DQA + FineWeb-Edu mix preserves the DQA aggregate and slightly exceeds the specialist reference, while sequential mixing and the orthogonal code mixture also retain most of the specialist advantage. The main tradeoff is that broader mixtures help coverage more than they help the QA-focused aggregate itself.
Table 20:Mixed-corpus memory study on Qwen3.5-2B. Positive gains are shown ingreenand negative changes inred. The two rightmost columns report aggregate scores: DQA-agg==mean(BoolQ, RTE, SciQ); Br-agg==mean(OBQA, TruthfulQA, RACE).DesignBoolQ↑\uparrowRTE↑\uparrowOBQA↑\uparrowSciQ↑\uparrowTQA↑\uparrowRACE↑\uparrowDQA-agg↑\uparrowBr-agg↑\uparrowSingle-corpus referencesHQ-DQA ref.+15.3±\pm0.3+8.9±\pm4.7+0.3±\pm0.2+0.9±\pm3.0+0.3±\pm0.2+1.4±\pm0.4+8.4+0.7FW-Edu ref.+2.5±\pm1.7+1.9±\pm0.2-0.0±\pm0.2+3.2±\pm0.6-0.2±\pm0.2+0.1±\pm0.1+2.5-0.1Mixed or broadened memories50/50 mix+14.3±\pm1.1+11.7±\pm3.5+0.4±\pm0.4-0.5±\pm3.8+0.3±\pm0.3+0.5±\pm0.3+8.5+0.4Sequential HQ→\toFW+13.6±\pm1.0+11.9±\pm2.8-0.0±\pm0.2-1.7±\pm2.7+0.1±\pm0.4+0.2±\pm0.3+7.9+0.1Orthogonal + Code+14.4±\pm0.9+9.8±\pm2.6+0.4±\pm0.3-0.4±\pm3.4-0.2±\pm0.7+1.3±\pm0.4+7.9+0.5
D.5Out-of-Domain Side Effects
Table21checks whether a reader trained in the WikiText-103 transfer setting systematically distorts language-modeling behavior on unrelated corpora, without additional adaptation. LAMBADA emphasizes long-range narrative completion, WikiText-103 provides a separately curated language-modeling test set, and C4 represents broad web text; together, they probe distribution shift away from the Phase 2 training corpus.
Table 21:Out-of-domain evaluation: test PPL (↓\downarrow) for baseline vs. transferred memory. Negative�\Deltaindicates improvement.Target ModelDatasetBaselineTransferred�\Delta(%)Pythia-410MLAMBADA41.041.041.041.0+0.0%WikiText-10322.622.622.1-2.4%C424.924.924.8-0.3%TinyLlama-1.1BLAMBADA23.623.623.5-0.7%WikiText-10310.610.610.1-5.0%C411.711.711.6-0.2%The OOD effect is mostly neutral. Both targets get a modest improvement on WikiText-103[Merity et al., 2017], while LAMBADA[Paperno et al., 2016]and C4[Raffel et al., 2020]stay very close to the baseline in either direction. This supports the paper’s boundary claim that transferred memory is not universally helpful, but it is also usually not catastrophically harmful outside the aligned regime.
D.6Reader–Evaluation Distribution Mismatch
To isolate whether transfer requires identical reader-training and evaluation distributions, we fix a WikiText-trained Pythia-160M source memory and fit Pythia-410M readers on either WikiText-103 or C4. We then evaluate each reader on both corpora.
Table 22:Cross-corpus reader fitting and evaluation. Lower PPL is better.Reader corpusEval. corpusNo memoryTransfer�\DeltaWikiTextWikiText22.61622.113-0.502WikiTextC424.72624.663-0.062C4WikiText22.61622.538-0.078C4C424.72624.604-0.122Results inTable22show that transfer improves over the no-memory target in all four conditions, so reader-training and evaluation corpora need not be identical. However, the substantially larger matched WikiText gain shows that the amount of recoverable signal remains distribution-sensitive. We therefore distinguishportability, which persists under moderate corpus mismatch, fromtransfer magnitude, which depends on alignment among the stored memory, reader-fitting distribution, and evaluation distribution.
Appendix EMultilingual Transfer
Word-boundary units are the default addressing granularity in our English-like experiments, but they are not appropriate for writing systems without reliable whitespace segmentation. For such inputs, we replace word units with NFKC-normalized, case-folded non-whitespace Unicode character events and apply the same deterministic hashing procedure to character spans. This preserves the model-independent addressing principle while changing only the canonical unit used to formnn-grams. This character-span variant restores non-degenerate cross-tokenizer addressing for Chinese and Japanese and yields positive transfer gains in both languages, as shown inTable23. We then trained Qwen2.5-0.5B source memories and Llama-3.2-1B target readers separately on Chinese and Japanese Wikipedia, using character-span hashing, four branches at layers 2/10, fixed held-out 262K-token test splits, and seeds 42/137/2024.
Table 23:Character-span addressing for non-segmented languages. Results are averaged over three seeds.LanguageNo-memory PPLTransfer PPLRel. improvementChinese20.124819.6718±0.003319.6718\pm 0.00332.251%Japanese15.355515.1985±0.025115.1985\pm 0.02511.023%
Appendix FComputational Cost Analysis
We count multiply-adds (MAdds) for dense projections and report memory addressing separately, since a table lookup is a memory-traffic operation rather than a dense matrix multiplication. LetNmaxN_{\max}denote the maximum n-gram order and letKKdenote the number of independent hash heads used for each ordern∈{2,…,Nmax}n\in\{2,\ldots,N_{\max}\}. The total number of hash heads is therefore
H=(Nmax−1)K.H=(N_{\max}-1)K.(15)RRbe the number of reader branches at one injection site, andSSbe the number of injection sites. For the default memory configuration,Nmax=3N_{\max}=3,K=4K=4,H=8H=8, and the dimension of memorydmem=Hdhead=512d_{\text{mem}}=Hd_{\text{head}}=512, wheredheadd_{\text{head}}is the dimension of the embedding row retrieved by one hash head. anddmemd_{\text{mem}}is the memory vector. Each token, therefore, requiresHHdeterministic hash indices andHHembedding-row reads, together supplyingHdhead=512Hd_{\text{head}}=512memory values. This cost isO(Nmax+H)O(N_{\max}+H)hash operations andO(Hdhead)O(Hd_{\text{head}})memory traffic per site and does not grow with the table sizeMM; in particular, it does not perform a nearest-neighbor search over theM=65,536M=65{,}536rows.
At one injection site of the target-side reader, the shared-value,RR-branch reader performs onedmem×dBd_{\text{mem}}\!\times\!d_{B}value projection andRRkey projections of the same shape, anddBd_{B}for the hidden dimension of target modelBB. If not, change it. The remaining reader operations include RMS normalization,RRkey-query dot products, sigmoid gates, scalar–vector products, and branch aggregation. These operations requireO(RdB)O(Rd_{B})work. We refer to this as the non-projection, lower-order term because it scales linearly withdBd_{B}, whereas the dense projections scale asdmemdBd_{\text{mem}}d_{B}. Withdmem=512d_{\text{mem}}=512, the projection term is substantially larger in the evaluated configurations. Lower than what? Thus, the dominant reader cost for a sequence of lengthTTand batch sizeBBis
Creader=BTS[(R+1)dmemdB+O(RdB)]MAdds,C_{\mathrm{reader}}=BTS\left[(R+1)d_{\text{mem}}d_{B}+O(Rd_{B})\right]\quad\text{MAdds},(16)or, per token and ignoring the lower-order gate terms,S(R+1)dmemdBS(R+1)d_{\text{mem}}d_{B}. The single-layer, single-branch reader hasR=S=1R=S=1, giving2⋅512⋅dB2\cdot 512\cdot d_{B}MAdds per token:1.051.05M atdB=1024d_{B}=1024(Pythia-410M) and4.194.19M atdB=4096d_{B}=4096(the 7B configurations). The dual-layer, four-branch QA reader hasS=2S=2andR=4R=4, so its dominant projection cost is2⋅5⋅512⋅4096=20.972\cdot 5\cdot 512\cdot 4096=20.97M MAdds per token. The shared value projection is counted once per injection site, not once per branch.
For completeness, the corresponding trainable reader parameter count is
Preader=S[(R+1)dmemdB+(R+1)dB+R],P_{\mathrm{reader}}=S\left[(R+1)d_{\text{mem}}d_{B}+(R+1)d_{B}+R\right],(17)where the second term accounts for the RMSNorm weights and the final term for the branch-specific gate biases. This gives approximately4.204.20M parameters for the one-site, one-branch reader atdB=4096d_{B}=4096, and21.0121.01M parameters for the two-site, four-branch reader used in the QA configuration. These reader parameters are the only additional trainable parameters in Phase 2; the transferred33.633.6M-parameter memory table and the target backbone remain frozen.
The comparison with attention should be read as a projection-only lower bound. A standard self-attention layer has four dense projections, giving4dB24d_{B}^{2}MAdds per token before counting the sequence-dependent attention score and value products. This is4.194.19M atdB=1024d_{B}=1024and67.1167.11M atdB=4096d_{B}=4096, compared with1.051.05M and4.194.19M for the single-branch reader. The additional attention products contributeO(TdB)O(Td_{B})work per token for a length-TTsequence, whereas the reader contribution remains constant per token. Consequently, the reader adds a small fixed overhead to each injected layer and does not change the transformer’s quadratic dependence on sequence length. During Phase 2, freezing the backbone and memory removes their optimizer-state and parameter-update costs, although their forward computation and the backward path needed to train the reader are still executed.
Amortization across consumers.
The practical advantage of a transferable artifact emerges when the same provider memory is reused across multiple targets. Using the measured payload times above, the cumulative cost of transfer toNNadapted targets is
Ctransfer(N)=0.653+3.798N,C_{\mathrm{transfer}}(N)=0.653+3.798N, whereas training a fresh target-specific memory for each consumer costs
Cfresh(N)=3.871N.C_{\mathrm{fresh}}(N)=3.871N. Under these measurements, the one-time provider cost is amortized after approximately nine adapted consumers by payload time. Using allocated GPU-hours gives a corresponding break-even point of approximately eleven consumers. These numbers are system- and configuration-specific rather than universal thresholds, but they illustrate the lifecycle regime in which reusable memory artifacts become advantageous. When the provider reader is directly compatible with the consumer, target-side adaptation is unnecessary: the measured consumer cost is zero training tokens and zero parameter updates, while mean QA rises from 32.12 to 38.31. The optional reader fitting raises it further to 38.50.
F.1Measured System and Lifecycle Cost
Table 24:Measured cost for the LLaMA-2-7B→\rightarrowMistral-7B-v0.3 using QA experiment configuration on LUMI supercomputer.ConditionTokensTrain timeGPU-hoursPeak inf. memoryPrefillNo memory00013.658±0.02013.658\pm 0.020GiB55.10±0.0655.10\pm 0.06msPhase-1 source artifact4.096M0.653 h3.154––Fresh Mistral memory19.968M3.871 h15.760––Matched FFN19.968M3.680 h14.92913.736±0.02013.736\pm 0.020GiB55.32±0.4955.32\pm 0.49msTransfer, incl. Phase 14.096M+19.968M4.451 h18.62313.881±0.02013.881\pm 0.020GiB57.69±0.4057.69\pm 0.40ms
SectionF.1shows the cost for transfer experiments from LLaMA-2-7B to Mistral-7B-v0.3. Relative to the no-memory target, the transferred configuration adds 0.223 GiB of peak inference memory and 2.60 ms of prefill latency in this measured setup. The Phase-1 source artifact is a one-time provider cost: the recorded run processes 4.096M tokens in 0.653 hours. Each adapted consumer then trains only the 21.01M-parameter reader, while the memory and backbone remain frozen.
Appendix GEthics and Broader Impact
Transferred memory raises both opportunities and risks. On the positive side, a frozen external memory can reduce repeated backbone retraining and make factual updates more modular. On the negative side, the same mechanism can propagate memorized or provenance-sensitive content across multiple target models if the source memory is not audited. Engram’s advantage is that its stored knowledge is explicit and addressable: entries can be inspected, removed, or replaced without retraining the backbone.
Appendix HLimitations
Scale and availability.Our experiments cover source and target models up to 9B parameters, but do not systematically characterize how target-side reader capacity should scale with backbone size, memory size, training data, or source–target heterogeneity. In particular, our two-phase protocol first trains the memory with the source model and then freezes both the transferred memory and the target backbone while adapting only the reader. We therefore do not explore joint training of the backbone, memory, and reader, nor establish a scaling law relating reader depth, branch count, injection placement, and parameter count to model scale. Although the present results show that stronger readers substantially improve frozen-memory extraction, it remains unclear whether the same reader configurations remain sufficient for much larger models or whether reader capacity must grow with the backbone or memory. We equip an existing multi-billion-parameter model with Engram and train it. So saying those multi-billion-parameter Engram models don’t exist is not a good argument.
Addressing and language diversity.Our default word-boundary addressing is effective for the evaluated English-like settings, but it is not universally tokenizer- or language-independent. In particular, whitespace-based word units degenerate for non-segmented Chinese and Japanese. We show that replacing word units with normalized character spans repairs this specific failure mode and produces positive transfer on Chinese and Japanese Wikipedia. These experiments demonstrate that the model-independent addressing principle can extend beyond whitespace-delimited text, but they do not establish a universally optimal canonical unit. Code, noisy Unicode, mixed-script text, byte-level tokenizers, and additional writing systems may require character-, byte-, or task-specific span definitions. More generally, portability requires agreement on both the canonicalization procedure and the granularity from which memory addresses are formed. More aggressive tokenizer mismatch may require a stronger key pipeline than the word-boundary canonicalization used here.
Reader scope.Our capacity diagnostics further show that both interface width and reader placement affect extraction quality; the defaultdmem=512d_{\mathrm{mem}}=512setting should therefore be viewed as an evaluated operating point rather than a fixed architectural requirement. The target-side reader is a central component of our transfer framework, but the present study evaluates only a deliberately lightweight subset of the possible reader design space. The main transfer matrix uses a single-branch linear reader, while the QA experiments extend it with shared-value multi-branch reading and multi-layer injection. These controlled designs make it possible to isolate the role of target-side extraction while keeping the number of trainable parameters small. Nevertheless, more heterogeneous source–target pairs may benefit from nonlinear transformations, adaptive branch aggregation, layer-dependent value projections, or learned injection placement. Such extensions could improve extraction capacity, but would also increase computation and make it harder to distinguish reusable memory content from capacity introduced by the reader itself. In addition, our results indicate that the usefulness of context-dependent gating varies with the source and target distributions. A broader study across corpus shifts and model families is needed to characterize when gating is essential and when a simpler reader is sufficient.
相似文章
MemoWM:世界模型如何改变智能体的记忆需求
MemoWM 提出一种以世界模型为条件的记忆分配框架,通过共享预测压缩智能体长期经验存储,在五个长时程智能体记忆基准上达到42.42%平均准确率,同时将经验特定存储降低53.9%。研究还揭示了共享模型容量与重复存储成本之间的权衡关系。
@Pluvio9yte: 好家伙,终于有人正经对 模型上下文一长就开始都失忆 这个痛点动手了。 一个一直被行业视作理所当然、但实际极度畸形的默认设定是:大模型在出厂的那一刻,它的能力就被彻底冻结了。 大家今天为了让模型「记住」,所有的工程努力几乎全部发生在模型外面…
MemTensor 团队发布 MemT 1.0,将记忆机制从外挂式 RAG/Agent Memory 推入模型计算核心,让模型在解决任务过程中自行沉淀经验,官方测试显示 Coding、Working 场景提升 5% 以上,源码已在 Hugging Face 开源。
@ba_niu80557: https://x.com/ba_niu80557/status/2073362883728302125
斯坦福大学提出AutoMem方法,通过让模型学会管理记忆(选择性遗忘)而非扩大参数,使320亿参数小模型性能翻倍并追平顶级大模型,揭示了记忆管理比模型规模更重要。
我们训练了一个把记忆保存在自身前向传播中的模型。
团队发布了 MemT 2.0——一个 293B 的记忆原生 MoE 模型,将历史信息压缩为持久内部记忆状态,在前向传播中通过 memory attention 完成读写更新,无需梯度或重新训练,36B 与 293B 版本已上线 Hugging Face。
@seclink: 陈天桥再不努力一把, 大模型记忆就要被字节偷家了... 赶了个大早, 很努力,但是执行的人不行 ... OpenViking 开源的 cli 工具做了好多迭代优化... 早晚你们会想起,用AI 编程改造复杂项目时, 一定会用上大模型记忆的…
OpenViking是一个开源的CLI工具,旨在通过大模型记忆功能优化复杂项目的AI编程体验并节省token。文章评论了其在执行层面的表现以及与字节跳动等竞争者在LLM记忆领域的动态。