five

EquiBench

收藏
arXiv2025-02-18 更新2025-02-27 收录
下载链接:
http://arxiv.org/abs/2502.12466v1
下载链接
链接失效反馈
官方服务:
资源简介:
EquiBench是一个包含2400对程序的数据集,这些程序对涵盖了四种编程语言(Python、C、CUDA和x86-64汇编)和六个等价类别。该数据集通过程序分析、编译器调度和超优化技术自动生成,包含了要求深度语义推理的复杂结构转换。数据集旨在评估大型语言模型在代码推理方面的能力,特别是在等价性检查任务上。

EquiBench is a dataset comprising 2400 program pairs covering four programming languages (Python, C, CUDA, and x86-64 assembly) and six equivalence categories. Automatically generated via program analysis, compiler scheduling, and superoptimization techniques, this dataset contains complex structural transformations that require deep semantic reasoning. It is designed to evaluate the code reasoning capabilities of large language models, particularly in equivalence checking tasks.
提供机构:
斯坦福大学, MIT, 谷歌, 南京大学, DeepSeek, 英特尔, Visa Research
创建时间:
2025-02-18
搜集汇总
数据集介绍
main_image_url
构建方式
EquiBench数据集的构建采用了程序分析、编译器调度和超级优化等技术,旨在生成2400对程序,涵盖四种编程语言和六种等价类别。这些程序对是通过自动化的方式生成的,包括语法变化、结构修改和算法等价。例如,通过编译器的死代码消除(DCE)过程来生成C语言程序对,通过不同的调度策略生成CUDA程序对,以及使用超级优化器生成x86-64汇编程序对。此外,还从在线编程平台收集了具有算法差异的Python程序对。
特点
EquiBench数据集的特点在于其多样性、挑战性和自动生成性。它涵盖了四种编程语言(Python、C、CUDA和x86-64汇编)和六种等价类别,包括DCE、CUDA、x86-64、OJ_A、OJ_V和OJ_VA。这些程序对不仅包括简单的语法变化,还包括深层次的语义推理,如结构修改和算法等价。数据集的生成过程完全自动化,可以生成更多对来满足需求。
使用方法
使用EquiBench数据集的方法包括准备程序对、定义等价性、评估模型和结果分析。首先,需要准备程序对,包括等价程序对和不等价程序对。然后,为每个类别定义等价性。接下来,评估大型语言模型(LLM)在等价性检查任务上的表现。最后,分析不同等价类别和提示策略下的模型性能,以找出LLM的代码推理能力。
背景与挑战
背景概述
在人工智能领域,大型语言模型(LLM)在代码推理能力的评估方面一直是一个重要课题。EquiBench数据集正是为了评估LLM在代码推理方面的能力而创建的。该数据集由斯坦福大学、麻省理工学院、谷歌等机构的研究人员合作开发,旨在通过等价性检查来评估LLM的代码推理能力。EquiBench数据集包含了2400对程序,涵盖了Python、C、CUDA和x86-64汇编语言,以及六个等价性类别。这些程序对是通过程序分析、编译器调度和超优化等技术生成的,涵盖了非平凡的程序结构转换,这些转换需要深度语义推理,而不仅仅是简单的语法变化。EquiBench数据集的发布为LLM代码推理能力的评估提供了新的基准,对相关领域的研究产生了深远的影响。
当前挑战
EquiBench数据集的创建和评估面临诸多挑战。首先,等价性检查本身是一个具有挑战性的任务,需要LLM对所有可能的输入进行推理,而不仅仅是单个输入。其次,构建一个既包含等价程序对也包含不等价程序对的大型数据集需要自动化生成过程,而现有的方法大多依赖于局部的语法变化,这不足以评估LLM的等价推理能力。此外,EquiBench数据集的评估结果表明,即使是表现最好的LLM模型,在等价性检查任务上的准确率也仅为78.0%,在最具挑战性的类别中,最佳准确率仅为62.3%和68.8%,这表明当前的模型在代码推理能力方面仍有很大的提升空间。最后,LLM模型在预测程序等价性时存在偏差,倾向于将具有显著结构变化的程序对分类为不等价。这些挑战表明,LLM代码推理能力的提升需要更深入的语义理解和更复杂的推理策略。
常用场景
经典使用场景
EquiBench数据集被广泛应用于评估大型语言模型(LLMs)的代码推理能力。该数据集包含2400个程序对,跨越四种编程语言和六个等价性类别,旨在测试LLMs是否能够判断两个程序对于所有可能的输入是否产生相同的输出。EquiBench数据集的挑战性在于其要求LLMs进行深入的语义推理,而不仅仅是简单的语法变化。
衍生相关工作
EquiBench数据集衍生了一系列相关工作,包括基于LLMs的代码等价性检查工具和优化算法。这些工作利用EquiBench数据集来训练和评估LLMs,从而提高LLMs在代码推理和优化方面的能力。此外,EquiBench数据集还为开发更强大的代码分析工具和优化算法提供了新的思路和方法。
数据集最近研究
最新研究方向
在编程语言和验证领域,等价性检查是确定两个程序是否对所有可能的输入产生相同输出的关键任务。EquiBench数据集的提出为评估大型语言模型(LLMs)的代码推理能力提供了一个新的视角。该数据集包含2400个程序对,涵盖四种编程语言和六种等价性类别,这些程序对通过程序分析、编译器调度和超优化技术系统生成,需要深度语义推理。研究评估了17个最先进的LLMs,发现OpenAI o3-mini在所有模型中整体准确率最高,达到78.0%,但在最具挑战性的类别中,最佳准确率仅为62.3%和68.8%,略高于50%的随机基线。这表明当前模型的代码推理能力仍有很大的提升空间。EquiBench数据集的引入,为LLMs在代码推理方面的研究和改进提供了一个重要的基准。
相关研究论文
  • 1
    EquiBench: Benchmarking Code Reasoning Capabilities of Large Language Models via Equivalence Checking斯坦福大学, MIT, 谷歌, 南京大学, DeepSeek, 英特尔, Visa Research · 2025年
以上内容由遇见数据集搜集并总结生成
5,000+
优质数据集
54 个
任务类型
进入经典数据集
二维码
社区交流群

面向社区/商业的数据集话题

二维码
科研交流群

面向高校/科研机构的开源数据集话题

数据驱动未来

携手共赢发展

商业合作