ArXiv-to-Model: A Practical Study of Scientific LM Training
AI 摘要
该论文详细介绍了从原始arXiv LaTeX数据训练小型科学语言模型的完整流程和经验。
主要贡献
- 构建科学语言模型的端到端pipeline
- 分析预处理决策对模型训练的影响
- 揭示存储和I/O约束的重要性
方法论
采用端到端的训练pipeline,包含数据预处理、tokenization和transformer训练,并在受限计算资源下进行了24次实验。
原文摘要
While frontier large language models demonstrate strong reasoning and mathematical capabilities, the practical process of training domain-specialized scientific language models from raw sources remains under-documented. In this work, we present a detailed case study of training a 1.36B-parameter scientific language model directly from raw arXiv LaTeX sources spanning mathematics, computer science, and theoretical physics. We describe an end-to-end pipeline covering metadata filtering, archive validation, LaTeX extraction, text normalization, domain-aware tokenization, and dense transformer training under constrained compute (2xA100 GPUs). Through 24 experimental runs, we analyze training stability, scaling behavior, data yield losses, and infrastructure bottlenecks. Our findings highlight how preprocessing decisions significantly affect usable token volume, how tokenization impacts symbolic stability, and how storage and I/O constraints can rival compute as limiting factors. We further analyze convergence dynamics and show stable training behavior in a data-rich regime (52B pretraining tokens). Rather than proposing a novel architecture, this work provides an engineering-grounded, transparent account of training a small scientific language model from scratch. We hope these insights support researchers operating under moderate compute budgets who seek to build domain-specialized models.