遇见数据集

feynman-bench

收藏
github2026-05-26 更新2026-05-27 收录
官方服务:

资源简介:

该数据集包含56,102个经过处理的IBP(积分约简)约简记录,涵盖22个多圈Feynman拓扑(维度从2D到15D),由FIRE 7在PRIME模式下使用固定素数(2017)和每个拓扑的固定运动学生成。它用于训练和评估积分-部分求解器,包括机器学习驱动的种子函数,以选择哪些积分被输入IBP关系。数据集分为训练集(13个拓扑,56,102条记录)和测试集(9个拓扑,913条记录),所有记录使用固定的素数和每个拓扑的固定运动学点。

This dataset contains 56,102 processed IBP (Integration-by-Parts) reduction records, covering 22 multi-loop Feynman topologies with spacetime dimensions ranging from 2D to 15D. It was generated by FIRE 7 in PRIME mode using a fixed prime (2017) and fixed kinematic points for each topology. This dataset is intended for training and evaluating IBP solvers, including machine learning-driven seed functions that select which integrals are fed into IBP relations. The dataset is split into a training set (13 topologies, 56,102 records) and a test set (9 topologies, 913 records). All records use the fixed prime and fixed kinematic points for each topology.

创建时间:
2026-05-26
原始信息汇总

Feynman IBP Benchmark 数据集概述

基本信息

该数据集包含 56,102 个已完成的 IBP(Integration-By-Parts)约化结果,覆盖 22 个多圈费曼拓扑结构(维度范围 2D – 15D),由 FIRE 7 在 PRIME 模式下以固定素数 2017 和每个拓扑固定运动学点生成。数据集旨在用于训练和评估 IBP 求解器,特别是机器学习驱动的种子函数(seeding functions)。

设计目的

IBP 约化是高阶费曼计算中的速率限制步骤:每个多圈积分必须通过 IBP 恒等式重写为少量"主积分"的线性组合。现代求解器通过将种子积分代入 IBP 关系并求解来构建线性系统,而种子的选择决定了系统规模和计算成本。该数据集提供了 56k 个 (目标积分,拓扑)→ 主积分约化 配对,并记录了 FIRE 的 num_steps(实际消耗的 IBP 方程数量),用于训练种子函数并评估其性能。

数据集结构

训练/测试划分

  • 训练集ground_truth_train.jsonl):13 个拓扑,56,102 条记录
  • 测试集ground_truth_test.jsonl):9 个保留拓扑,913 条记录,均为非空约化结果

训练拓扑(13个)

  • 2D/bub, 3D/bub2l, 4D/box1l, 5D/{bl2, bl2em}, 6D/{vac3lBN, vac3lNO}, 7D/tri2l, 9D/{banana3L, grav2l, grav2lx, p3lBenz, p3lLA}

测试拓扑(9个)

  • 4D/box1lc, 5D/ver2l, 6D/vac3lO4, 7D/tri2lx, 9D/p3lO4, 10D/{vac4lBN, vac4lNP}, 15D/{gravity3l, gravity3lsec}

记录格式(JSONL,每行一个 JSON 对象)

json { "solver": "fire", "topology_path": "5D/ver2l", "params": {"d": 7646, "m1sq": 962, "s1": 3547}, "integrals": [[1, 0, 2, 0, 2]], "reductions": {"[1, 0, 2, 0, 2]": {"[1, 0, 1, 0, 1]": 1478, "[1, 0, 1, 0, 2]": 1357}}, "num_steps": 111, "finite_field": 2017 }

每条记录包含一个目标积分及其在 FIRE 选定的主积分基下的约化结果。num_steps 表示 FIRE 日志中记录的实际使用的方程数量

关键特性

  • 固定素数:所有记录使用 FIRE 默认 PRIME 模式域 finite_field = 2017,约化系数为 [0, 2017) 范围内的整数
  • 固定运动学:每个拓扑仅使用一个 params 字典(例如 {d, m1sq, qsq}),以 --seed 42[2, 9999] 范围内采样得到,训练集和测试集共用相同运动学点
  • 预生成 FIRE 配置文件:FIRE 的启动文件(.start.lbases.sbases)已预生成并提交,无需 Mathematica 或 LiteRed

评估指标

指标 说明
Validity(有效性) 求解器的约化结果与真实值在模运算下精确匹配的积分比例(按拓扑平均)
Step ratio(步数比) Σ steps_solver / Σ steps_GT,仅计算求解器覆盖的积分,数值越低越好
n_missing(缺失数) 求解器未覆盖的积分数量,单独报告以防不完整覆盖受到惩罚

排行榜

当前基线:

  • FIRE 7p baseline:有效性 100.00%,步数比 1.000(作为参考基线)
  • FIRE 6p:有效性 100.00%,步数比 1.068

可通过 leaderboard/submit.py 提交结果,包含 score.json(指标+出处)和 predictions.jsonl(原始输出)。

使用方法

快速开始

bash git clone https://github.com/project-numina/feynman-bench cd feynman-bench pip install -e . docker pull ghcr.io/project-numina/fire6:latest cp config.example.yaml config.yaml ./run_eval.py --topology 5D/ver2l

评估运行选项

  • ./run_eval.py:运行全部 9 个测试拓扑
  • --topology 5D/ver2l:指定单个拓扑
  • --topologies 5D/ver2l,9D/p3lO4:指定多个拓扑
  • --ground-truth ground_truth_train.jsonl:在训练集上评分
  • --max-parallel 32 --threads 4:调优并行度

评分与提交

bash python3 score.py --predictions my_results.jsonl --ground-truth ground_truth_test.jsonl --solver my_solver --output my_score.json python3 leaderboard/submit.py --score my_score.json --name "My solver v0.1" --solver my_solver

仓库结构

run_eval.py 评估入口与评分 score.py 有效性与步数比计算 check_validity.py 隔离的比较原语 reductions.py 共享辅助函数 generate_ground_truth.py 生成真实值数据 solvers/fire/ FIRE 包装器 docker/fire6/ Docker 构建文件 topologies/<dim>/<name>/ 各拓扑的 FIRE 配置文件 leaderboard/ 排行榜提交与渲染

引用

bibtex @dataset{feynman-bench-2026, author = {Thibaut Barroyer and Shovon Biswas and Yann Fleureau and Jia Li and Julio Parra-Martinez and Mathis Reymond and Marina Vinyes}, title = {Feynman IBP benchmark}, year = {2026}, publisher = {GitHub}, url = {https://github.com/project-numina/feynman-bench} }

许可证

MIT 许可证(详见 LICENSE

搜集汇总
数据集介绍
feynman-bench 数据集图片
构建方式
在量子场论的高阶圈图计算中,积分-by-分部(IBP)约化是决定计算效率的关键瓶颈。为系统评估和优化IBP求解策略,Feynman-bench数据集应运而生。该数据集基于FIRE 7求解器在PRIME模式下,选取固定素数(2017)作为有限域,并为每个拓扑固定一组运动学点,生成了涵盖22种多圈费曼拓扑结构(维度从2维到15维)的共计56,102条IBP约化记录。每条记录包含一个目标积分、其约化至主积分的线性组合系数、以及FIRE求解过程中实际消耗的方程步数。数据集按拓扑划分为训练集(13种拓扑,56,102条记录)与测试集(9种外推拓扑,913条记录),确保训练的泛化性评估。
特点
该数据集具备多重鲜明特色。其一,它首次将机器学习与IBP求解器结合,提供了可直接用于学习播种函数(seeding function)的标注数据,目标在于通过智能选择输入IBP关系的种子积分来压缩系统规模。其二,数据记录在固定素数域和固定运动学点下生成,确保了约化系数的精确性和可复现性,同时规避了浮点误差。其三,每条记录附带了FIRE求解器实际使用的方程步数,为评估求解效率提供了基准。最后,数据集预先生成了FIRE配置文件和拓扑描述文件,使得研究人员无需依赖Mathematica或LiteRed等重型符号计算软件即可直接使用。
使用方法
使用该数据集开展研究时,用户首先通过pip安装Python包并配置FIRE求解器环境(支持Docker容器化运行或本地二进制文件)。随后,可利用提供的评估脚本run_eval.py在测试集上评测自定义播种函数,系统会报告有效性(Validity)和步数比(Step ratio)两项核心指标,其中步数比越低表示计算效率越高。用户亦可基于训练集数据训练机器学习模型,将预测的种子积分以.hint文件形式输入FIRE,再通过评估脚本验证加速效果。此外,数据集提供了完整的提交与排行榜机制,研究人员可将其结果提交至GitHub仓库的leaderboard,与FIRE 7基准及其他方案进行公开比较。
背景与挑战
背景概述
在粒子物理学的微扰量子色动力学与电弱理论计算中,多圈费曼积分的约化是迈向高精度理论预言的核心瓶颈。该过程依赖积分-分部关系将任意多圈积分表示为少量主积分的线性组合,而种子积分的选取直接影响线性系统的规模与求解时间。2026年,由Project Numina团队与法国高等科学研究所的Julio Parra-Martinez联合创建的feynman-bench数据集应运而生,它系统性地提供了涵盖22种拓扑结构、跨度从2维到15维的56102个已约化积分对,每个积分均采用FIRE 7在固定素数域与运动学点下的约化结果作为基准。该数据集的核心研究问题在于评估与提升机器学习驱动的种子函数策略,其发布标志着高能物理计算领域首次拥有可用于训练与标准化评测学习型积分约化算法的基准框架,对推动自动计算工具的效率革新具有深远意义。
当前挑战
该数据集所面临的挑战体现在领域问题与构建过程两个层面。在领域问题层面,传统IBP约化中种子积分的暴力枚举常导致线性系统极其庞大,求解耗时呈指数增长,而现有启发式策略在复杂拓扑下效率低下且缺乏泛化性,急需一种能够自动学习最优种子选择策略的算法来突破性能瓶颈。在构建过程中,数据生成依赖FIRE软件在特定素数域与固定运动学点下的精确运行,需为每种拓扑预生成复杂的配置脚本,并确保不同维度、不同扇区间的约化结果一致性与可复现性;同时,面对22种拓扑中高达15维的复杂结构,如何设计高效的采样策略以覆盖足够多的非平凡积分,并在训练集与测试集之间保持拓扑分离,从而避免数据泄露,是本数据集构造中的关键技术难点。
常用场景
经典使用场景
在高能物理领域,多圈费曼图的计算是精确理论预言的核心挑战,而积分约化(IBP reduction)作为将多圈积分表示为少量主积分线性组合的关键步骤,其计算效率直接制约着高阶微扰论的进展。feynman-bench数据集提供了涵盖从2维到15维共22种多圈拓扑结构的56102组已完成的IBP约化结果,每组数据包含目标积分、拓扑信息、约化系数以及FIRE求解器实际消耗的方程步数。该数据集最经典的使用场景是作为积分约化中种子函数(seeding function)学习的基准测试平台:研究者可利用训练集(13种拓扑)训练机器学习模型来预测哪些种子积分应被馈入IBP关系,进而在测试集(9种未见拓扑)上评估其约化效率与正确性,以超越FIRE官方基线为目标。
实际应用
在实际应用中,feynman-bench数据集直接服务于高能物理理论计算流程的加速与优化。粒子物理标准模型的精确检验(如强耦合常数的提取、希格斯玻色子产生截面的高阶修正、以及超出标准模型信号的搜索)均依赖多圈费曼图的计算,而IBP约化是其中计算量最密集的环节之一。借助该数据集,研究团队能够训练出可部署于FIRE等主流求解器的智能种子选择模型,将单次约化的方程步数降低至官方基线以下,从而将原本需要数天甚至数周的高圈计算压缩至数小时。此外,该数据集的模块化设计允许研究者无缝集成自定义求解器,通过统一的评估管道(validity和step-ratio指标)直接比较不同策略的优劣,为最终产出更快的通用IBP求解器提供实证基础。
衍生相关工作
feynman-bench数据集的发布已催生了一系列衍生工作,主要集中在学习型种子函数的构建与优化方面。直接意义上的衍生工作包括:基于图神经网络的拓扑感知种子评分器,利用拓扑结构图中的传播子连接关系预测候选种子的约化贡献;基于变换器架构的序列到序列模型,将费曼积分的指数向量作为输入直接输出最优种子集合;以及强化学习框架,其中智能体通过与环境(FIRE求解器)的交互逐步学习选取最小化方程步数的种子策略。更广泛地,该数据集推动了IBP约化中关键启发式模块的改进,如对称性折叠策略的替代方案、IBP关系预排序的动态算法,以及基于种子-IBP二元分类器的选择性方程生成技术。这些工作共同构建了一个以数据驱动提升高能理论计算效率的新兴研究方向。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务