SOTAlign: Semi-Supervised Alignment of Unimodal Vision and Language Models via Optimal Transport
About
The Platonic Representation Hypothesis posits that neural networks trained on different modalities converge toward a shared statistical model of the world. Recent work exploits this convergence by aligning frozen pretrained vision and language models with lightweight alignment layers, but typically relies on contrastive losses and millions of paired samples. In this work, we ask whether meaningful alignment can be achieved with substantially less supervision. We introduce a semi-supervised setting in which pretrained unimodal encoders are aligned using a small number of image-text pairs together with large amounts of unpaired data. To address this challenge, we propose SOTAlign, a two-stage framework that first recovers a coarse shared geometry from limited paired data using a linear teacher, and then refines the alignment on unpaired samples via an optimal-transport-based divergence that transfers relational structure without overconstraining the target space. SOTAlign effectively leverages unpaired images and text, learning robust joint embeddings across datasets and encoder pairs, and significantly outperforming supervised and semi-supervised baselines. Code is available at https://github.com/ExplainableML/SOTAlign.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | Food-101 | -- | 590 | |
| Image Classification | DTD | Top-1 Accuracy42.4 | 26 | |
| Image-to-Text Retrieval | COCO 100 Pairs (test) | Recall@535.8 | 4 | |
| Image-to-Text Retrieval | Flickr30k 500 Pairs (test) | Recall@559.8 | 4 | |
| Image-to-Text Retrieval | Polyvore 500 Pairs (test) | Recall@555.3 | 4 | |
| Image-to-Text Retrieval | SkyScript 10k (test) | Recall@1027.4 | 4 | |
| Speech-to-Text Retrieval | LibriSpeech 2.6k pairs (test) | Recall@178.4 | 4 | |
| Text-to-Image Retrieval | COCO 100 Pairs (test) | Recall@535 | 4 | |
| Text-to-Image Retrieval | Flickr30k 500 Pairs (test) | Recall@563.3 | 4 | |
| Text-to-Image Retrieval | Polyvore 500 Pairs (test) | Recall@555.3 | 4 |