AI Agents 相关度: 9/10

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

Yujie Zheng, Zhuo Li, Shengtao Zhang, Hanjing Wang, Junjie Sheng, Jiaqian Wang, Junchi Yan, Weinan Zhang, Ying Wen, Bo Tang, Muning Wen
arXiv: 2603.10846v1 发布: 2026-03-11 更新: 2026-03-11

AI 摘要

EvoKernel通过价值驱动的记忆机制,使LLM在NPU内核合成中实现冷启动并持续优化。

主要贡献

  • 提出了EvoKernel框架,实现内核合成的自动化
  • 引入了价值驱动的记忆检索机制,提升学习效率
  • 通过跨任务记忆共享,增强了模型的泛化能力

方法论

将内核合成视为基于记忆的强化学习任务,通过Q值评估经验,优化drafting和refining。

原文摘要

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.

标签

内核合成 强化学习 记忆检索 NPU

arXiv 分类

cs.LG cs.AI cs.CL