SWE-Fixer Dataset
收藏资源简介:
SWE-Fixer数据集由上海人工智能实验室等机构创建,旨在解决GitHub上的软件工程问题。该数据集包含11万条GitHub问题及其对应的补丁,数据来源于真实的GitHub仓库。数据集的构建过程采用了严格的过滤技术,确保数据的广泛性和可靠性。数据集主要用于训练和评估开源大语言模型在代码文件检索和代码编辑任务中的表现,旨在提高模型在解决实际软件工程问题中的效率和效果。
The SWE-Fixer dataset was developed by institutions including the Shanghai AI Laboratory and others, with the objective of addressing software engineering problems on GitHub. It contains 110,000 GitHub issues and their corresponding patches, with all data sourced from real GitHub repositories. Strict filtering techniques were employed during the dataset construction process to guarantee the breadth and reliability of the collected data. This dataset is primarily utilized for training and evaluating the performance of open-source large language models (LLMs) on code file retrieval and code editing tasks, aiming to enhance the efficiency and effectiveness of models when resolving practical software engineering issues.
SWE-Fixer 数据集概述
数据集简介
SWE-Fixer 是一个用于解决真实世界 GitHub 问题的开源大语言模型(LLM)训练解决方案。它采用了一个简化的检索-编辑流程,包含两个核心组件:代码文件检索器和代码编辑器。该模型通过微调 Qwen2.5-7b 和 Qwen2.5-72b 来实现,并利用了一个包含 110K 实例的精选数据集。
模型与数据集
模型
-
SWE-Fixer-Retriever-7B
- 用途:用于代码文件检索任务。
- 输入:问题描述和 BM25 检索结果。
- 输出:识别与问题相关的缺陷文件。
-
SWE-Fixer-Editor-72B
- 用途:用于代码编辑任务。
- 输入:问题描述和相应的文件内容。
- 输出:生成用于解决问题的修改补丁。
数据集
-
SWE-Fixer-Train-110K
- 内容:包含从真实世界 GitHub 仓库收集的近 110K 详细实例,用于模型训练。
-
SWE-Fixer-Eval
- 内容:包含 SWE-Bench Lite 和 Verified 实例、BM25 检索结果以及每个实例的代码结构,便于评估。
性能表现
SWE-Fixer 在开源解决方案中达到了最先进的性能,具体表现为:
- 23.3% 在 SWE-Bench Lite 上。
- 30.2% 在 SWE-Bench Verified 上。
运行流程
环境设置
- 下载并安装推理环境包 SWE_Fixer.tar.gz。
- 解压并激活环境。
准备模型和评估数据集
- 下载模型和数据集并保存到默认位置。
- 可选:修改脚本中的路径以匹配本地设置。
运行检索模型
- 运行检索管道(默认使用
lite数据集)。 - 可选:使用
verified数据集。
运行编辑模型
- 在完成检索步骤后,基于检索结果运行编辑管道。
- 可选:使用
verified数据集。
评估结果
使用 all-hands 评估方法 评估管道结果。
引用
bibtex @article{xie2025swefixer, title={SWE-Fixer: Training Open-Source LLMs for Effective and Efficient GitHub Issue Resolution}, author={Xie, Chengxing and Li, Bowen and Gao, Chang and Du, He and Lam, Wai and Zou, Difan and Chen, Kai}, journal={arXiv preprint arXiv:2501.05040}, year={2025} }
致谢

- 1SWE-Fixer: Training Open-Source LLMs for Effective and Efficient GitHub Issue Resolution上海人工智能实验室, 西安电子科技大学, 香港中文大学, 香港大学 · 2025年



