跳到正文
arXiv cs.AI· Yansong Sun, Shenxiu Wu, Siyuan Chen, Runlin Hou, Junhao Qiu, Junming Cao, Shudi Shao, Zhichao Lu, Qingfu Zhang·· 2 天前AI 评分39

不重新生成,而是调试:面向加速器内核修复的领域专用智能体

Don't Regenerate, Debug: A Domain-Specific Agent for Repairing Near-Miss Accelerator Kernels

AI 导读

针对 GPU、NPU 等加速器内核生成中大量"近失"候选被丢弃的问题,研究者提出领域专用调试智能体,通过检索模式、诊断插桩、反作弊检测与收敛保护实现自主修复。其 Debug Pass@1 达 66.7%,高于重新生成的平均 Pass@1 25.9% 和 Pass@3 40.7%,且每次成功消耗的 token 比三次重新生成少 92.8%。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:面向 GPU、NPU 等硬件加速器的内核生成已成为大语言模型(LLM)的试验场,最先进的系统通过将 LLM 与智能体强化学习和进化搜索相结合的流水线来提升正确性。此类流水线会生成、编译并执行大量候选内核,丢弃其中大部分,从而错失了将失败提炼为可复用知识的机会。许多被丢弃的候选是接近成功的算子,它们能编译并运行,但未通过数值验证;每一个都蕴含着真实的领域知识,以及在 LLM 推理、交叉编译和硬件执行上的不小投入。我们主张一种范式转变:与其重新生成,不如调试。调试比从零生成受约束得多:搜索空间小,反馈密集。我们提出一个领域专用的调试智能体,解决自主修复中的三个核心挑战:通过检索模式和诊断插桩缓解知识稀缺,通过反作弊检测和全覆盖评估确保完整性,以及通过收敛保护和有界迭代控制成本。调试发挥两种互补作用:它通过恢复反复重新生成无法产出的算子来拓展能力边界,同时降低每个可交付算子的成本。Debug Pass@1 达到 66.7%,而 Regenerate Avg Pass@1 为 25.9%、Regenerate Pass@3 为 40.7%,同时每次成功消耗的 token 比三次试验重新生成少 92.8%。组件消融表明,知识库驱动了恢复,而完整性门控拒绝了工作流自身所接受的 12.5-33.3% 的成功案例。
学科: 软件工程(cs.SE);人工智能(cs.AI)
引用为: arXiv:2608.02712 [cs.SE]
  (或 arXiv:2608.02712v2 [cs.SE] 用于此版本)
  https://doi.org/10.48550/arXiv.2608.02712

arXiv 颁发的 DOI,经由 DataCite

提交历史

来自:Yansong Sun [查看邮箱]
[v1] 2026年8月3日星期一 17:59:48 UTC(586 KB)
[v2] 2026年10月15日星期四 12:33:08 UTC(586 KB)

来源:arXiv cs.AI · arxiv.org