Learning Cross-Model Activation Alignments with Explicit Many-to-Many Layer Maps
Organizations: Technion · Technion, NVIDIA Research
Abstract
LLMs are released at a rapid pace, raising a natural question: how do two independently trained models relate, both in which layers correspond and in how features transform between them? We study this by learning an activation alignment, a map from a source model's layerwise activations to a target's. Our method, MATCHA, factors this map into a layer map, whose output is an explicit target-by-source matrix that can be extracted and inspected, and a layer-shared feature map between hidden spaces. Most of prior work fixes the layer correspondence in advance, pairing layers at roughly the same relative depth; in contrast, we learn both factors jointly from prompts. Across 42 pairs of seven models spanning three different families, MATCHA reconstructs the target's activations more faithfully and improves retrieval-based metrics substantially, w.r.t. previous approaches. The recovered maps are broadly monotone in depth but, in contrast with most previous approaches, are consistently many-to-many: each target layer draws on a band of source layers. Our alignments also enable transfer of activation-space interventions, allowing steering vectors and probes developed for one model to transfer to another.
Figures & tables
| comparable-size | Bigger Smaller | Smaller Bigger | ||||||||||
| Method | TJ@5 | R@1 | P@1 | J@5 | TJ@5 | R@1 | P@1 | J@5 | TJ@5 | R@1 | P@1 | J@5 |
| Terminal | 0.449 | 0.716 | 0.915 | 0.376 | 0.407 | 0.759 | 0.874 | 0.381 | – | – | – | – |
| Relative Depth | 0.447 | 0.727 | 0.917 | 0.377 | 0.411 | 0.757 | 0.918 | 0.395 | 0.435 | 0.686 | 0.856 | 0.381 |
| CKA | 0.361 | 0.544 | 0.457 | 0.204 | 0.283 | 0.634 | 0.561 | 0.256 | 0.311 | 0.567 | 0.464 | 0.217 |
| Command-V | 0.316 | 0.541 | 0.395 | 0.321 | 0.294 | 0.617 | 0.517 | 0.347 | 0.281 | 0.627 | 0.465 | 0.325 |
| HOT | 0.091 | 0.030 | 0.343 | 0.253 | 0.093 | 0.041 | 0.434 | 0.259 | 0.101 | 0.037 | 0.401 | 0.263 |
| Method | Same-size | Bigger Smaller | Smaller Bigger |
|---|---|---|---|
| Terminal | 0.647 | 0.634 | – |
| Relative Depth | 0.646 | 0.654 | 0.664 |
| CKA | 0.549 | 0.593 | 0.597 |
| Command-V | 0.589 | 0.605 | 0.601 |
| HOT | 0.642 | 0.639 | 0.652 |
| MATCHA (Ours) | 0.683 | 0.658 | 0.693 |
| Variant | R@1 | P@1 | J@5 | TJ@5 |
|---|---|---|---|---|
| MATCHA-K | 0.715 | 0.834 | 0.368 | 0.416 |
| MATCHA-FL | 0.707 | 0.794 | 0.340 | 0.412 |
| MLP | 0.460 | 0.817 | 0.318 | 0.364 |
| MATCHA (Ours) | 0.829 | 0.961 | 0.422 | 0.438 |
Appendix figures & tables36 assets
Supplementary material from the paper’s appendix.
Appendix
| Hyperparameter | # | Values swept |
| MATCHA | ||
| learning_rate | 2 | 0.0003, 0.001 |
| attention_temperature | 4 | 0.5, 1, 4, 8 |
| bottleneck_dim | 3 | 256, 512, 1024 |
| num_epochs | 3 | 10, 30, 60 |
| MATCHA-K | ||
| Target | LR | Temp. | Bottleneck | Epochs | Val loss |
| Source: Llama-3.2-1B | |||||
| Llama-3.2-3B | 0.0003 | 1 | 512 | 30 | 0.1736 |
| Llama-3.1-8B | 0.0003 | 1 | 256 | 30 | 0.1450 |
| Qwen2.5-1.5B | 0.001 | 1 | 1024 | 30 | 0.0391 |
| Qwen2.5-3B | 0.001 | 0.5 | 256 | 30 | 0.0426 |
| Qwen2.5-7B | 0.001 | 0.5 | 512 | 60 | 0.0703 |
| Target | LR | Temp. | Bottleneck | Epochs | Val loss |
| Source: Llama-3.2-1B | |||||
| Llama-3.2-3B | 0.001 | 0.5 | 1024 | 60 | 0.2092 |
| Llama-3.1-8B | 0.001 | 0.5 | 1024 | 60 | 0.1731 |
| Qwen2.5-1.5B | 0.001 | 0.5 | 1024 | 60 | 0.0445 |
| Qwen2.5-3B | 0.001 | 0.5 | 1024 | 60 | 0.0514 |
| Qwen2.5-7B | 0.001 | 0.5 | 1024 | 60 | 0.0879 |
| Target | LR | Temp. | Bottleneck | Epochs | Val loss |
| Source: Llama-3.2-1B | |||||
| Llama-3.2-3B | 0.001 | 0.5 | 512 | 30 | 0.1878 |
| Llama-3.1-8B | 0.0003 | 0.5 | 512 | 30 | 0.1584 |
| Qwen2.5-1.5B | 0.001 | 0.5 | 512 | 30 | 0.0431 |
| Qwen2.5-3B | 0.001 | 0.5 | 512 | 30 | 0.0480 |
| Qwen2.5-7B | 0.001 | 0.5 | 512 | 30 | 0.0778 |
| Target | LR | Temp. | Bottleneck | Epochs | Val loss |
| Source: Llama-3.2-1B | |||||
| Llama-3.2-3B | 0.001 | 1 | 512 | 10 | 0.2186 |
| Llama-3.1-8B | 0.001 | 0.5 | 256 | 10 | 0.1975 |
| Qwen2.5-1.5B | 0.001 | 8 | 256 | 30 | 0.0696 |
| Qwen2.5-3B | 0.001 | 0.5 | 256 | 30 | 0.0807 |
| Qwen2.5-7B | 0.001 | 0.5 | 512 | 30 | 0.1073 |
| Relative depth | Terminal | |||||
| Target | LR | Ep. | Val | LR | Ep. | Val |
| Source: Llama-3.2-1B | ||||||
| Llama-3.2-3B | 0.0003 | 30 | 0.2169 | 0.0003 | 30 | 0.3659 |
| Llama-3.1-8B | 0.0003 | 30 | 0.2023 | 0.0003 | 30 | 0.3629 |
| Qwen2.5-1.5B | 0.001 | 30 | 0.0698 | 0.001 | 30 | 0.1403 |
| Qwen2.5-3B | 0.001 | 30 | 0.0952 | 0.001 | 30 | 0.2019 |
| Target | LR | Epochs | Val loss |
| Source: Llama-3.2-1B | |||
| Llama-3.2-3B | 0.0003 | 60 | 0.2725 |
| Llama-3.1-8B | 0.0003 | 30 | 0.2567 |
| Qwen2.5-1.5B | 0.001 | 30 | 0.1350 |
| Qwen2.5-3B | 0.001 | 30 | 0.1276 |
| Qwen2.5-7B | 0.0003 | 60 | 0.1587 |
| Target | LR | Epochs | Val loss |
| Source: Llama-3.2-1B | |||
| Llama-3.2-3B | 0.0003 | 60 | 0.1740 |
| Llama-3.1-8B | 0.0003 | 60 | 0.1448 |
| Qwen2.5-1.5B | 0.0003 | 60 | 0.0394 |
| Qwen2.5-3B | 0.0003 | 60 | 0.0428 |
| Qwen2.5-7B | 0.0003 | 60 | 0.0720 |
| Target | ( routing_top_k ) | Val loss |
| Source: Llama-3.2-1B | ||
| Llama-3.2-3B | 3 | 0.1384 |
| Llama-3.1-8B | 3 | 0.1721 |
| Qwen2.5-1.5B | 10 | 5.4011 |
| Qwen2.5-3B | 10 | 6.6036 |
| Qwen2.5-7B | 5 | 5.3680 |
| Model | Family | Layers | Hidden |
|---|---|---|---|
| Llama-3.2-1B | Llama | 16 | 2048 |
| Gemma-2-2B | Gemma | 26 | 2304 |
| Qwen2.5-1.5B | Qwen | 28 | 1536 |
| Llama-3.2-3B | Llama | 28 | 3072 |
| Qwen2.5-7B | Qwen | 28 | 3584 |
| Llama-3.1-8B | Llama | 32 | 4096 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | 0.453 | 0.455 | 0.327 | 0.455 | 0.427 | 0.411 | 0.457 | 0.433 | 0.431 | 0.188 | 0.418 |
| Llama-3.1-8B | 0.461 | 0.442 | 0.259 | 0.461 | 0.413 | 0.410 | 0.464 | 0.413 | 0.393 | 0.182 | 0.400 |
| Qwen2.5-1.5B | 0.474 | 0.468 | 0.378 | 0.493 | 0.490 | 0.484 | 0.495 | 0.443 | 0.315 | 0.020 | 0.466 |
| Qwen2.5-3B | 0.431 | 0.435 | 0.339 | 0.455 | 0.428 | 0.431 | 0.455 | 0.421 | 0.317 | 0.052 | 0.402 |
| Qwen2.5-7B | 0.436 | 0.435 | 0.322 | 0.428 | 0.422 | 0.425 | 0.434 | 0.412 | 0.297 | 0.104 | 0.416 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | 0.397 | 0.783 | 0.737 | 0.857 | 0.745 | 0.758 | 0.853 | 0.744 | 0.904 | 0.063 | 0.728 |
| Llama-3.1-8B | 0.338 | 0.800 | 0.735 | 0.862 | 0.777 | 0.779 | 0.863 | 0.730 | 0.898 | 0.053 | 0.745 |
| Qwen2.5-1.5B | 0.445 | 0.612 | 0.498 | 0.762 | 0.735 | 0.679 | 0.770 | 0.593 | 0.500 | 0.037 | 0.610 |
| Qwen2.5-3B | 0.330 | 0.591 | 0.435 | 0.765 | 0.640 | 0.630 | 0.758 | 0.488 | 0.451 | 0.040 | 0.554 |
| Qwen2.5-7B | 0.412 | 0.668 | 0.573 | 0.856 | 0.619 | 0.693 | 0.862 | 0.620 | 0.557 | 0.046 | 0.632 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | 0.606 | 0.928 | 0.702 | 0.976 | 0.868 | 0.867 | 0.977 | 0.948 | 0.669 | 0.702 | 0.922 |
| Llama-3.1-8B | 0.594 | 0.948 | 0.651 | 0.989 | 0.896 | 0.853 | 0.989 | 0.960 | 0.608 | 0.739 | 0.947 |
| Qwen2.5-1.5B | 0.611 | 0.748 | 0.358 | 0.937 | 0.825 | 0.761 | 0.943 | 0.746 | 0.336 | 0.210 | 0.864 |
| Qwen2.5-3B | 0.529 | 0.821 | 0.347 | 0.951 | 0.718 | 0.712 | 0.956 | 0.699 | 0.273 | 0.199 | 0.846 |
| Qwen2.5-7B | 0.614 | 0.854 | 0.462 | 0.976 | 0.765 | 0.797 | 0.977 | 0.857 | 0.297 | 0.372 | 0.897 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | 0.271 | 0.443 | 0.320 | 0.464 | 0.415 | 0.390 | 0.467 | 0.473 | 0.385 | 0.324 | 0.380 |
| Llama-3.1-8B | 0.251 | 0.435 | 0.306 | 0.455 | 0.414 | 0.371 | 0.460 | 0.460 | 0.369 | 0.326 | 0.385 |
| Qwen2.5-1.5B | 0.299 | 0.364 | 0.177 | 0.380 | 0.345 | 0.312 | 0.384 | 0.341 | 0.292 | 0.242 | 0.313 |
| Qwen2.5-3B | 0.262 | 0.352 | 0.178 | 0.372 | 0.300 | 0.292 | 0.373 | 0.311 | 0.268 | 0.236 | 0.298 |
| Qwen2.5-7B | 0.292 | 0.381 | 0.217 | 0.398 | 0.317 | 0.320 | 0.399 | 0.371 | 0.293 | 0.261 | 0.326 |
| Target | Terminal | Relative depth | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CommandV | HOT |
|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | ||||||||||
| Llama-3.2-3B | 0.077 | 0.073 | 0.120 | 0.071 | 0.079 | 0.072 | 0.070 | 0.077 | 0.516 | 0.145 |
| Llama-3.1-8B | 0.090 | 0.113 | 0.195 | 0.085 | 0.090 | 0.088 | 0.083 | 0.098 | 0.627 | 0.180 |
| Qwen2.5-1.5B | 0.576 | 0.564 | 1.576 | 0.451 | 0.501 | 0.463 | 0.446 | 0.611 | 16.772 | 5.424 |
| Qwen2.5-3B | 0.932 | 1.481 | 1.655 | 0.724 | 0.797 | 0.738 | 0.718 | 0.953 | 49.502 | 6.687 |
| Qwen2.5-7B | 1.214 | 1.278 | 3.237 | 1.113 | 1.269 | 1.149 | 1.104 | 1.370 | 35.587 | 5.432 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | – | 0.731 | 0.696 | 0.723 | 0.701 | 0.668 | 0.726 | 0.733 | 0.653 | 0.734 | 0.696 |
| Llama-3.1-8B | – | 0.730 | 0.705 | 0.722 | 0.704 | 0.703 | 0.724 | 0.731 | 0.663 | 0.743 | 0.693 |
| Qwen2.5-1.5B | – | 0.662 | 0.648 | 0.671 | 0.669 | 0.656 | 0.668 | 0.644 | 0.641 | 0.630 | 0.659 |
| Qwen2.5-3B | – | 0.662 | 0.607 | 0.691 | 0.635 | 0.620 | 0.683 | 0.659 | 0.645 | 0.638 | 0.662 |
| Qwen2.5-7B | – | 0.679 | 0.638 | 0.680 | 0.619 | 0.653 | 0.663 | 0.642 | 0.602 | 0.631 | 0.660 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | – | 0.700 | 0.696 | 0.723 | 0.683 | 0.646 | 0.716 | 0.728 | 0.592 | 0.695 | 0.678 |
| Llama-3.1-8B | – | 0.705 | 0.663 | 0.715 | 0.692 | 0.677 | 0.714 | 0.716 | 0.607 | 0.722 | 0.672 |
| Qwen2.5-1.5B | – | 0.555 | 0.592 | 0.619 | 0.589 | 0.579 | 0.623 | 0.596 | 0.606 | 0.545 | 0.654 |
| Qwen2.5-3B | – | 0.633 | 0.599 | 0.681 | 0.633 | 0.605 | 0.665 | 0.657 | 0.571 | 0.598 | 0.649 |
| Qwen2.5-7B | – | 0.640 | 0.622 | 0.662 | 0.580 | 0.616 | 0.638 | 0.647 | 0.562 | 0.547 | 0.628 |
| Target | Term | Rel-dep | CKA | MATCHA | MATCHA-K | MATCHA-FL | MATCHA-avg | MATCHA-no FF | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||||||
| Llama-3.2-3B | – | 20 | 13 | 13 | 21 | – | – | – | 21 | 19 | 21 |
| Llama-3.1-8B | – | 21 | 16 | 18 | 17 | – | – | – | 29 | 18 | 21 |
| Qwen2.5-1.5B | – | 16 | 18 | 20 | 18 | – | – | – | 21 | 8 | 21 |
| Qwen2.5-3B | – | 18 | 27 | 25 | 18 | – | – | – | 27 | 10 | 24 |
| Qwen2.5-7B | – | 18 | 5 | 18 | 5 | – | – | – | 21 | 2 | 8 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 89.5 1.1 | 0.3 0.0 | – | 31.5 4.0 | 84.0 | 62.9 | 80.4 7.9 |
| Llama-3.1-8B | 88.3 0.4 | 0.3 0.0 | – | 55.0 21.2 | 85.1 | 42.5 | 82.7 6.9 |
| Qwen2.5-1.5B | 33.8 5.1 | 57.6 3.0 | – | 67.4 13.3 | 72.7 | 0.5 | 38.6 22.8 |
| Qwen2.5-3B | 49.1 7.2 | 0.3 0.0 | – | 34.0 5.5 | 33.5 | 0.5 | 30.6 13.3 |
| Qwen2.5-7B | 90.5 0.8 | 44.2 12.5 | – | 35.7 12.8 | 77.3 | 51.3 | 80.9 10.1 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 89.1 1.1 | 2.3 0.4 | – | 49.6 4.8 | 87.5 | 63.0 | 79.4 9.0 |
| Llama-3.1-8B | 89.2 1.7 | 2.4 0.2 | – | 65.4 15.3 | 82.0 | 51.0 | 83.8 6.4 |
| Qwen2.5-1.5B | 37.3 4.8 | 68.8 6.2 | – | 57.8 7.4 | 61.0 | 3.0 | 54.0 12.3 |
| Qwen2.5-3B | 51.1 7.5 | 2.5 0.2 | – | 44.0 7.2 | 40.0 | 3.5 | 49.1 11.7 |
| Qwen2.5-7B | 89.9 1.1 | 61.3 4.4 | – | 48.1 7.4 | 73.5 | 65.0 | 80.4 11.5 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 83.6 1.6 | 1.7 0.5 | – | 39.0 3.8 | 86.0 | 55.0 | 75.2 9.2 |
| Llama-3.1-8B | 84.7 1.2 | 1.3 0.5 | – | 57.7 15.6 | 82.0 | 45.0 | 77.9 5.9 |
| Qwen2.5-1.5B | 36.3 5.5 | 63.0 4.3 | – | 57.3 11.0 | 61.0 | 1.0 | 44.4 14.4 |
| Qwen2.5-3B | 47.1 8.8 | 1.4 0.5 | – | 40.1 6.0 | 33.0 | 1.0 | 40.2 12.4 |
| Qwen2.5-7B | 84.0 1.3 | 49.6 6.6 | – | 43.3 7.6 | 64.0 | 61.0 | 74.3 10.4 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 89.5 1.1 | 0.3 0.0 | – | 42.3 16.3 | 82.2 | 62.9 | 80.4 7.9 |
| Llama-3.1-8B | 87.0 1.3 | 0.3 0.0 | – | 40.7 23.3 | 76.0 | 4.1 | 79.6 7.9 |
| Qwen2.5-1.5B | 15.3 13.6 | 38.5 5.6 | – | 41.4 7.7 | 18.8 | 0.0 | 31.1 18.7 |
| Qwen2.5-3B | 17.4 10.8 | 0.3 0.0 | – | 32.9 8.5 | 12.1 | 0.3 | 21.9 13.3 |
| Qwen2.5-7B | 19.0 5.9 | 44.2 12.5 | – | 3.7 1.5 | 77.3 | 0.8 | 66.8 13.3 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 89.1 1.1 | 2.3 0.4 | – | 50.5 6.6 | 83.0 | 63.0 | 79.4 9.0 |
| Llama-3.1-8B | 84.0 1.3 | 2.4 0.2 | – | 57.9 14.9 | 71.0 | 12.0 | 79.9 6.8 |
| Qwen2.5-1.5B | 20.4 15.1 | 51.8 4.6 | – | 42.4 6.2 | 28.5 | 2.5 | 43.8 12.2 |
| Qwen2.5-3B | 28.6 11.4 | 2.5 0.2 | – | 38.7 8.4 | 23.5 | 0.0 | 35.4 12.8 |
| Qwen2.5-7B | 31.7 6.1 | 61.3 4.4 | – | 14.4 3.4 | 73.5 | 4.5 | 65.2 10.2 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | 83.6 1.6 | 1.7 0.5 | – | 45.3 11.2 | 86.0 | 55.0 | 75.2 9.2 |
| Llama-3.1-8B | 80.2 3.2 | 1.3 0.5 | – | 47.7 15.3 | 65.0 | 9.0 | 74.7 8.1 |
| Qwen2.5-1.5B | 20.0 14.1 | 48.9 6.8 | – | 40.4 6.8 | 25.0 | 1.0 | 35.2 15.9 |
| Qwen2.5-3B | 24.7 15.2 | 1.4 0.5 | – | 40.3 5.8 | 12.0 | 1.0 | 26.4 14.7 |
| Qwen2.5-7B | 30.0 7.0 | 49.6 6.6 | – | 9.2 2.4 | 64.0 | 1.0 | 59.0 9.2 |
| Target | Ours | Rel-dep | Term | CKA | CmdV | HOT | MLP |
|---|---|---|---|---|---|---|---|
| Source: Llama-3.2-1B | |||||||
| Llama-3.2-3B | -10 | -0.5 | – | -4 | -6 | -10 | -10 |
| Llama-3.1-8B | -8 | -0.5 | – | -6 | -2 | -8 | -8 |
| Qwen2.5-1.5B | 6 | 8 | – | -8 | 1.5 | -0.5 | 8 |
| Qwen2.5-3B | -8 | -0.5 | – | 6 | 2 | -0.5 | -8 |
| Qwen2.5-7B | -4 | -1 | – | -6 | -1 | -8 | -8 |