用AI智能调度多种优化器,让机器学习调参更快更准
ASAP: Agent-System Co-Design for Wall-Clock-Centered Auto HPO Research for ML Experiments

- 让大模型融合多种优化器,动态选最优方案
- 实测比传统方法快3倍以上,且精度不降
- 适合追求高效调参的算法工程师和研究者
超参数优化(HPO)对提升机器学习模型性能至关重要,核心挑战是样本效率:在有限预算内找到优质配置。现有工具依赖特定先验,当问题多样性增加时表现不佳。虽然近期工作尝试用大模型提升每轮性能,但存在两大局限:其一,大模型仅作为单一工具替代,受预训练目标限制,无法覆盖多样问题;其二,仅以迭代次数评估,忽略实际运行中大模型推理与工具执行带来的串行开销,导致迭代优势无法转化为端到端加速。本文提出ASAP,一种代理-系统协同设计框架:代理侧利用大模型整合多样化优化器并动态选择提案;系统侧重构优化循环,通过前缀稳定提示最大化键值缓存复用、推测并行隐藏剩余延迟,并引入自调优模块根据执行日志调整推测阈值。在多种现代HPO任务上的实验表明,ASAP持续优于基线,验证了工具集成与协同设计的价值。
原文摘要 · Abstract (English)
Hyperparameter Optimization (HPO) is essential for maximizing machine learning model performance, and its core challenge is sample efficiency: finding strong configurations within a limited budget. Because every HPO tool relies on a surrogate prior that imparts its own inductive bias, individual tools struggle once problems become sufficiently diverse and drift from these priors. Motivated by the reasoning and generalization capabilities of LLMs, recent work has explored using LLMs for HPO and reports improved per-iteration performance. Yet these methods share two limitations with a common origin: they use the LLM as a single-tool replacement evaluated by iteration count. (i) Deployed in place of prior tools, the LLM is itself constrained by its pretraining objective to one family of inductive-biased proposals; this single-source setup still fails to handle the full diversity of problems. (ii) Per-iteration evaluation ignores that, in real runs, LLM inference or tool execution is paid serially on top of model evaluation every round, so iteration-count gains do not translate into end-to-end wall-clock gains. We present ASAP, an agent-system co-design that addresses both limitations. On the agent side, ASAP uses the LLM to integrate a diverse pool of inductive-biased optimizers and to select among their proposals each round. On the system side, ASAP re-architects the loop to reduce end-to-end wall-clock while preserving regret quality: a prefix-stable prompt maximizes KV-cache reuse across rounds; speculation parallelism hides the remaining LLM and tool latency under model evaluation via a relative-error accept test; and a Self-Tuner adapts the speculation threshold from execution logs off the critical path. Extensive experiments on diverse modern HPO tasks show that ASAP consistently outperforms baselines, underscoring the value of tool integration and agent-system co-design.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。