Dual-Model Weight Selection
- 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 | |
| Two-network weight-ensemble | Model pair | |
| Data-free model merge | Global scaling | |
| Medical transfer initialization | Complementary masks | |
| Reversible low-rank merge | Basis and coefficients | |
| Dual-weight learning | Excitatory/inhibitory components |
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 , two regressors , prediction vectors 0 and 1 with 2 and 3, and response vector 4, the objective is
5
Equivalently, with
6
the loss reduces to the univariate quadratic
7
Differentiation yields the unconstrained minimizer
8
and the constrained solution is
9
Because the leading coefficient is 0 and the feasible set 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 2 and 3, leading to time complexity 4 and memory 5 extra. The same framework extends to weighted data points by replacing
6
with convexity preserved. If desired, an 7 penalty 8 shifts the solution to
9
again clipped to 0 (Echtenbruck et al., 2022).
Empirically, the paper reports that on a 4-D “MSG” test function with 1 Latin-Hypercube points and leave-one-out CV, the best single Gaussian-kernel Kriging model achieved CV-RMSE 2, the spline-kernel model achieved 3, and the dual-model QP ensemble achieved 4, a reduction of approximately 5 versus the best single model. On the CHEMBL4159 drug-activity dataset (6), combining a random-forest and a LASSO regression increased ROC-AUC from 7 for the random forest alone to 8 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,
9
and forms a dual-model weight-ensemble
0
Selection of the pair 1 can be posed as maximizing
2
or simply maximizing 3 when only in-distribution labels are available. To guide pair selection, the paper defines functional diversity by the ratio-error metric
4
where 5 is the number of examples both models get wrong and 6 is the number of examples exactly one of them gets wrong, and weight-space diversity by
7
For two-model soups, the “greedy” and “greedier” procedures coincide; a ranked-diversity heuristic first sorts candidates by descending 8 or 9 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 0–1 ID accuracy gain over greedy’s choice and up to approximately 2–3 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 4 and 5 with parameters 6 and 7, a standard linear merge introduces a global scaling parameter 8:
9
Weight Weaving replaces one-shot selection of a single 0 with pooling over a search grid 1. For each 2, 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 3 to 4, MagMax from 5 to 6, TIES from 7 to 8, PCB from 9 to 0, TSV from 1 to 2, and ISO-C from 3 to 4. The recommended default is 5 in steps of 6, extending to 7 for methods such as PCB; arithmetic mean is the default pooling choice, and 8–9 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
0
Two smaller student models 1 and 2 with identical architectures but fewer channels are initialized from disjoint subsets of the mapped teacher tensor 3. For each layer 4, index sets
5
define binary masks 6 and 7, and the initialization is
8
By construction, every weight of the teacher is used exactly once across 9 and 0. 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 1 (Tsutsumi et al., 28 Aug 2025).
After initialization, the method freezes 2 as an auxiliary model updated only by EMA and trains 3 using classification and self-knowledge-distillation losses. With logits 4, temperature-scaled softmax outputs are
5
The distillation loss is
6
the classification loss is standard cross-entropy,
7
and the total loss is
8
The main student is updated by SGD or Adam, while the auxiliary model follows
9
with 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 01 images and 02 classes, Lung CT with 03 images and 04 classes, and Brain MRI with 05 images and 06 classes. Teacher-to-student configurations include ViT-S to ViT-T and ConvNeXt-T to ConvNeXt-F. With 07 training data, reported test-accuracy gains include Chest X-ray with ViT-T from 08 for CM1 to 09 for the proposed method, Chest X-ray with ConvNeXt-F from 10 to 11, Lung CT with ViT-T from 12 to 13, and Brain MRI with ConvNeXt-F from 14 to 15. The reported overhead is approximately 16–17 additional GPU memory and approximately 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 19 and the Data Weighting Model parameters 20. The lower-level objective is
21
and the upper-level objective is
22
The DWM takes a mini-batch of 23 token sequences, embeds each sequence using a frozen or shallow-finetuned LM encoder, applies one self-attention block over the batch embedding matrix 24, and maps the resulting rows to raw weights 25 via two feed-forward layers. A softmax or ReLU normalization produces non-negative weights summing to 26. Training alternates between updating 27 with 28 frozen and updating 29 with 30 frozen, over 31 stages in the reported experiments. On a 32M model trained on 33B tokens, zero-shot average accuracy rose from 34 to 35 and two-shot average accuracy from 36 to 37 for RANDOM versus RANDOM + DWM. The learned weighting model also transferred from RANDOM to DSIR and QuRating, and on a 38B model imposed only approximately 39 extra FLOPs because large-model training used only forward passes through 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 41 and 42, with 43 and 44. For one row or column position, define task vectors 45 and stack them into
46
Rather than storing both vectors separately, Reversible Model Merging seeks a basis 47 and coefficients 48 minimizing
49
For the two-model case, choosing 50 allows exact reconstruction. The optimal basis is given by the top-51 right singular vectors of 52, equivalently the top-53 eigenvectors of 54, and the coefficients are orthogonal projections
55
Because 56, one has 57 exactly. The reported computational cost is 58 per row or column position and 59 per layer, with storage 60 instead of 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.
6. Limitations, misconceptions, and related directions
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 62 then 63 can become very small, leading to numerical instability; the paper notes that one may add a small ridge term 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 65 to balance labelled and unlabelled likelihood terms,
66
and proposed selecting
67
The same paper reports that BIC and ICL are monotone in 68 and pick a boundary, whereas 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 70 and 71 with effective weight 72 and reports that inference complexity remains the same as in the standard 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.