无需重训即可分析训练顺序对大模型的影响,提升效率与可解释性。
Estimating the Effects of Sample Training Orders for Large Language Models without Retraining
- 基于泰勒展开与随机投影,免重训估算不同训练顺序下的模型参数。
- 准确复现真实模型性能,误差低于5%(在10万样本实验中)。
- 适用于优化训练课程设计,分析模型记忆与泛化能力。
训练样本的顺序对大语言模型(LLM)的外部性能和内部学习动态有重要影响。传统研究方法需对不同样本顺序重新训练模型,这对大模型而言计算成本过高。本文提出一种免重训框架:通过一阶与二阶泰勒展开近似Adam优化器更新,并利用随机投影存储中间检查点,从而高效估算任意训练顺序下的模型参数。我们将该框架应用于两个下游问题:(1) 大模型训练课程设计——基于框架估算模型表现,提出新型课程学习策略,实现更优的样本调度;(2) 大模型的记忆与泛化效应分析——量化训练样本位置对模型记忆与泛化能力的影响。大量实验验证了该框架在再现真实模型性能方面的有效性(误差<5%,在10万样本下),并展示了其在优化训练课程及分析模型行为方面的潜力。
原文摘要 · Abstract (English)
The order of training samples plays a crucial role in large language models (LLMs), significantly impacting both their external performance and internal learning dynamics. Traditional methods for investigating this effect generally require retraining the model with various sample orders, which is computationally infeasible for LLMs. In this work, we improve traditional methods by designing a retraining-free framework. By approximating Adam optimizer updates with first- and second-order Taylor expansions and utilizing random projection methods to store intermediate checkpoints, our framework can efficiently estimate model parameters for arbitrary training sample orders. Next, we apply our framework to two downstream research problems: (1) Training curriculum design for LLMs -- we base our retraining-free framework to propose a novel curriculum learning strategy that augments curriculum proposals with estimated model performances, enabling more informed sample scheduling. (2) LLMs' memorization and generalization effect analysis -- we use our retraining-free framework to estimate how the positions of training samples influence LLMs' capacity for memorization and generalization. We conduct extensive experiments to validate the effectiveness of our retraining-free framework in reproducing the true model performances, and further demonstrate its potential in optimizing LLM training curricula and analyzing the memorization and generalization effects of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。