一次训练,同时完成十项实验,大幅降低大模型预训练成本
Train Once, Answer All: Many Pretraining Experiments for the Cost of One
- 同一训练过程并行开展多个预训练实验
- 2100亿token数据上训练27亿参数模型,结果可复现多项旧研究
- 适合想高效验证数据影响、模型行为的科研人员
近期研究表明,受控预训练实验是探究训练数据与大语言模型行为关系的强大工具。但预训练的计算成本构成显著约束。为此,我们提出一种新方法:在单次训练过程中同步进行多项实验。通过在2100亿个标记的数据上训练最大达27亿参数的模型,我们完成了十项实验,涵盖数据污染、中毒攻击和记忆现象等主题。尽管仅训练一次,仍可复现多项先前工作的成果,并开展知识获取、数学推理和水印等新研究。例如,动态调整训练数据直至模型掌握特定知识。令人惊讶的是,这些实验对模型训练动态和整体性能影响极小。然而,实验间可能存在交互干扰。为此,我们提出持续预训练依赖性测试(CPDT),用于检测此类交互,在本设置下发现其可忽略不计。总体而言,该方法使在有限算力预算下对大模型进行严谨科学实验成为可能。
原文摘要 · Abstract (English)
Recent work has demonstrated that controlled pretraining experiments are a powerful tool for studying the relationship between training data and large language model (LLM) behavior. However, the computational cost of pretraining presents a significant constraint. To overcome this constraint, we propose a new approach where multiple experiments are conducted simultaneously during a single training run. We validate our approach by performing ten experiments while training on 210B tokens, with models of up to 2.7B parameters. Although models are trained only once, we can replicate the results of multiple previous works on data contamination, poisoning, and memorization. We also conduct novel investigations into knowledge acquisition, mathematical reasoning, and watermarking. For example, we dynamically update the training data until a model acquires a particular piece of knowledge. Remarkably, the influence of the experiments on the model's training dynamics and overall performance is minimal. However, interactions between experiments may act as a confounder in our approach. We propose continual pretraining dependence testing (CPDT), a novel technique to test for interactions with continual pretraining experiments, finding them to be negligible in our setup. Overall, our results suggest that performing multiple pretraining experiments within a single training run can enable rigorous scientific experimentation with large models on a compute budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。