← 返回 2026-05-28

ResearchMath-14K:通过智能体扩展研究级数学数据 ResearchMath-14K: Scaling Research-Level Mathematics via Agents

Guijin Son, Seungyeop Yi, Minju Gwak, Hyunwoo Ko, Wongi Jang, Youngjae Yu 📅 2026-05-27 👍 50 2026-07-13 08:36
LLM推理行为 数学数据集 智能体系统 研究级数学

构建最大规模研究级数学问题数据集,分析LLM推理行为

前置知识

研究级数学问题

指尚未解决或部分解决的数学难题,通常以公开问题、猜想或研究问题的形式出现在学术论文、研讨会和数学文献中。这类问题不同于竞赛数学题,没有已知的标准答案,可能需要数年甚至数十年的研究才能解决。例如千禧年大奖问题、BSD猜想等。

本文的核心贡献就是构建一个大规模的研究级数学问题数据集,理解这个概念是理解整个工作的前提

智能体管道

一种通过多个AI智能体协作完成复杂任务的自动化流程。在本论文中,智能体管道由Extractor和Refiner两个智能体组成:Extractor负责从原始文档中提取候选问题,Refiner负责补充缺失的定义和背景信息,将问题转化为自包含形式。每个智能体通过特定的prompt和工具完成其职责。

本文采用双智能体管道构建数据集,理解这个概念有助于理解数据集的构建过程和质量保证

Elo评分系统

最初用于国际象棋棋手实力评分的方法,在本文中被用于评估数学问题的难度。通过让LLM判断两个问题中哪一个更难,构建两两比较的胜负关系,然后使用Elo算法计算每个问题的难度评分。初始分为1500分,获胜加分,失败扣分。这种方法可以捕捉问题的多维难度特征。

本文使用Elo评分证明RESEARCHMATH-14K比现有数据集难约400分,这是数据集质量的重要证据

自包含问题

指数学问题陈述包含理解该问题所需的所有定义、符号、假设和背景信息,读者不需要查阅源文档或其他外部材料就能理解问题的完整含义。将嵌入在论文上下文中的问题转化为自包含形式需要智能体主动搜索和补充相关定义。

这是数据集质量的关键指标,论文报告自包含率从67.2%提升到94.2%,显示Refiner智能体的效果

引证行为分析

通过规则计数器和智能体审查两种方法分析LLM在推理过程中的引用行为。规则计数器统计包含特定关键词(如paper、article)的文本片段;智能体审查使用搜索工具验证模型引用的论文、书籍等参考文献是否真实存在。这种方法可以区分真实引用和幻觉引用。

本文的核心发现是新模型在研究级数学问题上生成更多虚假引用,这是理解当前LLM局限性的关键

研究动机

当前数学推理LLM训练面临严重的资源瓶颈。公开可用的数学训练数据大多集中在竞赛级别或更低难度,如GSM8K(8.5k题)、MATH(12.5k题)、NuminaMath(860k题)和OpenMathInstruct(1.8M题),这些问题的答案已知且可验证。而真正达到研究前沿的数学数据集非常稀缺,通常规模较小(小于3k项)、由专家手工编写、私有或受限访问以防止污染,如FrontierMath(25题)、Riemann-Bench(300题)、Soohak(439题)。这种数据稀缺性使得LLM难以学习研究级数学推理所需的深层技能,如将问题分解为引理、测试反例、在真正不确定性下推理等。

本文的目标是本文的目标是构建一个大规模、高质量的研究级数学问题数据集,解决现有数据集在规模和研究深度之间的矛盾。具体目标包括:从数学文献中自动提取14,000+个研究级问题,生成对应的推理轨迹,分析LLM在这些问影题上的行为特征,并验证这些数据作为训练信号的有效性。最终目标是推动研究级数学推理能力的发展,使LLM能够真正参与前沿数学研究。

与已有工作不同的是,本文的独特切入角度是利用数学文献中已有的丰富开放问题资源,而不是从头合成问题或依赖专家手工编写。关键创新在于通过智能体管道自动将嵌入在论文上下文中的问题转化为自包含的训练数据,这打破了研究级数据可扩展性的瓶颈。与仅关注正确答案的传统方法不同,本文探索了错误但合理的推理轨迹作为训练信号的可能性,这为研究级训练提供了新的思路。

核心方法

本文采用双智能体管道从1,233个源文档中自动构建RESEARCHMATH-14K数据集。整体思路是:首先从zbMATH、arXiv和学术仓库收集开放问题列表和研究论文,然后使用Extractor智能体识别候选开放问题并提取原始引用,接着使用Refiner智能体补充缺失的定义和假设,将问题转化为自包含形式,最后进行去重和人工审核。这个过程产生了14,056个研究级数学问题,随后使用两个开源模型生成220,000个推理轨迹。整个方法的核心在于如何自动地从技术文档中提取和转换问题,使其成为可用的训练数据。

核心创新点是双智能体协同构建数据集的管道,特别是将嵌入在论文上下文中的问题转化为自包含问题的能力。Extractor智能体使用Codex配合GPT-5.5进行高推理力度的提取,能够识别问题并跳转到论文的其他部分拉取相关定义;Refiner智能体使用Claude Code配合Opus 4.7进行中等推理力度的精炼,不仅重读原始论文内联所需定义,还搜索最多10篇后续论文以检查问题是否已被解决。这种自动化的提取-精炼流程使得从数学文献中大规模获取问题成为可能,这是与依赖专家手工编写方法的关键区别。

方法步骤详情

数据集构建包含五个主要步骤。第一步是源文档收集,从三个流收集1,233个文档:arXiv开放问题论文(524文档,8,182问题)通过搜索标题和摘要中的open problems或unsolved关键词;开放问题网页(161文档,5,331问题)通过Google搜索发现,覆盖academia.edu、MathOverflow和Wikipedia等;问题会话单页和策展列表(548文档,7,322问题)包括AIM式研讨会问题会话和会议开放问题轮次。第二步是Extractor智能体处理,每个源文档运行一次,首先确认文档确实包含问题列表,然后端到端读取论文,提取每个开放问题的逐字引用和一级重写,平均每篇文档提取16.9个问题(中位数10,最大358)。第三步是Refiner智能体精炼,重读原始论文内联每个问题所需的定义和假设,搜索最多10篇引用或扩展该源文档的后续论文,将问题标记为开放、部分解决、已解决或未知,最终产生平均1,192字符的自包含陈述,比Extractor阶段增加4.1倍。第四步是去重过滤,使用Qwen3-Embedding-8B对所有问题进行嵌入,计算原始陈述和自包含重写的成对相似度,设置0.9的相似度阈值,保留arXiv或论文源的版本,丢弃Google搜索发现的版本,最终得到14,056个问题。第五步是难度验证,与AceMath、AIME、HLE-Verified和NuminaMath进行比较,从每个数据集采样90个问题,考虑所有10个数据集对的1,000个随机交叉问题对,使用GPT-5-mini沿知识性、新颖性和程序性三个轴进行两两比较,计算Elo评分。

技术新颖性

技术新颖性体现在多个方面。首先是规模与质量的突破:RESEARCHMATH-14K是最大的研究级数学问题集合,比FrontierMath(25题)大562倍,比Soohak(439题)大32倍,同时保持高自包含率(94.2%)。其次是自动化提取的创新:首次实现从数学文献中大规模提取和转换问题的自动化流程,避免了人工编写的高昂成本。第三是训练信号的新视角:探索使用错误但合理的推理轨迹作为训练数据,打破了必须完全正确的传统观念。第四是行为分析的新方法:结合规则计数器和智能体审查两种方法系统分析LLM的引用行为和真实性,发现了新模型倾向于生成更多虚假引用的令人担忧的趋势。最后是多领域覆盖:数据集涵盖11个一级领域(分析、数学物理、离散数学、几何拓扑、数论、代数与表示理论、应用与计算数学、概率统计与机器学习、逻辑基础、计算机科学、跨学科)和28个宏观主题,覆盖8,313个开放问题(59.14%)、2,489个未知状态问题(17.71%)、2,083个部分解决问题(14.82%)和1,171个已解决问题(8.33%)。

Agentic construction pipeline for RESEARCHMATH-14K
Figure 2: Agentic construction pipeline for RESEARCHMATH-14K

实验结果

本文报告了四个主要实验结果。第一,数据集质量评估显示Refiner智能体将自包含问题比例从67.2%提升到94.2%,问题长度从290字符增加到1,192字符(4.1倍扩展)。Elo难度评估显示RESEARCHMATH-14K在所有三个轴(知识性、新颖性、程序性)上比现有数学数据集高约400 Elo点,证明这是最难的开源数学问题集。第二,教师模型轨迹分析显示100个随机采样轨迹中约30%存在明显问题:21个不尝试问题仅列出已知相关引用并输出open,4个在判定问题开放后缩小条件求解狭窄版本或仅列出相关引用,25个模型完全放弃。第三,在8个开放权重模型(DeepSeek R1/V4-Pro、Kimi K2/K2.6、Qwen3 30B/235B和Qwen3.5 35B/397B)上的大规模行为分析发现,新模型产生的引用类提及平均增加5.6倍,虚假引用增加5.0倍。在720个RESEARCHMATH-14K轨迹中,629个(87.4%)包含至少一个引用类对象,389个(54.0%)包含至少一个虚假引用。例如DeepSeek R1到V4-Pro的每轨迹引用数从4.9增至57.8,虚假引用从0.5增至11.6。第四,微调实验显示在RESEARCHMATH-REASONING-FILTERED(5,000条过滤后轨迹)上微调Qwen3-4B/8B/30B-A3B-base模型,在9个模型乘基准设置中平均提升9.2个百分点,相比DASD-Thinking数据集在8/9个设置上表现更好,在HLE和SOOHAK上平均比DASD高2.6个百分点。最有意义的是,这些轨迹中的大多数问题模型并未真正解决,但过滤后的尝试仍然提供了有效的监督信号。

Representative public math datasets
Table 1: Representative public math datasets
LLM-judge lemma-decomposition positives by model and benchmark
Table 2: LLM-judge lemma-decomposition positives by model and benchmark
Domain distribution of the RESEARCHMATH-14K
Figure 1: Domain distribution of the RESEARCHMATH-14K
Elo Ratings for Difficulty Comparison
Figure 3: Elo Ratings for Difficulty Comparison
Fine-tuning results by benchmark
Figure 5: Fine-tuning results by benchmark
查看结构化数据
任务指标本文基线提升
研究级数学推理 平均准确率提升 +9.2pp (RESEARCHMATH-REASONING-FILTERED) +5.6pp (DASD-Thinking) 相对提升64%
HLE-Verified (n=315) Qwen3-30B-A3B准确率 11.9% 7.9% +4.0pp
SOOHAK (n=501) Qwen3-30B-A3B准确率 24.6% 25.0% -0.4pp (略低)
AIME (n=90) Qwen3-30B-A3B准确率 13.3% 24.4% -11.1pp (DASD更优)
数据集规模 问题数量 14,056 (RESEARCHMATH-14K) 439 (Soohak) 32倍
虚假引用率 包含虚假引用的轨迹比例 54.0% (720个RESEARCHMATH-14K轨迹) 未报告(基线无此分析) 首次系统性报告

局限与改进

作者承认的局限性包括:过滤过程是保守的,可能删除了一些不同但密切相关的问题,尽管作者也发布了原始种子集以减轻这一影响;自包含审计是自动化的,可能遗漏一些细微的上下文依赖;引用真实性检查仅验证引用对象是否存在,不判断引用是否支持模型的主张;微调实验的规模有限(5,000条轨迹),未来需要在大规模上验证该信号。自己观察到的局限性包括:数据集的去重阈值0.9可能仍然引入一些假阳性或假阴性,尽管作者通过人工检查边界情况进行了验证;模型行为分析仅在8个开放权重模型上进行,可能不适用于所有LLM;Elo评分使用GPT-5-mini作为评委,可能存在模型偏见;研究级问题的状态标记(开放/部分解决/已解决/未知)依赖于在线搜索,可能存在滞后或不准确;生成的推理轨迹质量高度依赖于教师模型(GPT-OSS-120B和Qwen3-30B-A3B)的能力,如果教师模型本身有局限,学生模型也会继承这些局限。

独立分析的弱点

本文存在几个可以改进的弱点。第一,问题状态检查不够及时可靠,Refiner智能体搜索后续论文来检查问题是否已被解决,但最新进展可能尚未发表或未被索引,导致状态标记滞后。改进方向是建立动态更新机制,定期重新检查问题状态。第二,去重过滤可能过于激进,0.9的相似度阈值虽然经过人工验证,但仍可能删除一些不同但表述相似的变体问题。改进方向是分层去重,保留不同难度级别或表述变体的问题。第三,教师轨迹过滤依赖昂贵的智能体审查(包含多次网络搜索API调用),限制了过滤规模。改进方向是开发更高效的自动过滤器,减少对付费API的依赖。第四,微调实验仅在Qwen3系列模型上验证,缺乏跨模型族的泛化性验证。改进方向是在更多模型族(如DeepSeek、Kimi)上验证训练信号的有效性。第五,行为分析关注引用真实性,但未深入分析推理质量的维度。改进方向是开发更细粒度的推理质量评估指标,如论证逻辑性、步骤完整性、反例测试等。

未来方向

作者提出的未来工作方向包括:在更大规模上测试过滤后推理轨迹的效用,当前5,000条轨迹的规模相对较小;探索何时需要完全正确的轨迹来实现可靠的证明行为,因为目前还不清楚在哪些任务中部分正确的轨迹足够,在哪些任务中必须完全正确。基于本文成果可延伸的方向包括:将RESEARCHMATH-14K与形式证明环境(如Lean)结合,构建可验证的推理数据集;开发更智能的问题难度预测和自适应训练机制;研究如何利用错误但合理的轨迹进行强化学习,奖励合理的推理路径而非仅奖励正确答案;探索跨领域泛化,测试在数学研究级数据上训练的模型是否能在其他科学领域的研究级推理任务上受益;构建动态更新的研究级数学问题数据库,跟踪问题解决进展;开发更精细的模型行为分析工具,不仅检测虚假引用,还检测推理谬误、逻辑漏洞等。另一个有前景的方向是研究如何让模型真正学会研究级数学推理的关键技能,如问题分解、引理构建、反例搜索、不确定性下的推理等,而不是仅仅模仿研究数学的表面形式。

复现评估

复现评估:RESEARCHMATH-14K和RESEARCHMATH-REASONING已在Hugging Face公开发布(MIT许可证),包含完整的问题陈述、领域元数据、来源和解决方案字段(如果适用)。Extractor和Refiner智能体的详细prompt在附录H中提供。去重过程使用的模型(Qwen3-Embedding-8B)和阈值(0.9)明确说明。行为分析的规则计数器关键词列表在附录C.1中完整列出。微调实验使用LoRA在Qwen3-4B/8B/30B-A3B-base模型上进行,训练配置在附录G中描述。评估在AIME 2024-2026(n=90)、HLE-Verified(n=315)和SOOHAK(n=501)上进行,使用math-verify进行评分。主要缺失的信息包括:Extractor使用的Codex版本未明确说明,Refiner使用的Claude Code和Opus 4.7的具体配置未提供,网络搜索API的具体服务未说明,训练超参数(如学习率、批量大小、训练轮数)在正文中未报告。总体而言,复现难度中等,主要挑战在于获取与论文相同的闭源模型(Codex、GPT-5.5、Claude Code Opus 4.7、GPT-OSS-120B)和付费API(网络搜索),但数据集和开源部分的可复现性良好。