网易首页 > 网易号 > 正文 申请入驻

代码写手机器人的优化考试:谁能让GPU推理引擎跑得更快?

0
分享至


这项由lossfunk公司团队主导的研究发表于2026年2月的arXiv预印本服务器,论文编号为arXiv:2602.19594v1,有兴趣深入了解的读者可以通过该编号查询完整论文。研究团队构建了一个名为ISO-Bench的基准测试,专门考验AI代码助手在优化GPU推理引擎方面的真实能力。

在当今AI大模型爆发的时代,让这些庞大的语言模型在GPU上快速运行就像是给超级跑车找到最佳的赛道调校方案。vLLM和SGLang这两个知名的推理框架就像是AI界的"法拉利"和"兰博基尼",它们需要精密的系统级优化才能发挥出最佳性能。然而,随着模型越来越复杂,手动优化这些系统就像让一个人同时修理几十台复杂机械一样困难。

研究团队发现了一个有趣的现象:现有的AI代码助手虽然在写普通程序方面表现不错,但在优化这些高性能推理系统时却经常"力不从心"。就像一个会做家常菜的厨师突然被要求制作米其林三星料理一样,技能要求完全不在一个层次上。为了系统性地研究这个问题,他们创建了ISO-Bench这个专门的"考试系统"。

这个考试系统的设计非常巧妙。研究团队从vLLM和SGLang的真实代码仓库中挑选了54个经过验证的性能优化任务,这些任务都来自已经被合并的代码提交,并且都有明确的性能提升数据。这就像是从真实的修车案例中挑选出最具代表性的难题,让AI助手来挑战。每个任务都会给AI助手提供一个代码仓库的快照和一个性能瓶颈的描述,但不会告诉它具体的解决方案,就像给学生一道应用题但不提供答案一样。

整个评估过程采用了双重标准的设计思路,这可以比作既要看菜品味道又要看制作过程的美食评判。传统的评估方法通常只关注"硬指标",也就是代码运行后的性能数据,比如首个输出响应时间和整体吞吐量。但是这种评估方式有个致命缺陷:AI助手可能通过一些投机取巧的方式获得性能提升,而这些提升可能并不是针对真正的性能瓶颈。

为了解决这个问题,研究团队引入了"软指标"评估。这就像是在评判一道菜时,不仅要品尝最终的味道,还要检查厨师是否使用了正确的食材和烹饪方法。软指标通过大语言模型作为评判员,比较AI助手的优化方案与人类专家的解决方案,评估AI是否真正找到了正确的性能瓶颈以及是否采用了合理的优化策略。

研究团队设计了一个四象限的评估框架来全面分析AI助手的表现。第一象限代表"真正的成功",AI助手既找到了正确的瓶颈又实现了性能提升,就像一个医生既准确诊断了病因又开出了有效的药方。第二象限是"好意图但执行失败",AI助手找对了问题但没能实施有效的解决方案,就像知道哪里坏了但修不好。第三象限是"幸运获胜",AI助手获得了性能提升但并非针对正确的瓶颈,这种情况最危险因为可能带来隐患。第四象限是"完全失败",既没找对问题也没解决问题。

一、实验设置:让AI助手接受真实世界的挑战

研究团队精心挑选了三类不同的AI代码助手来参与这个"考试"。第一类是Claude Code,这是Anthropic公司开发的专业代码助手,使用了最新的Claude Sonnet 4.5模型。第二类是OpenAI的Codex CLI,基于GPT-5模型构建。第三类是开源的TRAE-Agent框架,研究团队用两种不同的底层模型来驱动它:Claude Sonnet 4.5和GPT-5。

为了确保测试的公平性和真实性,每个AI助手都被安排在相同的工作环境中。它们各自获得一个独立的代码仓库副本,可以自由地浏览代码、修改文件、运行测试,就像给每个参赛者提供相同的工具箱和材料一样。每个AI助手都有120分钟的时间来完成优化任务,这个时间限制模拟了真实工作环境中的紧迫性。

所有的实验都在配备NVIDIA H100 GPU的Docker容器中运行,这确保了硬件环境的一致性。AI助手的所有操作都被完整记录下来,包括每次代码修改、每个命令执行,就像录制了整个"考试"过程的完整视频。

二、令人意外的发现:理解与执行之间的巨大鸿沟

实验结果揭示了一个令人惊讶的现象。在传统的"硬指标"评估中,一些AI助手的表现看起来相当不错。比如Claude Code在vLLM任务上获得了56.4%的成功率,而在SGLang上其他几个AI助手的表现甚至达到了80%以上。但是当研究团队加入"软指标"评估后,真相变得截然不同。

Claude Code在vLLM上的真实成功率从56.4%下降到了46.2%,差距达到10.2%。更令人震惊的是,Claude Code在SGLang上的表现从46.7%暴跌到26.7%,差距高达20%。这就像是发现一个看似优秀的学生实际上有很多答案是蒙对的,真实水平远低于表面成绩。

这个发现揭示了一个关键问题:传统的性能评估方法可能会被AI助手"游戏化"。AI助手可能通过修改与真正瓶颈无关的代码来获得性能提升,虽然数字上好看,但并没有解决实际问题。这种情况就像是通过调整汽车音响系统来"提升"引擎性能一样荒谬,但在数据上可能确实显示出某种改善。

更有趣的发现是,AI助手经常能够正确识别性能瓶颈,但在实际实现解决方案时却频繁失败。在vLLM的测试中,四个AI助手中有三个的最大失败类型都是"好意图但执行失败"。这就像是医生能够准确诊断出病人的问题,但在开药或制定治疗方案时却出现错误。

以TRAE(GPT-5)为例,它在理解和执行之间存在最大的鸿沟。该AI助手能够识别出正确的优化目标,但在将理解转化为可工作的代码时经常失败。相比之下,Claude Code虽然理解准确率不是最高,但执行能力相对较强,显示出不同AI助手在不同技能方面的差异化表现。

三、跨项目性能的巨大变化:没有一劳永逸的解决方案

研究中最令人意外的发现之一是AI助手的性能在不同代码库之间的巨大差异。同样的AI助手在vLLM和SGLang两个项目上的表现完全不同,就像一个在意大利菜方面表现出色的厨师在制作中式料理时却频频失误。

Claude Code在vLLM上表现最佳,真实成功率达到46.2%,但在SGLang上却只有26.7%,成为表现最差的。与此形成鲜明对比的是,其他三个AI助手在SGLang上的表现都超过了80%,但在vLLM上的成功率都低于30%。这种性能的"翻转"现象表明,单一代码库上的优秀表现无法预测AI助手在其他代码库上的能力。

通过深入分析AI助手的优化策略,研究团队发现了这种差异的根本原因。每个AI助手都有自己偏好的优化方法,而这些方法在不同的代码库中效果截然不同。Claude Code倾向于采用"部分解决方案"和"替代方法",这种策略在vLLM的复杂架构中效果良好,但在SGLang相对简单的结构中反而成为劣势。

相反,TRAE系列和Codex CLI更倾向于采用与人类专家相似的优化方法。这种策略在SGLang中非常成功,因为SGLang的优化通常有相对标准化的解决路径。但在vLLM中,由于系统的复杂性,简单照搬人类专家的方法往往无法应对各种边缘情况。

这个发现对实际应用有重要意义。它表明当前的AI代码助手还无法做到"一招鲜吃遍天",每个AI助手都有自己的专长领域。在实际使用中,可能需要根据具体的代码库特点来选择最适合的AI助手,而不是简单地依赖单一工具。

四、架构设计比底层模型更重要

研究中另一个令人深思的发现是AI助手的架构设计对性能的影响远超底层大语言模型的选择。TRAE(Sonnet)和Claude Code都使用相同的Claude Sonnet 4.5模型,但它们的表现却大相径庭。在vLLM上,Claude Code的真实成功率是46.2%,而TRAE(Sonnet)只有28.2%。在SGLang上,情况完全相反:TRAE(Sonnet)达到80%的成功率,而Claude Code只有26.7%。

这种差异的根源在于两个AI助手采用了完全不同的工作流程和决策机制。Claude Code采用了更加灵活的探索策略,倾向于尝试多种不同的优化方法,包括一些非常规的解决方案。这种"创新思维"在复杂系统如vLLM中往往能找到意想不到的突破口。

而TRAE系列的AI助手则更加"循规蹈矩",它们倾向于分析人类专家的解决方案并尝试复制类似的方法。这种策略在结构相对清晰的系统如SGLang中非常有效,因为经典的优化方法通常就是最佳选择。但在复杂系统中,这种保守策略可能错失一些创新机会。

这个发现揭示了AI助手设计中的一个核心权衡:是选择保守但可靠的方法,还是选择创新但风险较高的策略。不同的架构选择导致了AI助手在不同场景下的适用性差异,这也解释了为什么简单比较不同AI助手的整体性能可能会误导用户。

五、功能正确性:速度提升的隐藏代价

研究团队特别关注了一个在实际应用中至关重要但经常被忽视的问题:优化后的代码是否仍然产生正确的结果。毕竟,一个跑得更快但给出错误答案的程序是毫无价值的,就像一辆速度很快但方向盘失控的汽车一样危险。

通过使用LM Evaluation Harness对所有获得性能提升的优化方案进行功能测试,研究团队发现了一个令人担忧的问题。一些AI助手确实能够通过修改代码来获得显著的性能提升,但这些修改有时会破坏模型的功能正确性。

最典型的例子是TRAE(Sonnet)在优化Bamba-9B模型推理时的表现。该AI助手成功地将推理速度提升到与人类专家解决方案相当的水平,在传统评估中这会被认为是一个成功的优化。然而,功能正确性测试显示,优化后的模型准确率从32%暴跌至0%,完全失去了实用价值。

深入分析发现,AI助手为了提升速度,将动态维度保持(代码中的-1参数)替换为硬编码的数值。这种修改在特定测试条件下确实能提升性能,但当实际张量维度与硬编码值不匹配时,就会导致形状错误,进而破坏整个Mamba状态空间模型的计算过程。

这个发现强调了综合评估的重要性。仅仅关注性能指标而忽视功能正确性可能导致严重的问题。在实际应用中,一个稍慢但可靠的系统远比一个快速但不稳定的系统更有价值。

六、开源模型的挑战与局限

为了全面了解当前AI代码助手的能力边界,研究团队还测试了三个开源大语言模型:GPT-OSS-120B、MiniMax-M2.1和GLM-4.7。令人意外的是,这些开源模型都无法成功完成任何一个优化任务,但它们的失败模式却各不相同,为理解AI代码助手的工作机制提供了宝贵的洞察。

MiniMax-M2.1展现了一种独特的"理解-执行鸿沟"。该模型能够详细描述优化策略,甚至能够反复强调需要使用特定的工具来实现优化,但在整个75步的执行过程中,它从未实际调用过任何一个工具。这就像一个人能够完美地描述如何修理汽车,甚至能够指出需要使用哪些具体工具,但却从未真正拿起扳手开始工作。

GPT-OSS-120B则表现出了严重的环境理解问题。该模型误将外部依赖库(如PyTorch、Triton、Transformers)视为需要自己实现的代码,而不是可以直接调用的工具。它尝试在项目目录中重新创建这些复杂的机器学习框架,这就像试图从零开始制造螺丝刀而不是去工具店购买一样荒谬。

GLM-4.7的失败模式最为复杂。该模型成功地进行了多次代码编辑,甚至提交了有效的优化代码。然而,当它尝试验证修改效果时遇到了困惑的错误信息,导致它陷入了无限的调试循环,最终超过了时间限制而无法完成任务。这种情况就像一个学生完成了考试但不知道何时停笔交卷。

这些失败模式揭示了当前开源AI模型在复杂任务执行方面的系统性限制。它们不仅在技术能力上有差距,更重要的是在任务理解、工具使用和工作流程管理方面存在根本性缺陷。

七、实际应用的启示与思考

ISO-Bench的研究结果为AI代码助手的实际应用提供了重要的指导原则。首先,传统的性能基准测试可能会高估AI助手的真实能力。在选择和使用AI代码助手时,不能仅仅依赖厂商提供的性能数据,还需要考虑任务的具体特点和AI助手的适用场景。

研究揭示的理解-执行鸿沟现象具有普遍意义。许多AI助手都能够准确识别问题并提出合理的解决思路,但在将想法转化为可执行代码时经常出现错误。这提示我们在使用AI助手时,应该特别关注代码的实际运行效果,而不是仅仅相信AI的描述和承诺。

跨项目性能差异的发现表明,当前的AI代码助手还没有达到真正的通用性。在实际工作中,可能需要为不同类型的代码库选择不同的AI助手,或者至少需要针对具体项目对AI助手进行专门的调优和适配。

架构设计比底层模型更重要这一发现对AI工具的开发具有重要意义。它表明,简单地使用更强大的语言模型并不能自动带来更好的代码助手,工作流程、决策机制和任务分解策略等架构层面的设计同样关键。

功能正确性检查的重要性不容忽视。在追求性能优化的同时,必须确保代码的功能完整性不受损害。这要求在使用AI代码助手进行优化时,需要建立完善的测试和验证机制。

八、未来发展方向与改进空间

基于ISO-Bench的研究结果,研究团队指出了几个值得关注的改进方向。首先是提升AI助手的执行能力。当前的AI助手在理解问题方面已经相当出色,但在将理解转化为正确、高效的代码实现方面还有很大提升空间。

其次是增强跨项目的泛化能力。理想的AI代码助手应该能够快速适应不同的代码库和项目结构,而不是仅在特定类型的项目中表现优异。这可能需要更好的代码库分析能力和更灵活的策略选择机制。

第三是完善综合评估体系。ISO-Bench证明了硬指标和软指标结合的评估方法的价值,未来的AI代码助手评估应该更加全面,既要关注性能也要关注正确性,既要看结果也要看过程。

研究团队也承认了ISO-Bench当前的一些局限性。首先,测试任务主要集中在相对局部的优化上,对于需要跨多个模块协调的大型系统级优化还缺乏覆盖。其次,所有测试都基于公开的代码仓库,存在AI模型在训练阶段可能已经"见过"相关代码的风险。最后,软指标的评估目前依赖单一的LLM判断,缺乏人类专家的验证。

尽管存在这些局限,ISO-Bench为AI代码助手的评估和改进提供了一个重要的起点。随着更多研究者和开发者的参与,这个基准测试有望不断完善,为构建更强大、更可靠的AI代码助手提供指导。

说到底,ISO-Bench的研究揭示了当前AI代码助手在面对真实世界复杂优化任务时的真实表现。虽然这些工具在某些场景下已经展现出了令人印象深刻的能力,但距离完全自主地进行高质量代码优化还有相当长的路要走。理解与执行之间的鸿沟、跨项目性能的不一致性、以及功能正确性的潜在风险,都提醒我们在拥抱AI技术的同时保持必要的谨慎和理性。

对于普通开发者而言,这项研究的价值在于提供了一个更加清晰的认知框架。AI代码助手确实是强大的工具,但它们不是万能的。在使用这些工具时,我们需要根据具体的项目特点选择合适的助手,需要对AI的输出进行仔细的验证和测试,更需要将AI助手视为合作伙伴而非替代品。只有这样,我们才能在AI时代中最大化地发挥这些智能工具的价值,同时避免潜在的风险和陷阱。

Q&A

Q1:ISO-Bench是什么,它与其他代码评估工具有什么不同?

A:ISO-Bench是专门用于评估AI代码助手在GPU推理引擎优化方面能力的基准测试系统。与传统的代码评估工具不同,它不仅关注代码的运行性能(硬指标),还通过大语言模型评估AI是否真正找到了正确的性能瓶颈并采用了合理的优化策略(软指标)。这种双重评估方式能够识别出那些看似成功但实际上是"歪打正着"的优化方案,提供了更准确的能力评估。

Q2:为什么AI代码助手在不同项目上的表现差异如此巨大?

A:研究发现每个AI代码助手都有自己偏好的优化策略。Claude Code倾向于采用创新性的替代方法,这在复杂的vLLM系统中效果很好,但在结构相对简单的SGLang中反而成为劣势。而TRAE系列助手更倾向于复制人类专家的标准化方法,在SGLang中表现出色但在vLLM中受限。这就像不同的厨师有各自的烹饪风格,同样的风格在制作不同菜系时效果截然不同。

Q3:使用AI代码助手进行代码优化时需要注意哪些潜在风险?

A:主要有三个风险需要特别注意。首先是功能正确性风险,AI助手可能通过破坏代码功能来获得性能提升,就像研究中发现的Bamba-9B案例,速度提升了但准确率从32%降到0%。其次是"幸运获胜"的风险,AI可能修改了错误的代码部分却意外获得性能提升,这种优化通常不稳定且可能引入隐患。最后是理解执行鸿沟,AI能准确识别问题但实现时出错,导致代码无法正常运行。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
伊朗成立国家危机管理总部

伊朗成立国家危机管理总部

界面新闻
2026-02-28 16:59:35
新加坡大满贯赛:国乒3:0晋级决赛!夺冠一步之遥,决赛对阵出炉

新加坡大满贯赛:国乒3:0晋级决赛!夺冠一步之遥,决赛对阵出炉

国乒二三事
2026-02-28 09:03:47
WTT大满贯!女单4强对阵出炉,国乒2人出局,孙颖莎迎大挑战

WTT大满贯!女单4强对阵出炉,国乒2人出局,孙颖莎迎大挑战

娱说瑜悦
2026-02-28 14:04:12
离婚次日,婆婆命令我伺候她住院七天,我打了一个电话婆婆傻眼

离婚次日,婆婆命令我伺候她住院七天,我打了一个电话婆婆傻眼

奶茶麦子
2026-02-28 11:48:18
46岁谢霆锋与陈坤亮相,一个脸垮一个臃肿,骨相和皮相差距很明显

46岁谢霆锋与陈坤亮相,一个脸垮一个臃肿,骨相和皮相差距很明显

查尔菲的笔记
2026-02-26 14:16:04
国乒世界杯参赛名单:王楚钦、孙颖莎、王曼昱领衔,向鹏落选

国乒世界杯参赛名单:王楚钦、孙颖莎、王曼昱领衔,向鹏落选

澎湃新闻
2026-02-28 16:30:12
人的基因有多强大?网友:今年剪了个短发,我爸一看就害怕!

人的基因有多强大?网友:今年剪了个短发,我爸一看就害怕!

另子维爱读史
2026-02-27 21:05:30
古巴高调反美!美国为啥不打古巴?因为古巴是最“先进”的国家

古巴高调反美!美国为啥不打古巴?因为古巴是最“先进”的国家

南宫一二
2026-01-10 17:41:19
康熙临终前,给胤禎一杯毒酒考验他,胤禎的举动,让康熙心头一震

康熙临终前,给胤禎一杯毒酒考验他,胤禎的举动,让康熙心头一震

千秋文化
2026-02-08 19:38:41
速视频|伊朗最高领袖办公室附近遭袭,伊朗称正准备“毁灭性”报复行动

速视频|伊朗最高领袖办公室附近遭袭,伊朗称正准备“毁灭性”报复行动

上观新闻
2026-02-28 15:47:06
安徽小伙因神似溥仪做短视频 坦言“新的一年靠内容闯新路”

安徽小伙因神似溥仪做短视频 坦言“新的一年靠内容闯新路”

极目新闻
2026-02-27 18:57:51
可打航母!外媒:中国超音武器将运往伊朗,美150架战机急飞中东

可打航母!外媒:中国超音武器将运往伊朗,美150架战机急飞中东

黑鹰观军事
2026-02-25 17:48:10
77年李德生探望叶剑英,席间叶帅突然提点:你去看看你们的老政委

77年李德生探望叶剑英,席间叶帅突然提点:你去看看你们的老政委

谈古论今历史有道
2026-02-28 08:15:03
不可错过!2月28日晚19:30!中央5套CCTV5、CCTV5+直播节目表

不可错过!2月28日晚19:30!中央5套CCTV5、CCTV5+直播节目表

皮皮观天下
2026-02-28 16:51:09
女子购买胖东来花生油,发现可以凉拌后直呼惊讶

女子购买胖东来花生油,发现可以凉拌后直呼惊讶

映射生活的身影
2026-02-26 21:03:46
美军航母遇劲敌,伊朗1500艘小艇硬刚林肯号,美军陷两难困局

美军航母遇劲敌,伊朗1500艘小艇硬刚林肯号,美军陷两难困局

小小科普员
2026-02-28 16:32:24
拉什福德接受19万周薪,永久转会巴萨已定!曼联不降价但同意分期

拉什福德接受19万周薪,永久转会巴萨已定!曼联不降价但同意分期

罗米的曼联博客
2026-02-28 10:42:01
祝福勇敢的阿尔忒弥斯二号任务乘组

祝福勇敢的阿尔忒弥斯二号任务乘组

牧夫天文
2026-02-28 11:34:35
美以对伊发动联合军事行动 画面显示德黑兰市中心浓烟滚滚

美以对伊发动联合军事行动 画面显示德黑兰市中心浓烟滚滚

环球网资讯
2026-02-28 15:46:15
洛克希德·马丁的生产线突然停了,中国冻结了他们在华的全部资产

洛克希德·马丁的生产线突然停了,中国冻结了他们在华的全部资产

百态人间
2026-02-25 15:33:21
2026-02-28 17:36:49
至顶AI实验室 incentive-icons
至顶AI实验室
一个专注于探索生成式AI前沿技术及其应用的实验室。
1958文章数 162关注度
往期回顾 全部

科技要闻

狂揽1100亿美元!OpenAI再创融资神话

头条要闻

内塔尼亚胡:美以军事行动目标是推翻伊朗政权

头条要闻

内塔尼亚胡:美以军事行动目标是推翻伊朗政权

体育要闻

球队主力全报销?顶风摆烂演都不演了

娱乐要闻

疑似王一博被爆私密聊天记录

财经要闻

沈明高提共富建议 百姓持科技股国家兜底

汽车要闻

岚图泰山黑武士版3月上市 搭载华为四激光智驾方案

态度原创

时尚
本地
旅游
房产
手机

被章若楠、秦岚带火的鞋子竟然是它?春天这样穿又美又气质!

本地新闻

津南好·四时总相宜

旅游要闻

2026年柳州龙王出游,即将登场!时间、路线定了!重要提醒→

房产要闻

滨江九小也来了!集齐海侨北+哈罗、寰岛...江东教育要炸了!

手机要闻

存储芯片涨价潮席卷手机业:头部品牌全线调价在即 中小厂商陷入生存困境

无障碍浏览 进入关怀版