mlsanityData to 14 Oct 2026

Towards Cold-Start Drafting and Continual Refining: A Value-Driven Memory Approach with Application to NPU Kernel Synthesis

arXiv 2603.10846 ↗First posted 11 Mar 2026cs.LG cs.AI cs.CLcited by 3

Not on the map. Why?

Abstract (arXiv)

Deploying Large Language Models to data-scarce programming domains poses significant challenges, particularly for kernel synthesis on emerging Domain-Specific Architectures where a "Data Wall" limits available training data. While models excel on data-rich platforms like CUDA, they suffer catastrophic performance drops on data-scarce ecosystems such as NPU programming. To overcome this cold-start barrier without expensive fine-tuning, we introduce EvoKernel, a self-evolving agentic framework that automates the lifecycle of kernel synthesis from initial drafting to continual refining. EvoKernel addresses this by formulating the synthesis process as a memory-based reinforcement learning task. Through a novel value-driven retrieval mechanism, it learns stage-specific Q-values that prioritize experiences based on their contribution to the current objective, whether bootstrapping a feasible draft or iteratively refining latency. Furthermore, by enabling cross-task memory sharing, the agent generalizes insights from simple to complex operators. By building an NPU variant of KernelBench and evaluating on it, EvoKernel improves frontier models' correctness from 11.0% to 83.0% and achieves a median speedup of 3.60x over initial drafts through iterative refinement. This demonstrates that value-guided experience accumulation allows general-purpose models to master the kernel synthesis task on niche hardware ecosystems. Our official page is available at https://evokernel.zhuo.li.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 45 of its 54 references that are in the map. First posted on arXiv 11 Mar 2026.

Builds on

45 of 54 references.

References · 45

  1. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2020Scaling Laws for Neural Language Modelscited by 6,586
  4. 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
  5. 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,568
  6. 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
  7. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  8. 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
  9. 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,191
  10. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  11. 2023MemGPT: Towards LLMs as Operating Systemscited by 1,389
  12. 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 952
33 more
  1. 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 867
  2. 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
  3. 2025Mem0: Building Production-Ready AI Agents with Scalable Long-Term Memorycited by 779
  4. 2023The Flan Collection: Designing Data and Methods for Effective Instruction Tuningcited by 739
  5. 2024Large Language Models: A Surveycited by 525
  6. 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
  7. 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 202
  8. 2018Multi-task Deep Reinforcement Learning with PopArtcited by 141
  9. 2024Seven Failure Points When Engineering a Retrieval Augmented Generation Systemcited by 139
  10. 2025MemOS: A Memory OS for AI Systemcited by 138
  11. 2025EvolveR: Self-Evolving LLM Agents through an Experience-Driven Lifecyclecited by 137
  12. 2026MemRL: Self-Evolving Agents via Runtime Reinforcement Learning on Episodic Memorycited by 129
  13. 2025Memento: Fine-tuning LLM Agents without Fine-tuning LLMscited by 127
  14. 2025Memp: Exploring Agent Procedural Memorycited by 103
  15. 2025mHC: Manifold-Constrained Hyper-Connectionscited by 88
  16. 2024Generalization v.s. Memorization: Tracing Language Models' Capabilities Back to Pretraining Datacited by 85
  17. 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
  18. 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
  19. 2024A Survey on LLM-based Code Generation for Low-Resource and Domain-Specific Programming Languagescited by 66
  20. 2023Settling the Sample Complexity of Online Reinforcement Learningcited by 45
  21. 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 40
  22. 2023CodeScope: An Execution-based Multilingual Multitask Multidimensional Benchmark for Evaluating LLMs on Code Understanding and Generationcited by 39
  23. 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 38
  24. 2025STARK: Strategic Team of Agents for Refining Kernelscited by 25
  25. 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
  26. 2024Beyond Sparse Rewards: Enhancing Reinforcement Learning with Language Model Critique in Text Generationcited by 22
  27. 2023A Survey on Deep Learning Hardware Accelerators for Heterogeneous HPC Platformscited by 21
  28. 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 19
  29. 2026Towards Automated Kernel Generation in the Era of LLMscited by 11
  30. 2025LLLMs: A Data-Driven Survey of Evolving Research on Limitations of Large Language Modelscited by 9
  31. 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesiscited by 7
  32. 2024RAGSys: Item-Cold-Start Recommender as RAG Systemcited by 4
  33. 2025Sample-Efficient Reinforcement Learning from Human Feedback via Information-Directed Samplingcited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026NLPG: Natural-Language Policy Gradients for Self-Evolving Language Agents
  2. 2026Compiled Memory: Not More Information, but More Precise Instructions for Language Agents
  3. 2026Learning How and What to Memorize: Cognition-Inspired Two-Stage Optimization for Evolving Memory
  4. 2026CONTRAMEM: Learning Self-Evolving Procedural Memory from Contrasting Multi-Model Trajectories
  5. 2026KernelBlaster: Continual Cross-Task CUDA Optimization via Memory-Augmented In-Context Reinforcement Learning
  6. 2026Beyond Scaling: Self-Evolving LLM Agents for Hardware Kernel Optimization via an Experience-Driven Workflow and Experience Graph Memory
  7. 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generation
  8. 2026Rosetta Memory: Adaptive Memory for Cross-LLM Agents