Towards Cold-Start Drafting and Continual Refining: A Value-Driven Memory Approach with Application to NPU Kernel Synthesis
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
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2025Qwen3 Technical Reportcited by 8,820
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
- 2023Generative Agents: Interactive Simulacra of Human Behaviorcited by 3,568
- 2023Self-Refine: Iterative Refinement with Self-Feedbackcited by 3,442
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
- 2023Voyager: An Open-Ended Embodied Agent with Large Language Modelscited by 2,191
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2023MemGPT: Towards LLMs as Operating Systemscited by 1,389
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 952
33 more
- 2025AlphaEvolve: A coding agent for scientific and algorithmic discoverycited by 867
- 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
- 2025Mem0: Building Production-Ready AI Agents with Scalable Long-Term Memorycited by 779
- 2023The Flan Collection: Designing Data and Methods for Effective Instruction Tuningcited by 739
- 2024Large Language Models: A Surveycited by 525
- 2025KernelBench: Can LLMs Write Efficient GPU Kernels?cited by 225
- 2023TPU v4: An Optically Reconfigurable Supercomputer for Machine Learning with Hardware Support for Embeddingscited by 202
- 2018Multi-task Deep Reinforcement Learning with PopArtcited by 141
- 2024Seven Failure Points When Engineering a Retrieval Augmented Generation Systemcited by 139
- 2025MemOS: A Memory OS for AI Systemcited by 138
- 2025EvolveR: Self-Evolving LLM Agents through an Experience-Driven Lifecyclecited by 137
- 2026MemRL: Self-Evolving Agents via Runtime Reinforcement Learning on Episodic Memorycited by 129
- 2025Memento: Fine-tuning LLM Agents without Fine-tuning LLMscited by 127
- 2025Memp: Exploring Agent Procedural Memorycited by 103
- 2025mHC: Manifold-Constrained Hyper-Connectionscited by 88
- 2024Generalization v.s. Memorization: Tracing Language Models' Capabilities Back to Pretraining Datacited by 85
- 2025Kevin: Multi-Turn RL for Generating CUDA Kernelscited by 78
- 2025TritonBench: Benchmarking Large Language Model Capabilities for Generating Triton Operatorscited by 75
- 2024A Survey on LLM-based Code Generation for Low-Resource and Domain-Specific Programming Languagescited by 66
- 2023Settling the Sample Complexity of Online Reinforcement Learningcited by 45
- 2025AutoTriton: Automatic Triton Programming with Reinforcement Learning in LLMscited by 40
- 2023CodeScope: An Execution-based Multilingual Multitask Multidimensional Benchmark for Evaluating LLMs on Code Understanding and Generationcited by 39
- 2025MultiKernelBench: A Multi-Platform Benchmark for Kernel Generationcited by 38
- 2025STARK: Strategic Team of Agents for Refining Kernelscited by 25
- 2025TritonRL: Training LLMs to Think and Code Triton Without Cheatingcited by 24
- 2024Beyond Sparse Rewards: Enhancing Reinforcement Learning with Language Model Critique in Text Generationcited by 22
- 2023A Survey on Deep Learning Hardware Accelerators for Heterogeneous HPC Platformscited by 21
- 2025QiMeng-Kernel: Macro-Thinking Micro-Coding Paradigm for LLM-Based High-Performance GPU Kernel Generationcited by 19
- 2026Towards Automated Kernel Generation in the Era of LLMscited by 11
- 2025LLLMs: A Data-Driven Survey of Evolving Research on Limitations of Large Language Modelscited by 9
- 2025AKG kernel Agent: A Multi-Agent Framework for Cross-Platform Kernel Synthesiscited by 7
- 2024RAGSys: Item-Cold-Start Recommender as RAG Systemcited by 4
- 2025Sample-Efficient Reinforcement Learning from Human Feedback via Information-Directed Samplingcited by 4
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generationcited by 1
- 2026CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limitscited by 1
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesiscited by 0
- 2026From Experience to Expertise: Adoption-Aware Memory Learning for Data-Scarce NPU Kernel Synthesis28 Sep 2026
- 2026CANN Bench: Benchmarking Agent Generated Kernels against Real NPU and Algorithmic Limits8 Jul 2026
- 2026Hawk: Harnessing Hardware-Aware Knowledge for High-Performance NPU Kernel Generation2 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026NLPG: Natural-Language Policy Gradients for Self-Evolving Language Agents
- 2026Compiled Memory: Not More Information, but More Precise Instructions for Language Agents
- 2026Learning How and What to Memorize: Cognition-Inspired Two-Stage Optimization for Evolving Memory
- 2026CONTRAMEM: Learning Self-Evolving Procedural Memory from Contrasting Multi-Model Trajectories
- 2026KernelBlaster: Continual Cross-Task CUDA Optimization via Memory-Augmented In-Context Reinforcement Learning
- 2026Beyond Scaling: Self-Evolving LLM Agents for Hardware Kernel Optimization via an Experience-Driven Workflow and Experience Graph Memory
- 2026Towards Feedback-to-Plan Decisions for Self-Evolving LLM Agents in CUDA Kernel Generation
- 2026Rosetta Memory: Adaptive Memory for Cross-LLM Agents