arXiv:2606.11499cs.CLcs.AI2026-06被引 1

用网页图中心性筛选预训练数据,提升模型性能

Hubs or Fringes: Pretraining Data Selection via Web Graph Centrality

论文配图:Hubs or Fringes: Pretraining Data Selection via Web Graph Centrality
图 1 · 摘自论文原文
  • 基于网页主机图的中心性计算,区分核心与边缘内容
  • 1:1混合核心与边缘文档,平均得分提升至41.4%
  • 无需标注数据或模型训练,适合大规模数据筛选

现代语言模型性能高度依赖预训练数据构成。现有方法多依赖辅助分类器打分或混合优化,增加计算开销并依赖标注数据。本文提出WebGraphMix,一种轻量级数据筛选框架,通过在Common Crawl主机级网页图上计算结构中心性分数,调节预训练语料中核心与边缘文档的比例。假设核心主机传递可复用抽象知识,边缘主机包含专业长尾知识。WebGraphMix可在网页规模高效计算中心性,无需模型训练、标注数据或下游监督。将该方法集成至DataComp-LM流程,分别训练400M和1B参数模型,使用8B和28B token数据,在23项任务(涵盖事实知识到符号推理)上评估。实验表明,核心与边缘网络区域编码互补能力。1:1比例混合时平均得分达41.4%,优于均匀采样(39.8%)。结合结构分数与文档质量分类器分数后性能进一步提升至43.8%。结果表明,网页图拓扑是预训练数据筛选的有效维度,其信息与现有内容导向方法基本正交。

原文摘要 · Abstract (English)

The performance of modern language models depends critically on pretraining data composition. Yet existing data selection methods rely on auxiliary classifiers for document scoring or mixture optimization, adding computational overhead and dependence on labeled data. We propose WebGraphMix, a lightweight data selection framework that computes structural centrality scores over the Common Crawl host-level web graph and uses them to vary the proportion of central versus peripheral documents in the pretraining mixture. We hypothesize that central hosts expose models to reusable abstractions, while peripheral hosts encode specialized, long-tail knowledge. WebGraphMix computes centrality scores efficiently at web scale, requiring no model training, labeled data, or downstream supervision. We integrate WebGraphMix into the DataComp-LM pipeline and train models at 400M and 1B parameter scales with 8B and 28B tokens respectively, evaluating on 23 tasks ranging from factual knowledge to symbolic reasoning. Our experiments show that central and peripheral web regions encode complementary capabilities. Mixture combining both at a ratio of 1:1 achieves 41.4% on average, compared to 39.8% for uniform sampling. Combining structural scores with document-level quality classifier scores further improves performance to 43.8%. These findings demonstrate that web graph topology is a meaningful axis for pretraining data curation, capturing information that is largely orthogonal to existing content-based approaches.

数据筛选网页图预训练中心性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。