Stitching of Experts (SoE) Strategies
- Stitching of Experts (SoE) is a modular composition strategy that integrates independently trained expert models via lightweight learned interfaces.
- It fuses experts at various levels—hidden states, intermediate features, or outputs—to retain domain-specific strengths while enhancing generalist performance.
- The plug-and-play design of SoE supports scalable, privacy-preserving model fusion by allowing experts to be added or removed without full retraining.
Searching arXiv for papers and the specific SoE-related works to ground the article. Stitching of Experts (SoE) denotes a class of methods for combining pre-trained expert models through lightweight learned interfaces rather than full joint retraining. In the literature, the term covers several closely related mechanisms: hidden-state stitching between frozen LLM experts in Branch-Train-Stitch (BTS), bidirectional intermediate-feature translation between independently trained segmentation networks, deep blockwise fusion of audio and video generators, and, in a broader usage, lightweight supervised fusion over expert outputs (Zhang et al., 31 Jan 2025, Guijt et al., 19 Dec 2025, Wang et al., 7 Sep 2025, Wang et al., 2023). This suggests that SoE is best understood as a family of modular composition strategies whose common objective is to preserve expert specialization while improving generalist or cross-domain behavior.
1. Conceptual scope and defining characteristics
Across the cited works, SoE is organized around a recurring pattern: expert models are first trained independently or specialized after branching from a shared initialization, and a comparatively small set of additional parameters is then learned to reconcile their internal representations or outputs. In BTS, a single seed LLM is branched into domain-specific experts and later reconnected by stitch layers inserted between frozen experts and the seed LLM (Zhang et al., 31 Jan 2025). In medical segmentation, a stitching layer is a small parametric function that translates the representation of one expert into the feature space expected by another, typically using a convolution or a linear projection (Guijt et al., 19 Dec 2025). In UniVerse-1, SoE deeply fuses the corresponding blocks of pre-trained video and music generation experts models through cross-modal attention and small projection adapters (Wang et al., 7 Sep 2025). In "Fusing Models with Complementary Expertise," the broader SoE/FoE formulation treats fusion as supervised learning over expert outputs, using a convex combination with a lightweight gating network (Wang et al., 2023).
A common source of ambiguity is whether SoE denotes a single architecture. The reported literature does not support that interpretation. Instead, the concrete implementations differ in where fusion occurs—hidden activations, matched intermediate vertices in a directed acyclic graph, paired multimodal Transformer blocks, or expert outputs—while sharing an emphasis on lightweight learned coupling.
2. Branch-Train-Stitch and the LLM formulation
In "BTS: Harmonizing Specialized Experts into a Generalist LLM," the SoE mechanism is instantiated as Branch-Train-Stitch over a seed Transformer LLM of $2.7$ B parameters with layers, hidden dimension , FFN dimension $4d=12288$, $24$ attention heads, SwiGLU activation, RoPE positional embeddings, and vocab size $128$ K (Zhang et al., 31 Jan 2025). The branch step clones the seed into 0 identical copies 1 for code, math, and multilingual specialization; each copy starts with exactly the same weights as 2, so total parameters 3 B 4 B.
The train step performs continual pretraining of each expert on a domain mixture 5 of 6 B total tokens using standard next-token prediction: 7 The optimization uses 8 k steps, batch size 9 M tokens, and initial LR 0 warmed-up then cosine-decayed to 1 over training. The result is three specialized experts 2 that outperform 3 in their domain but may forget other domains.
The stitch step keeps 4 completely frozen and introduces 5 stitch layers per model, placed after every 6 Transformer blocks. At stitch point 7, hidden states are collected from the hub 8 and each expert: 9 Two alternating stitch types are used. In Experts-into-Hub (E0H),
1
In Hub-into-Experts (H2E),
3
After fusion, the modified states 4 feed into the next Transformer block of each model, and the final output is taken from the hub model’s last layer.
The modular expansion rule is explicit. To add expert 5, one branches a fresh copy of 6, trains it on 7, expands 8 and 9 in every stitch layer to handle the new expert, and trains only the expanded $2.7$0 on a small mix that includes the new domain. To remove expert $2.7$1, one simply drops its $2.7$2 and removes its hidden track in fusion; no retraining of other experts or seed is needed. The paper identifies this as a “plug-and-play” property yielding a fully modular SoE system.
3. Stitch layers, placement, and architectural reconciliation
The most explicit formalization of a stitch appears in the medical segmentation setting. For independently trained expert networks $2.7$3 and $2.7$4, with feature maps $2.7$5 and $2.7$6, a stitching layer $2.7$7 is defined as
$2.7$8
and is implemented in most experiments as a $2.7$9 convolution or as a linear projection when 0. In either case the total parameter count of one stitch is on the order of 1 (Guijt et al., 19 Dec 2025).
Because nnU-Net-style segmentation models form directed acyclic graphs with skip-connections, stitch placement is handled by representing each network as a DAG, topologically sorting it, and assigning each vertex a normalized depth
2
Vertices are matched only if they operate at the same spatial scale and 3 is small; a modified Hirschberg-style algorithm is then used to find a maximum-similarity bipartite matching under the acyclicity constraint. Empirically, stitches placed in the mid-to-late decoder 4 yield the best performance gains, whereas early stitches 5 give negligible benefit and very late stitches 6 give little room for corrective interaction.
UniVerse-1 addresses a different placement problem: the two experts do not initially have equal depth. If the video expert has 7 Transformer blocks and the audio expert has 8 blocks with 9, the method inserts 0 new video blocks at uniform fractional offsets between the original ones and initializes each inserted block by linear interpolation,
1
After this step both streams have the same depth 2, block 3 of video is paired with block 4 of audio, and each fused block performs unimodal attention together with cross-modal adapter injection and cross-attention (Wang et al., 7 Sep 2025).
A plausible implication is that SoE design is governed less by a single canonical stitch operator than by the representational geometry of the experts being coupled: sequential Transformer stacks admit periodic stitch points, DAG segmentation networks require scale- and depth-aware matching, and multimodal generators require layer-depth reconciliation before blockwise fusion.
4. Training objectives and optimization regimes
The optimization strategy in SoE depends on the level at which experts are stitched. In BTS, only stitch parameters 5 are updated while all constituent LLMs remain frozen. The stitch-layer training uses a data mix of 6 code, 7 math, 8 multilingual, and 9 remaining seed-domain text, corresponding to $4d=12288$0 B tokens over $4d=12288$1 k steps. The objective minimizes the next-token loss of the hub output only: $4d=12288$2 Hyperparameters are batch size $4d=12288$3 M tokens, LR warmup $4d=12288$4 over $4d=12288$5 k steps, then cosine decay to $4d=12288$6, with only $4d=12288$7 updated (Zhang et al., 31 Jan 2025).
In the medical setting, SoE proceeds in two phases. Phase A trains $4d=12288$8 and $4d=12288$9 solitarily on disjoint datasets $24$0 and $24$1. Phase B freezes $24$2 and optimizes only the stitches $24$3. Two alternative losses are studied. The direct matching loss is
$24$4
An optional fine-tuning term is
$24$5
The paper emphasizes a double-batched, MSE with downstream awareness scheme in which each batch is doubled so that both “reference” and “stitched” activations can be propagated through downstream layers, allowing all stitches $24$6 to be back-propagated in one shot (Guijt et al., 19 Dec 2025).
UniVerse-1 departs from the frozen-expert pattern. After copying pretrained weights from WAN2.1 into the video stream and from Ace-step into the audio stream, and after inserting interpolated video blocks plus randomly initialized adapter projections and a shared LayerNorm, the model is trained with
$24$7
where $24$8. Optimization uses AdamW, $24$9, batch $128$0, $128$1 K steps, grad-accum $128$2, and FSDP, with no progressive unfreezing: all parameters are trained from step $128$3 (Wang et al., 7 Sep 2025).
The FoE formulation treats stitching or fusion as supervised learning over expert outputs. Given expert outputs concatenated into $128$4, the fusion net $128$5 is trained by minimizing
$128$6
with a convex combination
$128$7
where $128$8 is produced by a small feed-forward MLP followed by softmax (Wang et al., 2023).
5. Empirical results across domains
The reported results span language modeling, medical image segmentation, multimodal generation, and generic expert fusion. The following table lists representative outcomes exactly as reported.
| Setting | SoE instantiation | Reported result |
|---|---|---|
| Dense $128$9 B LLM generalist | BTS with three experts and four stitch layers | avg 00, MBPP 01, MATH 02 |
| Russian-math cross task | BTS with 03 B tokens of in-domain Russian math data | Ru-MGSM 04 |
| Pelvic MRI | SoE (best-single-stitch) | on A: Dice 05 / HD95 06 mm; on B: Dice 07 / HD95 08 mm |
| Polyp segmentation | SoE | on CVC: Dice 09 / HD95 10 px; on Hyper: Dice 11 / HD95 12 px |
| Verse-Bench joint generation | UniVerse-1 (SoE) | ID 13, CS 14, AV-A 15 |
| CIFAR-100 super-class classification | FoE (full fusion of 16 experts) | Final accuracy 17 |
| MMLU | FoE select best expert | overall accuracy 18 |
In BTS, the main zero-/few-shot benchmark suite includes MMLU (5-shot), BBH (3-shot), MBPP (3-shot), HumanEval (0-shot), FLORES (1-shot, 7 langs, S19T & T20S), GSM8K (8-shot), and MATH (4-shot). The Seed Model finishes at average 21; individual experts trade off across domains with average 22; expert-upcycling MoE baselines reach average 23; expert-merging baselines average 24; BTS achieves the best average 25. Parameter efficiency is also explicit: total params 26 B and training params 27 M, whereas expert-upcycling MoEs train 28 B params and other merging methods train 29 B but underperform. Ablations show that one stitch layer lags with average 30 versus 31 or 32 layers at average 33, alternating fusion is crucial for cross-capability tasks such as Russian34math, and using the seed 35 as hub outperforms using any expert as hub with average 36 versus 37 (Zhang et al., 31 Jan 2025).
In medical segmentation, SoE is evaluated on pelvic MRI from LUMC vs. AUMC and endoscopy polyp segmentation from HyperKvasir vs. CVC-ClinicDB, with Dice Score and 38 Hausdorff Distance HD95 as metrics. The pelvic MRI merge-datasets baseline reaches Dice 39 / HD95 40 mm on both, while SoE closes 41 of the gap to “Merge” on cross-domain and retains 42 of solo expert accuracy. For polyp segmentation, the average relative improvement over solo experts is 43 on cross-domain error. Multi-objective plots place SoE close to the Pareto front jointly over performance on A and performance on B, whereas naïve ensembles fall inside the front and solo experts lie at the extreme corners (Guijt et al., 19 Dec 2025).
In UniVerse-1, the end-to-end comparison on Verse-Bench reports that SVG yields ID 44, CS 45, AV-A 46, whereas UniVerse-1 yields ID 47, CS 48, AV-A 49. The paper reports a 50 relative gain in ID, 51 in CS, and more than doubling of AV-A consistency. Ablations show that removing LQLS drops ID from 52, AS from 53, and worsens AV-A from 54; removing INSS changes CS from 55, FD from 56, and WER from 57. The reported convergence statement is that SoE converges stably in 58 K steps by leveraging the frozen priors of two 59B+60B models, while training from scratch on the same data took 61 K steps to reach comparable FM loss but still trailed on ID and CS metrics by 62 (Wang et al., 7 Sep 2025).
In FoE, the reported task families include CIFAR-100 super-class classification, sentiment analysis, text summarization, MMLU, and automatic summarization evaluation. On CIFAR-100, FoE reaches final accuracy 63 versus ensemble 64, confidence-based fusion 65, average single expert 66, and oracle 67. With only 68 of experts on average, FrugalFoE matches full-fusion accuracy 69. On sentiment analysis, FoE average accuracy is 70 versus best single expert average 71, with expert selection accuracy 72. On MMLU, FoE achieves 73 overall accuracy versus average expert 74 and best single expert 75 (Wang et al., 2023).
6. Modularity, interpretability, and relation to adjacent paradigms
SoE is often discussed near model merging, mixtures of experts, ensembles, and federated learning, but the cited works draw sharper distinctions. BTS is compared directly to expert-upcycling MoE baselines such as BTX and BAM and to expert-merging baselines such as Model Soup, BTM, Expert Routing, and BAM Adapters; its reported advantage is that it does not alter the constituent LLMs, and experts can be easily removed or new experts can be added with only a small amount of training (Zhang et al., 31 Jan 2025). This suggests that SoE, in this formulation, is a modular interface over intact experts rather than a weight-space merger or a sparse-routing MoE.
In the medical report, SoE is positioned against federated learning by emphasizing asynchronous collaboration, where only already trained models are shared. The workflow requires only one exchange of model weights and uses unlabeled data from whichever party is doing the stitching; labels are never shared. The paper characterizes this as asynchronous and privacy-safe, with practical relevance in medical and other privacy-sensitive domains (Guijt et al., 19 Dec 2025). A plausible implication is that SoE occupies a distinct point in the design space between post hoc ensembling and synchronous federated optimization.
Interpretability appears in BTS through gate-value visualizations showing that BTS dynamically shifts solver weights token by token, relying on the correct expert when generating code versus math versus translation, and smoothly context-switching across tasks in one prompt (Zhang et al., 31 Jan 2025). In the medical setting, the explanation is framed in terms of intermediate feature ensembling, correction of representation symmetries, double-batched downstream awareness, and the absence of catastrophic forgetting because expert weights are frozen during the majority of Phase B (Guijt et al., 19 Dec 2025). In FoE, the gating network gives an explicit expert-weight decomposition 76, while FrugalFoE casts expert selection as a sequential decision problem with a graph-search view (Wang et al., 2023).
A recurring misconception is that the only useful fusion point is the final output. The medical study states that ensembling at the final softmax cannot correct mis-detected features early in the network, whereas SoE aligns semantically high-level features through learnable bridges and averages aligned features rather than merely class probabilities (Guijt et al., 19 Dec 2025). Conversely, the FoE results show that output-level fusion can still be highly effective when the fusion module is trained rather than naïvely averaged (Wang et al., 2023). Taken together, the literature indicates that SoE is not tied to a single fusion depth; instead, the operative question is which level of representation offers the best trade-off among modularity, trainable parameter count, and cross-domain transfer.