Transporting Task Vectors across Different Architectures without Training
About
Adapting large pre-trained models to downstream tasks often produces task-specific parameter updates that are expensive to relearn for every model variant. While recent work has shown that such updates can be transferred between models with identical architectures, transferring them across models of different widths remains unexplored. In this work, we introduce Theseus, a training-free method for transporting task updates across heterogeneous-width models. Rather than matching parameters, we characterize a task update by the functional effect it induces on intermediate representations. We formalize task-vector transport as a functional matching problem on observed activations and show that, after aligning representation spaces via orthogonal Procrustes analysis, it admits a stable closed-form solution that preserves the geometry of the update. We evaluate Theseus on vision and language models across different widths, showing consistent improvements over baselines without additional training or backpropagation. Our results show that task updates can be meaningfully transferred across architectures when task identity is defined functionally rather than parametrically. Code is available at https://github.com/apanariello4/merge-and-rebase.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | Stanford Cars | Accuracy85.51 | 660 | |
| Classification | Cars | Accuracy88.58 | 492 | |
| Image Classification | DTD | Accuracy72.71 | 487 | |
| Image Classification | RESISC45 | Accuracy75.02 | 472 | |
| Image Classification | SUN397 | Accuracy77.16 | 450 | |
| Image Classification | SVHN | Accuracy82.82 | 395 | |
| Image Classification | GTSRB | Accuracy73.59 | 291 | |
| Image Classification | 8-task vision benchmark | Average Accuracy74.95 | 193 | |
| Image Classification | SVHN | Top-1 Accuracy71.05 | 186 | |
| Image Classification | SUN397 | Accuracy73.97 | 116 |