Papers
Topics
Authors
Recent
Search
2000 character limit reached

Dual-Model Weight Selection

Updated 9 July 2026
  • The paper introduces dual-model weight selection as a novel framework that blends two model-derived weight sources to optimize downstream criteria like RMSE and validation accuracy.
  • Dual-model weight selection is defined as combining weights via convex mixtures, direct averaging, or reversible basis constructions to leverage complementary model strengths.
  • Practical implementations demonstrate actionable gains, such as a 5% RMSE reduction in regression tasks and improved ROC-AUC in drug activity and medical imaging benchmarks.

Dual-model weight selection denotes a class of procedures in which two model-derived weight sources are combined, selected, or reconstructed to optimize a downstream criterion. In current arXiv usage, the phrase encompasses convex mixtures of two regression predictors, direct averaging of two neural networks, data-free interpolation between two expert models, complementary partitioning of a teacher’s weights into two students, and reversible basis constructions for two low-rank updates (Echtenbruck et al., 2022, Rojas et al., 2024, Chaves et al., 15 Oct 2025, Tsutsumi et al., 28 Aug 2025, Alipour et al., 15 Oct 2025). The common structure is an optimization over a low-dimensional selection variable—typically a scalar weight, a model pair, a basis, or a pair of complementary masks—while the performance criterion ranges from RMSE and validation accuracy to exact reconstruction and data-free robustness. This suggests that dual-model weight selection is better understood as a family of two-source parameter-allocation problems than as a single canonical algorithm.

1. Scope of the term and representative formulations

The literature uses closely related language for several technically distinct constructions. In each case, two models, two updates, or two complementary weight sets are the primitive objects, but the selected quantity differs. The table summarizes representative formulations (Echtenbruck et al., 2022, Rojas et al., 2024, Chaves et al., 15 Oct 2025, Tsutsumi et al., 28 Aug 2025, Alipour et al., 15 Oct 2025, Wang, 15 Mar 2025).

Setting Selected object Representative formulation
Two-regressor ensemble Convex scalar weight f^w(x)=wf^1(x)+(1w)f^2(x)\hat f_w(\mathbf{x}) = w\,\hat f_1(\mathbf{x}) + (1-w)\,\hat f_2(\mathbf{x})
Two-network weight-ensemble Model pair (i,j)(i,j) wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)
Data-free model merge Global scaling λ\lambda θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B
Medical transfer initialization Complementary masks WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T
Reversible low-rank merge Basis and coefficients xiWciTx_i \approx W c_i^T
Dual-weight learning Excitatory/inhibitory components W=W1W2W = W_1 - W_2

A central distinction is whether the procedure produces a single merged object or retains the ability to recover two specialized objects. Convex ensembling and direct weight averaging collapse two sources into one predictor or one parameter vector. By contrast, reversible model merging explicitly constructs a compact basis from which the original task-specific models can be recovered by linear combination (Alipour et al., 15 Oct 2025). Complementary teacher-weight partitioning occupies a third category: two lightweight models are initialized from disjoint subsets of a larger pretrained model, after which one model is trained and the other is maintained by EMA in a self-knowledge-distillation pipeline (Tsutsumi et al., 28 Aug 2025).

This diversity of usage matters because “selection” can mean at least four different operations: choosing a scalar coefficient, choosing a second model to average with an anchor model, choosing a basis dimension or subspace, or choosing complementary subsets of pretrained weights. Confusion between these regimes is a common source of category error.

2. Exact convex selection for two regression models

In the most classical formulation, dual-model weight selection is the problem of finding a convex mixture of two base regressors that minimizes mean squared error. Given training data {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n, two regressors f^1,f^2\hat f_1,\hat f_2, prediction vectors (i,j)(i,j)0 and (i,j)(i,j)1 with (i,j)(i,j)2 and (i,j)(i,j)3, and response vector (i,j)(i,j)4, the objective is

(i,j)(i,j)5

Equivalently, with

(i,j)(i,j)6

the loss reduces to the univariate quadratic

(i,j)(i,j)7

Differentiation yields the unconstrained minimizer

(i,j)(i,j)8

and the constrained solution is

(i,j)(i,j)9

Because the leading coefficient is wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)0 and the feasible set wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)1 is convex, the problem is a convex program with a unique global minimizer (Echtenbruck et al., 2022).

For exactly two models, no general QP solver is required. A single pass over the data is sufficient to accumulate wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)2 and wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)3, leading to time complexity wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)4 and memory wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)5 extra. The same framework extends to weighted data points by replacing

wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)6

with convexity preserved. If desired, an wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)7 penalty wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)8 shifts the solution to

wavg:=12(wi+wj)w_{\mathrm{avg}} := \tfrac12(w_i+w_j)9

again clipped to λ\lambda0 (Echtenbruck et al., 2022).

Empirically, the paper reports that on a 4-D “MSG” test function with λ\lambda1 Latin-Hypercube points and leave-one-out CV, the best single Gaussian-kernel Kriging model achieved CV-RMSE λ\lambda2, the spline-kernel model achieved λ\lambda3, and the dual-model QP ensemble achieved λ\lambda4, a reduction of approximately λ\lambda5 versus the best single model. On the CHEMBL4159 drug-activity dataset (λ\lambda6), combining a random-forest and a LASSO regression increased ROC-AUC from λ\lambda7 for the random forest alone to λ\lambda8 for the two-model ensemble in stratified CV (Echtenbruck et al., 2022).

3. Direct weight averaging, diversity, and data-free interpolation

A second major usage arises in neural network weight-ensembling. Here one starts from a pool of pre-trained or fine-tuned networks of identical architecture,

λ\lambda9

and forms a dual-model weight-ensemble

θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B0

Selection of the pair θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B1 can be posed as maximizing

θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B2

or simply maximizing θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B3 when only in-distribution labels are available. To guide pair selection, the paper defines functional diversity by the ratio-error metric

θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B4

where θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B5 is the number of examples both models get wrong and θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B6 is the number of examples exactly one of them gets wrong, and weight-space diversity by

θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B7

For two-model soups, the “greedy” and “greedier” procedures coincide; a ranked-diversity heuristic first sorts candidates by descending θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B8 or θmerged(λ)=(1λ)θA+λθB\theta_{\mathrm{merged}}(\lambda) = (1-\lambda)\,\theta_A + \lambda\,\theta_B9 and evaluates them in that order. In OfficeHome experiments with ResNet-50 and 40 fine-tuned models per trial across 10 trials, greedier’s first addition yielded approximately WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T0–WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T1 ID accuracy gain over greedy’s choice and up to approximately WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T2–WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T3 OOD gain. Euclidean-ranked behaved similarly to ratio-error-ranked, and the authors conclude that high diversity enhances weight-ensembling while qualifying the extent to which diversity alone improves accuracy (Rojas et al., 2024).

A related but distinct problem appears in data-free model merging. Given two fine-tuned expert networks WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T4 and WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T5 with parameters WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T6 and WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T7, a standard linear merge introduces a global scaling parameter WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T8:

WlSMlSWˉlT,  WlSMlSWˉlTW_l^S \leftarrow M_l^S \odot \bar W_l^T,\; W_l^{S'} \leftarrow M_l^{S'} \odot \bar W_l^T9

Weight Weaving replaces one-shot selection of a single xiWciTx_i \approx W c_i^T0 with pooling over a search grid xiWciTx_i \approx W c_i^T1. For each xiWciTx_i \approx W c_i^T2, one forms a candidate merge and then applies a pooling function to the resulting set of weight vectors. The paper gives three exemplar pooling operators: arithmetic mean, random uniform per-parameter selection, and max-magnitude pooling (MagMax). In the reported three-scenario benchmarks, Weight Weaving improved several state-of-the-art merges, including Breadcrumbs from xiWciTx_i \approx W c_i^T3 to xiWciTx_i \approx W c_i^T4, MagMax from xiWciTx_i \approx W c_i^T5 to xiWciTx_i \approx W c_i^T6, TIES from xiWciTx_i \approx W c_i^T7 to xiWciTx_i \approx W c_i^T8, PCB from xiWciTx_i \approx W c_i^T9 to W=W1W2W = W_1 - W_20, TSV from W=W1W2W = W_1 - W_21 to W=W1W2W = W_1 - W_22, and ISO-C from W=W1W2W = W_1 - W_23 to W=W1W2W = W_1 - W_24. The recommended default is W=W1W2W = W_1 - W_25 in steps of W=W1W2W = W_1 - W_26, extending to W=W1W2W = W_1 - W_27 for methods such as PCB; arithmetic mean is the default pooling choice, and W=W1W2W = W_1 - W_28–W=W1W2W = W_1 - W_29 is reported as sufficient (Chaves et al., 15 Oct 2025).

These two lines of work share the same outer problem—how to exploit two pretrained models without additional inference cost at runtime—but they optimize different objects. Weight-ensembling selects a partner model, whereas Weight Weaving marginalizes over interpolation coefficients.

4. Complementary teacher-weight partitioning and self-knowledge distillation

In medical image classification, dual-model weight selection has been used as an initialization strategy for lightweight students derived from a larger pretrained teacher. Let the teacher be

{(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n0

Two smaller student models {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n1 and {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n2 with identical architectures but fewer channels are initialized from disjoint subsets of the mapped teacher tensor {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n3. For each layer {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n4, index sets

{(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n5

define binary masks {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n6 and {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n7, and the initialization is

{(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n8

By construction, every weight of the teacher is used exactly once across {(xi,yi)}i=1n\{(\mathbf{x}_i,y_i)\}_{i=1}^n9 and f^1,f^2\hat f_1,\hat f_20. In the reported experiments, the index sets were chosen by a uniform-slice strategy, and no explicit loss is introduced at the selection stage beyond the structural coverage constraint f^1,f^2\hat f_1,\hat f_21 (Tsutsumi et al., 28 Aug 2025).

After initialization, the method freezes f^1,f^2\hat f_1,\hat f_22 as an auxiliary model updated only by EMA and trains f^1,f^2\hat f_1,\hat f_23 using classification and self-knowledge-distillation losses. With logits f^1,f^2\hat f_1,\hat f_24, temperature-scaled softmax outputs are

f^1,f^2\hat f_1,\hat f_25

The distillation loss is

f^1,f^2\hat f_1,\hat f_26

the classification loss is standard cross-entropy,

f^1,f^2\hat f_1,\hat f_27

and the total loss is

f^1,f^2\hat f_1,\hat f_28

The main student is updated by SGD or Adam, while the auxiliary model follows

f^1,f^2\hat f_1,\hat f_29

with (i,j)(i,j)00 and stop-gradient on the auxiliary branch (Tsutsumi et al., 28 Aug 2025).

The paper evaluates the approach on chest X-ray images, lung computed tomography scans, and brain magnetic resonance imaging scans. The reported datasets are Chest X-ray with (i,j)(i,j)01 images and (i,j)(i,j)02 classes, Lung CT with (i,j)(i,j)03 images and (i,j)(i,j)04 classes, and Brain MRI with (i,j)(i,j)05 images and (i,j)(i,j)06 classes. Teacher-to-student configurations include ViT-S to ViT-T and ConvNeXt-T to ConvNeXt-F. With (i,j)(i,j)07 training data, reported test-accuracy gains include Chest X-ray with ViT-T from (i,j)(i,j)08 for CM1 to (i,j)(i,j)09 for the proposed method, Chest X-ray with ConvNeXt-F from (i,j)(i,j)10 to (i,j)(i,j)11, Lung CT with ViT-T from (i,j)(i,j)12 to (i,j)(i,j)13, and Brain MRI with ConvNeXt-F from (i,j)(i,j)14 to (i,j)(i,j)15. The reported overhead is approximately (i,j)(i,j)16–(i,j)(i,j)17 additional GPU memory and approximately (i,j)(i,j)18 additional training time (Tsutsumi et al., 28 Aug 2025).

5. Bi-level and reversible formulations

Dual-model weight selection also appears in formulations where one model learns how to weight inputs for another model. In “LLM Data Selection and Utilization via Dynamic Bi-level Optimization,” the two learned objects are the LLM parameters (i,j)(i,j)19 and the Data Weighting Model parameters (i,j)(i,j)20. The lower-level objective is

(i,j)(i,j)21

and the upper-level objective is

(i,j)(i,j)22

The DWM takes a mini-batch of (i,j)(i,j)23 token sequences, embeds each sequence using a frozen or shallow-finetuned LM encoder, applies one self-attention block over the batch embedding matrix (i,j)(i,j)24, and maps the resulting rows to raw weights (i,j)(i,j)25 via two feed-forward layers. A softmax or ReLU normalization produces non-negative weights summing to (i,j)(i,j)26. Training alternates between updating (i,j)(i,j)27 with (i,j)(i,j)28 frozen and updating (i,j)(i,j)29 with (i,j)(i,j)30 frozen, over (i,j)(i,j)31 stages in the reported experiments. On a (i,j)(i,j)32M model trained on (i,j)(i,j)33B tokens, zero-shot average accuracy rose from (i,j)(i,j)34 to (i,j)(i,j)35 and two-shot average accuracy from (i,j)(i,j)36 to (i,j)(i,j)37 for RANDOM versus RANDOM + DWM. The learned weighting model also transferred from RANDOM to DSIR and QuRating, and on a (i,j)(i,j)38B model imposed only approximately (i,j)(i,j)39 extra FLOPs because large-model training used only forward passes through (i,j)(i,j)40 (Yu et al., 22 Jul 2025).

A different generalization is reversible merging for low-rank weights. Let two fine-tuned low-rank updates be (i,j)(i,j)41 and (i,j)(i,j)42, with (i,j)(i,j)43 and (i,j)(i,j)44. For one row or column position, define task vectors (i,j)(i,j)45 and stack them into

(i,j)(i,j)46

Rather than storing both vectors separately, Reversible Model Merging seeks a basis (i,j)(i,j)47 and coefficients (i,j)(i,j)48 minimizing

(i,j)(i,j)49

For the two-model case, choosing (i,j)(i,j)50 allows exact reconstruction. The optimal basis is given by the top-(i,j)(i,j)51 right singular vectors of (i,j)(i,j)52, equivalently the top-(i,j)(i,j)53 eigenvectors of (i,j)(i,j)54, and the coefficients are orthogonal projections

(i,j)(i,j)55

Because (i,j)(i,j)56, one has (i,j)(i,j)57 exactly. The reported computational cost is (i,j)(i,j)58 per row or column position and (i,j)(i,j)59 per layer, with storage (i,j)(i,j)60 instead of (i,j)(i,j)61. The paper’s central claim is that exact reconstruction implies no drop in task performance because the recovered low-rank updates coincide with the originals (Alipour et al., 15 Oct 2025).

Taken together, these works broaden weight selection beyond static scalar interpolation. One line learns weights dynamically by validation-aligned bi-level optimization; another replaces irreversible collapse with basis selection that preserves recoverability.

Several limitations recur across the literature. In the convex two-regressor setting, dual-model mixing cannot capture higher-order interactions beyond two predictors, and if (i,j)(i,j)62 then (i,j)(i,j)63 can become very small, leading to numerical instability; the paper notes that one may add a small ridge term (i,j)(i,j)64 in practice (Echtenbruck et al., 2022). In neural weight-ensembling, high diversity is beneficial but not sufficient by itself to guarantee the best pair, and ranked-diversity often underperforms greedier even when it matches or slightly outperforms greedy on OOD evaluation (Rojas et al., 2024). In reversible merging, the argument is explicitly not that one merged model can dominate all specialized models; rather, the paper states that no merged model can consistently outperform one specialized for its task, motivating recoverability rather than unconditional collapse (Alipour et al., 15 Oct 2025).

A related misconception is that “weight selection” always refers to model coefficients. Earlier statistical work on fractionally-supervised classification instead used a single weight parameter (i,j)(i,j)65 to balance labelled and unlabelled likelihood terms,

(i,j)(i,j)66

and proposed selecting

(i,j)(i,j)67

The same paper reports that BIC and ICL are monotone in (i,j)(i,j)68 and pick a boundary, whereas (i,j)(i,j)69 is the most stable of the criteria examined (Gallaugher et al., 2017). At the opposite end of the design space, the dual-weight learning framework of Wang and Shimazaki decomposes each neuron’s parameters into nonnegative (i,j)(i,j)70 and (i,j)(i,j)71 with effective weight (i,j)(i,j)72 and reports that inference complexity remains the same as in the standard (i,j)(i,j)73 setup, because one may collapse the representation after training (Wang, 15 Mar 2025).

These adjacent formulations indicate that the phrase “dual-model weight selection” sits within a broader methodological landscape of two-source allocation, decomposition, and interpolation. What unifies the area is not a single algorithmic template but the recurring attempt to exploit complementarity between two parameterized objects while controlling optimization cost, inference cost, or reconstruction error.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Dual-Model Weight Selection.