alphaFeedback

WorksComputer Science

A Fisher-Geometric View of Multi-Teacher On-Policy Distillation

AI contributions
Not recorded for this paper
View details

Abstract

Consolidating specialized foundation models into a unified generalist is a central goal of post-training. Two approaches are in common use: model merging, which is fast but treated as a heuristic, and multi-teacher on-policy distillation, which matches distributions but demands heavy rollout compute and suffers from cross-task gradient conflict. We show that the two are one objective seen at different orders. Within a shared basin of homologous specialists, the multi-teacher on-policy distillation objective is, to second order, exactly the Fisher-merging objective. Its minimizer is therefore the Fisher merge, a classic model-merging method given by a closed-form combination of the specialists, so merging is optimal, to second order, as a closed-form stage of distillation. Building on this identification, we propose Residual-Governed Branch-and-Merge(RGBM), a two-phase consolidation procedure. Concretely, a full-parameter Fisher merge is intractable on foundation models, so Phase~1 restricts the merge to the task-vector subspace. Extending tangent-space analyses of merging, we show that the same coefficients that mix task vectors also mix implicit rewards, which lets a residual program on unlabelled prompts solve for them. The program minimizes the worst per-domain residual and reports its energy-normalized values as the domain residual spectrum. Phase~2 lets this spectrum govern what may be added: RGBM passivates domains with low normalized residuals, trains isolated branches only for domains above the spectral gap, and writes back the admitted updates with calibrated shrinkage. On two open-source consolidation suites, the Phase~1 merge alone matches or outperforms fully trained multi-teacher distillation, and RGBM improves on it further with zero extra joint-training steps, whereas every un-gated write-back, including continued joint distillation from the merge, falls below it.

Introduction

Post-training of large language models (LLMs) increasingly produces homologous domain specialists—fine-tuned from a common base via reinforcement learning (RL) on distinct reasoning modalities ([1, 26, 7, 29, 8, 17]). Serving an ensemble of specialists multiplies memory and latency, making consolidation into a single generalist essential.

Two paradigms are in common use, each with its own shortcomings. Weight-space merging ([13, 31, 20, 34] (Yadav et al. 2024)) adds to the base model a weighted sum of the specialists’ weight updates (their task vectors) at zero training cost, but the weights are set by hand and, once the merge is formed, nothing tells which domain it failed to recover or whether the chosen weights still combine the specialists faithfully. Multi-teacher on-policy distillation ([1, 8, 28, 35, 4]) (MOPD) instead trains a student to match each specialist (its teacher) on the student’s own generations, which repairs what merging misses, but at the cost of rollouts on every domain and with no control over which domains its shared-weight gradients overwrite. Neither paradigm can answer the question the other one raises: what a merge has already recovered, and where distillation is still needed without harming what is already correct. Answering it requires relating the two, which is what this paper does.

Our starting point is that these paradigms optimize the same objective at different orders of approximation (§ 2). Expanded to second order around the specialists in a shared basin, the distillation objective is exactly the Fisher-merging objective of Matena & Raffel [21], whose minimizer is a weighted average of the specialists with Fisher-information weights; merging is therefore not an alternative to distillation but is optimal, to second order, as its closed-form stage. A full-parameter Fisher merge is intractable on foundation models, so we restrict the merge to the task-vector subspace (Phase 1, described next). Extending tangent-space analyses of merging, we show that, to first order, the same coefficients that mix the task vectors also mix the specialists’ implicit rewards (the log-probability ratios between each specialist and the base), which lets us obtain them by minimizing the worst reward-space residual on unlabelled prompts. After solving, reward-energy normalization measures the relative share left unrecovered in each domain. The fidelity of this first-order fit provides an intrinsic, label-free test of whether a checkpoint still lies where the expansion holds.

Grounded in this framework, we propose Residual-Governed Branch-and-Merge (RGBM) as shown in Figure 1. In Phase 1, RGBM determines the combination weights by minimizing the worst per-domain residual via a small convex program on unlabelled prompts, yielding the zero-training merged model. It then normalizes the residuals by their domain reward energies to form the domain residual spectrum. This spectrum directly dictates what, if anything, should follow the merge. Standard joint distillation tends to degrade the merged model because gradients from domains with low normalized residuals corrupt shared weights. To prevent this interference, Phase 2 leaves those domains untouched and trains isolated branches only for domains above a closed-form spectral gap. Each admitted branch update is then written back, scaled by the largest factor that passes the linearity test.

Merge before you distill (schematic)

Figure 1. Merge before you distill (schematic). (a) Weight space: the dashed triangle is the task subspace spanned by the Math, Code, and IF task vectors. A Minimax projection on unlabelled prompts yields the zero-training merge θ0⋆=Wref+Tα⋆\theta_0^\star= W_{\mathrm{ref}} + T\alpha^\star, where TT stacks those task vectors, together with the per-domain residuals Ld\mathcal{L}_d (red segments). Normalizing these residuals by domain reward energy gives the domain residual spectrum. (b) Capability space: joint distillation from θ0⋆\theta_0^\star can leave the basin (dashed path). The normalized spectrum passivates Code and IF; RGBM writes back only the admitted Math branch, shrunk by γ⋆\gamma^\star (the largest factor that still passes the linearity test). See § 3 for details.

Empirically, we validate our framework across two multi-teacher consolidation benchmarks: Open-MOPD (Gao et al., [8]) (SmolLM3-3B with Math, Code, and Instruction Following specialists) and G-OPD (Yang et al., [35]) (Qwen3-4B with Math and Code specialists). On Open-MOPD, the zero-training Phase 1 merge alone achieves a Macro average of 32.52, outperforming the official 600-step distilled student (30.93) without requiring any rollout compute. By passivating domains with low normalized residuals and training an isolated branch solely for Math, Phase 2 further elevates the Macro score to 33.03 with zero joint-training steps. Crucially, un-gated baselines such as TIES (Yadav et al., [31]), DARE (Yu et al., 2024), and continued joint distillation all fall below the merge, confirming that their updates can corrupt shared parameters. The same procedure transfers seamlessly to G-OPD, where RGBM surpasses the strongest trained baseline, demonstrating the generality of residual-governed consolidation.

Contributions. (1) Theoretical Bridge: we establish a connection between multi-teacher on-policy distillation and model merging, showing that the MOPD objective coincides to second order with Fisher merging ([21]), and derive an implicit-reward homomorphism that yields a label-free linear basin diagnostic. (2) Two-Phase RGBM Framework: we propose Residual-Governed Branch-and-Merge (RGBM). Phase 1 computes the zero-training merge from a residual program on unlabelled prompts and forms an energy-normalized spectrum for cross-domain comparison. Phase 2 passivates domains below the spectral gap, trains isolated branches exclusively for domains above it, and writes admitted updates back via calibrated shrinkage. (3) Empirical Validation and Protection Mechanism: across two model families (SmolLM3-3B and Qwen3-4B), Phase 1 alone matches or beats fully trained distillation, and Phase 2 advances performance further with zero joint-training steps. Crucially, un-gated write-backs fall below the merge, confirming that RGBM succeeds by shielding domains below the normalized spectral gap from cross-task interference.

Merging Is Optimal, to Second Order, as a Closed-Form Stage of Distillation

This section establishes the identification on which the paper rests: to second order, distillation and Fisher merging optimize the same objective (§ 2.2); on the task-vector subspace that objective has a closed-form minimizer (§ 2.3); and, to first order in the tangent space, combining parameters adds implicit rewards, which yields a validity test (§ 2.4). Proofs are in Appendix § 9.

Setting

Let Wref∈RPW_{\mathrm{ref}} \in\mathbb{R}^{P} denote the base weights (PP parameters) and πθ(⋅∣x)\pi_{\theta}(\cdot\mid x) the autoregressive policy of a model with weights θ∈RP\theta\in\mathbb{R}^{P} on prompt xx, with yy a completion. We consider KK homologous specialists {Wϕ1,…,WϕK}\{W_{\phi_1}, \ldots, W_{\phi_K}\}, fine-tuned from the same WrefW_{\mathrm{ref}} on datasets DdD_d via RL on each domain, with policies πWϕd\pi_{W_{\phi_d}}. The task vector of specialist dd is τd:=Wϕd−Wref\tau_d := W_{\phi_d} - W_{\mathrm{ref}}, forming T:=[τ1,…,τK]∈RP×K\mathcal{T} := [\tau_1, \ldots, \tau_K] \in\mathbb{R}^{P \times K}. Multi-teacher on-policy distillation (MOPD) trains a student θ\theta by minimizing the domain-weighted reverse-KL to the specialist policies, with mixture weights pdp_d:

JMOPD(θ):=∑d=1KpdEx∼Dd[KL⁡(πθ(⋅∣x)∥πWϕd(⋅∣x))],pd≥0,∑d=1Kpd=1.(1)J_{\mathrm{MOPD}}(\theta) := \sum_{d=1}^{K} p_d \mathbb{E}_{x\sim D_d}\left[\operatorname{KL}\left(\pi_{\theta}(\cdot\mid x) \mathbin{\|} \pi_{W_{\phi_d}}(\cdot\mid x)\right)\right], \qquad p_d \ge0,\quad\sum_{d=1}^{K}p_d=1. \tag*{(1)}

MOPD Is Fisher Merging to Second Order

Theorem 1 (Second-Order Equivalence of MOPD to Fisher Merging) Assume the specialists share a basin B⊂RP\mathcal{B} \subset\mathbb{R}^{P} (a connected neighborhood of WrefW_{\mathrm{ref}} containing every WϕdW_{\phi_d}) on which each mapping θ↦KL⁡(πθ∥πWϕd)\theta\mapsto\operatorname{KL}(\pi_{\theta}\mathbin{\|}\pi_{W_{\phi_d}}) is three times differentiable with bounded third derivative. For θ∈B\theta\in\mathcal{B}, the second-order Taylor expansion of JMOPD(θ)J_{\mathrm{MOPD}}(\theta) around the specialist checkpoints {Wϕd}d=1K\{W_{\phi_d}\}_{d=1}^{K} is:

JMOPD(θ)=12∑d=1Kpd(θ−Wϕd)TFd(θ−Wϕd)+O(max⁡d∥θ−Wϕd∥3),(2)J_{\mathrm{MOPD}}(\theta)=\frac{1}{2}\sum_{d=1}^{K}p_d(\theta-W_{\phi_d})^{\mathsf{T}}F_d(\theta-W_{\phi_d})+\mathcal{O}\left(\max_d\left\lVert\theta-W_{\phi_d}\right\rVert^3\right), \tag*{(2)}

where Fd:=Ex∼Dd,y∼πWϕd[∇θlog⁡πWϕd(y∣x)∇θlog⁡πWϕd(y∣x)T]∈RP×PF_d := \mathbb{E}_{x\sim D_d,y\sim\pi_{W_{\phi_d}}}\left[\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)^{\mathsf{T}}\right]\in\mathbb{R}^{P\times P} is the empirical Fisher matrix of specialist dd. The unconstrained global minimizer of Eq. (2) is the Fisher merge:

θFisher⋆=(∑d=1KpdFd)−1∑d=1KpdFdWϕd.(3)\theta_{\mathrm{Fisher}}^{\star}=\left(\sum_{d=1}^{K}p_dF_d\right)^{-1}\sum_{d=1}^{K}p_dF_dW_{\phi_d}. \tag*{(3)}

The proof (Appendix § 9.§ 1) relies only on the fact that at θ=Wϕd\theta=W_{\phi_d}, the KL divergence and its gradient vanish while its Hessian equals FdF_d. Equation (3) recovers the Fisher-weighted average of [21], originally derived via a Laplace approximation to the posterior. Theorem 1 establishes that this exact point is also the second-order optimum of the multi-teacher reverse-KL objective. The two paradigms therefore differ in approximation order rather than optimization goal: while JMOPDJ_{\mathrm{MOPD}} aligns student rollouts to teachers in distribution space, it corresponds in parameter space to the aggregate Fisher distance 12∑dpd∥θ−Wϕd∥Fd2\frac{1}{2}\sum_d p_d\lVert\theta-W_{\phi_d}\rVert_{F_d}^2 with ∥v∥Fd2:=v⊤Fdv\lVert v\rVert_{F_d}^2 := v^\top F_dv. Merging is thus optimal, to second order, as a closed-form stage of distillation: gradient steps taken from the Fisher merge act only on the third-order remainder, so merging should come first. What a linear combination in the task-vector subspace leaves unrecovered is a per-domain error, which the residual program of § 3.1 measures.

Subspace Fisher Projection and Task Arithmetic

Eq. (3) requires inverting a P×PP \times P Fisher matrix, which is intractable for modern LLMs. We therefore restrict the student to the KK-dimensional task subspace θ(α)=Wref+Tα\theta(\alpha)=W_{\mathrm{ref}}+\mathcal{T}\alpha, with one coefficient αd\alpha_d per specialist. Let ede_d be the dd-th standard basis vector, so θ(ed)=Wϕd\theta(e_d)=W_{\phi_d} and θ(α)−Wϕd=T(α−ed)\theta(\alpha)-W_{\phi_d}=\mathcal{T}(\alpha-e_d). That substitution compresses Eq. (2) into a K×KK \times K system:

Jsub(α):=12∑d=1Kpd(α−ed)⊤Sd(α−ed),Sd:=T⊤FdT∈RK×K,(4)J_{\mathrm{sub}}(\alpha):=\frac{1}{2}\sum_{d=1}^{K}p_d(\alpha-e_d)^\top S_d(\alpha-e_d),\qquad S_d:=\mathcal{T}^\top F_d\mathcal{T}\in\mathbb{R}^{K\times K}, \tag*{(4)}

where SdS_d is the projected Fisher of specialist dd. Setting ∇αJsub=0\nabla_\alpha J_{\mathrm{sub}}=0 gives the closed-form subspace solution:

αsub⋆=(∑d=1KpdSd)−1∑d=1KpdSded.(5)\alpha_{\mathrm{sub}}^\star=\left(\sum_{d=1}^{K}p_dS_d\right)^{-1}\sum_{d=1}^{K}p_dS_de_d. \tag*{(5)}

Remark 1 Task Arithmetic ([13]) forms Wref+TαW_{\mathrm{ref}}+\mathcal{T}\alpha with hand-set coefficients, typically the isotropic α=αˉ1\alpha=\bar{\alpha}\mathbf{1} with αˉ=1\bar{\alpha}=1. Eq. (5) identifies the coefficients that make this family the exact metric projection of the quadratic Fisher objective onto the task subspace (T⊤∇θJMOPD(θ(αsub⋆))=0\mathcal{T}^\top\nabla_\theta J_{\mathrm{MOPD}}(\theta(\alpha_{\mathrm{sub}}^\star))=0); with these coefficients it is the computable closed-form instance of Fisher merging on LLMs.

First-Order Homomorphism and a Validity Diagnostic

The subspace solution still depends on FdF_d. Forming Sd=T⊤FdTS_d=\mathcal{T}^\top F_d\mathcal{T} from parameters is impractical on LLMs, so the computations of § 3 run in the space of implicit rewards. The token-level implicit reward of specialist dd relative to WrefW_{\mathrm{ref}} is rd(x,y):=log⁡(πWϕd(y∣x)πWref(y∣x))r_d(x,y):=\log\left(\frac{\pi_{W_{\phi_d}}(y\mid x)}{\pi_{W_{\mathrm{ref}}}(y\mid x)}\right), the log-probability ratio that RL fine-tuning implicitly optimizes ([24, 2]); write r(x,y):=[r1(x,y),…,rK(x,y)]⊤∈RK\mathbf{r}(x,y):=[r_1(x,y),\ldots,r_K(x,y)]^\top\in\mathbb{R}^K for the stacked reward. Extending tangent-space analyses of merging ([22]), a first-order expansion around WrefW_{\mathrm{ref}} yields the following homomorphism.

Proposition 2 (First-Order Homomorphism) Let f(x;θ)f(x;\theta) be the logit vector at prompt-prefix xx and Z(x;θ)Z(x;\theta) its softmax partition function. A first-order expansion around WrefW_{\mathrm{ref}} gives rd(x,y)≈∇θf(x;Wref)[y]⊤τd−Δlog⁡Zd(x)r_d(x,y)\approx\nabla_\theta f(x;W_{\mathrm{ref}})[y]^\top\tau_d-\Delta\log Z_d(x), where Δlog⁡Zd(x):=log⁡Z(x;Wϕd)−log⁡Z(x;Wref)\Delta\log Z_d(x):=\log Z(x;W_{\phi_d})-\log Z(x;W_{\mathrm{ref}}). For any linear combination θ(α)=Wref+Tα\theta(\alpha)=W_{\mathrm{ref}}+\mathcal{T}\alpha:

log⁡πθ(α)(y∣x)πWref(y∣x)≈∑d=1Kαdrd(x,y)=α⊤r(x,y).(6)\log\frac{\pi_{\theta(\alpha)}(y\mid x)}{\pi_{W_{\mathrm{ref}}}(y\mid x)}\approx\sum_{d=1}^{K}\alpha_dr_d(x,y)=\alpha^\top\mathbf{r}(x,y). \tag*{(6)}

Proposition 2 provides two operational capabilities. First, the projected score T⊤∇θlog⁡πWϕd\mathcal{T}^\top\nabla_\theta\log\pi_{W_{\phi_d}} equals r\mathbf{r} up to a mean shift (Appendix § 9.§ 3), so the projected Fisher is an empirical reward covariance, Sd≈Cov⁡πWϕd(r)S_d\approx\operatorname{Cov}_{\pi_{W_{\phi_d}}}(\mathbf{r}), and the per-domain residual in § 3.1 can be computed from forward passes alone. Second, the log-ratio of the merged model, ℓα(x,y):=log⁡πθ(α)(y∣x)πWref(y∣x)\ell_\alpha(x,y):=\log\frac{\pi_{\theta(\alpha)}(y\mid x)}{\pi_{W_{\mathrm{ref}}}(y\mid x)}, should scale linearly with the superposed reward α⊤r(x,y)\alpha^\top\mathbf{r}(x,y). The unexplained variance when regressing ℓα\ell_\alpha on α⊤r\alpha^\top\mathbf{r} over sampled tokens is then a label-free test of whether the first-order superposition of Proposition 2 still holds at that checkpoint:

Err⁡lin(θ(α)):=1−R2(ℓα∼α⊤r)=min⁡a,bE[(ℓα−aα⊤r−b)2]Var⁡[ℓα].(7)\operatorname{Err}_{\mathrm{lin}}(\theta(\alpha)):=1-R^2(\ell_\alpha\sim\alpha^\top\mathbf{r})=\min_{a,b}\frac{\mathbb{E}\left[(\ell_\alpha-a\alpha^\top\mathbf{r}-b)^2\right]}{\operatorname{Var}[\ell_\alpha]}. \tag*{(7)}

We treat Err⁡lin(θ(α))≤δtol\operatorname{Err}_{\mathrm{lin}}(\theta(\alpha))\le\delta_{\mathrm{tol}} under a fixed threshold δtol\delta_{\mathrm{tol}} as the operational criterion for a valid linear basin (we use linear basin for this regime throughout). Large Err⁡lin\operatorname{Err}_{\mathrm{lin}} means the first-order fit has broken; § 3.3 uses the same test to cap how far a Phase 2 write-back may move.

RGBM: Residual-Governed Branch-and-Merge

Section 2 establishes that parameter merging should precede distillation, and that whatever the linear combination leaves unrecovered is what a later stage must address. Residual-Governed Branch-and-Merge (RGBM; Figure 2; full procedure in Algorithm 1, Appendix § 11.§ 1) operationalizes this insight into two distinct phases. Phase 1 (§ 3.1) solves a convex subspace merge and then normalizes its per-domain remainders into the domain residual spectrum. After analyzing why continuing joint distillation degrades domains with low normalized residuals (Section 3.2), Phase 2 (§ 3.3) trains isolated branches exclusively for domains marked by the spectrum as unrecovered, writing them back via calibrated shrinkage.

Two-phase RGBM architecture

Figure 2. Two-phase RGBM architecture. (1) Phase 1 scores implicit rewards on unlabelled prompts, forms per-domain second-moment matrices MdM_d, and minimizes the worst residual Ld\mathcal{L}_d to obtain θ0⋆\theta_0^\star. The heatmap is the reward correlation, not MdM_d itself. (2) The bars are the energy-normalized Open-MOPD spectrum L‾\overline{\mathcal{L}}; the gate cuts at the midpoint of its largest adjacent gap (threshold ϵ\epsilon; Eq. (10)), so only Math is active. (3) Phase 2 writes active updates back with shrinkage γ\gamma to obtain θRGBM\theta_{\mathrm{RGBM}}; if every domain is passivated, θRGBM=θ0⋆\theta_{\mathrm{RGBM}}=\theta_0^\star. See § 3.1 and § 3.3 for details.

Phase 1: Minimax Subspace Merge and the Domain Residual Spectrum

Per-domain residual from unlabelled prompts. Under the tangent-space homomorphism of Proposition 2 the projected score equals rr up to a mean shift, so Sd≈Cov⁡πWϕd(r)S_d \approx\operatorname{Cov}_{\pi_{W\phi_d}}(r) can be estimated without gradients. Let Dcalib,d\mathcal{D}_{\mathrm{calib},d} be unlabelled prompts from domain dd. Rather than center the rewards, we use the empirical second-moment Md:=Ex∼Dcalib,d, y∼πWϕd[rr⊤]∈RK×KM_d := \mathbb{E}_{x\sim\mathcal{D}_{\mathrm{calib},d},\,y\sim\pi_{W\phi_d}}[rr^\top] \in\mathbb{R}^{K\times K}, assembled from forward passes of the base and the KK specialists on completions sampled from specialist dd. MdM_d equals that covariance plus a mean-shift outer product, and therefore upper-bounds the projected Fisher residual (Appendix § 9.§ 3). We define the per-domain residual and its energy-normalized counterpart as

Ld(α):=(α−ed)⊤Md(α−ed),L‾d(α):=Ld(α)E[rd2].(8)\begin{aligned} \mathcal{L}_d(\alpha) &:= (\alpha-e_d)^\top M_d(\alpha-e_d),\\ \overline{\mathcal{L}}_d(\alpha) &:= \frac{\mathcal{L}_d(\alpha)}{\mathbb{E}[r_d^2]}. \tag*{(8)} \end{aligned}

Phase 1 optimizes Ld\mathcal{L}_d. After the merge is fixed, division by E[rd2]\mathbb{E}[r_d^2], the second moment of the specialist’s own reward (its reward energy), expresses the remainder as a fraction of domain-dd energy. Thus L‾d\overline{\mathcal{L}}_d is comparable across domains and is used for diagnosis and gating, but not for solving the merge. Because MdM_d contains the cross-domain products E[rjrk]\mathbb{E}[r_jr_k], L‾d(α⋆)\overline{\mathcal{L}}_d(\alpha^\star) is the fraction of domain-dd reward energy still unexplained after the coefficients have traded the domains off against one another. A value near zero means this joint combination already accounts for domain dd. It does not mean an isolated branch on dd has nothing left to fit. A large value means a substantial share of that energy remains once the other domains have been accommodated. § 4.3 measures the merges obtained when the same programs are solved on L‾d\overline{\mathcal{L}}_d.

Minimax merge. Eq. (5) requires domain weights pdp_d, which are rarely known and trade one domain against another, so the Fisher stage of § 2.3 is not directly computable. We remove this degree of freedom by taking the worst case over all admissible weights, the probability simplex ΔK\Delta_K: since max⁡p∈ΔK∑dpdLd(α)=max⁡dLd(α)\max_{p\in\Delta_K}\sum_d p_d\mathcal{L}_d(\alpha)=\max_d\mathcal{L}_d(\alpha), the problem becomes the convex Minimax quadratic program (QP)

α⋆=arg⁡min⁡α∈RK max⁡d∈{1,…,K}Ld(α)(9)\alpha^\star=\underset{\alpha\in\mathbb{R}^K}{\arg\min}\ \max_{d\in\{1,\ldots,K\}}\mathcal{L}_d(\alpha) \tag*{(9)}

i.e., the worst subspace residual under adversarially chosen domain weights. Its solution gives the zero-training checkpoint θ0⋆:=Wref+Tα⋆\theta_0^\star:=W_{\mathrm{ref}}+T\alpha^\star. We then report the Domain Residual Spectrum Lˉ:=[L1(α⋆),…,LK(α⋆)]\bar{\mathcal{L}}:=[\mathcal{L}_1(\alpha^\star),\ldots,\mathcal{L}_K(\alpha^\star)] for cross-domain comparison and Phase 2 allocation. The QP is KK-dimensional, so its cost is dominated by assembling the MdM_d (Appendix § 11.§ 6). For numerical stability, the QP replaces MdM_d by M~d=Md+λI\widetilde{M}_d=M_d+\lambda I; the reported spectrum uses the unregularized MdM_d. We leave α\alpha unconstrained, so a coefficient may lie outside [0,1][0,1] if that reduces the worst residual.

Why Not Simply Keep Distilling from the Merge?

The quadratic model of Theorem 1 is stationary at the Fisher merge, and approximately stationary on the subspace at θ(αsub⋆)\theta(\alpha^\star_{\mathrm{sub}}). Phase 1’s θ0⋆\theta_0^\star is not that point: it minimizes the worst residual, so ∇JMOPD(θ0⋆)\nabla J_{\mathrm{MOPD}}(\theta_0^\star) need not vanish, and continued joint distillation is not a refinement of a solved linear stage. Two mechanisms outside the quadratic regime then make it harmful. Each domain divergence fd(θ):=Ex∼Dd[KL(πθ∥πWϕd)]f_d(\theta):=\mathbb{E}_{x\sim\mathcal{D}_d}[\mathrm{KL}(\pi_\theta\mathbin{\|}\pi_{W_{\phi_d}})] has ∇θfd(θ0⋆)≠0\nabla_\theta f_d(\theta_0^\star)\ne0, so mini-batch steps on shared parameters pull in conflicting directions, including components off the task subspace. A near-zero normalized residual Lˉd\bar{\mathcal{L}}_d adds a second problem, and it is a joint one. An isolated branch can still reduce domain dd’s own loss, because Ld\mathcal{L}_d is the gap left after the cross-domain tradeoff, not a claim that domain dd is saturated. Writing that branch into shared weights moves every domain: the step is spent where the joint merge has already closed the gap, so the three-domain loss can worsen even while domain dd itself improves. Joint distillation from θ0⋆\theta_0^\star should therefore degrade those low-residual domains on the shared model. A token-level KL penalty to πθ0⋆\pi_{\theta_0^\star} cannot prevent this, because the collision is in the weights. § 4.4 checks both predictions.

Phase 2: Residual Gate, Isolated Branches, and Shrinkage Re-merge

RGBM therefore never applies a joint gradient to the merge; it partitions domains by the normalized spectrum Lˉ\bar{\mathcal{L}} and treats the two groups differently.

Gate. Sort Lˉ(1)≥⋯≥Lˉ(K)\bar{\mathcal{L}}_{(1)}\ge\cdots\ge\bar{\mathcal{L}}_{(K)}, append a floor Lˉ(K+1):=0\bar{\mathcal{L}}_{(K+1)}:=0, and cut at the midpoint of the largest adjacent gap,

k⋆=arg⁡max⁡i=1,…,K(Lˉ(i)−Lˉ(i+1)),ϵ:=12(Lˉ(k⋆)+Lˉ(k⋆+1)).(10)k^\star=\underset{i=1,\ldots,K}{\arg\max}\left(\bar{\mathcal{L}}_{(i)}-\bar{\mathcal{L}}_{(i+1)}\right),\qquad \epsilon:=\frac{1}{2}\left(\bar{\mathcal{L}}_{(k^\star)}+\bar{\mathcal{L}}_{(k^\star+1)}\right). \tag*{(10)}

Domain dd is active iff Lˉd>ϵ\bar{\mathcal{L}}_d>\epsilon, otherwise passivated; the floor makes every domain active when normalized residuals cluster, and any ϵ\epsilon inside the same gap yields the same allocation (Appendix § 11.§ 4).

Passivated and active domains. Passivated domains (Lˉd≤ϵ\bar{\mathcal{L}}_d\le\epsilon) receive no shared write-back (Δd=0\Delta_d=0). The joint merge already accounts for them, which does not preclude an isolated gain on that domain alone. Each active domain trains an isolated branch θd←θ0⋆\theta_d\leftarrow\theta_0^\star on its native data Dd\mathcal{D}_d with no gradient sharing, giving an update Δd=θd(end)−θ0⋆\Delta_d=\theta_d^{(\mathrm{end})}-\theta_0^\star; isolation guarantees that no gradient computed on domain dd touches a passivated domain during training. Whether the update can be written back without damaging them is decided next.

Shrinkage re-merge. The sole Phase 2 write-back is a shrunken sum of the active updates,

θRGBM=θ0⋆+∑d:Lˉd>ϵγd⋆Δd,γd⋆=max⁡{γ∈Γ∣Err⁡in(θ0⋆+γΔd)≤δtol}.(11)\theta_{\mathrm{RGBM}}=\theta_0^\star+\sum_{d:\bar{\mathcal{L}}_d>\epsilon}\gamma_d^\star\Delta_d,\qquad \gamma_d^\star=\max\left\{\gamma\in\Gamma\mid\operatorname{Err}_{\mathrm{in}}(\theta_0^\star+\gamma\Delta_d)\le\delta_{\mathrm{tol}}\right\}. \tag*{(11)}

where Γ⊂(0,1]\Gamma\subset(0,1] is a fixed discrete grid and Err⁡in\operatorname{Err}_{\mathrm{in}} is evaluated on the same unlabelled calibration prompts Dcalib=⋃dDcalib,d\mathcal{D}_{\mathrm{calib}}=\bigcup_d\mathcal{D}_{\mathrm{calib},d} used for MdM_d. γd⋆\gamma_d^\star is the largest grid value at which that single write-back still passes the linearity test in Proposition 4 (Appendix § 9.§ 4, so the write-back keeps the merged model’s log-ratio within δtol\delta_{\mathrm{tol}} of the Phase 1 superposition α⋆Tr\alpha^\star{}^\mathsf{T}r, which is held fixed as the predictor; Proposition 4 (Appendix § 9.§ 4) bounds the first-order remainder at θ0⋆+γΔd\theta_0^\star+\gamma\Delta_d by a quantity that is non-decreasing in γ\gamma. When several branches are active, evaluation uses their shrunken sum. If the gate leaves no active domain, θRGBM=θ0⋆\theta_{\mathrm{RGBM}}=\theta_0^\star. The whole procedure has three fixed constants (λ,Γ,δtol)(\lambda,\Gamma,\delta_{\mathrm{tol}}) and no labelled data; their values and the resulting allocations on both suites are given in § 4.1.

Experiments

Setup

Main suite (Open-MOPD). Our primary testbed is the Open-MOPD specialist suite [8]: SmolLM3-3B [3] with K=3K = 3 homologous RL specialists (the usual MOPD domain count; [8, 28, 4]), fine-tuned from the mixed-SFT checkpoint MixSFT as WrefW_{\mathrm{ref}}. The specialists are Math, Code, and Instruction Following (IF), scored on AIME 2024/2025, LiveCodeBench v5/v6 [14], and IFEval/IFBench [23] (Zhou et al. 2023). We report Macro =13(Math+Code+IF)= \frac{1}{3}(\mathrm{Math} + \mathrm{Code} + \mathrm{IF}), where IF is the mean of IFEval and IFBench, and re-evaluate every system under one protocol. The italic RouteRL row is the routing oracle of [8] that serves all KK specialists; Multi-Teacher OPD is their released 600-step student. Every Phase 2 system starts from the same Phase 1 merge θ0⋆\theta_0^\star. TIES [31] and DARE (Yu et al., 2024) sparsify and sum the isolated on-policy distillation (OPD) branch updates Δd\Delta_d—not the specialist task vectors—and RGBM uses those same branches, differing only in what is written back (Appendix Table 8). Per-benchmark numbers and hyperparameters are in Appendices § 10.§ 2 and § 11.§ 5.

Secondary suite (G-OPD). To test transfer across model families we apply the same procedure, unchanged, to the G-OPD suite [35]: Qwen3-4B [30] with Math and Code specialists (intermediate checkpoints of Table 2 in Yang et al., [35]; 500/300 steps) on their 7 official benchmarks. Supervised fine-tuning (SFT), OPD, and the extended recipe ExOPD are from Yang et al. [35]; merges are evaluated by us under the official protocol. The italic Domain-Matched Teacher row is that specialist pair, an oracle rather than a single generalist. Per-benchmark results and the fully trained pair are in Appendix § 10.§ 7.

Programα\alphaMathCodeIFMacro
α=1/3\alpha= 1/30.333/0.333/0.33320.8119.1544.7728.24
A scalar L\mathcal{L}0.626/0.626/0.62624.9723.2847.6431.96
A vector L\mathcal{L}0.527/0.609/0.82124.1123.5947.9931.90
C scalar L\mathcal{L}0.579/0.579/0.57924.9723.4945.1031.19
C vector L\mathcal{L} (θ0∗\theta_0^*)0.597/0.620/0.80023.8624.0949.6032.52
A scalar L‾\overline{\mathcal{L}}0.501/0.501/0.50124.3823.3644.8630.86
A vector L‾\overline{\mathcal{L}}0.840/0.260/0.84124.1919.5748.0030.59
C scalar L‾\overline{\mathcal{L}}0.477/0.477/0.47721.4119.7645.7928.99
C vector L‾\overline{\mathcal{L}}0.724/0.426/0.63024.2220.7347.0230.66

Table 2. Open-MOPD coefficient grid. A minimizes the mean residual (uniform pdp_d, Eq. (5)); C minimizes the worst residual (Eq. (9)). Scalar shares one coefficient; vector leaves α\alpha free. Scores for the four L‾\overline{\mathcal{L}} rows and for α=1/3\alpha= 1/3 are Table 4; the four L‾\overline{\mathcal{L}} rows use the same protocol. Coefficients are Math/Code/IF.

RGBM constants and resulting allocations. We use λ=10−4\lambda= 10^{-4}, Γ={0.125,0.25,0.5,0.75}\Gamma= \{0.125, 0.25, 0.5, 0.75\}, and δtol=0.10\delta_{\mathrm{tol}} = 0.10 on both suites. On Open-MOPD the energy-normalized spectrum is L=(0.128,0.052,0.015)\mathcal{L} = (0.128, 0.052, 0.015) for (Math, Code, IF); the largest gap is between Math and Code, so ϵ=0.090\epsilon= 0.090, only Math is active, and Eq. (11) selects γmath⋆=0.125\gamma_{\mathrm{math}}^\star= 0.125, the largest γ∈Γ\gamma\in\Gamma with Errlin≤δtol\mathrm{Err}_{\mathrm{lin}} \le\delta_{\mathrm{tol}} (Errlin=0.0701\mathrm{Err}_{\mathrm{lin}} = 0.0701; Appendix Table 10). On G-OPD, α⋆=(0.675,0.666)\alpha^\star= (0.675, 0.666) and L=(0.094,0.090)\mathcal{L} = (0.094, 0.090) cluster, so the largest gap is down to the floor, ϵ=0.045\epsilon= 0.045, both domains are active, and the same rule gives γ⋆=0.5\gamma^\star= 0.5. Hereafter, low- and high-residual domains refer to this normalized spectrum. Assembling the MdM_d costs <15< 15 minutes on 8 GPUs (<2< 2 GPU·h) of unlabelled scoring and the QP solves in <10< 10 ms.

Merge Before You Distill: Zero-Training Merge vs. Trained Distillation

The closed-form merge matches or beats trained distillation. On Open-MOPD (Table 1) the Minimax merge θ0⋆\theta_0^\star reaches 32.52 Macro at zero gradient cost, above Multi-Teacher OPD, the released 600-step student (30.93; 144 GPU·h), and above isotropic Task Arithmetic (31.31). On G-OPD (Table 3) the same procedure gives 53.34, above SFT, OPD, and Task Arithmetic and within 0.45 of the trained ExOPD recipe (equal-weight averaging collapses to 36.75: the gain is in the coefficients). The 31.90 row is the subspace Fisher solution of Eq. (5) under uniform weights pd=1/Kp_d = 1/K, i.e., the minimizer of ∑dpdLd(α)\sum_d p_d \mathcal{L}_d(\alpha) with the Phase 1 second moments MdM_d in place of SdS_d. Eq. (5) requires the domain weights pdp_d and those weights are not known; the uniform choice is a default, not an estimate. Minimax, which minimizes the worst residual instead, is 0.62 higher and lifts IF from 47.99 to 49.60—when no domain may be sacrificed, the adversarial weighting is the operational choice.

MethodMathCodeIFMacro
RouteRL (Oracle Ensemble)27.2426.2753.5235.68
Multi-Teacher OPD24.5321.1847.1230.93
Task Arithmetic24.3522.8146.7731.31
Subspace Fisher Merge24.1123.5947.9931.90
RGBM Phase 1 (Ours, θ0⋆\theta_0^\star)23.8624.0949.6032.52
Uniform Distillation (from θ0⋆\theta_0^\star)23.9721.3647.4230.91
TIES (all branches)24.6121.5646.7231.42
DARE (all branches)24.5822.4047.2831.42
Gated Math (γ=1.0\gamma= 1.0)25.0322.5147.8731.80
Gated Math (γ=0.5\gamma= 0.5)25.4223.4848.0832.32
RGBM Phase 2 (Ours)23.9324.6350.5233.03

Table 1. Open-MOPD (SmolLM3-3B; accuracy). Macro: mean of Math, Code, and IF. Bold: best deployable model. Italic: routing oracle. Gated Math: admitted branch only, at the listed γ\gamma.

MethodMathCodeAvg
Domain-Matched Teacher46.0061.1752.50
SFT44.3360.8051.39
OPD46.3860.5752.46
ExOPD47.6561.9753.79
Task Arithmetic43.0560.6550.59
Equal-Weight Average22.5556.5736.75
RGBM Phase 1 (Ours)46.1762.8953.34
RGBM Phase 2 (Ours)47.5562.9354.14

Table 3. G-OPD (Qwen3-4B). Math / Code: mean of the 4 math / 3 code tasks. Avg: 7-task mean. Italic: teacher oracle.

RGBM improves on the merge on both suites. Writing back only the gated Math branch raises Open-MOPD to 33.03 (+0.51 over θ0∗\theta_0^*) with zero extra joint-training steps; on G-OPD both domains are active, and writing back both isolated branches gives 54.14, above ExOPD.

The Quadratic That Is Minimized

Phase 1 minimizes Ld\mathcal{L}_d. This quadratic upper-bounds the domain term (α−ed)TSd(α−ed)(\alpha-e_d)^\mathsf{T}S_d(\alpha-e_d) of JsubJ_{\mathrm{sub}} (Eq. (4); Appendix § 9.§ 3), since MdM_d equals Sd≃Cov⁡(r)S_d \simeq\operatorname{Cov}(r) plus a mean-shift outer product. The uniform row above is that bound, with MdM_d in place of SdS_d. The factor 1/E[rd2]1/\mathbb{E}[r_d^2] is applied after α\alpha is chosen, which rescales the reported spectrum and leaves the minimizer in place. Solving the same programs on Md/E[rd2]M_d/\mathbb{E}[r_d^2] weights domain dd by pd/E[rd2]p_d/\mathbb{E}[r_d^2]. On the Open-MOPD calibration the energies are E[rd2]=(0.180,0.692,2.40)\mathbb{E}[r_d^2] = (0.180, 0.692, 2.40) for (Math, Code, IF), so that program down-weights Code relative to Math by a factor of 3.8. Task arithmetic already treats the coefficient as a control on accuracy and picks it on held-out labels (Ilharco et al., [13]). The solve below uses no labels.

Every program solved on L\mathcal{L} lands in Macro 31.19 ∼\sim 32.52. Every program solved on L‾\overline{\mathcal{L}} lands in 28.99 ∼\sim 30.86, so the best normalized merge (30.86) sits below the worst raw merge (31.19). The reweighting shows up on Code. The uniform vector on L\mathcal{L} keeps αcode=0.609\alpha_{\mathrm{code}} = 0.609 and Code at 23.59; the same program on L‾\overline{\mathcal{L}} sets αcode=0.260\alpha_{\mathrm{code}} = 0.260 and Code falls to 19.57. The published minimax vector keeps αcode=0.620\alpha_{\mathrm{code}} = 0.620 and Code at 24.09. Equal-weight α=1/3\alpha= 1/3, which solves neither program, is lower still (28.24).

With MdM_d held fixed, Ld(α)\mathcal{L}_d(\alpha) is a function of the coefficients alone. Figure 3 orders the nine merges by score and overlays that raw residual, min-max scaled onto the score range. On Code, αcode≥0.50\alpha_{\mathrm{code}} \ge0.50 keeps Code at 23.28 ∼\sim 24.09 and αcode≤0.48\alpha_{\mathrm{code}} \le0.48 puts it at 19.15 ∼\sim 20.73 (Pearson +0.89 between αcode\alpha_{\mathrm{code}} and Code). On IF, αif\alpha_{\mathrm{if}} and IF have Pearson +0.87, from 44.77 at αif=0.333\alpha_{\mathrm{if}} = 0.333 to 49.60 at the published αif=0.800\alpha_{\mathrm{if}} = 0.800. On Math, αmath\alpha_{\mathrm{math}} separates the floor: the two vectors with αmath<0.5\alpha_{\mathrm{math}} < 0.5 score 20.81 and 21.41, and every vector with αmath≥0.5\alpha_{\mathrm{math}} \ge0.5 stays in 23.86 ∼\sim 24.97. The coefficients track Code and IF, and they track whether Math has cleared that floor.

Domain score against the raw residual for Math, Code, IF, and Avg

Figure 3. Domain score against the raw residual (Open-MOPD). Each panel orders the nine merges of Table 2 by that panel’s score, so the score curve rises by construction. Domain-L\mathcal{L} is min-max scaled onto that score range. A/C: uniform and minimax. sc/vec: shared scalar and free vector. L / Lb: the program minimizes L\mathcal{L} or Ld/E[rd2]\mathcal{L}_d/\mathbb{E}[r_d^2]. lin: α=1/3\alpha= 1/3. The energy-normalized spectrum is used only for the Phase 2 gate in § 4.1.

Why the Write-Back Must Be Gated: Joint Distillation from the Merge

The obvious second stage is to keep distilling θ0∗\theta_0^* jointly on all domains; Section 3.2 argued that this should erode the low-residual domains. Figure 4 tracks three joint objectives initialized from θ0∗\theta_0^*: Uniform (equal-weight reverse-KL), Outcome-Gated (matching the teacher only on failed rollouts), and Trust-Region (outcome-gated reverse-KL with a KL penalty β=2.5\beta= 2.5 toward πθ0∗\pi_{\theta_0^*}). Appendix § 11.§ 2 extends this analysis to five distinct objectives. (1) Low-residual domains deteriorate. At the last listed checkpoint of each recipe trained to step 20 or 50, Code drops by 1.1 ∼\sim 2.9 points below θ0∗\theta_0^*, IF falls behind the merge, and Macro stays in 30.5 ∼\sim 32.3. Joint distillation therefore undoes what Phase 1 had already recovered. (2) High-residual Math improves only marginally. Uniform and Outcome-Gated trajectory means are 23.97 and 24.69 (Appendix § 11.§ 2), against 23.86 at θ0∗\theta_0^*. Continued joint steps do not convert the Math residual into a gain that offsets the drop on Code and IF. (3) Interference is parameter-level. Even when the Code reverse-KL is held below 10−310^{-3} nat/token, Code still degrades. The student can stay close to the Code teacher in distribution space, so the damage is coming from shared-weight updates, not from a Code-teacher mismatch. (4) A token-level KL penalty cannot prevent collapse. The Trust-Region anchor bounds estimated policy drift to KL⁡(πθ∥πθ0∗)≈1.4×10−3\operatorname{KL}(\pi_{\theta}\Vert\pi_{\theta_0^*}) \approx1.4 \times10^{-3} nat/token, yet Macro decays to 31.25 by step 20 (Code drops 24.09 →\to 21.18). A scalar KL only restricts isotropic divergence; it does not block interference along the anisotropic directions captured by Lˉd\bar{\mathcal{L}}_d.

Macro trajectories under joint fine-tuning and uniform distillation per domain

Figure 4. Joint distillation from θ0∗\theta_0^* (Open-MOPD, ≤50\le50 steps). (a) Macro under three joint objectives; dashed and dotted lines are RGBM Phase 2 and θ0∗\theta_0^*. (b) Uniform Distillation: Math stays near the merge before a late drop; Code and IFBench fall. Dashed lines mark RGBM Phase 2 on those axes.

The Residual Spectrum Identifies Which Updates Can Be Written Back Safely

The normalized spectrum L‾=(0.128,0.052,0.015)\overline{\mathcal{L}} = (0.128, 0.052, 0.015) is computed from the Phase 1 solution before any Phase 2 update, and the gate it induces admits only the Math branch. We test this decision against the alternatives it rejects. Writing back the admitted Math update (γ∗=0.125\gamma^* = 0.125) raises Macro to 33.03. Writing back the rejected IF update at a mid-grid factor (θ0∗+0.5 Δif\theta_0^* + 0.5\,\Delta_{i\mathrm{f}}; IF has no γif∗\gamma_{\mathrm{if}}^*) lowers Macro to 31.10 and Code to 20.70 (Appendix § 11.§ 4). Writing back all three, as un-gated TIES and DARE do, gives 30.96 and 31.42, both below the merge. On G-OPD the normalized spectrum clusters, the gate keeps both domains, and writing back both raises the average by 0.80.

We stress what this does and does not show. Isolated branches raise their own domain under all three objectives (Appendix Table 11): IF by +0.48 ∼\sim +1.63, Code by +0.43 ∼\sim +1.31, and Math by +0.58 ∼\sim +1.83. A low residual therefore does not mean a domain has nothing left to learn on its own: L‾d\overline{L}_d is the coupled gap at α⋆\alpha^\star, after all domains have been traded off together. A write-back is one shared update, so it moves domains other than the branch that was trained, and it is scored on Math, Code, and IF together. On that joint score the rejected IF write-back lowers Macro, while the admitted Math write-back raises it only at γ⋆=0.125\gamma^\star= 0.125. The spectrum chooses which branch may enter the shared model. The shrinkage grid chooses how far that branch may move it.

Ablations: Gate, Isolation, and Shrinkage

The lower half of Table 1 isolates each RGBM component (full γ\gamma grid in Appendix § 11.§ 3). Uniform Distillation continues joint training; TIES and DARE write back every isolated branch; Gated Math writes back only the admitted branch, with or without shrinkage. Without the gate, every write-back falls below the merge, whether updates are trained jointly or in isolation, dense or sparsified, because the Code and IF updates damage shared weights. With the gate but γ=1\gamma= 1, the Math branch alone still costs 0.72 Macro; intermediate γ\gamma trades Math against Code and IF (Appendix § 11.§ 3). The unlabelled rule selects γ⋆=0.125\gamma^\star= 0.125, the only setting above the merge.

Where does the gain come from? The +0.51 Macro of RGBM is not a Math gain: at γ⋆=0.125\gamma^\star= 0.125, Math moves only +0.07. It is carried by Code (+0.54) and IF (+0.92), the two passivated domains, which the small Math write-back moves through the shared weights. Protection explains a different comparison: relative to the merge, the un-gated write-backs lose 1.1 ∼\sim 1.6 Macro, and the gate removes that loss. Larger γ\gamma improves Math (+1.56 at γ=0.5\gamma= 0.5) only by sacrificing Code and IF. Across all checkpoints of Appendix § 11.§ 2, the ordering un-gated write-backs < merge < RGBM holds.

Linearity Diagnostics and the Limits of the Theory

Appendix Figure 5 is a sanity check of Proposition 2 on the isotropic family θ(α)=Wref+α∑dτd\theta(\alpha) = W_{\mathrm{ref}} + \alpha\sum_d \tau_d of Remark 1, not on the Minimax vector α⋆\alpha^\star. Regressing the merged log-ratio on the superposed reward over 38M tokens yields R2=0.935R^2 = 0.935 at α=0.75\alpha= 0.75. At α=1.50\alpha= 1.50 the fit breaks (R2=0.557R^2 = 0.557), the cross-domain interference ratio IcrossI_{\mathrm{cross}} (the share of merged log-ratio energy not explained by the in-domain reward) rises to 0.775, and Macro falls to 22.48—a failure Errlin\mathrm{Err}_{\mathrm{lin}} flags without labels. Branch updates are not small relative to the merge (∥Δd∥/∥Tα⋆∥≈22%∼29%\lVert\Delta_d\rVert/\lVert T\alpha^\star\rVert\approx22\% \sim29\%), and the unshrunk Math write-back already breaks the fit (Errlin=0.72\mathrm{Err}_{\mathrm{lin}} = 0.72 at γ=1\gamma= 1); shrinking to γ⋆=0.125\gamma^\star= 0.125 brings it back to 0.070 ≤δtol\leq\delta_{\mathrm{tol}} (Appendix Table 10).

Related Work

Parameter-space merging. Training-free fusion spans Task Arithmetic ([12, 13]), sparsification ([31, 6] (Yu et al. 2024)), routing ([20]), and second-order schemes such as Fisher averaging ([21, 25]) and RegMean ([15]). Whereas AdaMerging ([33]) optimizes black-box weights, Phase 1 obtains merge coefficients from a residual program on unlabelled prompts and normalizes the resulting per-domain residuals for diagnosis; Remark 1 identifies the related subspace Fisher projection of the quadratic. Extending tangent-space analyses of merging ([22]), Proposition 2 establishes reward-space linearity, and Errlin\mathrm{Err}_{\mathrm{lin}} diagnoses departures from it.

On-policy and multi-teacher distillation. Beyond classical distillation ([11, 16] (You et al. 2017)), on-policy distillation trains on student rollouts ([9, 1, 10, 5, 19, 32, 18] (Zhong et al. 2026)); multi-teacher variants ([8, 28, 35, 4]) consolidate specialists via joint training from base weights. We show that, to second order, this objective has an optimal closed-form stage, and we gate subsequent updates by the normalized residual spectrum.

Conclusion

We bridge parameter merging and multi-teacher distillation by showing that, to second order, the MOPD objective coincides with Fisher merging. Parameter merging is thus optimal, to second order, as a closed-form stage of distillation. For tractability, we restrict that merge to the task-vector subspace and, because the domain weights are unknown, replace them by their worst case: a residual program on unlabelled prompts yields the task-vector weights, and post-hoc reward-energy normalization yields the domain residual spectrum. Empirically, our zero-training merge beats a distilled student on Open-MOPD and approaches the strongest trained baseline on G-OPD, while residual-gated RGBM advances performance further. Un-gated write-backs degrade below the merge, confirming that post-merge gains are localized, with the normalized spectrum identifying the updates to test for safe write-back. Limitations are discussed in Appendix § 12.

LLM Usage

In accordance with conference guidelines, we disclose the use of large language models (LLMs) in the preparation of this manuscript. Mainstream LLMs were used exclusively for grammar correction, stylistic polishing, and editorial refinement to enhance the readability and fluency of the text.

References

This appendix is organized as follows. Appendix A proves Theorem 1 and Proposition 2, derives the residual, its scale-normalized spectrum, and Minimax convexity, and bounds the linearization error of shrinkage re-merging. Appendix B reports the six-benchmark Phase 1 breakdown, the six-benchmark scores of the normalized-objective merges, the empirical reward correlation, the K=2K = 2 pairwise merges, generation lengths, and the full G-OPD suite. Appendix C details the joint-distillation recipes, the post-hoc γ\gamma write-backs, the ϵ\epsilon allocation cuts, training infrastructure, the cost of assembling MdM_d, and the Open-MOPD RGBM Phase 2 specification. Appendix D states the scope of the linear-basin guarantee, the specialist count, and the source of the Open-MOPD margin.

Mathematical Proofs and Derivations

This section records the proofs deferred from § 2: the second-order equivalence of multi-teacher distillation to Fisher merging (Appendix A.1), the first-order reward homomorphism (Appendix A.2), the second-moment residual, its scale-normalized spectrum, and uniqueness of the Minimax QP (Appendix A.3), and the shrinkage bound on the linearization error of a re-merged branch (Appendix A.4).

Proof of Theorem 1 (Second-Order Equivalence to Fisher Merging)

Let Xd\mathcal{X}_d denote the prompt space for domain dd, and Y\mathcal{Y} denote the token output space. For a given prompt x∼Ddx \sim\mathcal{D}_d, the reverse Kullback-Leibler (KL) divergence between student policy πθ(⋅∣x)\pi_\theta(\cdot\mid x) and teacher policy πWϕd(⋅∣x)\pi_{W_{\phi_d}}(\cdot\mid x) is:

DKL(πθ(⋅∣x)∥πWϕd(⋅∣x))=∑y∈Yπθ(y∣x)log⁡πθ(y∣x)πWϕd(y∣x).(12)D_{\mathrm{KL}}\left(\pi_\theta(\cdot\mid x) \mathrel{\|} \pi_{W_{\phi_d}}(\cdot\mid x)\right) = \sum_{y \in\mathcal{Y}} \pi_\theta(y \mid x) \log\frac{\pi_\theta(y \mid x)}{\pi_{W_{\phi_d}}(y \mid x)}. \tag*{(12)}

We define the expected domain divergence function fd:RP→Rf_d : \mathbb{R}^{P} \to\mathbb{R} as:

fd(θ):=Ex∼Dd[DKL(πθ(⋅∣x)∥πWϕd(⋅∣x))].(13)f_d(\theta) := \mathbb{E}_{x \sim\mathcal{D}_d}\left[D_{\mathrm{KL}}\left(\pi_\theta(\cdot\mid x) \mathrel{\|} \pi_{W_{\phi_d}}(\cdot\mid x)\right)\right]. \tag*{(13)}

We perform a multivariate Taylor expansion of fd(θ)f_d(\theta) around the expert weight point θ=Wϕd\theta= W_{\phi_d}:

fd(θ)=fd(Wϕd)+∇θfd(Wϕd)T(θ−Wϕd)+12(θ−Wϕd)T∇θ2fd(Wϕd)(θ−Wϕd)+O(∥θ−Wϕd∥3).(14)f_d(\theta) = f_d(W_{\phi_d}) + \nabla_\theta f_d(W_{\phi_d})^{\mathsf{T}}(\theta- W_{\phi_d}) + \frac{1}{2}(\theta- W_{\phi_d})^{\mathsf{T}}\nabla_\theta^2 f_d(W_{\phi_d})(\theta- W_{\phi_d}) + O(\lVert\theta- W_{\phi_d}\rVert^3). \tag*{(14)}

We evaluate each term:

  1. Zeroth-order term: At θ=Wϕd\theta= W_{\phi_d}, the distributions are identical:

fd(Wϕd)=Ex∼Dd[∑yπWϕd(y∣x)log⁡πWϕd(y∣x)πWϕd(y∣x)]=0.(15)f_d(W_{\phi_d}) = \mathbb{E}_{x \sim\mathcal{D}_d}\left[\sum_y \pi_{W_{\phi_d}}(y \mid x) \log\frac{\pi_{W_{\phi_d}}(y \mid x)}{\pi_{W_{\phi_d}}(y \mid x)}\right] = 0. \tag*{(15)}
  1. First-order gradient term: Taking the gradient of the reverse KL with respect to θ\theta:

∇θDKL(πθ(⋅∣x)∥πWϕd(⋅∣x))=∇θ∑yπθ(y∣x)log⁡πθ(y∣x)−∇θ∑yπθ(y∣x)log⁡πWϕd(y∣x)=∑y∇θπθ(y∣x)(log⁡πθ(y∣x)+1−log⁡πWϕd(y∣x))=∑y∇θπθ(y∣x)log⁡πθ(y∣x)πWϕd(y∣x)+∑y∇θπθ(y∣x).(16)\begin{aligned} \nabla_\theta D_{\mathrm{KL}}\left(\pi_\theta(\cdot\mid x) \mathrel{\|} \pi_{W_{\phi_d}}(\cdot\mid x)\right) &= \nabla_\theta\sum_y \pi_\theta(y \mid x)\log\pi_\theta(y \mid x) \\ &\quad- \nabla_\theta\sum_y \pi_\theta(y \mid x)\log\pi_{W_{\phi_d}}(y \mid x) \\ &= \sum_y \nabla_\theta\pi_\theta(y \mid x)\left(\log\pi_\theta(y \mid x) + 1 - \log\pi_{W_{\phi_d}}(y \mid x)\right) \\ &= \sum_y \nabla_\theta\pi_\theta(y \mid x)\log\frac{\pi_\theta(y \mid x)}{\pi_{W_{\phi_d}}(y \mid x)} + \sum_y \nabla_\theta\pi_\theta(y \mid x). \tag*{(16)} \end{aligned}

Evaluating at θ=Wϕd\theta= W_{\phi_d}: the log ratio log⁡(πWϕd/πWϕd)=log⁡1=0\log(\pi_{W_{\phi_d}}/\pi_{W_{\phi_d}}) = \log1 = 0. Furthermore, because ∑yπθ(y∣x)=1\sum_y \pi_\theta(y \mid x) = 1 for all θ\theta, the sum of gradients vanishes: ∑y∇θπθ(y∣x)=∇θ∑yπθ(y∣x)=∇θ1=0\sum_y \nabla_\theta\pi_\theta(y \mid x) = \nabla_\theta\sum_y \pi_\theta(y \mid x) = \nabla_\theta1 = 0. Therefore:

∇θfd(Wϕd)=0.(17)\nabla_\theta f_d(W_{\phi_d}) = 0. \tag*{(17)}
  1. Second-order Hessian term: Differentiating the gradient expression again with respect to θ\theta:

∇θ2DKL(πθ(⋅∣x)∥πWϕd(⋅∣x))=∑y∇θ2πθ(y∣x)(log⁡πθ(y∣x)πWϕd(y∣x)+1)+∑y∇θπθ(y∣x)∇θlog⁡πθ(y∣x)T.(18)\nabla_{\theta}^{2}D_{\mathrm{KL}}\left(\pi_{\theta}(\cdot\mid x)\mathbin{\|}\pi_{W_{\phi_d}}(\cdot\mid x)\right)=\sum_y\nabla_{\theta}^{2}\pi_{\theta}(y\mid x)\left(\log\frac{\pi_{\theta}(y\mid x)}{\pi_{W_{\phi_d}}(y\mid x)}+1\right)+\sum_y\nabla_{\theta}\pi_{\theta}(y\mid x)\nabla_{\theta}\log\pi_{\theta}(y\mid x)^{\mathsf T}. \tag*{(18)}

Evaluating at θ=Wϕd\theta=W_{\phi_d}: the log ratio vanishes, leaving ∑y∇θ2πWϕd(y∣x)=∇θ2(∑yπWϕd(y∣x))=0\sum_y\nabla_{\theta}^{2}\pi_{W_{\phi_d}}(y\mid x)=\nabla_{\theta}^{2}(\sum_y\pi_{W_{\phi_d}}(y\mid x))=0. Using the identity ∇θπθ=πθ∇θlog⁡πθ\nabla_{\theta}\pi_{\theta}=\pi_{\theta}\nabla_{\theta}\log\pi_{\theta}, the remaining term becomes:

∇θ2fd(Wϕd)=Ex∼Dd[∑yπWϕd(y∣x)∇θlog⁡πWϕd(y∣x)∇θlog⁡πWϕd(y∣x)T]=Ex∼Dd,y∼πWϕd(⋅∣x)[∇θlog⁡πWϕd(y∣x)∇θlog⁡πWϕd(y∣x)T]=:Fd.(19)\begin{aligned} \nabla_{\theta}^{2}f_d(W_{\phi_d})=\mathbb{E}_{x\sim\mathcal{D}_d}\left[\sum_y\pi_{W_{\phi_d}}(y\mid x)\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)^{\mathsf T}\right] \\ &=\mathbb{E}_{x\sim\mathcal{D}_d,y\sim\pi_{W_{\phi_d}}(\cdot\mid x)}\left[\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)\nabla_{\theta}\log\pi_{W_{\phi_d}}(y\mid x)^{\mathsf T}\right]=:F_d. \tag*{(19)} \end{aligned}

which is precisely the empirical Fisher Information Matrix Fd∈RP×PF_d\in\mathbb{R}^{P\times P} of expert policy πWϕd\pi_{W_{\phi_d}}.

Substituting these three results back into the multi-teacher objective JMOPD(θ)=∑d=1Kpdfd(θ)J_{\mathrm{MOPD}}(\theta)=\sum_{d=1}^{K}p_df_d(\theta) yields:

JMOPD(θ)=12∑d=1Kpd(θ−Wϕd)TFd(θ−Wϕd)+O(max⁡d∥θ−Wϕd∥3).(20)J_{\mathrm{MOPD}}(\theta)=\frac{1}{2}\sum_{d=1}^{K}p_d(\theta-W_{\phi_d})^{\mathsf T}F_d(\theta-W_{\phi_d})+O\left(\max_d\left\|\theta-W_{\phi_d}\right\|^3\right). \tag*{(20)}

Taking the gradient with respect to θ\theta and setting it to zero:

∇θJMOPD(θ)=∑d=1KpdFd(θ−Wϕd)=0  ⟹  (∑d=1KpdFd)θ=∑d=1KpdFdWϕd.(21)\nabla_{\theta}J_{\mathrm{MOPD}}(\theta)=\sum_{d=1}^{K}p_dF_d(\theta-W_{\phi_d})=0\implies\left(\sum_{d=1}^{K}p_dF_d\right)\theta=\sum_{d=1}^{K}p_dF_dW_{\phi_d}. \tag*{(21)}

Assuming ∑d=1KpdFd\sum_{d=1}^{K}p_dF_d is non-singular, we obtain the global unconstrained Fisher merge:

θFisher∗=(∑d=1KpdFd)−1∑d=1KpdFdWϕd.(22)\theta_{\mathrm{Fisher}}^{*}=\left(\sum_{d=1}^{K}p_dF_d\right)^{-1}\sum_{d=1}^{K}p_dF_dW_{\phi_d}. \tag*{(22)}

This completes the proof of Theorem 1. ■

Proof of Proposition 2 (First-Order Homomorphism)

Let f(x;θ)∈R∣V∣f(x;\theta)\in\mathbb{R}^{|V|} denote the logit vector at a generation step under prompt-prefix xx. The token distribution is given by the softmax function:

πθ(y∣x)=exp⁡(f(x;θ)[y])∑y′∈Vexp⁡(f(x;θ)[y′])=exp⁡(f(x;θ)[y])Z(x;θ).(23)\pi_{\theta}(y\mid x)=\frac{\exp(f(x;\theta)[y])}{\sum_{y'\in V}\exp(f(x;\theta)[y'])}=\frac{\exp(f(x;\theta)[y])}{Z(x;\theta)}. \tag*{(23)}

Taking the logarithm gives log⁡πθ(y∣x)=f(x;θ)[y]−log⁡Z(x;θ)\log\pi_{\theta}(y\mid x)=f(x;\theta)[y]-\log Z(x;\theta). Write Δlog⁡Zd(x):=log⁡Z(x;Wϕd)−log⁡Z(x;Wref)\Delta\log Z_d(x):=\log Z(x;W_{\phi_d})-\log Z(x;W_{\mathrm{ref}}) and Δlog⁡Zα(x):=log⁡Z(x;θ(α))−log⁡Z(x;Wref)\Delta\log Z_{\alpha}(x):=\log Z(x;\theta(\alpha))-\log Z(x;W_{\mathrm{ref}}).

Let θ=Wref+τ\theta=W_{\mathrm{ref}}+\tau. A first-order Taylor expansion of f(x;θ)[y]f(x;\theta)[y] around WrefW_{\mathrm{ref}} yields:

f(x;Wref+τ)[y]=f(x;Wref)[y]+∇θf(x;Wref)[y]Tτ+O(∥τ∥2).(24)f(x;W_{\mathrm{ref}}+\tau)[y]=f(x;W_{\mathrm{ref}})[y]+\nabla_{\theta}f(x;W_{\mathrm{ref}})[y]^{\mathsf T}\tau+O(\|\tau\|^2). \tag*{(24)}

Applying this to specialist weights Wϕd=Wref+τdW_{\phi_d}=W_{\mathrm{ref}}+\tau_d, the implicit reward is:

rd(x,y)=log⁡πWϕd(y∣x)πWref(y∣x)=∇θf(x;Wref)[y]Tτd−Δlog⁡Zd(x)+O(∥τd∥2).(25)r_d(x,y)=\log\frac{\pi_{W_{\phi_d}}(y\mid x)}{\pi_{W_{\mathrm{ref}}}(y\mid x)}=\nabla_{\theta}f(x;W_{\mathrm{ref}})[y]^{\mathsf T}\tau_d-\Delta\log Z_d(x)+O(\|\tau_d\|^2). \tag*{(25)}

The log-partition function log⁡Z(x;θ)=log⁡∑y′∈Vexp⁡(f(x;θ)[y′])\log Z(x;\theta)=\log\sum_{y'\in V}\exp(f(x;\theta)[y']) is a smooth function of θ\theta, so the same first-order expansion around WrefW_{\mathrm{ref}} gives

log⁡Z(x;Wref+τ)−log⁡Z(x;Wref)=∇θlog⁡Z(x;Wref)Tτ+O(∥τ∥2).(26)\log Z(x;W_{\mathrm{ref}}+\tau)-\log Z(x;W_{\mathrm{ref}})=\nabla_{\theta}\log Z(x;W_{\mathrm{ref}})^{\mathsf T}\tau+O(\|\tau\|^2). \tag*{(26)}

Taking τ=τd\tau= \tau_d gives Δlog⁡Zd(x)=∇θlog⁡Z(x;Wref)⊤τd+O(∥τd∥2)\Delta\log Z_d(x) = \nabla_\theta\log Z(x; W_{\mathrm{ref}})^\top\tau_d + O(\lVert\tau_d\rVert^2), and taking τ=Tα=∑d=1Kαdτd\tau= T\alpha= \sum_{d=1}^{K} \alpha_d\tau_d gives, by linearity of the inner product,

Δlog⁡Zα(x)=∑d=1Kαd∇θlog⁡Z(x;Wref)⊤τd+O(∥Tα∥2)=∑d=1KαdΔlog⁡Zd(x)+O(∥Tα∥2+∑d∣αd∣∥τd∥2).(27)\begin{aligned} \Delta\log Z_\alpha(x) &= \sum_{d=1}^{K} \alpha_d \nabla_\theta\log Z(x; W_{\mathrm{ref}})^\top\tau_d + O\left(\lVert T\alpha\rVert^2\right) \\ &= \sum_{d=1}^{K} \alpha_d \Delta\log Z_d(x) + O\left(\lVert T\alpha\rVert^2 + \sum_d |\alpha_d|\lVert\tau_d\rVert^2\right). \tag*{(27)} \end{aligned}

Now, for any linear combination θ(α)=Wref+∑d=1Kαdτd\theta(\alpha) = W_{\mathrm{ref}} + \sum_{d=1}^{K} \alpha_d\tau_d, by linearity of the gradient operator and Eq. (27):

log⁡πθ(α)(y∣x)πWref(y∣x)=∇θf(x;Wref)[y]⊤(∑d=1Kαdτd)−Δlog⁡Zα(x)+O(∥∑dαdτd∥2)=∑d=1Kαd(∇θf(x;Wref)[y]⊤τd−Δlog⁡Zd(x))+O(∥Tα∥2+∑d∣αd∣∥τd∥2)=∑d=1Kαdrd(x,y)+O(∥Tα∥2+∑d∣αd∣∥τd∥2)≈α⊤r(x,y),(28)\begin{aligned} \log\frac{\pi_{\theta(\alpha)}(y \mid x)}{\pi_{W_{\mathrm{ref}}}(y \mid x)} &= \nabla_\theta f(x;W_{\mathrm{ref}})[y]^\top\left(\sum_{d=1}^{K}\alpha_d\tau_d\right) - \Delta\log Z_\alpha(x) + O\left(\left\lVert\sum_d\alpha_d\tau_d\right\rVert^2\right) \\ &= \sum_{d=1}^{K}\alpha_d\left(\nabla_\theta f(x;W_{\mathrm{ref}})[y]^\top\tau_d-\Delta\log Z_d(x)\right) + O\left(\lVert T\alpha\rVert^2+\sum_d|\alpha_d|\lVert\tau_d\rVert^2\right) \\ &= \sum_{d=1}^{K}\alpha_d r_d(x,y) + O\left(\lVert T\alpha\rVert^2+\sum_d|\alpha_d|\lVert\tau_d\rVert^2\right) \approx\alpha^\top r(x,y), \tag*{(28)} \end{aligned}

where the last equality substitutes Eq. (25) for each rdr_d. This proves the First-Order Homomorphism between weight combinations and reward superposition. ■

Derivation of Second-Moment Residuals and Minimax Convexity

Connection to the Projected Fisher Information Metric. Under the first-order logit expansion (Proposition 2), for any specialist dd and prompt-prefix xx,

τ⊤∇θlog⁡πWϕd(y∣x)=r(x,y)−μd(x)+O(max⁡k∥τk∥2),(29)\tau^\top\nabla_\theta\log\pi_{W_{\phi_d}}(y\mid x)=r(x,y)-\mu_d(x)+O\left(\max_k\lVert\tau_k\rVert^2\right), \tag*{(29)}

where μd(x):=Ey′∼πWϕd(⋅∣x)[r(x,y′)]\mu_d(x):=\mathbb{E}_{y'\sim\pi_{W_{\phi_d}}(\cdot\mid x)}[r(x,y')]. The projected Fisher Information Matrix Sd:=τ⊤FdT∈RK×KS_d:=\tau^\top F_dT\in\mathbb{R}^{K\times K} satisfies:

Sd=Ex∼Dd,y∼πWϕd[(τ⊤∇θlog⁡πWϕd(y∣x))(τ⊤∇θlog⁡πWϕd(y∣x))⊤]≈Ex∼Dd,y∼πWϕd[(r(x,y)−μd(x))(r(x,y)−μd(x))⊤]=Cov⁡πWϕd(r).(30)\begin{aligned} S_d &= \mathbb{E}_{x\sim\mathcal{D}_d,y\sim\pi_{W_{\phi_d}}}\left[\left(\tau^\top\nabla_\theta\log\pi_{W_{\phi_d}}(y\mid x)\right)\left(\tau^\top\nabla_\theta\log\pi_{W_{\phi_d}}(y\mid x)\right)^\top\right] \\ &\approx\mathbb{E}_{x\sim\mathcal{D}_d,y\sim\pi_{W_{\phi_d}}}\left[(r(x,y)-\mu_d(x))(r(x,y)-\mu_d(x))^\top\right] = \operatorname{Cov}_{\pi_{W_{\phi_d}}}(r). \tag*{(30)} \end{aligned}

Now, expanding the second-moment matrix Md:=Ex∼Dd,y∼πWϕd[r(x,y)r(x,y)⊤]M_d:=\mathbb{E}_{x\sim\mathcal{D}_d,y\sim\pi_{W_{\phi_d}}}[r(x,y)r(x,y)^\top]:

Md=Cov⁡πWϕd(r)+Ex∼Dd[μd(x)μd(x)⊤]≈Sd+Ex∼Dd[μd(x)μd(x)⊤].(31)M_d=\operatorname{Cov}_{\pi_{W_{\phi_d}}}(r)+\mathbb{E}_{x\sim\mathcal{D}_d}[\mu_d(x)\mu_d(x)^\top]\approx S_d+\mathbb{E}_{x\sim\mathcal{D}_d}[\mu_d(x)\mu_d(x)^\top]. \tag*{(31)}

For any weight combination vector α∈RK\alpha\in\mathbb{R}^{K}, writing ed∈RKe_d\in\mathbb{R}^{K} for the dd-th standard basis vector, the residual optimized in Phase 1 is

Ld(α):=(α−ed)⊤Md(α−ed)≈(α−ed)⊤Sd(α−ed)+Ex∼Dd[((α−ed)⊤μd(x))2].(32)\mathcal{L}_d(\alpha):=(\alpha-e_d)^\top M_d(\alpha-e_d)\approx(\alpha-e_d)^\top S_d(\alpha-e_d)+\mathbb{E}_{x\sim\mathcal{D}_d}\left[((\alpha-e_d)^\top\mu_d(x))^2\right]. \tag*{(32)}

Ld\mathcal{L}_d upper-bounds the projected Fisher term 2Jsub,d(α)=(α−ed)⊤Sd(α−ed)2J_{\mathrm{sub},d}(\alpha)=(\alpha-e_d)^\top S_d(\alpha-e_d). After solving the Minimax problem, we form the normalized residual L‾d(α):=Ld(α)/E[rd2]\overline{\mathcal{L}}_d(\alpha):=\mathcal{L}_d(\alpha)/\mathbb{E}[r_d^2]. This post-hoc normalization expresses the remainder as a fraction of domain-dd reward energy and makes the spectrum comparable across domains.

Term-wise Decomposition. Writing out the quadratic with (Md)jk=E[rjrk](M_d)_{jk}=\mathbb{E}[r_jr_k] and then dividing by E[rd2]\mathbb{E}[r_d^2] yields the normalized diagnostic

L‾d(α)=(αd−1)2+∑j≠dαj2E[rj2]E[rd2]+2(αd−1)∑j≠dαjE[rdrj]E[rd2]+2∑j,k≠dj,k≠dαjαkE[rjrk]E[rd2].(33)\overline{\mathcal{L}}_d(\alpha)=(\alpha_d-1)^2+\sum_{j\ne d}\alpha_j^2\frac{\mathbb{E}[r_j^2]}{\mathbb{E}[r_d^2]}+2(\alpha_d-1)\sum_{j\ne d}\alpha_j\frac{\mathbb{E}[r_dr_j]}{\mathbb{E}[r_d^2]}+2\sum_{\substack{j,k\ne d\\j,k\ne d}}\alpha_j\alpha_k\frac{\mathbb{E}[r_jr_k]}{\mathbb{E}[r_d^2]}. \tag*{(33)}

The first term is squared relative under-recovery of the in-domain reward (it vanishes at αd=1\alpha_d=1). The remaining terms are cross-domain interference measured in the same units. The correlation structure of Appendix § 10.§ 4 governs the cross-terms.

Proposition 3 (Convexity of the Minimax QP). The objective function Φ(α):=max⁡d∈{1,…,K}Ld(α)\Phi(\alpha) := \max_{d \in\{1,\ldots,K\}} \mathcal{L}_d(\alpha) is strictly convex in α\alpha whenever each Md≻0M_d \succ0.

Proof. Each component has Hessian ∇2Ld(α)=2Md\nabla^{2}\mathcal{L}_d(\alpha) = 2M_d. Since the finite collection {Md}d=1K\{M_d\}_{d=1}^{K} is positive definite, all components share a positive strong-convexity lower bound. Their pointwise maximum is therefore strongly convex, so the Minimax problem has a unique global minimizer α⋆\alpha^{\star}. In practice, replacing MdM_d by Md+λIM_d + \lambda I supplies this condition. ■

Theoretical Justification of Branch Re-merging within Local Tangent Basins

A natural question is whether the branch updates Δd=θd−θ0⋆\Delta_d = \theta_d - \theta_0^{\star} obtained in Phase 2, once written back with shrinkage, remain within the regime in which the first-order reward superposition of Proposition 2 is accurate. The following statement makes precise the role of the contraction factor γ\gamma.

Proposition 4 (Shrinkage Tightens the Linearization Error). Fix a prompt-prefix xx and token yy, and let g(θ):=f(x;θ)[y]g(\theta) := f(x;\theta)[y] denote the corresponding logit. Assume gg is twice differentiable with ∥∇θ2g(θ)∥≤LH\lVert\nabla_{\theta}^{2}g(\theta)\rVert\le L_H on a convex neighborhood B\mathcal{B} containing WrefW_{\mathrm{ref}}, θ0⋆=Wref+Tα⋆\theta_0^{\star} = W_{\mathrm{ref}} + T\alpha^{\star}, and the branch endpoint θ0⋆+Δd\theta_0^{\star}+\Delta_d. Let g^(θ):=g(Wref)+∇θg(Wref)⊤(θ−Wref)\hat{g}(\theta) := g(W_{\mathrm{ref}})+\nabla_{\theta}g(W_{\mathrm{ref}})^{\top}(\theta-W_{\mathrm{ref}}) be the first-order model underlying Proposition 2. Then for every γ∈[0,1]\gamma\in[0,1], the re-merged point θγ:=θ0⋆+γΔd\theta_{\gamma} := \theta_0^{\star}+\gamma\Delta_d satisfies

∣g(θγ)−g^(θγ)∣≤LH2∥Tα⋆+γΔd∥2≤LH2(∥Tα⋆∥+γ∥Δd∥)2.(34)\left|g(\theta_{\gamma})-\hat{g}(\theta_{\gamma})\right| \le\frac{L_H}{2}\left\|T\alpha^{\star}+\gamma\Delta_d\right\|^{2} \le\frac{L_H}{2}\left(\left\|T\alpha^{\star}\right\|+\gamma\left\|\Delta_d\right\|\right)^{2}. \tag*{(34)}

and the right-hand side is non-decreasing in γ\gamma. In particular, the linearization error at the re-merged model is at most that of the un-shrunk branch endpoint (γ=1\gamma=1) and approaches that of θ0⋆\theta_0^{\star} as γ→0\gamma\to0; when ∥Δd∥≪∥Tα⋆∥\left\|\Delta_d\right\| \ll\left\|T\alpha^{\star}\right\|, the excess error over θ0⋆\theta_0^{\star} is LHγ∥Δd∥∥Tα⋆∥+O(γ2∥Δd∥2)L_H\gamma\left\|\Delta_d\right\|\left\|T\alpha^{\star}\right\|+O(\gamma^{2}\left\|\Delta_d\right\|^{2}), i.e., first order in γ∥Δd∥\gamma\left\|\Delta_d\right\|.

Proof. Since θγ\theta_{\gamma} is a convex combination of θ0⋆\theta_0^{\star} and θ0⋆+Δd\theta_0^{\star}+\Delta_d, it lies in B\mathcal{B}. Taylor’s theorem with Lagrange remainder along the segment from WrefW_{\mathrm{ref}} to θγ\theta_{\gamma} gives g(θγ)−g^(θγ)=12(θγ−Wref)⊤∇θ2g(ξ)(θγ−Wref)g(\theta_{\gamma})-\hat{g}(\theta_{\gamma})=\frac{1}{2}(\theta_{\gamma}-W_{\mathrm{ref}})^{\top}\nabla_{\theta}^{2}g(\xi)(\theta_{\gamma}-W_{\mathrm{ref}}) for some ξ∈B\xi\in\mathcal{B}, and θγ−Wref=Tα⋆+γΔd\theta_{\gamma}-W_{\mathrm{ref}}=T\alpha^{\star}+\gamma\Delta_d. Bounding the Hessian by LHL_H and applying the triangle inequality yields Eq. (34). Monotonicity in γ\gamma follows because ∥Tα⋆∥+γ∥Δd∥\left\|T\alpha^{\star}\right\|+\gamma\left\|\Delta_d\right\| is affine and increasing in γ\gamma. Expanding the square gives the excess-error expression. The same argument applies to the log-partition term log⁡Z(x;θ)\log Z(x;\theta), which is a smooth function of the logit vector, so the bound transfers to the log-ratio log⁡πθγ(y∣x)πWref(y∣x)\log\frac{\pi_{\theta_{\gamma}}(y\mid x)}{\pi_{W_{\mathrm{ref}}}(y\mid x)} up to a constant factor depending only on the softmax smoothness. ■

Proposition 4 does not by itself assert that ∥Δd∥\left\|\Delta_d\right\| is small; it states that, whatever displacement branch training produces, shrinkage can only reduce the bound on the linearization error, and that the reduction is first order in γ\gamma. On Open-MoPD the residual gate writes back only the Math branch at the unlabelled grid value γ⋆=0.125\gamma^{\star}=0.125; passivated Code/IF contribute Δd=0\Delta_d=0.

Phase 1 Diagnostics and Detailed Benchmark Results

This section supports § 2–§ 4 with the linearity diagnostics (Appendix § 10.§ 1), the six-benchmark Phase 1 breakdown (Appendix § 10.§ 2), the six-benchmark scores of the normalized-objective merges (Appendix § 10.§ 3), the empirical reward correlation (Appendix § 10.§ 4), the K=2K=2 pairwise merges (Appendix § 10.§ 5), generation lengths (Appendix § 10.§ 6), and the full G-OPD suite (Appendix § 10.§ 7).

Linearity Diagnostics on the Isotropic Family

Figure 5 reports the diagnostics discussed in § 4.7: the isotropic scan θ(α)=Wref+α∑dτd\theta(\alpha)=W_{\mathrm{ref}}+\alpha\sum_d\tau_d of Remark 1, the regression R2R^2 of the merged log-ratio on the superposed reward over 38M tokens, and the cross-domain interference ratio Icross:=∑dEd[(ℓα−αrd)2]/∑dEd[ℓα2]\mathcal{I}_{\mathrm{cross}} := \sum_d\mathbb{E}_d[(\ell_{\alpha}-\alpha r_d)^2]/\sum_d\mathbb{E}_d[\ell_{\alpha}^{2}], the share of merged log-ratio energy not explained by the in-domain reward. The branch displacements used in § 4.7 are ∥Tα⋆∥=1.025\left\|T\alpha^{\star}\right\|=1.025, ∥Δmath∥=0.229\left\|\Delta_{\mathrm{math}}\right\|=0.229, ∥Δcode∥=0.299\left\|\Delta_{\mathrm{code}}\right\|=0.299, ∥Δif∥=0.265\left\|\Delta_{\mathrm{if}}\right\|=0.265.

Linearity diagnostics on the isotropic family

Figure 5. Linearity diagnostics on the isotropic family (θ(α)=Wref+α∑dτd\theta(\alpha)=W_{\mathrm{ref}}+\alpha\sum_d\tau_d; not the Minimax vector α⋆\alpha^\star). (a) Macro vs. scalar α\alpha. Horizontal lines are RouteRL (35.68), the 600-step student (30.93), and MixSFT (25.35); the peak at α=0.75\alpha=0.75 is this scan, not α⋆\alpha^\star. (b) R2R^2 and Icross\mathcal{I}_{\mathrm{cross}} over 38M tokens.

Complete Per-Benchmark Phase 1 Results

Table 4 details the individual benchmark evaluations across all six downstream evaluation tasks: AIME 2024, AIME 2025 (Math), LiveCodeBench v5, LiveCodeBench v6 (Code), IFEval, and IFBench (Instruction Following). The Type column is the coefficient family: Scalar is a shared scale α1\alpha\mathbf{1}; 3D Vector is a free α∈R3\alpha\in\mathbb{R}^3. Official Distilled Model is the main-text Multi-Teacher OPD row. Forward-KL Merge uses forward KL in the same isotropic subspace program; Minimax Worst-Domain Merge (Scalar) is Eq. (9) restricted to α=α1\alpha=\alpha\mathbf{1}. Both Subspace Fisher Merge rows solve Eq. (5) under uniform pd=1/3p_d=1/3 with MdM_d in place of SdS_d; the main-text Subspace Fisher Merge is the 3D Vector row.

Method / ConfigurationTypeA24A25Avgv5v6AvgIFEvalIFBAvgMacro
MixSFT Base (WrefW_{\mathrm{ref}})–14.7918.1816.4915.6318.4617.0566.3618.6742.5225.35
RouteRL (Oracle Upper Bound)–26.8827.6027.2427.2725.2626.2781.7025.3353.5235.68
Official Distilled ModelDistill24.3824.6824.5321.0521.3121.1870.2424.0047.1230.93
Equal-Weight AverageScalar20.3121.3020.8117.6620.6319.1569.8719.6744.7728.24
Task ArithmeticScalar23.9124.7924.3522.9922.6322.8169.8723.6746.7731.31
Subspace Fisher MergeScalar25.4724.4824.9723.6522.9123.2873.9421.3347.6431.96
Subspace Fisher Merge3D Vector23.8524.3824.1124.4922.6923.5972.6423.3347.9931.90
Forward KL (FKL) MergeScalar24.5823.2823.9323.5922.3422.9771.5324.3347.9331.61
Minimax Worst-Domain MergeScalar25.1024.8424.9724.0722.9123.4969.8720.3345.1031.19
Minimax Merge (α⋆\alpha^\star)3D Vector23.9123.8023.8624.2723.9124.0973.8725.3349.6032.52
RGBM Phase 2Residual-Gated23.8024.0623.9325.1524.1124.6375.0526.0050.5233.03

Table 4. Comprehensive benchmark evaluation of Phase 1 candidates. Six-task breakdown. Bold: best zero-training candidate per column (or Phase 2 where it exceeds Phase 1).

Scores of the Normalized-Objective Merges

The four Open-MOPD programs of Table 2 that minimize Ld/E[rd2]\mathcal{L}_d/\mathbb{E}[r_d^2] are absent from Table 4; their six-benchmark scores are Table 5.

ProgramA24A25Mathv5v6CodeIFEvalIFBIFMacro
A scalar L‾\overline{\mathcal{L}}24.6424.1124.3823.2323.4923.3671.7218.0044.8630.86
A vector L‾\overline{\mathcal{L}}23.6524.7424.1917.6021.5419.5774.6821.3348.0030.59
C scalar L‾\overline{\mathcal{L}}21.2521.5621.4118.4421.0919.7670.2421.3345.7928.99
C vector L‾\overline{\mathcal{L}}23.5424.9024.2219.5221.9420.7375.0519.0047.0230.66

Table 5. Six-benchmark scores of the four normalized-objective merges. Same protocol as Table 4. The four L\mathcal{L} programs and α=1/3\alpha=1/3 are already in that table.

Empirical Reward Correlation Geometry

We sample 6,144 rollout sequences across domain training prompts and extract implicit rewards r=[rmath,rcode,rif]T\mathbf{r} = [r_{\mathrm{math}}, r_{\mathrm{code}}, r_{\mathrm{if}}]^{\mathsf{T}}. The empirical correlation matrix is:

Corr⁡(r)=[1.0000.3130.0430.3131.0000.0350.0430.0351.000].(35)\operatorname{Corr}(\mathbf{r}) = \begin{bmatrix} 1.000 & \mathbf{0.313} & 0.043 \\ \mathbf{0.313} & 1.000 & 0.035 \\ 0.043 & 0.035 & 1.000 \end{bmatrix}. \tag*{(35)}

The off-diagonal correlation between Math and Code is moderately positive (0.313), indicating that mathematical problem-solving and programming share underlying algorithmic reasoning representations. In contrast, the Instruction-Following specialist is strictly orthogonal to both reasoning domains (<0.045< 0.045). This geometric structure directly explains why the Minimax solver assigns αif⋆≈0.800\alpha_{\mathrm{if}}^{\star} \approx0.800 (as the IF domain experiences near-zero destructive interference) while compressing αmath⋆≈0.597\alpha_{\mathrm{math}}^{\star} \approx0.597 and αcode⋆≈0.620\alpha_{\mathrm{code}}^{\star} \approx0.620 to minimize cross-talk.

Two-Domain (K=2K = 2) Sub-Consolidations and Residual Tracking

To test whether the Fisher-geometric theory and the predictive validity of the residual spectrum hold across different expert subsets, we evaluate all three pairwise K=2K = 2 specialist consolidations formed from {Wϕmath,Wϕcode,Wϕif}\{W_{\phi_{\mathrm{math}}}, W_{\phi_{\mathrm{code}}}, W_{\phi_{\mathrm{if}}}\}:

  1. Math + Code (K=2K = 2): The implicit rewards share positive covariance (ρ=0.313\rho= 0.313). The Minimax solver allocates balanced weights α⋆=[0.635,0.652]T\alpha^{\star} = [0.635, 0.652]^{\mathsf{T}}, yielding zero-training performance of Math 24.38, Code 24.81 (Macro 24.60, higher than the 3-expert merge on Code due to zero interference from IF). The normalized spectrum ranks L~math>L~code\widetilde{\mathcal{L}}_{\mathrm{math}} > \widetilde{\mathcal{L}}_{\mathrm{code}}, correctly identifying Math as the remaining active bottleneck.

  1. Math + IF (K=2K = 2): Because the two specialists are nearly orthogonal (ρ=0.043\rho= 0.043), cross-domain interference is minimal. The Minimax weights are α⋆=[0.680,0.885]T\alpha^{\star} = [0.680, 0.885]^{\mathsf{T}}, achieving Math 24.82 and IF 48.95 (Macro 36.89). L~if\widetilde{\mathcal{L}}_{\mathrm{if}} is near the floor (almost complete recovery), while L~math\widetilde{\mathcal{L}}_{\mathrm{math}} marks Math for post-merge branch training.

  1. Code + IF (K=2K = 2): With near-zero cross-covariance (ρ=0.035\rho= 0.035), linear superposition incurs minimal interference. The Minimax weights α⋆=[0.720,0.890]T\alpha^{\star} = [0.720, 0.890]^{\mathsf{T}} achieve Code 25.02 and IF 49.10 (Macro 37.06). Both normalized residuals sit near the floor, predicting that both domains are passivated and require zero post-merge branch compute.

Across all K=2K = 2 sub-settings, the normalized residual spectrum L~\widetilde{\mathcal{L}} correctly ranks domain bottlenecks and tracks pairwise recovery.

Token Generation Length Distributions

Table 6 summarizes the average completion lengths across evaluation benchmarks.

ModelAIME 2024 (Math)LiveCodeBench v5 (Code)IFBench (IF)
MixSFT Base (WrefW_{\mathrm{ref}})3,420485310
Math Specialist (WϕmathW_{\phi_{\mathrm{math}}})14,850490280
Code Specialist (WϕcodeW_{\phi_{\mathrm{code}}})3,890720295
IF Specialist (WϕifW_{\phi_{\mathrm{if}}})2,950420315
Equal-Weight Task Arithmetic11,240560410
Official Distilled Model (600 steps)11,850580390
Zero-Training Merge (θ0⋆\theta_{0}^{\star})12,316615461

Table 6. Average generation length across models. Completion tokens on one representative task per domain (AIME 2024, LiveCodeBench v5, IFBench).

The zero-training merge θ0⋆\theta_{0}^{\star} preserves extended reasoning lengths (≈12.3k\approx12.3\mathrm{k} tokens on AIME), confirming that parameter merging successfully preserves reasoning traces without requiring expensive post-hoc length penalties.

Cross-Suite Generalization on G-OPD Benchmarks (Qwen3-4B)

To further establish that the Fisher-geometric projection generalizes across specialist regimes and model families, we evaluate RGBM on the official G-OPD benchmarks [35] using Qwen3-4B in two specialist regimes: the intermediate checkpoints of Table 2 in [35] (WϕmathW_{\phi\mathrm{math}} at 500 steps, WϕcodeW_{\phi\mathrm{code}} at 300 steps), which are the main-text G-OPD setting, and the fully trained pair of their Table 8 (both at 1200 steps). Table 7 reports the complete per-task benchmark results across all 7 evaluation datasets (AIME 2024, AIME 2025, HMMT Feb, HMMT Nov, HumanEval+ [HE+], MBPP+, and LiveCodeBench [LCB v6]) following the standardized evaluation protocol.

RegimeMethod / ConfigurationAIME24AIME25HMMT-FebHMMT-NovHE+MBPP+LCB7-Task Avg
Intermediate (Math-500 + Code-300)Teacher (WϕmathW_{\phi\mathrm{math}}, WϕcodeW_{\phi\mathrm{code}})58.0054.6032.5038.9086.0070.2027.3052.50
Intermediate (Math-500 + Code-300)Student Base (WrefW_{\mathrm{ref}})21.5021.9010.008.0074.7064.7017.9031.24
Intermediate (Math-500 + Code-300)SFT58.5053.3030.7034.8086.4069.6026.4051.39
Intermediate (Math-500 + Code-300)ExPO (Zheng et al., 2024)57.5054.5031.7036.3086.7072.0029.0052.53
Intermediate (Math-500 + Code-300)On-Policy Distillation (OPD)60.6054.1032.5038.3084.6069.5027.6052.46
Intermediate (Math-500 + Code-300)ExOPD [35]61.0056.0034.4039.2086.3070.6029.0053.79
Intermediate (Math-500 + Code-300)Task Arithmetic56.9850.4228.0236.7784.3068.5229.1450.59
Intermediate (Math-500 + Code-300)Equal-Weight Average33.7525.0015.5215.9479.8867.1320.0036.75
Intermediate (Math-500 + Code-300)RGBM Phase 1 (Ours)59.1755.3132.7137.5087.9671.4329.2953.34
Intermediate (Math-500 + Code-300)RGBM Phase 2 (Ours)60.4256.5633.9639.2786.9171.3030.5754.14
Converged (Math-1200 + Code-1200)Teacher (WϕmathW_{\phi\mathrm{math}}, WϕcodeW_{\phi\mathrm{code}})68.2059.3037.3042.9088.9072.5028.0056.73
Converged (Math-1200 + Code-1200)Student Base (WrefW_{\mathrm{ref}})21.5021.9010.008.0074.7064.7017.9031.24
Converged (Math-1200 + Code-1200)OPD (Single-Teacher)68.3058.7038.7041.2089.3071.3028.0056.50
Converged (Math-1200 + Code-1200)ExOPD (Single-Teacher)68.4059.2038.2042.8089.9073.7029.3057.36
Converged (Math-1200 + Code-1200)OPD (Multi-Teacher)68.2060.2038.5040.8086.4072.1027.6056.26
Converged (Math-1200 + Code-1200)ExOPD (Multi-Teacher)70.1059.6037.5042.7089.5073.9029.7057.57
Converged (Math-1200 + Code-1200)Task Arithmetic64.7959.4835.1041.5687.5074.7430.8656.29
Converged (Math-1200 + Code-1200)Equal-Weight Average40.9435.6220.1024.1780.7968.8521.5741.72
Converged (Math-1200 + Code-1200)RGBM Phase 1 (Ours)67.5062.9236.4643.3385.3774.8731.0057.35

Table 7. Per-benchmark results on the G-OPD suite (Qwen3-4B). Intermediate and converged specialist regimes under the official 7-task protocol. ExPO is a published G-OPD baseline. Single-Teacher distills from one specialist; Multi-Teacher uses both. Bold: best consolidated model per column (teacher excluded).

SystemInitTrainingWritten back
RouteRL (Gao et al., [8])KK specialistsnoneRouting oracle; serves KK models
Multi-Teacher OPD (Gao et al., [8])WrefW_{\mathrm{ref}}joint OPD, all domainsFull student checkpoint
Task Arithmetic (Ilharco et al., [13])WrefW_{\mathrm{ref}}none∑dτd\sum_d\tau_d (α=1\alpha=1)
Subspace Fisher MergeWrefW_{\mathrm{ref}}noneTαsub\mathcal{T}\alpha_{\mathrm{sub}} (Eq. (5), pd=1/3p_d=1/3, MdM_d for SdS_d)
RGBM Phase 1 (θ0⋆\theta_0^{\star})WrefW_{\mathrm{ref}}noneTα⋆\mathcal{T}\alpha^{\star}
Uniform Distillationθ0⋆\theta_0^{\star}joint OPD, pd=1/3p_d=1/3Shared weights
TIES (Yadav et al., [31]) / DARE (Yu et al., 2024)θ0⋆\theta_0^{\star}isolated branch per domainAll {Δd}d=1K\{\Delta_d\}_{d=1}^{K}, sparsified
Gated Math (γ=1.0\gamma=1.0)θ0⋆\theta_0^{\star}isolated Math branch onlyΔmath\Delta_{\mathrm{math}} (no shrinkage)
Gated Math (γ=0.5\gamma=0.5)θ0⋆\theta_0^{\star}isolated Math branch only0.5Δmath0.5\Delta_{\mathrm{math}}
RGBM Phase 2 (Ours)θ0⋆\theta_0^{\star}the same isolated Math branchγ⋆Δmath\gamma^{\star}\Delta_{\mathrm{math}}

Table 8. Consolidation systems compared on Open-MOPD. How each row of Table 1 is built: initialization, training, and what is written back.

Phase 1 coefficients and Phase 2 allocation. On the intermediate pair the Minimax weights are α⋆=(0.675,0.666)\alpha^{\star} = (0.675, 0.666) and the normalized spectrum L=(0.094,0.090)\mathcal{L} = (0.094, 0.090) has its largest gap down to 0, so Eq. (10) gives ϵ=0.045\epsilon= 0.045 and both Math and Code stay active. Isolated distillation branches re-merged at the unlabelled grid value γ=0.5\gamma= 0.5 reach 54.14 (Table 7), +0.80 over Phase 1 and +0.35 over trained ExOPD. On the converged pair we report only Phase 1 (α⋆=(0.511,0.505)\alpha^{\star} = (0.511, 0.505), average 57.35); there is no Phase 2 run.

Additional Experimental Details and Parameter Sensitivity

This section collects the full RGBM procedure and the systems compared (Appendix § 11.§ 1), the five joint-distillation recipes (Appendix § 11.§ 2), the post-hoc γ\gamma write-backs (Appendix § 11.§ 3), the ϵ\epsilon allocation cuts (Appendix § 11.§ 4), training infrastructure (Appendix § 11.§ 5), the cost and robustness of assembling MdM_d (Appendix § 11.§ 6), and the Open-MOPD RGBM Phase 2 specification (Appendix § 11.§ 7).

RGBM Procedure and Systems Compared

Algorithm 1 lists the complete RGBM procedure of § 3; Table 8 lists how every Open-MOPD system in Table 1 is built.

When evaluating Err⁡lin(θθ⋆+γΔd)\operatorname{Err}_{\mathrm{lin}}(\theta^{\star}_{\theta} + \gamma\Delta_d) in Algorithm 1, the response is ℓθγ=log⁡(πθγ/πWref)\ell_{\theta_{\gamma}} = \log(\pi_{\theta_{\gamma}}/\pi_{W_{\mathrm{ref}}}) and the predictor is the Phase 1 superposition α⋆Tr\alpha^{\star\mathsf{T}}r, held fixed. Both are scored token-wise on the same Phase 1 pairs (x,y)(x,y) used to assemble MdM_d, pooled over ⋃dDcalib,d\bigcup_d \mathcal{D}_{\mathrm{calib},d}. A single OLS with intercept then yields R2R^2, and Err⁡lin=1−R2\operatorname{Err}_{\mathrm{lin}} = 1 - R^2 as in Eq. (7).

Algorithm 1 Residual-Governed Branch-and-Merge (RGBM)

Require: Base weights WrefW_{\mathrm{ref}}, specialist weights {Wϕd}d=1K\{W_{\phi_d}\}_{d=1}^{K}, unlabelled calibration prompts Dcalib={Dcalib,d}d=1K\mathcal{D}_{\mathrm{calib}}=\{\mathcal{D}_{\mathrm{calib},d}\}_{d=1}^{K}, Tikhonov constant λ\lambda, discrete shrinkage grid Γ={0.125,0.25,0.5,0.75}\Gamma=\{0.125,0.25,0.5,0.75\}, linearity tolerance δtol\delta_{\mathrm{tol}}.

  1. For each dd: draw NdN_d pairs xi∼Dcalib,dx_i\sim\mathcal{D}_{\mathrm{calib},d}, yi∼πWϕd(⋅∣xi)y_i\sim\pi_{W_{\phi_d}}(\cdot\mid x_i); score r(xi,yi)=[log⁡πWϕk(yi∣xi)πWref(yi∣xi)]k=1K\mathbf{r}(x_i,y_i)=\left[\log\frac{\pi_{W_{\phi_k}}(y_i\mid x_i)}{\pi_{W_{\mathrm{ref}}}(y_i\mid x_i)}\right]_{k=1}^{K}; Md←1Nd∑irrTM_d\leftarrow\frac{1}{N_d}\sum_i\mathbf{r}\mathbf{r}^{\mathsf{T}}.

  2. Solve the residual Minimax QP (Eq. (9)) with M~d=Md+λI\widetilde{M}_d=M_d+\lambda I for α⋆\alpha^{\star}; set θ0⋆←Wref+Tα⋆\theta_0^{\star}\leftarrow W_{\mathrm{ref}}+\mathcal{T}\alpha^{\star}; using MdM_d, set Ld←(α⋆−ed)TMd(α⋆−ed)\mathcal{L}_d\leftarrow(\alpha^{\star}-e_d)^{\mathsf{T}}M_d(\alpha^{\star}-e_d) and Ld←Ld/E[rd2]\mathcal{L}_d\leftarrow\mathcal{L}_d/\mathbb{E}[r_d^2] (Eq. (8)).

  3. Sort L(1)≥⋯≥L(K)≥L(K+1):=0\mathcal{L}_{(1)}\ge\cdots\ge\mathcal{L}_{(K)}\ge\mathcal{L}_{(K+1)}:=0; k⋆←arg max⁡i(L(i)−L(i+1))k^{\star}\leftarrow\underset{i}{\operatorname{arg\,max}}(\mathcal{L}_{(i)}-\mathcal{L}_{(i+1)}); ϵ←12(L(k⋆)+L(k⋆+1))\epsilon\leftarrow\frac{1}{2}(\mathcal{L}_{(k^{\star})}+\mathcal{L}_{(k^{\star}+1)}) (Eq. (10)).

  4. for d=1,…,Kd=1,\ldots,K do

  5. if Ld≤ϵ\mathcal{L}_d\le\epsilon then

  6. Passivate: Δd←0\Delta_d\leftarrow0

  7. else

  8. Active branch: θd←θ0⋆\theta_d\leftarrow\theta_0^{\star}; train on Dd\mathcal{D}_d only; Δd←θd(end)−θ0⋆\Delta_d\leftarrow\theta_d^{(\mathrm{end})}-\theta_0^{\star}.

  9. Compute unlabelled shrinkage γd⋆←max⁡{γ∈Γ∣Err⁡lin(θ0⋆+γΔd)≤δtol}\gamma_d^{\star}\leftarrow\max\{\gamma\in\Gamma\mid\operatorname{Err}_{\mathrm{lin}}(\theta_0^{\star}+\gamma\Delta_d)\le\delta_{\mathrm{tol}}\} (Eq. (11)).

  10. end if

  11. end for

  12. Re-merge: θRGBM←θ0⋆+∑d:Ld>ϵγd⋆Δd\theta_{\mathrm{RGBM}}\leftarrow\theta_0^{\star}+\sum_{d:\mathcal{L}_d>\epsilon}\gamma_d^{\star}\Delta_d (Eq. (11)).

  13. return θRGBM\theta_{\mathrm{RGBM}}

Joint Multi-Task Distillation: Exploration of 5 Recipes

To thoroughly investigate whether any joint fine-tuning objective could maintain or improve upon θ0⋆\theta_0^{\star}, we implemented and evaluated five distinct joint training recipes:

  1. Standard Multi-Teacher Distillation (Uniform MOPD): Multi-teacher reverse-KL on student rollouts with equal domain sampling (pd=1/3p_d=1/3).

  2. Outcome-Gated Distillation: Outcome-gated on-policy distillation, where teacher distribution matching is active only when student reasoning fails verification.

  3. Outcome-Veto Regularized Distillation: Regularized distillation applying explicit negative KL penalties to suppress drift on non-target domains.

  4. Trust-Region Regularized Distillation: Outcome-gated distillation with explicit KL anchoring to θ0⋆\theta_0^{\star} (β=2.5\beta=2.5).

  5. Hybrid Teacher-Anchor Distillation: Convex combination of specialist teacher KL and student-to-θ0⋆\theta_0^{\star} anchor KL.

Empirical Findings across Checkpoints. On the five recipes in Table 9, Macro Total stays in the 30.5%∼32.3%30.5\%\sim32.3\% range; every listed checkpoint lies below residual-gated RGBM Phase 2 (33.03%33.03\%). At the last listed checkpoint of each recipe, Code is 1.1∼2.91.1\sim2.9 points below θ0⋆\theta_0^{\star}, and IF stays below the Phase 1 value 49.60. Within this testbed, no joint objective we tried preserved the low-residual domains while training on the shared weights, which is the behaviour anticipated by the argument of § 3.3.

Distillation FrameworkStepMathCodeIFMacroTotalΔ\Delta vs. θ0⋆\theta_0^\star
RGBM Phase 1 (θ0⋆\theta_0^\star)023.8624.0949.6032.52–
RGBM Phase 2 (Ours)023.9324.6350.5233.03+0.51
Uniform Distillation [8]Step 1024.3820.7847.7930.98-1.54
Step 3024.5821.4647.6231.22-1.30
Step 5022.9421.8346.8430.54-1.98
Trajectory Mean23.9721.3647.4230.91-1.61
Outcome-Gated DistillationStep 1024.4521.6248.1231.40-1.12
Step 3024.9722.1848.9132.02-0.50
Step 5024.6422.2546.9731.28-1.24
Trajectory Mean24.6922.0248.0031.57-0.95
Outcome-Veto DistillationStep 1025.2321.9347.1031.42-1.10
Step 3024.2421.5948.9931.61-0.91
Step 5025.0821.2047.7331.33-1.19
Trajectory Mean24.8521.5747.9431.45-1.07
Trust-Region Regularization (β=2.5\beta= 2.5)Step 523.5723.3349.9932.30-0.22
Step 1024.8423.6647.5232.01-0.51
Step 2024.9521.1847.6231.25-1.27
Trajectory Mean24.4522.7248.3831.85-0.67
Hybrid Teacher-AnchorStep 1024.4021.1547.7331.09-1.43
Step 3023.6522.7348.6931.69-0.83
Step 5024.6922.9946.3831.35-1.17
Trajectory Mean24.2522.2947.6031.38-1.14

Table 9. Trajectory evaluation of joint multi-teacher distillation from θ0⋆\theta_0^\star. Five joint objectives. Trajectory Mean is the average of the listed checkpoints. Trust-Region is scored at steps 5/10/20 because that run ends at step 20.

Write-Back Does Not Raise Macro above Residual-Gated RGBM

The main-text Phase 2 scheme is residual-gated RGBM Phase 2 with Math active and Code/IF passivated. Shrinkage is chosen by the unlabelled rule (Eq. (11)) on Γ={0.125,0.25,0.5,0.75}\Gamma= \{0.125, 0.25, 0.5, 0.75\}, which selects γ⋆=0.125\gamma^\star= 0.125 (Macro 33.03), above the Phase 1 merge (32.52). Table 10 is a post-hoc diagnostic: the remaining grid points, a γ=1\gamma= 1 (no-shrinkage) foil, and equal-mass alternatives are not used to choose γ\gamma; none of them exceed 33.03.

ConfigurationFormula / ShrinkageMathCodeIFMacroTotalErr⁡lin\operatorname{Err}_{\mathrm{lin}}
Part A: Grid Γ={0.125,0.25,0.5,0.75}\Gamma= \{0.125, 0.25, 0.5, 0.75\} and γ=1\gamma= 1 foil
γ=0.125\gamma= 0.125 (Ours, γ⋆\gamma^\star)θ0⋆+0.125 Δmath\theta_0^\star+ 0.125\,\Delta_{\mathrm{math}}23.9324.6350.5233.030.0701
γ=0.25\gamma= 0.25θ0⋆+0.25 Δmath\theta_0^\star+ 0.25\,\Delta_{\mathrm{math}}24.1924.0247.8232.010.1749
γ=0.50\gamma= 0.50θ0⋆+0.50 Δmath\theta_0^\star+ 0.50\,\Delta_{\mathrm{math}}25.4223.4848.0832.320.2594
γ=0.75\gamma= 0.75θ0⋆+0.75 Δmath\theta_0^\star+ 0.75\,\Delta_{\mathrm{math}}25.0822.8745.5131.150.6442
γ=1.00\gamma= 1.00θ0⋆+1.00 Δmath\theta_0^\star+ 1.00\,\Delta_{\mathrm{math}}25.0322.5147.8731.800.7231
Part B: Equal Total Weight Mass Alternatives
Single Point θmath(2)\theta_{\mathrm{math}}^{(2)}θ0⋆+23Δmath(2)\theta_0^\star+ \frac{2}{3}\Delta_{\mathrm{math}}^{(2)}25.5223.3347.5732.14–
Single Point θmath(1)\theta_{\mathrm{math}}^{(1)}θ0⋆+23Δmath(1)\theta_0^\star+ \frac{2}{3}\Delta_{\mathrm{math}}^{(1)}24.6423.8448.1832.22–
Two-Point Average13(θ0⋆+θmath(1)+θmath(2))\frac{1}{3}(\theta_0^\star+ \theta_{\mathrm{math}}^{(1)} + \theta_{\mathrm{math}}^{(2)})24.9023.5048.9132.44–

Table 10. Ablation of Math-branch write-back configurations. Part A: the grid Γ\Gamma plus a γ=1\gamma= 1 no-shrinkage foil. Err⁡lin\operatorname{Err}_{\mathrm{lin}} is evaluated on Dcalib\mathcal{D}_{\mathrm{calib}} and is the quantity in Eq. (11); δtol=0.10\delta_{\mathrm{tol}} = 0.10. Part B: θmath(1)\theta_{\mathrm{math}}^{(1)} and θmath(2)\theta_{\mathrm{math}}^{(2)} are two intermediate Math-branch checkpoints; the equal-mass rows match their total displacement rather than using Err⁡lin\operatorname{Err}_{\mathrm{lin}}.

Readout. Eq. (11) selects γ⋆=0.125\gamma^\star= 0.125 from Γ\Gamma using only Err⁡lin\operatorname{Err}_{\mathrm{lin}} on Dcalib\mathcal{D}_{\mathrm{calib}}. In Table 10 that value is 0.0701 ≤δtol\leq\delta_{\mathrm{tol}}, while γ=0.25\gamma= 0.25 already reaches 0.1749. Post-hoc, every larger grid point and every equal-mass alternative in Table 10 has Macro ≤32.44\le32.44. On G-OPD, where both domains require a separate OPD branch, the same unlabelled rule yields γ⋆=0.5\gamma^\star= 0.5.

Residual Threshold Sensitivity and Passivation Robustness

In Phase 2 the threshold ϵ\epsilon only matters through which domains it activates. Eq. (10) is a closed-form choice: sort the normalized spectrum L‾=(0.128,0.052,0.015)\overline{\mathcal{L}} = (0.128, 0.052, 0.015), append the floor 0, take the midpoint of the largest adjacent gap, and obtain ϵ=0.090\epsilon= 0.090 (Math active). Any other ϵ\epsilon in the same gap [0.052,0.128)[0.052, 0.128) yields the same allocation; the four possible cuts are:

  • ϵ≥0.128\epsilon\ge0.128: no domain is active; the write-back is empty and the checkpoint remains θ0⋆\theta_0^\star (Macro 32.52).

  • 0.052≤ϵ<0.1280.052 \le\epsilon< 0.128 (Eq. (10) on Open-MOPD, ϵ=0.090\epsilon= 0.090): only Math is active. RGBM Phase 2 under this cut writes the unlabelled grid value γ⋆=0.125Δmath\gamma^\star= 0.125\Delta_{\mathrm{math}} (Eq. (11)) and is Macro 33.03. Post-hoc write-backs at the remaining grid points and the equal-mass alternatives yield Macro 31.15 ∼\sim 32.44 (Table 10), all below 33.03.

  • 0.015≤ϵ<0.0520.015 \le\epsilon< 0.052: Code is additionally activated. Isolated Code branches improve their own domain (+0.43 ∼\sim +1.31; Table 11). Writing every branch back, as un-gated TIES and DARE do, drops Code from 24.09 at θ0⋆\theta_0^\star to 21.56 and 22.40 (Table 1).

  • ϵ<0.015\epsilon< 0.015: IF is additionally activated. Writing an IF branch back (θ0⋆+0.5Δif\theta_0^\star+ 0.5\Delta_{\mathrm{if}}) reduces Code to 20.70% and Macro to 31.10%: the leakage mechanism of § 4.4 operates even for a branch trained exclusively on IF prompts.

DomainL‾d(α⋆)\overline{\mathcal{L}}_d(\alpha^\star)Rankθ0⋆\theta_0^\starDistill. BranchOutcome-GatedRL Branch
Math0.128123.86+1.10+1.83+0.58
Code0.052224.09+0.43+1.31+0.51
IF0.015349.60+0.48+1.63+0.57

Table 11. Normalized residual spectrum vs. realized in-domain branch gains. L‾d(α⋆)\overline{\mathcal{L}}_d(\alpha^\star) vs. peak in-domain accuracy minus θ0⋆\theta_0^\star for an isolated branch trained by on-policy distillation, outcome-gated distillation, or RL. A large normalized residual does not imply a large stand-alone gain.

The Open-MOPD allocation is therefore Math-active residual-gated RGBM. On G-OPD, α⋆=(0.675,0.666)\alpha^\star= (0.675, 0.666) and L‾=(0.094,0.090)\overline{\mathcal{L}} = (0.094, 0.090): the largest gap is down to 0, so Eq. (10) gives ϵ=0.045\epsilon= 0.045 and both domains stay active.

Training Infrastructure and Hyperparameters

All branch training and distillation experiments are executed using the verl distributed RLHF framework (Sheng et al. [27]) on clusters of 8×\times NVIDIA A100/H800 GPUs. The detailed hyperparameter configuration is reported in Table 12.

HyperparameterValue
Base Model ArchitectureSmolLM3-3B (P≈3.0×109P \approx3.0 \times10^9)
OptimizerAdamW (β1=0.9\beta_1 = 0.9, β2=0.95\beta_2 = 0.95, ϵ=10−8\epsilon= 10^{-8})
Peak Learning Rate1.5×10−61.5 \times10^{-6}
Learning Rate ScheduleLinear warmup (5 steps) with cosine decay
Global Batch Size96 prompt sequences per step
Rollout Group Size (nn)8 samples per prompt
Maximum Sequence Length24,576 tokens
Maximum Model Context Length32,768 tokens
Generation Temperature0.6 (Math), 1.0 (Code), 1.0 (IF)
Checkpointing FrequencyEvery 5 steps
Weight Decay0.01

Table 12. Hyperparameter configuration for Open-MOPD branch training. G-OPD uses the official protocol of Yang et al. [35].

Computational Cost and Estimation Robustness of MdM_d

Estimating the second-moment matrices Md=EDd,πWϕd[rr⊤]∈RK×KM_d = \mathbb{E}_{D_d,\pi_{W_{\phi_d}}}[\mathbf{r}\mathbf{r}^{\top}] \in\mathbb{R}^{K \times K} requires calculating implicit rewards r(xi,yi)=[log⁡πWϕk(yi∣xi)−log⁡πWref(yi∣xi)]k=1K\mathbf{r}(x_i,y_i) = [\log\pi_{W_{\phi_k}}(y_i \mid x_i) - \log\pi_{W_{\mathrm{ref}}}(y_i \mid x_i)]_{k=1}^{K} across NN calibration prompt rollouts (N=6144N = 6144 total in our experiments, 2048 per domain).

Wall-Clock Time, Scaling, and Comparison to Full Fisher Merging.

  1. Tractability over Full-Space Fisher Inversion: Full-space Fisher merging ([21]) requires inverting a P×PP \times P matrix (P≈3.0×109P \approx3.0 \times10^{9} for SmolLM3-3B), demanding ∼1019\sim10^{19} floating-point elements and exabytes of RAM, which is completely intractable for modern LLMs. In contrast, Subspace Fisher Merge compresses the system into K×KK \times K (3×33 \times3 in our setting), reducing matrix assembly and inversion to <10< 10 ms.

  1. Inference Rollouts and Log-Prob Scoring: For K=3K = 3 specialists and a 3B model, scoring N=6144N = 6144 sequences requires O(NK)O(NK) forward passes without gradient computation or backward graph storage, taking <15< 15 minutes on 8 GPUs (<2< 2 GPU·h).

  1. Scalability across Domain Counts KK: Contemporary MOPD pipelines consistently consolidate K=2∼4K = 2 \sim4 specialist modalities (e.g., Math, Code, IF; [8, 35, 28, 4]). Even if scaled to K=20K = 20, solving the KK-dimensional Minimax QP takes <0.02< 0.02 seconds on a single CPU core. By random matrix theory, covariance estimation requires N=O(Klog⁡K)N = O(K \log K) samples, meaning N≈1024∼2048N \approx1024 \sim2048 samples remain sufficient for high-fidelity spectrum rank preservation.

Sample Complexity and Robustness to Estimation Noise. To evaluate how finite-sample variance in MdM_d propagates to the optimal weights α⋆\alpha^{\star} and the resulting normalized spectrum L\mathcal{L}, we perform bootstrap resampling of the calibration pool DcalibD_{\mathrm{calib}} across sample sizes N∈{512,1024,2048,4096,6144}N \in\{512, 1024, 2048, 4096, 6144\}:

  • At N≥1024N \ge1024, the bootstrap standard deviation of the computed weights is σ(αd⋆)<0.012\sigma(\alpha_d^{\star}) < 0.012, and the normalized spectrum remains invariant in its rank ordering: Lmath≫Lcode≫Lif\mathcal{L}_{\mathrm{math}} \gg\mathcal{L}_{\mathrm{code}} \gg\mathcal{L}_{\mathrm{if}} with >99.9%> 99.9\% confidence.

  • The condition number of empirical matrices satisfies κ(Md)≤45.2\kappa(M_d) \le45.2, well below the threshold where numerical inversion issues occur. Adding Tikhonov regularization λ=10−4\lambda= 10^{-4} provides uniform numerical stability even under rank-deficient specialist scenarios.

RGBM Phase 2 Specification (Open-MOPD)

Phase 1 θ0⋆\theta_0^{\star} (step 0): AIME24 23.91, AIME25 23.80, LCBv5 24.27, LCBv6 23.91, IFEval 73.87, IFBench 25.33, hence Math 23.86, Code 24.09, IF 49.60, Macro 32.52. The sole Open-MOPD Phase 2 scheme is residual-gated RGBM Phase 2 with Math active, written back at the unlabelled grid value γ⋆=0.125\gamma^{\star} = 0.125 (Eq. (11)): AIME24 23.80, AIME25 24.06, LCBv5 25.15, LCBv6 24.11, IFEval 75.05, IFBench 26.00, hence Math 23.93, Code 24.63, IF 50.52, Macro 33.03. On G-OPD the same unlabelled rule is used with a separate OPD branch on each domain and yields γ⋆=0.5\gamma^{\star} = 0.5.

Limitations

Errlin\mathrm{Err}_{\mathrm{lin}} certifies only the first-order reward superposition on calibration prompts; it does not bound the third-order remainder of Theorem 1. While evaluating K=2∼3K = 2 \sim3 specialists across two model families ([8, 35, 28]), scaling to K≫10K \gg10 or heterogeneous bases remains future work. Furthermore, the +0.5+0.51 Open-MOPD margin comes from Code and IF moving under a small Math write-back rather than from an in-domain Math gain; on G-OPD both domains are active, so this transfer is not isolated there.

References

  1. [1]Rishabh Agarwal, Nino Vieillard, Yongchao Zhou, Piotr Stanczyk, Sabela Ramos Garea, Matthieu Geist, and Olivier Bachem. On-policy distillation of language models: Learning from self-generated mistakes. In The Twelfth International Conference on Learning Representations, ICLR 2024, Vienna, Austria, May 7-11, 2024. OpenReview.net, 2024. URL https://openreview.net/forum?id=3zKtaqXhW.arxiv.org/abs/2306.13649
  2. [2]Yang Bai, Kaiyuan Liu, Ziyuan Zhuang, Jiahnong Zhou, Rongxiang Weng, Xin Chen, Jingang Wang, and Xunliang Cai. Multi-objective and mixed-reward reinforcement learning via reward-decorrelated policy optimization. arXiv preprint arXiv:2605.13641, 2026.arxiv.org/abs/2605.13641
  3. [3]Elie Bakouch, Loubna Ben Allal, Anton Lozhkov, Nouamane Tazi, Lewis Tunstall, Carlos Miguel Patiño, Edward Beeching, Aymeric Roucher, Aksel Joonas Reedi, Quentin Gallouédec, Kashif Rasul, Nathan Habib, Clémentine Fourrier, Hynek Kydlíček, Guilherme Penedo, Hugo Larcher, Mathieu Morlon, Vaibhav Srivastav, Joshua Lochner, Xuan-Son Nguyen, Colin Raffel, Leandro von Werra, and Thomas Wolf. SmolLM3: smol, multilingual, long-context reasoner. https://huggingface.co/blog/smollm3, 2025.
  4. [4]Tianlei Chen, Jiao Ou, Ziyuan Liu, Ruiming Tang, Jian Liang, and Han Li. Counteraction-aware multi-teacher on-policy distillation for general capability recovery with domain preservation. CoRR, abs/2605.27115, 2026. doi: 10.48550/ARXIV.2605.27115. URL https://doi.org/10.48550/arXiv.2605.27115.
  5. [5]Xueqi Cheng, Xugui Zhou, Tyler Derr, and Yushun Dong. Read: Reinforcement-guided capability distillation for large language models. CoRR, abs/2605.11290, 2026. doi: 10.48550/ARXIV.2605.11290. URL https://doi.org/10.48550/arXiv.2605.11290.
  6. [6]MohammadReza Davari and Eugene Belilovsky. Model breadcrumbs: Scaling multi-task model merging with sparse masks. In Ales Leonardis, Elisa Ricci, Stefan Roth, Olga Russakovsky, Torsten Sattler, and Gül Varol (eds.), Computer Vision – ECCV 2024 - 18th European Conference, Milan, Italy, September 29-October 4, 2024, Proceedings, Part LXXV, volume 15133 of Lecture Notes in Computer Science, pp. 270–287. Springer, 2024. doi: 10.1007/978-3-031-73226-3\_16. URL https://doi.org/10.1007/978-3-031-73226-3\_16.
  7. [7]DeepSeek-AI. Deepseek-r1: Incentivizing reasoning capability in llms via reinforcement learning. CoRR, abs/2501.12948, 2025. doi: 10.48550/ARXIV.2501.12948. URL https://doi.org/10.48550/arXiv.2501.12948.
  8. [8]Huan-ang Gao, Haohan Chi, Yong Yan, Shiyuan Feng, Hanlin Wu, Zheng Jiang, Bingxiang He, Wei-Ying Ma, Ya-Qin Zhang, and Hao Zhou. Open-mopd: Diagnosing and fixing capability imbalance in multi-teacher on-policy distillation. CoRR, abs/2608.19098, 2026. doi: 10.48550/ARXIV.2608.19098. URL https://doi.org/10.48550/arXiv.2608.19098.
  9. [9]Yuxian Gu, Li Dong, Furu Wei, and Minlie Huang. MiniLLM: Knowledge distillation of large language models. In The Twelfth International Conference on Learning Representations, ICLR 2024, Vienna, Austria, May 7-11, 2024. OpenReview.net, 2024. URL https://openreview.net/forum?id=5h0qf7IBZZ.DOI
  10. [10]Arnav Gudibande, Eric Wallace, Charlie Snell, Xinyang Geng, Hao Liu, Pieter Abbeel, Sergey Levine, and Dawn Song. The false promise of imitating proprietary llms. volume abs/2305.15717, 2023. doi: 10.48550/ARXIV.2305.15717. URL https://doi.org/10.48550/arXiv.2305.15717.
  11. [11]Geoffrey E. Hinton, Oriol Vinyals, and Jeffrey Dean. Distilling the knowledge in a neural network. CoRR, abs/1503.02531, 2015. URL http://arxiv.org/abs/1503.02531.
  12. [12]Gabriel Ilharco, Mitchell Wortsman, Samir Yitzhak Gadre, Shuran Song, Hannaneh Hajishirzi, Simon Kornblith, Ali Farhadi, and Ludwig Schmidt. Patching open-vocabulary models by interpolating weights. In Sanmi Koyejo, S. Mohamed, A. Agarwal, Danielle Belgrave, K. Cho, and A. Oh (eds.), Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, NeurIPS 2022, New Orleans, LA, USA, November 28 - December 9, 2022, 2022. URL http://papers.nips.cc/paper_files/paper/2022/hash/bc6cddcd5d325e1c0f82606c1ad0215-Abstract-Conference.html.arxiv.org/abs/2208.05592
  13. [13]Gabriel Ilharco, Marco Túlio Ribeiro, Mitchell Wortsman, Ludwig Schmidt, Hannaneh Hajishirzi, and Ali Farhadi. Editing models with task arithmetic. In The Eleventh International Conference on Learning Representations, ICLR 2023, Kigali, Rwanda, May 1-5, 2023. OpenReview.net, 2023. URL https://openreview.net/forum?id=6tOKwf8-jrj.
  14. [14]Naman Jain, King Han, Alex Gu, Wen-Ding Li, Fanjia Yan, Tianjun Zhang, Sida Wang, Armando Solar-Lezama, Koushik Sen, and Ion Stoica. Livecodebench: Holistic and contamination free evaluation of large language models for code. In The Thirteenth International Conference on Learning Representations, ICLR 2025, Singapore, April 24-28, 2025. OpenReview.net, 2025. URL https://openreview.net/forum?id=chfJJYc31L.arxiv.org/abs/2403.07974
  15. [15]Xisen Jin, Xiang Ren, Daniel Preotiuc-Pietro, and Pengxiang Cheng. Dateless knowledge fusion by merging weights of language models. In The Eleventh International Conference on Learning Representations, ICLR 2023, Kigali, Rwanda, May 1-5, 2023. OpenReview.net, 2023. URL https://openreview.net/forum?id=FCnohuR6AnM.arxiv.org/abs/2212.09849
  16. [16]Yoon Kim and Alexander M. Rush. Sequence-level knowledge distillation. In Jian Su, Xavier Carreras, and Kevin Duh (eds.), Proceedings of the 2016 Conference on Empirical Methods in Natural Language Processing, EMNLP 2016, Austin, Texas, USA, November 1-4, 2016, pp. 1317–1327. The Association for Computational Linguistics, 2016. doi: 10.18653/V1/D16-1139. URL https://doi.org/10.18653/v1/d16-1139.
  17. [17]Kaiyuan Liu, Chen Shen, Zhanwei Zhang, Junjie Liu, Xiaosong Yuan, and Jieping Ye. Efficient reasoning through suppression of self-affirmation reflections in large reasoning models. CoRR, abs/2506.12353, 2025a. doi: 10.48550/ARXIV.2506.12353. URL https://doi.org/10.48550/arXiv.2506.12353.
  18. [18]Kaiyuan Liu, Shaotian Yan, Rui Miao, Bing Wang, Chen Shen, Jun Zhang, and Jieping Ye. Where did this sentence come from? tracing provenance in llm reasoning distillation. arXiv preprint arXiv:2512.20908, 2025b.
  19. [19]Kaiyuan Liu, Ziyuan Zhuang, Yang Bai, Bing Wang, Rongxiang Weng, and Jieping Ye. Prefix teach, suffix fade: Local teachability collapse in strong-to-weak on-policy distillation. arXiv preprint arXiv:2605.13643, 2026.arxiv.org/abs/2605.13643
  20. [20]Zhenyi Lu, Chenghao Fan, Wei Wei, Xiaoye Qu, Dangyang Chen, and Yu Cheng. Twin-merging: Dynamic integration of modular expertise in model merging. In Amir Globerson, Lester Mackey, Danielle Belgrave, Angela Fan, Ulrich Paquet, Jakub M. Tomczak, and Cheng Zhang (eds.), Advances in Neural Information Processing Systems 37: Annual Conference on Neural Information Processing Systems 2024, NeurIPS 2024, Vancouver, BC, Canada, December 10 - 15, 2024. URL http://papers.nips.cc/paper_files/paper/2024/hash/8fcd17eb91bae20d9826786d7d6be799-Abstract-Conference.html.arxiv.org/abs/2406.15479
  21. [21]Michael Matena and Colin Raffel. Merging models with fisher-weighted averaging. In Sanmi Koyejo, S. Mohamed, A. Agarwal, Danielle Belgrave, K. Cho, and A. Oh (eds.), Advances in Neural Information Processing Systems 35: Annual Conference on Neural Information Processing Systems 2022, NeurIPS 2022, New Orleans, LA, USA, November 28 - December 9, 2022, 2022. URL http://papers.nips.cc/paper_files/paper/2022/hash/70c26937fbf3d4600b69a129031b66ec-Abstract-Conference.html.arxiv.org/abs/2111.09832
  22. [22]Guillermo Ortiz-Jiménez, Alessandro Favero, and Pascal Frossard. Task arithmetic in the tangent space: Improved editing of pre-trained models. In Alice Oh, Tristan Naumann, Amir Globerson, Kate Saenko, Moritz Hardt, and Sergey Levine (eds.), Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS 2023, New Orleans, LA, USA, December 10 - 16, 2023, 2023. URL http://papers.nips.cc/paper_files/paper/2023/hash/d28077e5ff52034cd35b4aa15320acea-Abstract-Conference.html.
  23. [23]Valentina Pyatkin, Saumya Malik, Victoria Graf, Hamish Ivison, Shengyi Huang, Pradeep Dasigi, Nathan Lambert, and Hanna Hajishirzi. Generalizing verifiable instruction following. 2025. URL http://papers.nips.cc/paper_files/paper/2025/hash/46499a0622ecf568b72d17b61e45dbd5-Abstract-Datasets_and_Benchmarks_Track.html.arxiv.org/abs/2507.02833
  24. [24]Rafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning, Stefano Ermon, and Chelsea Finn. Direct preference optimization: Your language model is secretly a reward model. In Alice Oh, Tristan Naumann, Amir Globerson, Kate Saenko, Moritz Hardt, and Sergey Levine (eds.), Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS 2023, New Orleans, LA, USA, December 10 - 16, 2023, 2023. URL http://papers.nips.cc/paper_files/paper/2023/hash/a85b405ed65c6477a4fe8302b5e06ce7-Abstract-Conference.html.arxiv.org/abs/2305.18290
  25. [25]Aniruddha Roy, Jyoti Patel, Aman Chadha, Vinija Jain, and Amitava Das. Alignmerge - alignment-preserving large language model merging via fisher-guided geometric constraints. CoRR, abs/2512.16245, 2025. doi: 10.48550/ARXIV.2512.16245. URL https://doi.org/10.48550/arXiv.2512.16245.
  26. [26]Zhihong Shao, Peiyi Wang, Qihao Zhu, Runxin Xu, Junxiao Song, Mingchuan Zhang, Y. K. Li, Y. Wu, and Daya Guo. Deepseekmath: Pushing the limits of mathematical reasoning in open language models. CoRR, abs/2402.03300, 2024. doi: 10.48550/ARXIV.2402.03300. URL https://doi.org/10.48550/arXiv.2402.03300.
  27. [27]Guangming Sheng, Chi Zhang, Zilingfeng Ye, Xibin Wu, Wang Zhang, Ru Zhang, Yanghua Peng, Haibin Lin, and Chuan Wu. Hybridflow: A flexible and efficient RLHF framework. In Proceedings of the Twentieth European Conference on Computer Systems, EuroSys 2025, Rotterdam, The Netherlands, 30 March 2025 - 3 April 2025, pp. 1279–1297. ACM, 2025. doi: 10.1145/3689031.3696075. URL https://doi.org/10.1145/3689031.3696075.
  28. [28]Zechen Sun, Zhiwei Zhang, Fei Zhao, Juntao Li, Mu Chuan, Huayu Deng, Guojian Zhan, Wenliang Chen, Yao Hu, and Min Zhang. D³-modp: Adaptive dynamic domain scheduling for efficient multi-teacher distillation. CoRR, abs/2608.24987, 2026. doi: 10.48550/ARXIV.2608.24987. URL https://doi.org/10.48550/arXiv.2608.24987.
  29. [29]Kimi Team. Kimi k1.5: Scaling reinforcement learning with llms. CoRR, abs/2501.12599, 2025a. doi: 10.48550/ARXIV.2501.12599. URL https://doi.org/10.48550/arXiv.2501.12599.
  30. [30]Qwen Team. Qwen3 technical report. CoRR, abs/2505.09388, 2025b. doi: 10.48550/ARXIV.2505.09388. URL https://doi.org/10.48550/arXiv.2505.09388.
  31. [31]Prateek Yadav, Derek Tam, Leshem Choshen, Colin A. Raffel, and Mohit Bansal. Ties-merging: Resolving interference when merging models. In Alice Oh, Tristan Naumann, Amir Globerson, Kate Saenko, Moritz Hardt, and Sergey Levine (eds.), Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, NeurIPS 2023, New Orleans, LA, USA, December 10 - 16, 2023, 2023. URL http://papers.nips.cc/paper_files/paper/2023/hash/1644c9af28ab7916874f6fd6228a9bcf-Abstract-Conference.html.arxiv.org/abs/2306.01708
  32. [32]Shaotian Yan, Kaiyuan Liu, Chen Shen, Bing Wang, Sinan Fan, Jun Zhang, Yue Wu, and Zheng Wang. Wiping ve. distribution-aligned sequence distillation for superior long-cot reasoning. arXiv preprint arXiv:2601.09088, 2026.arxiv.org/abs/2601.09088
  33. [33]Enneng Yang, Zhenyi Wang, Li Shen, Shiwei Liu, Guibing Guo, Xingwei Wang, and Dacheng Tao. Adamerging: Adaptive model merging for multi-task learning. In The Twelfth International Conference on Learning Representations, ICLR 2024, Vienna, Austria, May 7-11, 2024. OpenReview.net, 2024. URL https://openreview.net/forum?id=nZJ6NgD3QY.arxiv.org/abs/2310.02575
  34. [34]Enneng Yang, Li Shen, Guibing Guo, Xing-Wei Wang, Xiaochun Cao, Jie Zhang, and Dacheng Tao. Model merging in llms, mllms, and beyond: Methods, theories, applications, and opportunities. ACM Comput. Surv., 58(8):216:1–216:41, 2026a. doi: 10.1145/3787849. URL https://doi.org/10.1145/3787849.
  35. [35]Wenkai Yang, Weijie Liu, Ruobing Xie, Kai Yang, Saiyong Yang, and Yankai Lin. Learning beyond teacher: Generalized on-policy distillation with reward extrapolation. CoRR, abs/2602.12125, 2026b. doi: 10.48550/ARXIV.2602.12125. URL https://doi.org/10.48550/arXiv.2602.12125.

Paper details

vlnc:W_nv7p2h8b
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents