用低成本方法从网络数据中提炼高质量天文语料,提升大模型专业能力。
ORBIT: Cost-Effective Dataset Curation for Large Language Model Domain Adaptation with an Astronomy Case Study
- 基于噪声网页数据,通过筛选与重构生成100亿词的高质天文书语料
- 在天文任务上使大模型准确率从69%提升至76%,超越基线模型
- 方法可迁移至法律与医学领域,适合需要垂直知识训练的研究者
近期语言建模进展表明,专业任务需高质量领域特定训练数据。通用模型因缺乏专业知识深度,在专家级任务中表现不足。领域适配训练虽可增强模型,但需大量高质量数据。为此,我们提出ORBIT——一种从噪声网络源高效构建大规模高质量领域数据集的方法,专为训练专业大模型设计。以天文学为例,我们将1.3万亿词的FineWeb-Edu数据集精炼为聚焦天文学的100亿词子集。在10亿词的天文数据子集上微调LLaMA-3-8B,使其在MMLU天文基准上的表现从69%提升至76%,并在AstroBench(天文学专用基准)中取得最佳成绩。此外,我们的模型Orbit-LLaMA在GPT-4o评估中,于1000道天文专属问题中被偏好73%。我们还验证了ORBIT在法律和医学领域的泛化能力,相较未过滤基线显著提升数据质量。相关方法、数据集、代码及模型已开源,地址见https://github.com/ModeEric/ORBIT-Llama。
原文摘要 · Abstract (English)
Recent advances in language modeling demonstrate the need for high-quality domain-specific training data, especially for tasks that require specialized knowledge. General-purpose models, while versatile, often lack the depth needed for expert-level tasks because of limited domain-specific information. Domain adaptation training can enhance these models, but it demands substantial, high-quality data. To address this, we propose ORBIT, a cost-efficient methodology for curating massive, high-quality domain-specific datasets from noisy web sources, tailored for training specialist large language models. Using astronomy as a primary case study, we refined the 1.3T-token FineWeb-Edu dataset into a high-quality, 10B-token subset focused on astronomy. Fine-tuning \textsc{LLaMA-3-8B} on a 1B-token astronomy subset improved performance on the MMLU astronomy benchmark from 69\% to 76\% and achieved top results on AstroBench, an astronomy-specific benchmark. Moreover, our model (Orbit-LLaMA) outperformed \textsc{LLaMA-3-8B-base}, with GPT-4o evaluations preferring it in 73\% of cases across 1000 astronomy-specific questions. Additionally, we validated ORBIT's generalizability by applying it to law and medicine, achieving a significant improvement of data quality compared to an unfiltered baseline. We open-source the ORBIT methodology, including the curated datasets, the codebase, and the resulting model at \href{https://github.com/ModeEric/ORBIT-Llama}{https://github.com/ModeEric/ORBIT-Llama}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。