arxiv-ai-entity-graph

arXiv AI实体图谱是一个结构化知识图谱数据集，专门从arXiv计算机科学（CS）研究论文的标题和摘要中提取并规范化人工智能实体。其核心目标是将海量、非结构化的学术文本转化为一个可查询的分析数据库，用于追踪研究社区中AI模型、框架、基准和数据集的实际使用与采纳趋势。数据集覆盖从2024年1月至今的23个CS类别（如cs.CL、cs.LG、cs.CV等），并每日更新。数据内容经过精心处理，每个实体提及（例如GPT-4、gpt4）都被规范化为唯一的规范名称，并附带了丰富的元数据，包括实体类型（模型、框架、基准、数据集）、提供者（如OpenAI、Meta、Alibaba）、模型家族（如GPT、Llama、Qwen）、模型类型（如llm、vlm、image_gen）、置信度分数以及所属论文信息。核心数据表为`entities`，关联论文元数据表`papers`以及其他辅助表（如authors、citations）。数据集规模方面，在示例时间窗口（2024年1月至2026年6月）内，包含约65,000个模型实体、55,000个框架实体、15,000个基准实体和4,000个数据集实体。该数据集适用于研究情报分析、AI生态趋势监测、科技政策研究以及构建基于实体的学术检索或推荐系统等任务。用户可以通过提供的DuckDB文件（免费样本）或订阅完整数据集进行访问。

The arXiv AI Entity Graph is a structured knowledge graph dataset that specifically extracts and normalizes artificial intelligence entities from the titles and abstracts of arXiv computer science (CS) research papers. Its core objective is to transform massive, unstructured academic text into a queryable analytical database for tracking the actual usage and adoption trends of AI models, frameworks, benchmarks, and datasets within the research community. The dataset covers 23 CS categories (e.g., cs.CL, cs.LG, cs.CV) from January 2024 to the present, with daily updates. The data content is meticulously processed, with each entity mention (e.g., GPT-4, gpt4) normalized to a unique canonical name and enriched with extensive metadata, including entity type (model, framework, benchmark, dataset), provider (e.g., OpenAI, Meta, Alibaba), model family (e.g., GPT, Llama, Qwen), model type (e.g., llm, vlm, image_gen), confidence scores, and associated paper information. The core data table is `entities`, linked to the paper metadata table `papers` and other auxiliary tables (e.g., authors, citations). In terms of scale, within the example time window (January 2024 to June 2026), the dataset contains approximately 65,000 model entities, 55,000 framework entities, 15,000 benchmark entities, and 4,000 dataset entities. This dataset is suitable for tasks such as research intelligence analysis, AI ecosystem trend monitoring, science and technology policy research, and building entity-based academic search or recommendation systems. Users can access it via the provided DuckDB file (free sample) or by subscribing to the full dataset.

原始信息汇总

数据集概述

arXiv AI Entity Graph 是一个从 arXiv 计算机科学论文中提取的结构化实体数据集，专门聚焦于人工智能领域。它将论文标题和摘要中提及的 AI 模型、框架、基准测试和数据集，标准化为带有提供者（Provider）、家族（Family）和置信度（Confidence）元数据的规范名称。

核心特点

• 结构化与标准化: 将论文中的实体提及（如 GPT-4、gpt4、GPT-4-turbo）统一映射到同一规范条目，消除别名歧义。
• 每日更新: 覆盖 2024 年 1 月至今的 arXiv CS 论文，并持续更新。
• 分析型数据库: 旨在回答“研究人员实际使用哪些模型？”、“DeepSeek R1 何时成为主流？”等分析性问题，而非作为检索基准或语料库。

关键发现（2024年1月 – 2026年6月）

• GPT 持续主导: 在整个周期内被 5,652 篇论文引用。
• 阿里 Qwen 家族崛起: Qwen3、Qwen2.5 和 Qwen 合计在 2026 年引用量（4,723 篇）与 OpenAI 总影响力相当。
• DeepSeek R1 增速最快: 从 2025 年 1 月的 3 篇论文激增至 2 月的 48 篇（月增长 16 倍），此后维持在每月 50–90 篇。
• o1 进入前 20: 被 620 篇论文引用，表明推理模型已成为主流研究基础设施。
• cs.CL 超越 cs.LG: 语言研究（27,452 个实体行）的论文产出已超过纯机器学习（20,040 个实体行）。
• Stable Diffusion 稳固: 被 884 篇论文引用，图像生成已成为与 LLM 并行的严肃研究分支。

数据集架构

核心表为 entities，包含以下主要字段：

字段	类型	描述
paper_id	varchar	arXiv 论文 ID
entity_type	varchar	实体类型：model、framework、benchmark、dataset
entity	varchar	规范化的实体名称
confidence	double	置信度分数：1.0（已确认）、0.2（模糊匹配）、0.1（可能误报）
family	varchar	模型家族（如 GPT、Llama、Qwen）
provider	varchar	提供者（如 OpenAI、Meta、Alibaba）
model_type	varchar	模型类型（如 llm、reasoning_llm、vlm、image_gen）

此外还有 papers、authors、citations、categories、versions、links 等辅助表。

数据覆盖

• 时间范围: 2024年1月至今（每日更新）。
• 覆盖类别: 涵盖 23 个 CS 子类别，包括 cs.LG, cs.CL, cs.CV, cs.AI, cs.NE, cs.CR, cs.SE, cs.DC 等。
• 实体数量（2024年1月 – 2026年6月）:
  • 模型：约 65,000 个实体行，100+ 种唯一实体
  • 框架：约 55,000 个实体行，40+ 种唯一实体
  • 基准：约 15,000 个实体行，50+ 种唯一实体
  • 数据集：约 4,000 个实体行，30+ 种唯一实体

访问与定价

提供免费样本（3×10天窗口数据，DuckDB 格式）、一次性快照（$49，完整历史数据 Parquet 格式）和订阅服务（$29/月，包含每日增量更新）。数据通过 AWS S3 同步获取。

许可协议

元数据（标题、摘要、作者、类别）源自 arXiv，遵循 CC0 1.0 公共领域贡献。实体提取、标准化、置信度评分及架构为原创工作，采用 CC BY 4.0 许可。

搜集汇总

数据集介绍

构建方式

该数据集基于arXiv平台上计算机科学领域论文的标题与摘要，采用确定性规则驱动的流水线进行实体抽取。研究者首先构建了一个经过精心整理的YAML知识库，涵盖模型、框架、基准与数据集等实体及其别名，继而通过正负上下文窗口机制对模糊匹配结果进行消歧，最终将所有提及的实体归一化为规范名称，并附带提供者、家族与置信度等元数据。数据从2024年1月起每日更新，覆盖23个CS子类别，确保时间序列的连续性与覆盖的广度。

特点

本数据集的核心特色在于其结构化与可分析性。与传统的论文语料库不同，它提供了一个可直接查询的分析型数据库，使研究者能够轻松追踪模型、框架和基准在学术文献中的真实使用轨迹。例如，数据揭示了GPT系列在2024至2026年间累计被引用5,652篇论文，而DeepSeek R1在2025年初出现了前所未有的采纳速度跃升。此外，每个实体提及均经过规范化处理，消除了别名歧义，并辅以明确的置信度评分，其中1.0分表示最高精度的匹配结果。

使用方法

使用者可通过DuckDB或任何支持Parquet格式的数据分析工具直接连接本数据集。免费样本以DuckDB文件形式提供，允许执行即席查询，例如统计被引用最多的模型、分析特定基准的评估场景，或按月份与提供者追踪市场份额的变化趋势。完整的快照与订阅版本则以分区Parquet文件形式托管于Amazon S3，支持每日增量同步，方便用户将其无缝集成至现有的数据处理管线中，进行长期趋势分析与知识图谱构建。

背景与挑战

背景概述

随着人工智能研究的爆炸式增长，arXiv平台上已累积超过两百万篇计算机科学论文，其中模型、框架、基准和数据集等关键实体信息散落在海量文本中，难以被系统化地追踪与分析。对此，arxgraph团队于2024年发布了arXiv AI Entity Graph数据集，旨在将论文标题与摘要中提及的AI实体进行结构化提取与归一化处理，覆盖2024年1月至今的23个CS子领域并每日更新。该数据集不仅将GPT-4、gpt4等别名统一为规范名称，还提供了提供者、家族和置信度等元数据，从而将arXiv从纸本文库转化为可分析的“研究情报数据库”，使学界与产业界能够量化追踪模型采纳趋势、基准使用情况与研究热点的演变，对自然语言处理、知识图谱及科研情报分析领域产生了重要影响。

当前挑战

该数据集面临的核心挑战在于解决非结构化科学文本中的实体模糊性与歧义问题。学术论文中常出现缩写、别名和跨领域术语（如“Flux”在物理学与图像生成中含义迥异），传统的检索方法难以区分，而基于LLM的黑箱抽取又缺乏可审计性。构建过程中，团队需设计确定性规则管道与精心编排的YAML知识库，通过正负上下文窗口判别模糊匹配，并将结果赋予1.0、0.2或0.1的三级置信度分数，以确保高频精准分析（如过滤confidence=1.0的数据）。此外，每日处理来自多个CS子领域的增量论文、维护实体名称的时效性（如DeepSeek R1从寥寥数篇到每月超50篇的爆发式增长），以及在不引入破坏性变更的前提下稳定更新模式，均为构建与维护带来持续挑战。

常用场景

经典使用场景

arXiv AI Entity Graph 数据集将海量计算机科学论文中提及的模型、框架、基准测试与数据集进行结构化抽取与名称归一化，构建了一个动态更新的知识图谱。研究者可直接查询特定模型（如GPT-4o、Llama 3、Qwen3）在论文中的引用频率、时间趋势和跨领域渗透情况，也可追踪框架（PyTorch、vLLM）或基准（MMLU、HumanEval）的演进轨迹。该数据集尤其适合从事学术计量分析、研究趋势预测和人工智能发展史研究的学者，为他们提供了一个从论文层面精准度量技术采纳与影响力的利器。

解决学术问题

该数据集有效解决了人工智能研究中技术使用情况难以量化、跨论文的实体名称不统一、以及趋势分析缺乏结构化数据支撑等长期困扰学界的难题。通过提供权威的规范化实体名、家族归属和信任度评分，研究人员得以超越单纯的论文检索，进行严谨的假设检验，例如验证推理模型（如DeepSeek R1）的采纳速度是否超越此前所有模型，或分析开源模型（如Qwen）与闭源模型（如GPT）在不同子领域的竞争格局。这些分析为理解研究社区的集体行为、技术扩散模式与学术生态变迁提供了前所未有的可靠数据基础。

衍生相关工作

基于该数据集的结构化特性，已衍生出若干重要的后续工作方向。学术层面，研究者可构建研究影响力的动态地图，例如绘制不同模型家族（GPT、Llama、Qwen）在时间维度上的“领土扩张”动画，或分析基准测试（如MMLU）如何与特定模型形成共现生态。工程层面，该数据集可作为训练论文推荐系统、自动文摘工具或研究趋势预警系统的关键特征源。此外，基于实体置信度分数和上下文窗口的方法论，也为其他学术数据库的自动化实体抽取任务提供了可复用的实践范本，推动了全文级科研知识图谱构建技术的进步。