The Practical Side of Representation Geometry
Deciding which pretrained models to join, and how, before the training run

Two papers at NeurIPS 2026

MLBio lab, EPFL

Neural networks trained on different data and objectives converge, and our earlier work shows that what reliably converges is local: the models agree on which points are each other's nearest neighbors. In two new papers we connect this local signal to a practical decision about combining two pretrained models. COMPASS turns it into a score that predicts, before any training, whether a pair is compatible, whether you align their representation spaces or merge their weights. PORTAL finds that the projector connecting a vision encoder to a language model only has to roughly line up their two feature distributions, and constructs it in closed form instead of training it.

When a finding about representations becomes practical

The Platonic Representation Hypothesis argues that as networks scale, their representations converge to a shared model of the world, a trend reported across the usual representational similarity metrics. Our earlier work, the Aristotelian Representation Hypothesis [3], re-examines that claim. After correcting for two confounders that these similarity metrics have, the trend reported by global metrics largely flattens, while the trend from local neighborhood metrics holds. What reliably converges is local neighborhood structure: which points are each other's nearest neighbors.

Local neighborhood alignment is typically measured by mutual $k$-NN: embed a small probe set through each model and, for every sample, take the fraction of its $k$ nearest neighbors that both representation spaces share. This needs no training, and the cost grows with the number of models, not the number of pairs: each model is embedded once, and its neighbors are reused against every candidate partner.

Here we show how to make these convergence findings practical, by connecting what two models share locally to how you combine them. COMPASS predicts, before training, which pairs are worth combining. PORTAL uses the same understanding to replace the training of a projector with a closed-form construction.

Whether to connect them COMPASS · NeurIPS 2026

Combining two pretrained models can mean multiple things. You can align them, learning a map that lines up their two representation spaces so the same inputs land in the same place, as when a vision encoder and a text encoder are tied together for retrieval (cross-modal alignment). Or you can merge two models fine-tuned on different tasks into a single set of weights (task arithmetic). Either way, whether a given pair combines well is normally known only after you run that alignment or merge, so choosing a pair means a search over training runs. COMPASS [1] gives the answer first, as a score computed from forward passes.

COMPASS overview: candidate vision-language pairs, local neighborhood overlap per probe sample, and the ease and ceiling scores that rank each pair and sort it into one of four regimes.
COMPASS in one figure. (1) Before training, the compatibility of each candidate pair is unknown. (2) For each probe sample, compare its $k$ nearest neighbors across the two representation spaces. (3) Aggregate this local overlap into ease and ceiling, which rank the pairs and group them into four regimes.

COMPASS scores a pair on two numbers, both built from the same local overlap: for each sample in a small probe set, the fraction of its $k$ nearest neighbors that the two models share.

  • Ease is pair-specific. How much does this pair already agree, before any training (before encoder training in alignment, before merging in task arithmetic)? Two models that already agree on which points are nearest neighbors need little training to align fully. Two that start far apart need much more.
  • Ceiling is paradigm-level. How much agreement can the two training paradigms support at all, averaged over models drawn from them? Paradigms with very different objectives, say a reconstructive encoder and an autoregressive language model, can only agree so far, however well a single pair happens to start.

COMPASS: model compatibility diagnostic

Given a library of candidate models and a way to combine them (for example, cross-modal alignment or task-arithmetic merging):

  1. Compute ease and ceiling on the probe set. Ease is the pair-specific $k$-NN overlap, measured at the start of the relevant training dynamics. Ceiling is a paradigm-level mean $k$-NN overlap that estimates the upper bound on agreement the two paradigms can support. Both scores lie in $[0, 1]$, and higher is better.
  2. Rank pairs by the product of ease and ceiling. Combine the two scores into one ranking, and prioritize the pairs that score higher.
  3. Read the quadrant as a fallback diagnostic. Splitting each axis at its median gives four regimes: act now (high ease, high ceiling), budget-limited (low ease, high ceiling), ceiling-limited (high ease, low ceiling), and incompatible (low ease, low ceiling). For budget-limited, invest in more data, optimization, or a stronger partner. For ceiling-limited, switch to a paradigm whose target is closer to the downstream target. Deprioritize incompatible pairs.

The widget below runs step 1. Watch the local neighborhood overlap, and the ease and ceiling behind it, change as you move the two sliders.

Each dot is one probe item, the same items placed by the two models. Hover one to see its $k$ nearest neighbors in each space, with shared neighbors marked in red. Ease slides this pair from scrambled to aligned. Ceiling sets how many items the two paradigms can match at all, so the rest turn grey. The red count is the local overlap, and both sliders lower it.

representation space X · vision model

representation space Y · language model

shared neighbors of the highlighted point: 6 / 6

COMPASS score (ease × ceiling)0.49 regimeact now
ease this pair now · 0.70
ceiling paradigm cap · 0.70

Act now. High ease and high ceiling. Align or merge this pair now. The run is cheap and likely to succeed.

The four regimes read off the two axes. Act-now pairs (high ease, high ceiling) are the strongest immediate candidates. Budget-limited pairs (low ease, high ceiling) can still succeed, but need more data, more optimization, or a stronger partner. Ceiling-limited pairs (high ease, low ceiling) already agree, but their paradigms limit the agreement they can reach, so a paradigm whose objective is closer to the target helps more. Incompatible pairs are low on both axes and are deprioritized.

Bar charts of absolute correlation with downstream success for ease times ceiling, ease, ceiling, and global CKA, across Flickr30k, COCO, and task-arithmetic merging.
The score tracks what happens downstream. Across 72 vision-language pairs and 392 task-arithmetic merges, ease × ceiling predicts downstream success far better than global CKA, a standard global similarity measure. Pearson correlation reaches $0.90$ on Flickr30k and $0.91$ on COCO image-to-text R@1, against $0.63$ and $0.68$ for CKA, and $0.70$ on merge retention against $0.25$.

How to connect them PORTAL · NeurIPS 2026

A multimodal LLM is built in two stages. First a projector, a small module, is trained on hundreds of thousands of image-caption pairs to map visual tokens into the language model's input space (LLaVA and the models after it). Then the full model is trained together. This first stage is treated as essential, yet what it learns has stayed unclear, and earlier reports even disagree on whether it helps. PORTAL [2] makes a concrete claim: the stage performs only a coarse alignment, moving visual features into a region the language model can read, which many different projectors do equally well.

PORTAL in one figure, three panels: the projector alignment pipeline with a frozen vision encoder and language model, projectors trained on different datasets interpolating without loss of performance, and PORTAL as a closed-form optimal-transport map between the vision and language feature distributions.
PORTAL in one figure. (Left) The standard pipeline freezes the vision encoder and the language model and trains only the small projector between them. (Middle) Two projectors trained on different caption datasets stay strong at every blend along the straight line between them, and the midpoint blend does best of all, so the set of good projectors is a broad region, not a single point. (Right) PORTAL constructs the projector directly, as an optimal-transport map between the two feature distributions.

Two findings say the stage learns only a coarse alignment. First, the interpolation in the middle panel: if training reached one precise, unique projector, averaging two independently trained projectors would break it, yet the average holds and often beats both. Second, only $10$ to $20\%$ of the standard $558$k-pair corpus already recovers most of the downstream gain. Neither a broad interpolation path nor such early saturation fits a narrow target.

Since many projectors work, the stage may not need gradient descent to find one. PORTAL (Projector via Optimal-tRansport ALignment) initializes the projector in closed form from each modality's statistics. It keeps the standard two-layer MLP projector but fixes the second layer to the identity, so only the first affine map $(\mathbf{W}_1, \mathbf{b}_1)$ has to be chosen, as the Gaussian optimal-transport map between the vision and language feature distributions:

$$ \mathbf{W}_1 = \mathbf{V}_t \operatorname{diag}\!\left(\frac{\boldsymbol{\sigma}_t}{\boldsymbol{\sigma}_s}\right) \mathbf{V}_s^{\top}, \qquad \mathbf{b}_1 = \boldsymbol{\mu}_t - \mathbf{W}_1\,\boldsymbol{\mu}_s. $$

Here $\boldsymbol{\mu}_s, \boldsymbol{\mu}_t$ are the means, $\mathbf{V}_s, \mathbf{V}_t$ the top-$k$ principal directions, and $\boldsymbol{\sigma}_s, \boldsymbol{\sigma}_t$ the spread along each, for the vision (source) and language (target) sides. The map whitens the vision features, rotates them into the language model's principal directions, and rescales each to the language model's spread, and $\mathbf{b}_1$ recenters on its mean. It needs one PCA per side, no paired captions, and no gradient steps, and it removes the projector training stage rather than shortening it. The same construction also initializes other projector designs, a plain linear projector and the larger C-Abstractor, not only the two-layer MLP.

Across seven (vision encoder, language model) pairs spanning four language-model families (Vicuna, LLaMA, Qwen, OLMo) from 0.5B to 8B parameters and two vision encoders (CLIP, SigLIP), the training-free projector matches the standard pipeline trained on $558$k image-caption pairs, and beats it in three of the seven settings, peaking at $106\%$ of the trained baseline. The gap that remains is on OCR-heavy benchmarks, where fine character-level detail still rewards the paired supervision that per-modality statistics cannot capture.

90 95 100 105 trained baseline = 100% 106.2 LLaMA-3.2-1B CLIP 106.1 Qwen-2.5-0.5B CLIP 102.7 LLaMA-3.1-8B CLIP 99.3 Vicuna-1.5 SigLIP 98.8 Vicuna-1.5 CLIP 98.3 Qwen-2.5-7B CLIP 96.3 OLMo-2-1B CLIP
PORTAL against the standard projector trained on $558$k image-caption pairs, scored as a percentage of it across seven (vision encoder, language model) settings. The training-free projector stays within a few points everywhere, and beats the trained baseline in three settings (bars above the dashed line).
Average score against alignment training samples: PORTAL starts well above random initialization and reaches the trained baseline with 135 times fewer FLOPs.
Used as an initialization rather than a replacement, PORTAL reaches the trained baseline with $135\times$ fewer alignment FLOPs. It starts far above a random projector and needs only a little training to close the rest.

The common idea

COMPASS and PORTAL use the same idea twice. COMPASS compares the local neighborhoods of two models to predict, before training, which pairs are worth combining. PORTAL uses the feature geometry of the vision and language sides to build the projector that links them in closed form. In both, understanding the structure that representations share does practical work: it lets COMPASS rank pairs before training, and lets PORTAL build a projector without it.

References

  1. F. Gröger, P. Damjanović, M. Brbić. To Align or Not To Align: Check Your COMPASS Before You Train. Advances in Neural Information Processing Systems (NeurIPS), 2026. arXiv coming soon.
  2. F. Gröger, N. Ouerghemi, S. Wen, M. Brbić. Rethinking Projector Training in Multimodal LLMs. Advances in Neural Information Processing Systems (NeurIPS), 2026. arXiv coming soon.
  3. F. Gröger, S. Wen, M. Brbić. Revisiting the Platonic Representation Hypothesis: An Aristotelian View. International Conference on Machine Learning (ICML), 2026. project arXiv:2602.14486 code

BibTeX

@article{groger2026compass,
  title   = {{To Align or Not To Align: Check Your COMPASS Before You Train}},
  author  = {Gr{\"o}ger, Fabian and Damjanovi{\'c}, Petar and Brbi{\'c}, Maria},
  journal = {Advances in Neural Information Processing Systems (NeurIPS)},
  year    = {2026}
}

@article{groger2026projector,
  title   = {{Rethinking Projector Training in Multimodal LLMs}},
  author  = {Gr{\"o}ger, Fabian and Ouerghemi, Na{\"e}l and Wen, Shuo and Brbi{\'c}, Maria},
  journal = {Advances in Neural Information Processing Systems (NeurIPS)},
  year    = {2026}
}