用学习节奏优化训练,用少1/3数据达到更好模型效果
Accelerating Large Language Model Pretraining via LFR Pedagogy: Learn, Focus, and Review
- 根据模型学习进度动态重访难点数据,避免遗忘
- 仅用5%~19%训练数据,困惑度更低、准确率更高
- 适合追求高效预训练的团队,尤其资源有限时
传统大语言模型预训练依赖从网络级数据集中随机采样进行自回归建模。受人类间隔重复学习启发,我们假设随机采样导致训练成本高、模型质量低且存在显著数据遗忘。为此,提出学习-聚焦-回顾(LFR)范式——一种随模型学习进度动态调整的训练方法。LFR追踪模型在数据块(词元序列)上的表现,优先重访易遗忘的难点区域,提升记忆保留与训练效率。在SlimPajama和OpenWebText数据集上,分别用LFR预训练Llama和GPT模型。下游任务评估涵盖问答、推理、语言建模与翻译等多领域。相比全量数据训练基线,LFR始终实现更低困惑度与更高准确率,仅使用5%–19%训练词元。此外,仅用3.2%训练词元,即达到参数量两倍的行业标准Pythia模型性能,验证了其高效性与有效性。
原文摘要 · Abstract (English)
Traditional Large Language Model (LLM) pretraining relies on autoregressive language modeling with randomly sampled data from web-scale datasets. Inspired by human learning techniques like spaced repetition, we hypothesize that random sampling leads to high training costs, lower-quality models, and significant data forgetting. To address these inefficiencies, we propose the Learn-Focus-Review (LFR) paradigm -- a dynamic training approach that adapts to the model's learning progress. LFR tracks the model's learning performance across data blocks (sequences of tokens) and prioritizes revisiting challenging regions of the dataset that are more prone to being forgotten, enabling better retention and more efficient learning. Using the LFR paradigm, we pretrained Llama and GPT models on the SlimPajama and OpenWebText datasets, respectively. These models were evaluated on downstream tasks across various domains, including question answering, problem-solving, commonsense reasoning, language modeling, and translation. Compared to baseline models trained on the full datasets, LFR consistently achieved lower perplexity and higher accuracy, while using only 5%--19% of the training tokens. Furthermore, LFR matched the performance of industry-standard Pythia models with up to 2$\times$ the parameter count, using just 3.2% of the training tokens, demonstrating its effectiveness and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。