同时优化大模型训练数据与模型配置,打破数据与模型互相依赖的僵局。
The Chicken and Egg Dilemma: Co-optimizing Data and Model Configurations for LLMs
- 用基于缩放定律的性能预测器辅助贝叶斯优化,实现数据与模型配置联合调优。
- 在相同预算下,相比基线方法,显著提升多个大模型任务的性能表现。
- 适合需要高效调参的大模型训练场景,尤其适用于资源受限的实验设置。
大模型训练中的数据配置(如数据混合比例)与模型配置(如架构设计)存在相互依赖关系:最优数据配置依赖于模型选择,反之亦然。然而,联合优化二者通常被认为难以实现,现有方法多只关注数据或模型单独优化,忽略其交互影响。本文提出 JoBS,利用缩放定律启发的性能预测器,结合贝叶斯优化(BO),高效协同优化数据与模型配置。JoBS 将部分优化预算用于训练一个性能预测器,该预测器能通过少量训练步数预估配置的潜力;剩余预算则完全依赖预测器进行优化,有效摊销完整训练的开销。我们分析了平均遗憾并推导出最优预算分配策略。实验表明,JoBS 在相同优化预算下,优于多个多保真度贝叶斯优化基线,以及独立优化数据或模型的方法,在多种大模型任务中均取得更优结果。
原文摘要 · Abstract (English)
Co-optimizing data and model configurations for training LLMs presents a classic chicken-and-egg dilemma: The best training data configuration (e.g., data mixture) for a downstream task depends on the chosen model configuration (e.g., model architecture), and vice versa. However, jointly optimizing both data and model configurations is often deemed intractable, and existing methods focus on either data or model optimization without considering their interaction. We introduce JoBS, an approach that uses a scaling-law-inspired performance predictor to aid Bayesian optimization (BO) in jointly optimizing LLM training data and model configurations efficiently. JoBS allocates a portion of the optimization budget to learn an LLM performance predictor that predicts how promising a training configuration is from a small number of training steps. The remaining budget is used to perform BO entirely with the predictor, effectively amortizing the cost of running full-training runs. We study JoBS's average regret and devise the optimal budget allocation to minimize regret. JoBS outperforms existing multi-fidelity BO baselines, as well as data and model optimization approaches across diverse LLM tasks under the same optimization budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。