Paper

Cortex: Consolidating Context into Parametric Memory

Prospective-query distillation writes context gains into FFN/MoE memory, using mergeable low-rank or structural updates instead of keeping context at inference.

arXiv ↗Oct 14, 2026Code ↗

  • nlp
  • llm
  • cv
  • model editing
  • knowledge distillation
  • continual learning
Detailed method/pipeline diagram showing consolidation construction, settings, and teacher-student training approach
Detailed method/pipeline diagram showing consolidation construction, settings, and teacher-student training approach

What it is

Turns temporary context gains into persistent model parameters by distilling context-conditioned behavior into mergeable FFN updates while preserving other capabilities. Instead of relying on retrieval or task-specific editing interfaces, it consolidates context-induced behavior into FFN/MoE parameters. It generates prospective queries from the context, then distills context-conditioned outputs into mergeable low-rank or structural updates, with replay for sequential writes.

Key results

  • 50 sequential factual edits on 14B: 88.9% CR and 99.8% RR
  • Beats AlphaEdit by +66.2 CR points and +3.8 RR points on that setting
  • On 4B-120B dense, MoE, text and multimodal backbones, sits on the CR-RR frontier

Context-conditioned behavior can be internalized as targeted FFN updates with minimal interference to the model's original capabilities.

One useful ML email a week.

Worth your time, not all of your time.

We track new ML work across papers, code, and data, then keep only the pieces worth a closer look. Selected, not collected.

Papers.Data.Code

The Weekly ML Signal

Selected, not collected.

One useful ML email a week: the papers, repos and datasets worth your attention.

Worth your time, not all of your time.