SuperAI: Merging Capabilities Across Heterogeneous Foundation Models
Organizations: University of Sydney Sydney, Australia · National University of Singapore Singapore
Abstract
Enabling the reuse and composition of knowledge and capabilities distributed across foundation models is an important step toward building scalable artificial intelligence (AI) systems. However, architectural differences and incompatible representations make it difficult to transfer and integrate these capabilities through a unified cross-model interface. To address this challenge, we propose SuperAI, a cross-architecture multi-model fusion framework that integrates complementary capabilities from multiple source models into a host model through orthogonal alignment of low-dimensional task-relevant representations. Specifically, we first extract task-relevant representations from the source models and the host model using shared anchor samples, and apply principal component analysis (PCA) to construct low-dimensional functional subspaces with the same dimensionality. Next, we use orthogonal Procrustes mappings to align the source and host coordinate systems, establishing cross-model correspondences while preserving the internal geometry of each source subspace. Finally, we train a lightweight residual multilayer perceptron (MLP) with joint supervision on task outputs and source-model increments, learning input-dependent residual increments that are injected into the host model for multi-model capability fusion. Experiments on three host models across 15 capability domains show that SuperAI achieves higher average domain scores than the evaluated cross-architecture fusion methods and better preserves general capabilities as the number of source models increases. Ablation studies further support the effectiveness of orthogonal alignment, nonlinear residual mapping, and joint supervision.
Figures & tables
| Method | Math | Safety | Law | Phys. | Sci. | C.Sense | Social | Code | Multi. | Fin./Bus. | Astro. | Med. | Climate | Cyber | Chem. | Avg. |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Domain | 67.7 | 72.0 | 46.5 | 80.9 | 76.5 | 81.6 | 75.2 | 81.5 | 69.4 | 15.0 | 74.3 | 58.4 | 71.4 | 36.2 | 49.0 | 63.7 |
| Host model: Llama-3.2-3B | ||||||||||||||||
| Original | 16.7 | 56.8 | 38.0 | 68.3 | 67.9 | 66.2 | 58.7 | 39.7 | 56.1 | 3.7 | 60.5 | 50.9 | 42.3 | 31.4 | 44.0 | 46.7 |
| THESEUS | 24.1 | 57.3 | 40.3 | 62.9 | 63.5 | 60.0 | 53.8 | 42.1 | 44.0 | 12.7 | 54.2 | 40.6 | 38.4 | 34.7 | 32.1 | 44.0 ( ) |
| T&M | 17.6 | 59.5 | 41.2 | 61.4 | 64.8 | 61.7 | 54.2 | 41.7 | 44.3 | 10.9 | 49.7 | 40.3 | 35.8 | 31.2 | 43.4 | 43.8 ( ) |
| SuperAI | 40.7 | 70.9 | 40.8 | 89.1 | 69.8 | 68.2 | 63.3 | 28.3 | 55.0 | 14.7 | 57.9 | 48.2 | 72.5 | 48.9 | 44.0 | 54.2 ( ) |
| Objective | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| 15-domain average score | |||
| Original | 46.7 | 56.2 | 22.9 |
| 50.6 ( ) | 57.5 ( ) | 27.1 ( ) | |
| 44.8 ( ) | 56.3 ( ) | 24.0 ( ) | |
| Joint | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |
| Cosine-similarity gain | |||
| Mapping Method | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| Original | 46.7 | 56.2 | 22.9 |
| No Alignment | 47.6 ( ) | 56.1 ( ) | 21.0 ( ) |
| PCA Only | 47.8 ( ) | 56.9 ( ) | 28.6 ( ) |
| PCA + Linear | 53.2 ( ) | 57.5 ( ) | 27.2 ( ) |
| PCA + Orthogonal | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |
| Mapper | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| Original | 46.7 | 56.2 | 22.9 |
| Linear | 50.6 ( ) | 56.0 ( ) | 30.1 ( ) |
| MLP | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |
| Layer Pair | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| 15-domain average score | |||
| Original | 46.7 | 56.2 | 22.9 |
| Weakly Matched | 51.1 ( ) | 59.0 ( ) | 28.4 ( ) |
| Medium Matched | 53.5 ( ) | 60.4 ( ) | 31.6 ( ) |
| Best Matched | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |
| Post-alignment Match score | |||
| Host Model | Math | Safety | Law | Physics | Avg. |
|---|---|---|---|---|---|
| Qwen2.5-1.5B | 14.3 ( ) | 72.5 ( ) | 49.3 ( ) | 94.0 ( ) | 57.5 ( ) |
| Qwen2.5-3B | 45.7 ( ) | 76.6 ( ) | 53.5 ( ) | 94.9 ( ) | 67.7 ( ) |
| Qwen2.5-7B | 57.3 ( ) | 79.3 ( ) | 56.3 ( ) | 95.7 ( ) | 72.2 ( ) |
| Qwen2.5-14B | 57.7 ( ) | 80.2 ( ) | 59.2 ( ) | 97.9 ( ) | 73.8 ( ) |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
| Domain | Source Model | Training Dataset | Task Metric |
|---|---|---|---|
| Science | ScienceQA-OFT-8B | ARC-Challenge | Accuracy |
| Common Sense | CommonsenseQA-OFT-8B | CommonsenseQA | Accuracy |
| Social | Social-IQA-OFT-8B | BIG-bench Social IQA | Accuracy |
| Math | NuminaMath-OFT-8B | SVAMP | Numeric Answer Accuracy |
| Code | Qwen2.5-Coder-7B-Instruct | MBPP | Pass@1 |
| Multilingual | Llama-3.2-3B Multilingual | M-ARC | Averaged Accuracy |
| Symbol | Definition |
|---|---|
| Models and representations | |
| Host model and source models. | |
| Hidden dimensions of the host and source model . | |
| Shared anchor prompts for source model and the host. | |
| Layer index and last non-padding-token hidden state of model . | |
| Matrix of hidden states for prompts in . | |
| PCA Rank | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| Base | 46.7 | 56.2 | 22.9 |
| 16 | 56.3 ( ) | 60.2 ( ) | 28.7 ( ) |
| 32 | 55.3 ( ) | 61.3 ( ) | 28.3 ( ) |
| 64 | 54.5 ( ) | 61.2 ( ) | 29.0 ( ) |
| 128 | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |
| 256 | 55.2 ( ) | 60.8 ( ) | 28.9 ( ) |
| Examples/Domain | Llama-3.2-3B | Qwen2.5-3B | Pythia-2.8B |
|---|---|---|---|
| Base | 46.7 | 56.2 | 22.9 |
| 50 | 54.1 ( ) | 57.9 ( ) | 27.8 ( ) |
| 100 | 56.3 ( ) | 59.6 ( ) | 27.1 ( ) |
| 200 | 57.6 ( ) | 58.9 ( ) | 26.9 ( ) |
| 500 | 55.6 ( ) | 60.0 ( ) | 27.4 ( ) |
| 1000 | 54.2 ( ) | 61.3 ( ) | 32.8 ( ) |