第 224 题:超参搜索(HPO)的基础设施?Optuna、Ray Tune集成?
题目
超参搜索(HPO)的基础设施?Optuna、Ray Tune集成?
完整讲解
一、HPO 基础设施需求
超参搜索(HPO):在 超参空间 上采样或搜索(网格、随机、贝叶斯、NAS 等),并发 跑多组训练,按 metric 选优。基础设施需:任务编排(多 trial 的提交、排队、资源)、metric 收集与早停、搜索策略 与 结果存储。
二、Optuna、Ray Tune 集成
Optuna:Python 库,定义 objective 与 suggest 超参;scheduler(如 MultiObjective、Pruning)做早停与多目标;storage 存 trial 与 metric。与平台:Optuna 可 backend 到 DB/Redis;每个 trial 可提交为 平台上的训练 job(如 K8s Job、Ray task),平台负责调度与资源;callback 或 轮询 把 job 的 metric 回填 Optuna,供下一轮 suggest。Ray Tune:分布式 HPO,trial 以 Ray task 跑在 Ray 集群;scheduler(ASHA、HyperBand 等)与 search_algorithm 内置;与 训练脚本 通过 config + report 对接。集成:平台提供 Ray 集群 或 K8s + Optuna executor,训练镜像与资源由平台统一管;HPO 只负责「生成 trial、收 metric、选优」。
面试要点
- HPO 需:多 trial 编排与资源、metric 收集与早停、搜索策略、结果存储。
- Optuna:objective + suggest + storage;trial 可提交为平台 job,metric 回填。
- Ray Tune:trial 即 Ray task;scheduler + search 内置;与平台通过 Ray 或 K8s 集成。
记忆要点
- HPO = 多 trial 编排 + metric + 搜索策略。
- Optuna = 库;trial 可映射平台 job;Ray Tune = trial 即 Ray task。
- 平台提供资源与训练环境;HPO 负责 suggest 与选优。