arXiv:2410.20302cs.LGcs.AI2024-10被引 28

用大模型优化超参数,比传统方法更稳更省成本。

Sequential Large Language Model-Based Hyper-parameter Optimization

  • 结合大模型初始化与贝叶斯采样,动态调整搜索空间。
  • 14个表格任务中9项超越传统贝叶斯优化,减少调参耗时。
  • 适合想用大模型做自动调参的研究者和工程师。

本研究提出SLLMBO框架,利用大语言模型(LLM)进行超参数优化(HPO),融合动态搜索空间适应性、参数空间深度挖掘及新型的LLM树状帕岑估计器(LLM-TPE)采样器。该框架克服了纯大模型方法与传统贝叶斯优化(BO)的局限,在14个分类与回归任务中表现优异:在9项任务上超越经典贝叶斯优化方法。通过对比GPT-3.5-Turbo、GPT-4o、Claude-Sonnet-3.5和Gemini-1.5-Flash等多款主流大模型,首次系统性评估其在HPO中的性能差异。实验表明,引入大模型的参数初始化能力与TPE的探索优势,可在保证高效性的同时实现探索与利用的平衡,降低API调用成本,并缓解过早终止问题。在预算受限场景下,适度延长迭代次数仍能提升结果,表明此类方法对计算资源扩展具有显著收益。本工作为后续研究开源大模型在HPO中的应用、结果可复现性以及在图像分类、分割、机器翻译等复杂数据集上的拓展奠定基础。

原文摘要 · Abstract (English)

This study introduces SLLMBO, an innovative framework leveraging large language models (LLMs) for hyperparameter optimization (HPO), incorporating dynamic search space adaptability, enhanced parameter space exploitation, and a novel LLM-tree-structured parzen estimator (LLM-TPE) sampler. By addressing limitations in recent fully LLM-based methods and traditional bayesian optimization (BO), SLLMBO achieves more robust optimization. This comprehensive benchmarking evaluates multiple LLMs, including GPT-3.5-Turbo, GPT-4o, Claude-Sonnet-3.5, and Gemini-1.5-Flash, extending prior work and establishing SLLMBO as the first framework to benchmark a diverse set of LLMs for HPO. By integrating LLMs' established strengths in parameter initialization with the exploitation abilities demonstrated in this study, alongside TPE's exploration capabilities, the LLM-TPE sampler achieves a balanced exploration-exploitation trade-off, reduces API costs, and mitigates premature early stoppings for more effective parameter searches. Across 14 tabular tasks in classification and regression, the LLM-TPE sampler outperformed fully LLM-based methods and achieved superior results over BO methods in 9 tasks. Testing early stopping in budget-constrained scenarios demonstrated competitive performance, indicating that LLM-based methods generally benefit from extended iterations for optimal results. This work lays the foundation for future research exploring open-source LLMs, reproducibility of LLM results in HPO, and benchmarking SLLMBO on complex datasets, such as image classification, segmentation, and machine translation.

超参数优化大模型贝叶斯优化自动化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。