MOCHA: Multi-Objective Chebyshev Annealing for Agent Skill Optimization

algorithm 2605.19330
multi-objective-optimizationskill-optimizationchebyshev-scalarizationhypervolumepareto-frontagent-skills

MOCHA: Multi-Objective Chebyshev Annealing for Agent Skill Optimization — L2 #

§1 TL;DR #

MOCHA 将 agent skill 优化形式化为多目标问题(correctness vs. compliance),用 Chebyshev scalarization 覆盖 non-convex Pareto front,结合 HVC 探索与指数退火,6 任务上 +7.5% correctness,baselines 4/6 任务零进展。

§2 痛点 / 方法 / 结果 #

Q1 痛点 #

Agent skill 是结构化多字段 NL artifact(description ≤1024 chars for routing, body ≤5000 chars for execution, 共享 context budget)。现有 prompt optimizer(TextGrad、ProTeGi、GEPA)将多目标折叠为单标量或启发式选择,无法进入 non-convex Pareto 区域。实证表现:在 4/6 任务上 1000 rollout 后 baselines 返回原始 seed skill 不变——选择策略成为瓶颈。

Q2 方法 #

MOCHA 每轮迭代包含两阶段:

  1. Parent Selection: 从 $\Delta^{M-1}$ 均匀采样权重 $\mathbf{w} \sim \mathrm{Dirichlet}(\mathbf{1})$,以 Chebyshev scalarization $s_{\mathbf{w}}(p) = \max_{j} [w_j \cdot |m_j(p) - 1|]$ 选 parent——保证可达所有 Pareto-optimal 点(含 non-convex 区域)。
    1. Acceptance Criterion with Annealing: 阈值 $\tau(b) = \tau_0 \cdot \exp(-\lambda \cdot b/B)$ 控制探索/利用切换:
    2. $\tau(b) > 0$(exploration): 接受 HVC > $\tau(b)$ 的候选——方向无关地扩展 Pareto front
    3. $\tau(b) = 0$(exploitation): 接受 $s_{\mathbf{w}}(p') < s_{\mathbf{w}}(p_{\text{parent}})$ 的候选——沿最弱目标方向精炼
    4. 核心技术壁垒: Chebyshev scalarization 的完备性(Proposition 3.1, Miettinen 1999)保证可触达完整 Pareto front(线性加权无法到达 non-convex 区域),而 HVC 严格单调性(唯一满足 Pareto dominance monotonicity 的 unary indicator)使探索阶段的 accept 判据具有理论最优性。两者结合解决了有限 budget 下的 front 覆盖问题——这是纯 Chebyshev 或纯 HVC 各自无法单独达到的。

      Q3 结果 #

      • 6 task 平均 correctness: MOCHA .675 vs strongest baseline .628 (+7.5% relative)
      • FEVER: +14.9%, TheoremQA: +10.4%
      • Pareto-optimal variants: 3.6 vs 1.7 (2× more)
      • Hypervolume: .483 vs .454 (+6.4%)
      • Baselines 在 4/6 任务 1000 rollout 零进展

      §3 架构 / 方法图 #

      Figure 1: MOCHA overview — skill optimization trade-off and two-phase search

      Paper's Figure 1, verbatim (caption: "(a) Skill optimization produces a correctness–compliance trade-off: the optimized skill p gains correctness but may violate compliance limits. (b) MOCHA navigates this trade-off via two phases: exploration (green) expands the Pareto front, then exploitation (purple) refines the extremes.").*

      Figure 1(a) 展示 skill optimization 的核心张力:单目标优化器将 correctness gain 与 compliance violation 视为不可调和,因此 reject 所有 improvement candidate。Figure 1(b) 展示 MOCHA 的解法:exploration 阶段(绿色)通过 HVC gating 无方向扩展 front,exploitation 阶段(紫色)通过 Chebyshev acceptance 沿特定方向精炼。指数退火连接两阶段。

      stateDiagram-v2 [*] --> Init: seed skill p₀ Init --> SampleWeight: w ~ Dirichlet(1) SampleWeight --> SelectParent: argmin s_w(p) SelectParent --> Mutate: SkillMdProposer(parent, feedback) Mutate --> CheckThreshold: compute τ(b) CheckThreshold --> Explore: τ(b) > 0 CheckThreshold --> Exploit: τ(b) ≈ 0 Explore --> HVCGate: HVC(p', P) > τ(b)? HVCGate --> CommitExplore: yes → add to P HVCGate --> Buffer: no but >0 → speculative queue Buffer --> SampleWeight: next iteration Exploit --> ChebyGate: s_w(p') < s_w(parent)? ChebyGate --> CommitExploit: yes → add to P ChebyGate --> SampleWeight: no → next iteration CommitExplore --> Validate: eval on val set CommitExploit --> Validate Validate --> SampleWeight: b < B Validate --> [*]: b ≥ B → return P*

      §4 作者证明 #

      符号表 #

      符号含义范围
      $p \in \mathcal{P}$skill definition(NL 文本)所有候选 skill 集合
      $M$目标数量本文 $M=3$
      $m_j(p)$skill $p$ 在 metric $j$ 上的值$[0, 1]$
      $\mathbf{w} \in \Delta^{M-1}$权重向量单纯形
      $s_{\mathbf{w}}(p)$Chebyshev scalarization$\max_j [w_j \cdotm_j(p) - 1]$
      $\mathrm{HVC}(p, \mathcal{P})$hypervolume contribution$\mathrm{HV}(\mathcal{P} \cup \{p\}) - \mathrm{HV}(\mathcal{P})$
      $\tau(b)$阈值$\tau_0 \cdot \exp(-\lambda b/B)$
      $B$总 budget(rollout 数)实验中 1000
      $\lambda$退火速率实验中 10

      关键方程物理意义 #

      Eq. (1) $\mathcal{P}^{*} = \{p \in \mathcal{P} : \nexists p' \text{ s.t. } \mathbf{m}(p') \succ \mathbf{m}(p)\}$ — 优化目标:找到不被任何其他 skill 在所有 metric 上同时 dominate 的完整集合(a-posteriori MOO)。

      Eq. (3) $s_{\mathbf{w}}(p) = \max_{j \in [M]} [w_j \cdot |m_j(p) - z_j^*|]$ — 最小化最大加权偏差,天然倾向平衡各目标(penalize 最弱维度),且由 Proposition 3.1 保证完备性。

      Eq. (5) $\mathrm{HVC}(p, \mathcal{P}) = \mathrm{HV}(\mathcal{P} \cup \{p\}) - \mathrm{HV}(\mathcal{P})$ — 候选的独占贡献体积——$> 0$ 当且仅当 $p$ 是 non-dominated,为探索提供方向无关的 quality signal。

      Eq. (7) $\tau(b) = \tau_0 \cdot \exp(-\lambda \cdot b/B)$ — 指数衰减将 acceptance gate 从"任何方向都接受"平滑过渡到"只接受 Chebyshev improvement",mid-budget 后 $\tau \approx 0$。

      6 项检查 #

      #检查项结果
      1形式化声明是否有证明或引用Proposition 3.1 引用 Miettinen 1999 定理,HVC 单调性引用 Zitzler 2003
      2实验是否覆盖声明的全部条件6 task × 5 seeds × 4 methods + 2 ablation variants, 但仅 1 backbone (Claude Haiku 4.5)
      3负面结果是否被报告是: HotpotQA 上 MOCHA 略低于 ProTeGi (.600 vs .622); body compliance 大幅下降 (.33 mean)
      4消融实验是否隔离贡献是: w/o HVC 和 w/o Annealing 清晰隔离探索/利用两个贡献
      5实验设计是否公平是: 所有方法共享 identical SkillMdProposer mutation interface,唯一变量是选择策略
      6限制条件是否被明确声明是: 3 条 limitations(低冲突任务、固定退火、平台特定 compliance)

      §5 实验与数据 #

      Figure 2: Optimization dynamics — convergence curves across six skills

      Paper's Figure 2, verbatim (caption: correctness over optimization budget for all six skills). MOCHA 在 budget 前半段快速爬升(exploration 阶段 HVC 驱动),后半段趋于稳定(exploitation 阶段 Chebyshev 精炼)。Baselines 在 GPQA/HoVer/FEVER/DebugBench 上全程平坦——选择策略 reject 了所有 mutation candidate。

      Figure 3: 2D Pareto fronts for FEVER and TheoremQA

      Paper's Figure 3, verbatim (caption: "2D Pareto fronts (correctness vs. body compliance) for two representative tasks. MOCHA discovers diverse non-dominated skill variants while baselines remain near the initial prompt."). 关键观察:MOCHA 变体分散于 Pareto front 的不同区域(correctness 0.65–0.76 × body compliance 0.2–0.5),而三个 baseline 聚集在 seed skill 附近单一点。这可视化了核心论点——单目标选择器无法离开初始 prompt。

      Table 1: Main experimental results

      Paper's Table 1 (aggregated across 6 skills, 5 seeds). 核心数据点:MOCHA correctness .675 vs GEPA .619 / ProTeGi .628;hypervolume .483 vs .454;Pareto points 3.6 vs 1.7。值得注意的是 MOCHA 的 body compliance 从 .83 降到 .33——这是 correctness gain 的显式代价。

      Figure 14: Ablation heatmap — correctness delta over GEPA

      Paper's Figure 14, verbatim (caption: ablation heatmap showing correctness delta over GEPA for each MOCHA variant across six skills). 关键发现:w/o HVC(纯 exploitation)在 TheoremQA/FEVER 上 correctness gain 最高;w/o Annealing(纯 exploration)在 HoVer 上表现更好;full MOCHA 在 aggregated metrics 上最平衡。这印证了 exploration-exploitation spectrum 的设计合理性。

      关键数据摘要 #

      任务GEPAMOCHAΔ relative
      FEVER.632.726+14.9%
      TheoremQA.690.762+10.4%
      DebugBench.615.666+8.3%
      GPQA.592.636+7.4%
      HoVer.618.660+6.8%
      HotpotQA.622.600−3.5%

      Gain 与 objective conflict 程度正相关:FEVER/TheoremQA 中 correctness improvement 需要更长 instruction(body 超限),MOCHA 能接受此 trade-off;HotpotQA 无显著冲突,multi-objective machinery 反而成为 overhead。

      §6 论证链 #

      Step论点证据来源推理方式
      1Agent skill 优化本质上是多目标的:correctness 与 platform compliance 不可避免冲突§1 platform constraints (1024/5000 char limits), §4 FEVER case study (correctness +11% 伴随 body compliance 从 .99→.38)定义性论证 + 实证
      2单目标选择策略是现有方法失败的根本原因(非 mutation 质量问题)§4.2 baselines 使用相同 multi-objective mutation feedback 仍在 4/6 tasks 返回 seed unchanged; §C.1 identical SkillMdProposer 接口控制变量实验
      3Chebyshev scalarization 可达完整 Pareto front(含 non-convex 区域),线性加权不行Proposition 3.1 (Miettinen 1999 定理); §A.2 形式对比定理引用
      4有限 budget 下纯 Chebyshev 前沿覆盖不足,需要 HVC 探索§4.3 ablation: w/o HVC 的 Pareto points (2.8) < full MOCHA (3.6); HoVer 上 w/o Annealing(.614) 崩塌说明纯探索也不够消融实验
      5指数退火连接两种模式,实现 correctness × diversity 的最佳 trade-off§4.3 Table 3: full MOCHA HV .483 > both ablations (.477/.478); §C.7 HoVer 上 w/o Ann 退化 -4.6pp消融实验
      6MOCHA gain 与 objective conflict 强度正相关,低冲突任务无优势§4.2 Table 2: FEVER(+14.9%) vs HotpotQA(-3.5%); §5 Discussion 明确归因对比分析 + 作者声明

      §7 实现 cross-reference #

      [实现未公开]

      作者未公开 MOCHA 的代码实现。论文描述了统一优化框架(reimplements TextGrad/ProTeGi within same framework),但无公开仓库链接。

      关键实现细节 #

      1. Speculative Buffer: exploration 阶段使用 capacity-5 priority queue 缓存 HVC > 0 但 < $\tau(b)$ 的候选,仅当 queue 中最佳项 HVC > $\tau(b)$ 时 commit。这避免了 exploration 阶段因阈值过高而丢弃所有候选的 cold-start 问题。
        1. Two-Stage Evaluation: 先在小 minibatch 上 gate(低成本),通过后在 full validation set 上评估再 commit。Budget 计算为 $\Delta b = 2n + |\mathcal{D}_{\text{val}}| \cdot \mathbf{1}[\text{commit}]$,使得 reject 的候选消耗极少 budget。