ai-infra-interview-305

第 224 题:超参搜索(HPO)的基础设施?OptunaRay Tune集成?

题目

超参搜索(HPO)的基础设施?OptunaRay Tune集成?


完整讲解

一、HPO 基础设施需求

超参搜索(HPO):在 超参空间采样或搜索(网格、随机、贝叶斯、NAS 等),并发 跑多组训练,按 metric 选优。基础设施需:任务编排(多 trial 的提交、排队、资源)、metric 收集与早停搜索策略结果存储

二、Optuna、Ray Tune 集成

OptunaPython 库,定义 objectivesuggest 超参;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、选优」。

面试要点


记忆要点

  1. HPO = 多 trial 编排 + metric + 搜索策略。
  2. Optuna = 库;trial 可映射平台 job;Ray Tune = trial 即 Ray task。
  3. 平台提供资源与训练环境;HPO 负责 suggest 与选优。
返回模块 返回总览