Procedural Memory Distillation: Online Reflection for Self-Improving Language Models
On the map: Synthetic data and self-improvement for LLMs → Self-rewarding alignment and self-training
Abstract (arXiv)
Reinforcement learning with verifiable rewards (RLVR), along with recent selfdistillation variants such as SDPO, evaluates each rollout against a verifier and updates the policy from that episode-level signal. However, the richer procedural information in the rollout is rarely retained or reused. Across episodes and epochs, the model repeatedly encounters related problems under a changing policy, producing cross-episode signals that episode-local updates cannot capture: which strategies consistently pass verification, which failure modes persist, which patterns recur. We propose Procedural Memory Distillation (PMD), which converts these crossepisode signals into reusable procedural memory and distills it into the policy's weights during training. This memory functions as a training scaffold, absorbed into the policy itself, yielding a memory-free model at inference. PMD organizes the memory at three levels of abstraction: raw trajectories, self-reflected strategies and lessons, and higher-level behavioral patterns that recur across problems, all extracted online from the model's own trajectories. A memory-conditioned self-teacher draws on the accumulated experience to supervise the student on its own rollouts, enabling student to progressively internalize procedural knowledge within its parameters. The central design principle is co-evolution: the policy generates rollouts that update the memory, and memory shapes the supervision that updates the policy. Empirically, across Qwen3-8B and OLMo3-Instruct-7B, PMD improves over SDPO by 3.8-5.5% on SCIKNOWEVAL and 7.9-13.6% on LIVECODEBENCH. Co-evolution powers these gains: freezing either the memory or the policy trails PMD by more than 10% across SCIKNOWEVAL domains.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Self-rewarding alignment and self-training, one of 13 directions in Synthetic data and self-improvement for LLMs. Builds on 45 of its 54 references that are in the map. First posted on arXiv 1 Jul 2026.
Builds on
45 of 54 references.
Closest ancestors · 1
Same direction, newest first
Other references · 44
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
- 2025Qwen3 Technical Reportcited by 8,195
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,064
- 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,500
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,317
- 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,081
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
32 more
- 2023MemGPT: Towards LLMs as Operating Systemscited by 1,295
- 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,175
- 2025A-MEM: Agentic Memory for LLM Agentscited by 1,000
- 2023MemoryBank: Enhancing Large Language Models with Long-Term Memorycited by 539
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Modelscited by 386
- 2026Reinforcement Learning via Self-Distillationcited by 285
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2025ReasoningBank: Scaling Agent Self-Evolving with Reasoning Memorycited by 253
- 2026SkillRL: Evolving Agents via Recursive Skill-Augmented Reinforcement Learningcited by 252
- 2025Memory-R1: Enhancing Large Language Model Agents to Manage and Utilize Memories via Reinforcement Learningcited by 220
- 2026Self-Distillation Enables Continual Learningcited by 212
- 2025R-Zero: Self-Evolving Reasoning LLM from Zero Datacited by 210
- 2026On-Policy Context Distillation for Language Modelscited by 161
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
- 2025EvolveR: Self-Evolving LLM Agents through an Experience-Driven Lifecyclecited by 129
- 2025Memento: Fine-tuning LLM Agents without Fine-tuning LLMscited by 119
- 2026Self-Distilled RLVRcited by 109
- 2025MemGen: Weaving Generative Latent Memory for Self-Evolving Agentscited by 99
- 2026Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?cited by 87
- 2026SKILL0: In-Context Agentic Reinforcement Learning for Skill Internalizationcited by 82
- 2026Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agentscited by 75
- 2026Agentic Memory: Learning Unified Long-Term and Short-Term Memory Management for Large Language Model Agentscited by 72
- 2025MemoryGraft: Persistent Compromise of LLM Agents via Poisoned Experience Retrievalcited by 65
- 2026Self-Distillation Zero: Self-Revision Turns Binary Rewards into Dense Supervisioncited by 58
- 2025Mem-α: Learning Memory Construction via Reinforcement Learningcited by 49
- 2026Embarrassingly Simple Self-Distillation Improves Code Generationcited by 36
- 2026Online Experiential Learning for Language Modelscited by 24
- 2026SkillOrchestra: Learning to Route Agents via Skill Transfercited by 22
- 2026Mem2Evolve: Towards Self-Evolving Agents via Co-Evolutionary Capability Expansion and Experience Distillationcited by 10
- 2026Meta-Reinforcement Learning with Self-Reflection for Agentic Searchcited by 8
- 2025TokMem: One-Token Procedural Memory for Large Language Modelscited by 5
- 2026MemCoRe: Recovering Evidence from Progressively Compressed Factual Knowledge for Agent Memorycited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Recursive Self-Improvement in AI: From Bounded Self-Refinement to Autonomous Research Loopscited by 25
- 2026Privileged Likelihood Is Not Automatically Value: Three Checks for Token Credit in On-Policy Self-Distillationcited by 3
- 2026Shockingly Simple Self-retrospection Improves Agentic Models Without RLcited by 0
- 2026Shockingly Simple Self-retrospection Improves Agentic Models Without RL28 Sep 2026
- 2026Privileged Likelihood Is Not Automatically Value: Three Checks for Token Credit in On-Policy Self-Distillation10 Aug 2026
- 2026Recursive Self-Improvement in AI: From Bounded Self-Refinement to Autonomous Research Loops8 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026CONTRAMEM: Learning Self-Evolving Procedural Memory from Contrasting Multi-Model Trajectories
- 2026Self-Review Reinforcement Learning (SRRL) with Cross-Episode Memory and Policy Distillation
- 2026Verifiable Memory: Learning Unified Memory Management with Local and Global Verifiers for Large Language Model Agents
- 2026Continual Self-Improvement with Lightweight Experiential Latent Memories
- 2026PSD: Pseudo Self-Distillation of Memory Representation Capabilities for LLM Agents
- 2026COUNTERMEM: World-Model Verified Counter-Factual Memory for Language Agents
- 2026On-policy Distillation with Verifiable Reward
- 2026NLPG: Natural-Language Policy Gradients for Self-Evolving Language Agents