Richard Cheam / research
Embedding diagnostics Research case study / 2026

Can you trust
an embedding?

Models turn images and captions into vectors: lists of numbers used for classification and search. This study asks what the geometry of those vectors tells us about the tasks they can still perform.

Read the study ↓
Problem
A representation can look unusual geometrically while retaining useful information. It can also look broadly distributed while search results lose their relevance.
Approach
Controlled training, fixed-encoder stress tests, dimensional compression, numerical precision checks, and paired image–caption retrieval.
Finding
Distribution, recoverable labels, neighbour identity, and caption coverage are different properties. Evaluate the property the application actually needs.
FIG. 01 / FOUR QUESTIONS, ONE REPRESENTATIONConceptual illustration
Distribution

How do vectors spread?

Geometry reads the representation without using labels.

Labels

What can a classifier recover?

A fitted classifier tests whether label information is accessible.

Identity

Which neighbours stay nearby?

Exact identities can change even when coarse labels agree.

Relevance

How much of the target is found?

Finding one positive is different from recovering the positive set.

These are schematic examples, not measured embedding projections. The experiments below evaluate each property separately.
01 / Motivation

Search needs useful neighbours.

Imagine looking through a large collection of road images for situations similar to a rainy night on a city street. An embedding-based system ranks images by the similarity of their vectors.

The problem

Checking every result requires labels or human judgement. Geometric measurements are cheaper: they describe vector spread, direction, and dimensional structure without labels. But a warning from geometry does not tell us which task has stopped working.

The research question

Can label-free measurements tell us whether an embedding still supports classification and useful search?

The study deliberately separates what the vectors look like from what a classifier or retrieval system can do with them. Its contribution is an instrument comparison with explicit limitations, rather than a new embedding model or a universal quality score.

02 / Approach

Change the representation. Keep the question fixed.

Two complementary designs test the measurements: train controlled image models, then alter cached outputs from one pretrained model. They share the diagnostic questions, rather than a directly comparable score scale.

Controlled training

Seven variants. One training loop.

A Joint-Embedding Predictive Architecture (JEPA) predicts hidden image-region representations from visible regions. The encoder turns images into vectors; the predictor performs the hidden-region prediction. [1]

Change stop-gradient, a slowly updated target encoder, distribution regularization, or a projector. All seven conditions use the same training loop. Five matched seeds per condition give 35 completed runs, measured at the final 4,000-step checkpoint.

Fixed pretrained encoder

One model. Controlled vector changes.

EmbeddingGemma 2 maps images and text into a common 768-dimensional space. The study pins one revision, extracts and caches its outputs, and evaluates those vectors without retraining the encoder. [6]

Test imposed contraction, offsets, direction restriction and noise; compare compression; perturb numerical storage and scoring; then evaluate original image–caption pairs. These post-hoc changes do not establish what naturally occurs during training.

Images / captionsPinned encoderCached vectorsControlled changeGeometry + task checks

Inside the training experiment

Stop-gradient
Prevents the target side of a prediction from being updated through that prediction’s gradient.
EMA target
A separate target encoder updated by an exponential moving average, a slowly changing copy of the main encoder.
SIGReg
Sketched Isotropic Gaussian Regularization: a published loss that encourages the embedding distribution to resemble a standard Gaussian. [2]
Projector
A temporary neural-network layer between the encoder and SIGReg. Evaluation reads the encoder, not this layer.

Shared architecture

The shared JEPA training apparatus Masked images enter the context encoder and predictor. Full images enter the target branch. The predictor matches masked target latents. SIGReg, when enabled, acts on pooled context embeddings or projector outputs. The highlighting steps through the four operations of one training step for the selected condition; it shows which computations run, not any measured value. EMA copy gradient stops here Masked input Context encoder trained Predictor Latent prediction loss Full input Target branch EMA copy, frozen Projector / bypass SIGReg, when enabled

One optimizer step: the masked branch is encoded and predicted, the full branch supplies the target latents, and the prediction loss is formed from the two.

Architecture schematic. Arrows show computation and loss inputs; spacing does not encode compute cost. The highlighting traces one optimizer step of Trainer.train_step for the selected condition and encodes no measured quantity. Evaluation always reads the context encoder.
Intervention matrix · seven conditions
ConditionStop-gradientEMA targetSIGRegProjector
EMA reference●●NoNo
Stop-gradient only●NoNoNo
SIGReg + stop-gradient●No●No
SIGRegNoNo●No
Projector + SIGReg + stop-gradient●No●●
Projector + SIGRegNoNo●●
Contracted controlNoNoNoNo

Architecture motion explains computation in one training step. It is not a visualization of measured training dynamics. SIGReg follows a validated loss implementation, but this masked-prediction system is not a reproduction of LeJEPA. [2]

Data / Evaluation

Two datasets. Different definitions of similarity.

Driving attributes test coarse visual context. Recorded image–caption pairs test whether the corresponding image or captions are recovered. Neither is a complete definition of semantic relevance.

Driving context

BDD100K

Berkeley DeepDrive contains 100,000 driving videos; this project uses annotated still images. Weather, scene type, and time of day provide three label-based checks on the embeddings. [3]

The pretrained analysis uses a BDD100K-derived driving-scene subset with 2,000 training images and 983 validation images. Training images fit the classifier and PCA basis; validation images measure classification and retrieval.

Recorded correspondence

COCO image–caption pairs

COCO contains everyday-scene photographs with human annotations. This study samples 1,000 original-release validation image groups, five captions each, to test image-to-text and text-to-image retrieval. [8]

For a caption query, its recorded parent image is the positive. For an image query, its five recorded captions are positives. Other semantically valid matches may exist without being annotated.

No labels

Geometry

Variance: centered spread. Cosine: directional agreement. RankMe: raw singular-spectrum count. Participation ratio: centered covariance-spectrum count. Their definitions and blind spots differ. [4] Mathematical explanation ↓

Labels used

Classification

Freeze the encoder and fit a simple classifier, a linear probe. Accuracy counts correct predictions. Balanced accuracy averages class recall over the frozen eligible classes, so common classes do not dominate it.

Labels used

Search

Rank candidates by cosine similarity. BDD P@10 counts neighbours sharing the query’s attribute. COCO Hit@10 asks whether any recorded positive is found; caption recall@10 counts the fraction of all five captions recovered.

Training probes use raw embeddings with validation-selected regularization. Pretrained BDD probes standardize features. The studies also use different samples and output normalization: their absolute scores and variance scales are not an encoder benchmark. Chance baselines and class support are reported beside the relevant results. Training comparison ↓

03 / Training evidence

Less spread. Labels still recoverable.

Compare an EMA reference, with a slowly updated target and stop-gradient, against the same training loop with no prevention mechanism. These final-checkpoint results are means over five seeds. [5]

Embedding spread

Embedding spreadEMA reference: 108. Unprotected control: 6.55e-05.10^-510^-310^-110^110^3EMA reference108EMA reference: 108Unprotected control6.55e-05Unprotected control: 6.55e-05
Five-seed means. Logarithmic variance axis.

Labels recovered by a classifier

Labels recovered by a classifierEMA reference: 92.71%. Unprotected control: 91.37%.0255075100EMA reference92.71%EMA reference: 92.71%Unprotected control91.37%Unprotected control: 91.37%
Five-seed means. Percent; time-of-day labels.

Time-of-day agreement in search

Time-of-day agreement in searchEMA reference: 89.49%. Unprotected control: 67.52%.0255075100EMA reference89.49%EMA reference: 89.49%Unprotected control67.52%Unprotected control: 67.52%
Five-seed means. Percent; time-of-day labels.

The unprotected control still predicts time of day at 91.37% accuracy. Its vector spread is much smaller, and its time-of-day neighbour agreement is 67.52%, compared with 89.49% for the reference.

The probe fits a classifier to recover labels. Retrieval uses the existing neighbour order. Their different readings show why a geometric warning, label decodability, and search utility must be kept separate.

04 / Pretrained stress tests

A broader spectrum can contain less useful signal.

Hold the pretrained encoder fixed and change its output vectors. The severe interventions below use the .99 setting; direction restriction keeps eight directions fitted on training vectors only.

ContractionShrink each vector.
Common offsetMove all vectors together.
Rank restrictionKeep selected directions.
NoiseAdd unstructured variation.

Read the same row across both plots: the left measures spectrum breadth, the right measures weather agreement among neighbours. Red identifies the noise intervention. The dashed marks show random-neighbour agreement, 39.87%.

Spectrum breadth / RankMe

Spectrum breadth / RankMeUnchanged embeddings: 270.81. Shrink magnitude: 270.81. Add a common offset: 2.01. Keep eight directions: 4.62. Add severe noise: 667.31.0192384576768Unchanged embeddings270.81Unchanged embeddings: 270.81Shrink magnitude270.81Shrink magnitude: 270.81Add a common offset2.01Add a common offset: 2.01Keep eight directions4.62Keep eight directions: 4.62Add severe noise667.31Add severe noise: 667.31
Raw singular-spectrum count. Higher means a broader spectrum, not necessarily more useful information.

Weather matches / P@10

Weather matches / P@10Unchanged embeddings: 57.86%. Shrink magnitude: 57.86%. Add a common offset: 57.79%. Keep eight directions: 52.73%. Add severe noise: 40.05%.0255075100Unchanged embeddings57.86%Unchanged embeddings: 57.86%Shrink magnitude57.86%Shrink magnitude: 57.86%Add a common offset57.79%Add a common offset: 57.79%Keep eight directions52.73%Keep eight directions: 52.73%Add severe noise40.05%Add severe noise: 40.05%
P@10: fraction of ten neighbours sharing the weather label. For example, 57.86% means about 5.8 weather matches per ten results. Dashed line: random-neighbour floor 39.87%.

Noise raises RankMe while weather retrieval approaches chance. Spectral breadth describes a distribution; it does not identify whether its neighbours share useful labels.

Shrinking alone gives a different response.

Unchanged embeddings → severe positive contraction. Same sample, standardized classifier, and cosine retrieval.

Variance
0.257033→2.57033e-05
Weather balanced accuracy
63.02%→63.02%
Weather retrieval P@10
57.86%→57.86%

Positive scaling reduces variance quadratically, while cosine ordering and normalized rank ratios remain invariant. Standardization also removes positive feature scaling from the probe. A common offset leaves centered variance unchanged; standardization removes that offset too. These are established mathematical responses, demonstrated here on the fixed sample.

Raw RankMe and centered participation ratio can disagree because centering changes the spectrum being measured. Neither supplies a universal threshold. Absolute variance readings from the unnormalized training study do not transfer to normalized pretrained outputs.

BDD driving-scene subset: 983 validation images, fixed encoder and sample. Effects are descriptive, without training-seed intervals. Geometry and probe table ↓ · All tested severities and supports ↗.

05 / Dimensional compression

Fewer dimensions. More than one utility curve.

Compare two ways of reducing native 768-dimensional embeddings to 512, 256, or 128 dimensions. Measure classification and retrieval separately for weather, scene, and time of day.

FIG. / TWO WAYS TO REDUCE A VECTORMethod schematic

Learned prefixes / MRL

Matryoshka Representation Learning trains nested prefixes to be useful. Keep the leading dimensions, then renormalize the shortened vector to unit length. [7]

768
512
256
128

Fitted projection / PCA

Principal Component Analysis learns directions of greatest variance from the training vectors. Subtract the training mean, project training and validation vectors into the same reduced basis, then normalize.

Center with training meanProject into training-fitted directionsNormalize the reduced vector
Prefix widths encode the actual dimensional budgets. The PCA diagram shows operations, not measured computation time. Centering and coordinate construction also differ between the methods.
Native 768MRL / solid linePCA / dashed line

Weather / Balanced accuracy

Weather / Balanced accuracyNative 63.02%. MRL 128: 54.91%. MRL 256: 61.07%. MRL 512: 62.82%. PCA 128: 62.36%. PCA 256: 58.85%. PCA 512: 55.36%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 54.91%MRL 256: 61.07%MRL 512: 62.82%PCA 128: 62.36%PCA 256: 58.85%PCA 512: 55.36%Native: 63.02%
Native 63.02%. At 128 dimensions: MRL 54.91%, PCA 62.36%.

Scene / Balanced accuracy

Scene / Balanced accuracyNative 69.52%. MRL 128: 66.70%. MRL 256: 69.80%. MRL 512: 70.56%. PCA 128: 68.37%. PCA 256: 66.27%. PCA 512: 65.89%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 66.70%MRL 256: 69.80%MRL 512: 70.56%PCA 128: 68.37%PCA 256: 66.27%PCA 512: 65.89%Native: 69.52%
Native 69.52%. At 128 dimensions: MRL 66.70%, PCA 68.37%.

Time of day / Balanced accuracy

Time of day / Balanced accuracyNative 76.44%. MRL 128: 74.19%. MRL 256: 77.03%. MRL 512: 76.73%. PCA 128: 78.17%. PCA 256: 75.46%. PCA 512: 75.10%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 74.19%MRL 256: 77.03%MRL 512: 76.73%PCA 128: 78.17%PCA 256: 75.46%PCA 512: 75.10%Native: 76.44%
Native 76.44%. At 128 dimensions: MRL 74.19%, PCA 78.17%.

Weather / P@10

Weather / P@10Native 57.86%. MRL 128: 56.33%. MRL 256: 57.33%. MRL 512: 57.77%. PCA 128: 56.19%. PCA 256: 56.04%. PCA 512: 55.78%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 56.33%MRL 256: 57.33%MRL 512: 57.77%PCA 128: 56.19%PCA 256: 56.04%PCA 512: 55.78%Native: 57.86%
Native 57.86%. At 128 dimensions: MRL 56.33%, PCA 56.19%.

Scene / P@10

Scene / P@10Native 62.44%. MRL 128: 60.78%. MRL 256: 62.20%. MRL 512: 62.23%. PCA 128: 62.25%. PCA 256: 61.97%. PCA 512: 61.79%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 60.78%MRL 256: 62.20%MRL 512: 62.23%PCA 128: 62.25%PCA 256: 61.97%PCA 512: 61.79%Native: 62.44%
Native 62.44%. At 128 dimensions: MRL 60.78%, PCA 62.25%.

Time of day / P@10

Time of day / P@10Native 82.73%. MRL 128: 80.66%. MRL 256: 82.41%. MRL 512: 82.64%. PCA 128: 84.00%. PCA 256: 83.34%. PCA 512: 83.07%.-10-50+5Change from native / percentage points128256512768Embedding dimensionsMRL 128: 80.66%MRL 256: 82.41%MRL 512: 82.64%PCA 128: 84.00%PCA 256: 83.34%PCA 512: 83.07%Native: 82.73%
Native 82.73%. At 128 dimensions: MRL 80.66%, PCA 84.00%.

No method leads on every attribute and endpoint. MRL 512 and 256 retain much of the measured native utility; at 128 dimensions, the ordering depends on what is measured.

For weather at 128 dimensions, PCA has higher balanced accuracy, while MRL has slightly higher neighbour-label agreement. Read the two curves separately: a classifier can use a direction that does not preserve the same cosine neighbours.

This is a comparison of existing compression methods on a fixed sample. It does not isolate the cause of their differences or establish population robustness, a new compression algorithm, or operational search-speed gains.

Compression axes show change from native in percentage points, with the same −10 to +5 range for every panel. Zero is the native result. Lines guide the eye between the tested dimensions, rather than predicting untested settings. Fixed-sample effects; no training-seed error bars. Exact compression table ↓.

06 / Neighbour stability

Small score errors can change who makes the top ten.

Store cached vectors with fewer bits or change similarity arithmetic, then repeat exact search. FP16 uses 16-bit floating point; INT8 uses 8-bit integer quantization. Neither condition changes the encoder’s inference precision.

FIG. / THE TOP-TEN BOUNDARYConceptual illustration
Reference order

10 Candidate A included

11 Candidate B outside

Nearly tied scoresA small perturbation can swap their order.
Perturbed order

10 Candidate B included

11 Candidate A outside

The boundary margin is the reference score gap between ranks 10 and 11. This schematic has no measured score scale. A query can keep the same attribute agreement while returning a different image.

Read across each row: how many queries returned different neighbours, and how much their similarity scores changed. Compare storage changes against the native reference; compare offset + FP16 against the common-offset reference. Error uses a logarithmic scale.

Queries with changed neighbours

Queries with changed neighboursNative reference: 0. FP16 · gallery only: 5. FP16 · queries + gallery: 5. INT8 · gallery only: 60. INT8 · queries + gallery: 75. FP32 · arithmetic only: 0. Common-offset reference: 0. Offset + FP16 gallery: 195.050100150200Native reference0Native reference: 0FP16 · gallery only5FP16 · gallery only: 5FP16 · queries + gallery5FP16 · queries + gallery: 5INT8 · gallery only60INT8 · gallery only: 60INT8 · queries + gallery75INT8 · queries + gallery: 75FP32 · arithmetic only0FP32 · arithmetic only: 0Common-offset reference0Common-offset reference: 0Offset + FP16 gallery195Offset + FP16 gallery: 195
Out of 983 queries. Native and common-offset rows each use their own unperturbed reference. Zero changes are reported explicitly.

Largest cosine-score error

Largest cosine-score errorNative reference: 8.88e-16. FP16 · gallery only: 5.78e-05. FP16 · queries + gallery: 8.16e-05. INT8 · gallery only: 0.00395. INT8 · queries + gallery: 0.00482. FP32 · arithmetic only: 7.12e-07. Common-offset reference: 8.88e-16. Offset + FP16 gallery: 2.95e-07.10^-1610^-1210^-810^-410^-2Native reference8.88e-16Native reference: 8.88e-16FP16 · gallery only5.78e-05FP16 · gallery only: 5.78e-05FP16 · queries + gallery8.16e-05FP16 · queries + gallery: 8.16e-05INT8 · gallery only0.00395INT8 · gallery only: 0.00395INT8 · queries + gallery0.00482INT8 · queries + gallery: 0.00482FP32 · arithmetic only7.12e-07FP32 · arithmetic only: 7.12e-07Common-offset reference8.88e-16Common-offset reference: 8.88e-16Offset + FP16 gallery2.95e-07Offset + FP16 gallery: 2.95e-07
Largest absolute cosine-score difference from the relevant reference. Logarithmic axis; smaller error alone does not guarantee stable neighbours.

Error magnitude alone does not explain neighbour stability. Under the common-offset stress, FP16 changes 195 top-10 sets despite a smaller maximum score error than native FP16, which changes five.

Native FP16 changes all lie in the smallest reference-margin bin. Reconstruction-based sufficient conditions cover 576 native FP16-gallery queries with no observed exception, but cover no INT8 or stressed queries. These are empirical checks of sufficient conditions, not rigorous floating-point certification.

Of the 195 stressed identity changes, weather P@10 stays unchanged for 131 queries, scene for 119, and time of day for 149. Coarse attribute agreement can hide neighbour turnover; it does not establish unchanged object layout, instance meaning, or safety relevance.

Most comparisons reconstruct stored vectors and score in FP64; the arithmetic-only condition normalizes and accumulates in FP32. Approximate nearest-neighbour search and deployment throughput were not tested. All numerical conditions ↓ · Boundary-margin evidence ↗.

07 / Image–caption retrieval

Finding one caption is not recovering the whole description set.

Use original COCO image–caption pairs with the same pretrained encoder. Compare native 768-dimensional vectors with learned 256- and 128-dimensional prefixes. An image has five recorded captions, so retrieval can be measured in several ways.

FIG. / ONE IMAGE, FIVE CAPTIONSCOCO sample / illustrative ranking
A skier in a red jacket posing on a snowy slope.

COCO image 785 · Han, Nick J Webb · CC BY 2.0. Photograph unchanged.

Five recorded captions for this image
  1. AA woman posing for the camera standing on skis.
  2. Ba woman standing on skiis while posing for the camera
  3. CA woman in a red jacket skiing down a slope
  4. DA young woman is skiing down the mountain slope.
  5. Ea person on skis makes her way through the snow

Suppose the top ten return captions A, C, and E from this image, plus seven captions from other images. This example ranking explains the metrics; it is not a measured retrieval result.

Ten returned captions
  1. A
  2. other
  3. other
  4. C
  5. other
  6. other
  7. E
  8. other
  9. other
  10. other

Hit@10: yesAt least one positive was found.

Hit@1: yesThe first result is a positive.

Caption recall@10: 3/5Three of five captions were recovered.

The photograph and captions are from the evaluation sample. The illustrative ranking can score well on any-positive retrieval while missing some of the recorded caption set. Other returned captions may be semantically valid; recorded relevance is incomplete.
Native 768MRL 256MRL 128

Find at least one caption

Find at least one captionNative · 768 dimensions: 98.40%. Learned prefix · 256 dimensions: 98.10%. Learned prefix · 128 dimensions: 95.90%.0255075100Native · 768 dimensions98.40%Native · 768 dimensions: 98.40%Learned prefix · 256 dimensions98.10%Learned prefix · 256 dimensions: 98.10%Learned prefix · 128 dimensions95.90%Learned prefix · 128 dimensions: 95.90%
Image → text Hit@10. A query counts as a hit when any of its five recorded captions appears in the top ten.

Recover the first result

Recover the first resultNative · 768 dimensions: 84.30%. Learned prefix · 256 dimensions: 83.60%. Learned prefix · 128 dimensions: 73.70%.0255075100Native · 768 dimensions84.30%Native · 768 dimensions: 84.30%Learned prefix · 256 dimensions83.60%Learned prefix · 256 dimensions: 83.60%Learned prefix · 128 dimensions73.70%Learned prefix · 128 dimensions: 73.70%
Image → text Hit@1. The highest-ranked result must be one of the five recorded captions.

Recover more of the five captions

Recover more of the five captionsNative · 768 dimensions: 80.24%. Learned prefix · 256 dimensions: 78.50%. Learned prefix · 128 dimensions: 68.90%.0255075100Native · 768 dimensions80.24%Native · 768 dimensions: 80.24%Learned prefix · 256 dimensions78.50%Learned prefix · 256 dimensions: 78.50%Learned prefix · 128 dimensions68.90%Learned prefix · 128 dimensions: 68.90%
Image → text caption recall@10. Fraction of the five recorded captions recovered in the top ten.

At 128 dimensions, Hit@10 remains high while ranking and caption coverage lose more. Image-to-text Hit@10 falls 2.50 percentage points, compared with 10.60 points for Hit@1 and 11.34 points for caption recall@10.

Hit@10 change after compression

Hit@10 change after compression256d · Text → image: -0.20 [-0.54, +0.14] percentage points. 256d · Image → text: -0.30 [-1.10, +0.50] percentage points. 128d · Text → image: -3.34 [-4.06, -2.62] percentage points. 128d · Image → text: -2.50 [-3.60, -1.40] percentage points.-5-2.50+2256d · Text → image-0.20 [-0.54, +0.14]256d · Image → text-0.30 [-1.10, +0.50]128d · Text → image-3.34 [-4.06, -2.62]128d · Image → text-2.50 [-3.60, -1.40]
Compressed minus native, in percentage points. Dots: paired change; lines: conditional 95% bootstrap intervals. Left of zero means lower Hit@10; right means higher. Zero is the native result.
Reading the uncertainty

Compare the same image groups.

The intervals use 5,000 paired bootstrap draws of 1,000 parent image groups, carrying their captions together. They are conditional on the fixed gallery, prompts, and encoder.

A line spanning zero includes changes in both directions; it does not demonstrate equivalence. No intervals were declared for caption recall or identity overlap, so those changes remain descriptive.

These are evaluation-sample intervals, not encoder-training or independent-encoder uncertainty.

Geometry does not establish image–caption alignment.

A control reassigns the native image–caption correspondence while leaving vectors and rankings fixed. Hit@10 becomes 1.22% for text-to-image and 0.20% for image-to-text: unchanged geometry cannot establish paired relevance.

Category-level and paired utility broadly decline under stronger compression in this study. There is no strong opposing ordering where category utility is preserved while pairing is destroyed. Different retrieval directions also use different caption-role prefixes, so their scores cannot isolate modality asymmetry.

1,000 image groups and five captions each. Unknown pretraining overlap, near/event duplicates, and unrecorded valid matches remain limitations. Same encoder, independently sampled evaluation data. Both directions, identity overlap, and exact values ↓ · Dataset details ↗.

08 / Conclusions

Measure the property you need.

The useful outcome is a clearer evaluation practice: treat each diagnostic as an instrument with a definition and a scope, then check the task property being claimed.

  1. Read geometry as geometry.

    Spread and spectral counts can reveal contraction or directional concentration. They do not, alone, identify what labels are recoverable or what makes a useful neighbour.

  2. Separate utility endpoints.

    Classification, coarse neighbour agreement, exact neighbour identity, first-rank recovery, and multiple-caption coverage can respond differently to the same change.

  3. Keep the evidence scope attached.

    Training-seed variation, fixed-sample BDD effects, and conditional COCO bootstrap intervals answer different uncertainty questions.

What the study does not establish

Dataset scope

The BDD-derived validation subset contains 983 images. It has no foggy-weather, gas-station, or tunnel examples, so the results do not cover those categories.

Encoder scope

The pretrained results use one encoder, EmbeddingGemma 2. A second-encoder replication with Jina-CLIP-v2 could not be completed because its precision checks failed. The findings therefore have not been replicated across encoders.

Claim scope

The study establishes no universal quality predictor, full LeJEPA reproduction, independent encoder replication, operational search-speed benefit, or deployment certification. Familiar compression behavior is reported with its limitations.

Experiment records ↗ · Plot and table data ↗.

Sources / Methods and evidence

Bibliography

Method and dataset sources are listed below. This project’s numerical results come from its tracked, checksum-verified experiment records.

  1. Assran, M., et al. (2023). Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture. CVPR. Masked representation prediction and the I-JEPA training design.
  2. Balestriero, R., & LeCun, Y. (2025). LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristics. arXiv:2511.08544. Source of SIGReg; this project studies a different training configuration.
  3. Yu, F., et al. (2020). BDD100K: A Diverse Driving Dataset for Heterogeneous Multitask Learning. CVPR. Driving imagery and annotations used for the scenario-attribute evaluation.
  4. Garrido, Q., Balestriero, R., Najman, L., & LeCun, Y. (2023). RankMe: Assessing the Downstream Performance of Pretrained Self-Supervised Representations by Their Rank. ICML, PMLR 202, 10929–10974. Source of the singular-value entropy diagnostic.
  5. Cheam, R. (2026). Embedding diagnostics. Experiment code and tracked run logs. Evaluation methods and limitations; per-seed endpoints and paired intervals.
  6. Google DeepMind (2026). EmbeddingGemma 2 model card. Pinned model revision used for the fixed-encoder experiments, including image and text embeddings and normalized learned prefixes.
  7. Kusupati, A., et al. (2022). Matryoshka Representation Learning. NeurIPS. Learning representations with useful nested dimension prefixes.
  8. Lin, T.-Y., et al. (2014). Microsoft COCO: Common Objects in Context. ECCV. Everyday-scene imagery; the paired study uses original COCO2017 validation images and caption annotations.
Appendix / Supporting detail

Explore the complete evidence.

All seven training variants, extension tables, both caption-retrieval directions, paired training uncertainty, and the mathematics behind the dimensionality measurements.

All seven training variants

Explore the measurements

The plots show all seven conditions in the same order as the table. Each row is a model variant, including SIGReg with and without a projector. Each symbol is an individual seed; the short vertical marker is the five-seed mean. Use the selector to compare the three driving attributes.

EMA referenceStop-gradient onlySIGReg variantsContracted control

BDD100K, five training seeds, final step 4,000. Variance has a log axis; the semantic plots use 0–100% axes. Dashed lines show the majority floor for raw accuracy and random-neighbour expectation for retrieval. Balanced accuracy excludes classes below the ten-sample support floor.

HighLowHighest and lowest displayed means in each column, including ties.
All seven conditions · five-seed means · time of day
ConditionTotal varianceRankMeLinear-probe accuracyRetrieval P@10
EMA reference10878.3692.71%High89.49%High
Stop-gradient only39.67.4591.67%86.70%
SIGReg + stop-gradient178High103.96High91.76%87.06%
SIGReg3.2424.4591.11%Low86.79%
Projector + SIGReg + stop-gradient11348.0291.94%87.53%
Projector + SIGReg30.69.7491.40%85.83%
Contracted control6.55e-05Low1.06Low91.37%67.52%Low

What the table shows

The highest geometry readings do not coincide with the highest task scores. SIGReg + stop-gradient has the largest total variance (178) and RankMe (103.96), while the EMA reference has the highest displayed time-of-day classification accuracy (92.71%) and retrieval P@10 (89.49%). Larger spread or a broader spectrum alone does not identify the most useful embeddings.

The control separates classification from retrieval. Its accuracy is 1.34 percentage points below the EMA reference, but its neighbour agreement is 21.97 points below. Recovering a label with a fitted classifier and finding neighbours that share that label test different properties.

A projector does not give a uniform retrieval increase. With SIGReg and stop-gradient, adding it raises the displayed mean P@10 by 0.47 percentage points; with SIGReg alone, it lowers the mean by 0.95 points. These are descriptive five-seed averages. See the paired uncertainty plots in the appendix for the pre-declared comparisons. [5]

Higher accuracy and P@10 mean more correct labels or greater neighbour agreement. Geometry columns describe distributions; their highest values do not automatically identify the most useful model. These are five-seed averages on one fixed evaluation split. The appendix shows paired differences and their 95% intervals for the five pre-declared comparisons.

Full result tables

Ochre High/Low marks the highest and lowest displayed values within each table column, including ties. These are numerical extrema, not significance or uniform model-quality verdicts.

Geometry and task responses

BDD driving-scene subset: Unchanged embeddings and severe interventions; 983 validation rows
ConditionVarianceCosineRankMePRWeather BA %Weather P@10 %
Unchanged embeddings0.257030.742967270.8138.2663.02High57.86High
Shrink magnitude (.99)2.5703e-05Low0.742967270.8138.2663.02High57.86High
Eight directions (.99)0.0795850.9050204.624.39Low30.7552.73
Common offset (.99)0.257030.999974High2.01Low38.2663.02High57.79
Severe noise (.99)9799High0.000156Low667.31High430.99High18.85Low40.05Low

Dimensional compression

BDD driving-scene subset: standardized probe and attribute retrieval, 983 validation rows
RepresentationWeather BA %Scene BA %Time BA %Weather P@10 %Scene P@10 %Time P@10 %
Native 768d63.02High69.5276.4457.86High62.44High82.73
MRL 512d62.8270.56High76.7357.7762.2382.64
PCA 512d55.3665.89Low75.1055.78Low61.7983.07
MRL 256d61.0769.8077.0357.3362.2082.41
PCA 256d58.8566.2775.4656.0461.9783.34
MRL 128d54.91Low66.7074.19Low56.3360.78Low80.66Low
PCA 128d62.3668.3778.17High56.1962.2584.00High

Numerical storage and scoring

BDD driving-scene subset: storage and arithmetic conditions; deltas use each condition's reference
ConditionMax score errorChanged queries / 983Overlap@10Δ weather P@10 ppΔ scene ppΔ time pp
Native reference8.88e-16Low0Low1.00000High+0.000+0.000+0.000
FP16 gallery5.78e-0550.99949-0.020+0.020High-0.010
FP16 queries + gallery8.16e-0550.99949-0.020+0.020High-0.010
INT8 gallery0.00395600.99390-0.041Low-0.010Low+0.000
INT8 queries + gallery0.00482High750.99237-0.020-0.010Low-0.031Low
FP32 arithmetic7.12e-070Low1.00000High+0.000+0.000+0.000
Common-offset reference8.88e-16Low0Low1.00000High+0.000+0.000+0.000
Offset + FP16 gallery2.95e-07195High0.97996Low+0.041High+0.020High+0.183High

The geometry table shows the pre-declared severe settings, not a selected best severity. Full fixed grid and class supports · Classifier fit diagnostics.

Both caption-retrieval directions and neighbour overlap

Text → image / Hit@10

Text → image / Hit@10Native · 768 dimensions: 97.16%. Learned prefix · 256 dimensions: 96.96%. Learned prefix · 128 dimensions: 93.82%.0255075100Native · 768 dimensions97.16%Native · 768 dimensions: 97.16%Learned prefix · 256 dimensions96.96%Learned prefix · 256 dimensions: 96.96%Learned prefix · 128 dimensions93.82%Learned prefix · 128 dimensions: 93.82%
5,000 caption queries, 1,000 image candidates; one recorded parent per caption.

Text → image / Hit@1

Text → image / Hit@1Native · 768 dimensions: 73.86%. Learned prefix · 256 dimensions: 72.14%. Learned prefix · 128 dimensions: 62.50%.0255075100Native · 768 dimensions73.86%Native · 768 dimensions: 73.86%Learned prefix · 256 dimensions72.14%Learned prefix · 256 dimensions: 72.14%Learned prefix · 128 dimensions62.50%Learned prefix · 128 dimensions: 62.50%
Recorded parent recovered at the first rank.

Image → text / neighbour overlap

Image → text / neighbour overlapNative · 768 dimensions: 100.00%. Learned prefix · 256 dimensions: 83.75%. Learned prefix · 128 dimensions: 65.24%.0255075100Native · 768 dimensions100.00%Native · 768 dimensions: 100.00%Learned prefix · 256 dimensions83.75%Learned prefix · 256 dimensions: 83.75%Learned prefix · 128 dimensions65.24%Learned prefix · 128 dimensions: 65.24%
Percent of native top-10 identities retained. This is not relevance.

Text → image / neighbour overlap

Text → image / neighbour overlapNative · 768 dimensions: 100.00%. Learned prefix · 256 dimensions: 83.63%. Learned prefix · 128 dimensions: 64.78%.0255075100Native · 768 dimensions100.00%Native · 768 dimensions: 100.00%Learned prefix · 256 dimensions83.63%Learned prefix · 256 dimensions: 83.63%Learned prefix · 128 dimensions64.78%Learned prefix · 128 dimensions: 64.78%
Percent of native top-10 identities retained. This is not relevance.
COCO: 1,000 image groups and five captions per image; separate text roles by direction
RepresentationT→I Hit@1 %T→I Hit@10 %I→T Hit@1 %I→T Hit@10 %I→T caption recall@10 %Category P@10 %
Native 768d73.86High97.16High84.30High98.40High80.24High48.84High
MRL 256d72.1496.9683.6098.1078.5048.12
MRL 128d62.50Low93.82Low73.70Low95.90Low68.90Low44.88Low

Image-to-text uses document-role captions; text-to-image uses query-role captions. Directions differ in candidate count and number of recorded positives. Category P@10 is a separate image-only comparator over 75 eligible object categories. All category supports and retrieval results · Interpretation and controls.

How uncertain are these differences?

Compare the same seed, then average the differences.

Each row changes one mechanism. For each of the five seeds, subtract model B’s score from model A’s score: dᵢ = score(A, i) − score(B, i). This pairing removes variation shared by both runs, rather than treating them as unrelated experiments.

Read from the zero line. Right means a higher score for A; left means a lower score for A. The small dots are the five seed differences, the diamond is their mean, and the horizontal line is the 95% interval. A line spanning zero includes differences in both directions. A wider line means a less precise estimate under this protocol.

Classification accuracy

stop-gradient effect, no SIGReg. A: Stop-gradient only; B: Contracted control. Mean A minus B: +0.30 percentage points. 95% interval: -0.08 to +0.68 percentage points. stop-gradient effect, no projector. A: SIGReg + stop-gradient; B: SIGReg. Mean A minus B: +0.65 percentage points. 95% interval: +0.21 to +1.09 percentage points. stop-gradient effect, with projector. A: Projector + SIGReg + stop-gradient; B: Projector + SIGReg. Mean A minus B: +0.54 percentage points. 95% interval: +0.12 to +0.96 percentage points. projector effect, with stop-gradient. A: Projector + SIGReg + stop-gradient; B: SIGReg + stop-gradient. Mean A minus B: +0.18 percentage points. 95% interval: -0.09 to +0.45 percentage points. projector effect, without stop-gradient. A: Projector + SIGReg; B: SIGReg. Mean A minus B: +0.29 percentage points. 95% interval: -0.17 to +0.75 percentage points.-3-2-10+1+2+3stop-gradient, no SIGRegA: Stop-gradient onlyB: Contracted controlSeed 0: +0.175 percentage pointsSeed 1: -0.175 percentage pointsSeed 2: +0.387 percentage pointsSeed 3: +0.562 percentage pointsSeed 4: +0.538 percentage pointsStop-gradient only minus Contracted control: mean +0.30; 95% interval [-0.08, 0.68] percentage pointsstop-gradient, no projectorA: SIGReg + stop-gradientB: SIGRegSeed 0: +0.525 percentage pointsSeed 1: +0.925 percentage pointsSeed 2: +0.075 percentage pointsSeed 3: +0.887 percentage pointsSeed 4: +0.838 percentage pointsSIGReg + stop-gradient minus SIGReg: mean +0.65; 95% interval [0.21, 1.09] percentage pointsstop-gradient, with projectorA: Projector + SIGReg + stop-gradientB: Projector + SIGRegSeed 0: +0.412 percentage pointsSeed 1: +0.075 percentage pointsSeed 2: +0.613 percentage pointsSeed 3: +0.613 percentage pointsSeed 4: +1.000 percentage pointsProjector + SIGReg + stop-gradient minus Projector + SIGReg: mean +0.54; 95% interval [0.12, 0.96] percentage pointsprojector, with stop-gradientA: Projector + SIGReg + stop-gradientB: SIGReg + stop-gradientSeed 0: -0.038 percentage pointsSeed 1: +0.025 percentage pointsSeed 2: +0.450 percentage pointsSeed 3: +0.100 percentage pointsSeed 4: +0.375 percentage pointsProjector + SIGReg + stop-gradient minus SIGReg + stop-gradient: mean +0.18; 95% interval [-0.09, 0.45] percentage pointsprojector, without stop-gradientA: Projector + SIGRegB: SIGRegSeed 0: +0.075 percentage pointsSeed 1: +0.875 percentage pointsSeed 2: -0.087 percentage pointsSeed 3: +0.375 percentage pointsSeed 4: +0.213 percentage pointsProjector + SIGReg minus SIGReg: mean +0.29; 95% interval [-0.17, 0.75] percentage pointsLower score for AHigher score for Astop-gradient effect, no SIGReg. A: Stop-gradient only; B: Contracted control. Mean A minus B: +0.30 percentage points. 95% interval: -0.08 to +0.68 percentage points. stop-gradient effect, no projector. A: SIGReg + stop-gradient; B: SIGReg. Mean A minus B: +0.65 percentage points. 95% interval: +0.21 to +1.09 percentage points. stop-gradient effect, with projector. A: Projector + SIGReg + stop-gradient; B: Projector + SIGReg. Mean A minus B: +0.54 percentage points. 95% interval: +0.12 to +0.96 percentage points. projector effect, with stop-gradient. A: Projector + SIGReg + stop-gradient; B: SIGReg + stop-gradient. Mean A minus B: +0.18 percentage points. 95% interval: -0.09 to +0.45 percentage points. projector effect, without stop-gradient. A: Projector + SIGReg; B: SIGReg. Mean A minus B: +0.29 percentage points. 95% interval: -0.17 to +0.75 percentage points.-3-2-10+1+2+3stop-gradient, no SIGRegA: Stop-gradient onlyB: Contracted controlSeed 0: +0.175 percentage pointsSeed 1: -0.175 percentage pointsSeed 2: +0.387 percentage pointsSeed 3: +0.562 percentage pointsSeed 4: +0.538 percentage pointsStop-gradient only minus Contracted control: mean +0.30; 95% interval [-0.08, 0.68] percentage pointsstop-gradient, no projectorA: SIGReg + stop-gradientB: SIGRegSeed 0: +0.525 percentage pointsSeed 1: +0.925 percentage pointsSeed 2: +0.075 percentage pointsSeed 3: +0.887 percentage pointsSeed 4: +0.838 percentage pointsSIGReg + stop-gradient minus SIGReg: mean +0.65; 95% interval [0.21, 1.09] percentage pointsstop-gradient, with projectorA: Projector + SIGReg + stop-gradientB: Projector + SIGRegSeed 0: +0.412 percentage pointsSeed 1: +0.075 percentage pointsSeed 2: +0.613 percentage pointsSeed 3: +0.613 percentage pointsSeed 4: +1.000 percentage pointsProjector + SIGReg + stop-gradient minus Projector + SIGReg: mean +0.54; 95% interval [0.12, 0.96] percentage pointsprojector, with stop-gradientA: Projector + SIGReg + stop-gradientB: SIGReg + stop-gradientSeed 0: -0.038 percentage pointsSeed 1: +0.025 percentage pointsSeed 2: +0.450 percentage pointsSeed 3: +0.100 percentage pointsSeed 4: +0.375 percentage pointsProjector + SIGReg + stop-gradient minus SIGReg + stop-gradient: mean +0.18; 95% interval [-0.09, 0.45] percentage pointsprojector, without stop-gradientA: Projector + SIGRegB: SIGRegSeed 0: +0.075 percentage pointsSeed 1: +0.875 percentage pointsSeed 2: -0.087 percentage pointsSeed 3: +0.375 percentage pointsSeed 4: +0.213 percentage pointsProjector + SIGReg minus SIGReg: mean +0.29; 95% interval [-0.17, 0.75] percentage pointsLower score for AHigher score for A
Difference in percentage points. Dots: five paired seeds. Diamond: mean difference. Horizontal line: paired Student-t 95% interval. Zero: equal scores. Axis limits stay fixed across attributes for this metric.

Neighbour agreement / P@10

stop-gradient effect, no SIGReg. A: Stop-gradient only; B: Contracted control. Mean A minus B: +19.18 percentage points. 95% interval: +9.10 to +29.27 percentage points. stop-gradient effect, no projector. A: SIGReg + stop-gradient; B: SIGReg. Mean A minus B: +0.28 percentage points. 95% interval: -0.94 to +1.50 percentage points. stop-gradient effect, with projector. A: Projector + SIGReg + stop-gradient; B: Projector + SIGReg. Mean A minus B: +1.70 percentage points. 95% interval: +1.14 to +2.26 percentage points. projector effect, with stop-gradient. A: Projector + SIGReg + stop-gradient; B: SIGReg + stop-gradient. Mean A minus B: +0.47 percentage points. 95% interval: -0.00 to +0.95 percentage points. projector effect, without stop-gradient. A: Projector + SIGReg; B: SIGReg. Mean A minus B: -0.95 percentage points. 95% interval: -2.14 to +0.23 percentage points.-30-20-100+10+20+30stop-gradient, no SIGRegA: Stop-gradient onlyB: Contracted controlSeed 0: +21.035 percentage pointsSeed 1: +5.169 percentage pointsSeed 2: +23.469 percentage pointsSeed 3: +25.815 percentage pointsSeed 4: +20.436 percentage pointsStop-gradient only minus Contracted control: mean +19.18; 95% interval [9.10, 29.27] percentage pointsstop-gradient, no projectorA: SIGReg + stop-gradientB: SIGRegSeed 0: +0.148 percentage pointsSeed 1: +0.071 percentage pointsSeed 2: +0.841 percentage pointsSeed 3: +1.474 percentage pointsSeed 4: -1.154 percentage pointsSIGReg + stop-gradient minus SIGReg: mean +0.28; 95% interval [-0.94, 1.50] percentage pointsstop-gradient, with projectorA: Projector + SIGReg + stop-gradientB: Projector + SIGRegSeed 0: +2.323 percentage pointsSeed 1: +1.759 percentage pointsSeed 2: +1.709 percentage pointsSeed 3: +1.655 percentage pointsSeed 4: +1.058 percentage pointsProjector + SIGReg + stop-gradient minus Projector + SIGReg: mean +1.70; 95% interval [1.14, 2.26] percentage pointsprojector, with stop-gradientA: Projector + SIGReg + stop-gradientB: SIGReg + stop-gradientSeed 0: +0.376 percentage pointsSeed 1: +0.162 percentage pointsSeed 2: +0.078 percentage pointsSeed 3: +0.804 percentage pointsSeed 4: +0.936 percentage pointsProjector + SIGReg + stop-gradient minus SIGReg + stop-gradient: mean +0.47; 95% interval [-0.00, 0.95] percentage pointsprojector, without stop-gradientA: Projector + SIGRegB: SIGRegSeed 0: -1.799 percentage pointsSeed 1: -1.525 percentage pointsSeed 2: -0.790 percentage pointsSeed 3: +0.623 percentage pointsSeed 4: -1.275 percentage pointsProjector + SIGReg minus SIGReg: mean -0.95; 95% interval [-2.14, 0.23] percentage pointsLower score for AHigher score for Astop-gradient effect, no SIGReg. A: Stop-gradient only; B: Contracted control. Mean A minus B: +19.18 percentage points. 95% interval: +9.10 to +29.27 percentage points. stop-gradient effect, no projector. A: SIGReg + stop-gradient; B: SIGReg. Mean A minus B: +0.28 percentage points. 95% interval: -0.94 to +1.50 percentage points. stop-gradient effect, with projector. A: Projector + SIGReg + stop-gradient; B: Projector + SIGReg. Mean A minus B: +1.70 percentage points. 95% interval: +1.14 to +2.26 percentage points. projector effect, with stop-gradient. A: Projector + SIGReg + stop-gradient; B: SIGReg + stop-gradient. Mean A minus B: +0.47 percentage points. 95% interval: -0.00 to +0.95 percentage points. projector effect, without stop-gradient. A: Projector + SIGReg; B: SIGReg. Mean A minus B: -0.95 percentage points. 95% interval: -2.14 to +0.23 percentage points.-30-20-100+10+20+30stop-gradient, no SIGRegA: Stop-gradient onlyB: Contracted controlSeed 0: +21.035 percentage pointsSeed 1: +5.169 percentage pointsSeed 2: +23.469 percentage pointsSeed 3: +25.815 percentage pointsSeed 4: +20.436 percentage pointsStop-gradient only minus Contracted control: mean +19.18; 95% interval [9.10, 29.27] percentage pointsstop-gradient, no projectorA: SIGReg + stop-gradientB: SIGRegSeed 0: +0.148 percentage pointsSeed 1: +0.071 percentage pointsSeed 2: +0.841 percentage pointsSeed 3: +1.474 percentage pointsSeed 4: -1.154 percentage pointsSIGReg + stop-gradient minus SIGReg: mean +0.28; 95% interval [-0.94, 1.50] percentage pointsstop-gradient, with projectorA: Projector + SIGReg + stop-gradientB: Projector + SIGRegSeed 0: +2.323 percentage pointsSeed 1: +1.759 percentage pointsSeed 2: +1.709 percentage pointsSeed 3: +1.655 percentage pointsSeed 4: +1.058 percentage pointsProjector + SIGReg + stop-gradient minus Projector + SIGReg: mean +1.70; 95% interval [1.14, 2.26] percentage pointsprojector, with stop-gradientA: Projector + SIGReg + stop-gradientB: SIGReg + stop-gradientSeed 0: +0.376 percentage pointsSeed 1: +0.162 percentage pointsSeed 2: +0.078 percentage pointsSeed 3: +0.804 percentage pointsSeed 4: +0.936 percentage pointsProjector + SIGReg + stop-gradient minus SIGReg + stop-gradient: mean +0.47; 95% interval [-0.00, 0.95] percentage pointsprojector, without stop-gradientA: Projector + SIGRegB: SIGRegSeed 0: -1.799 percentage pointsSeed 1: -1.525 percentage pointsSeed 2: -0.790 percentage pointsSeed 3: +0.623 percentage pointsSeed 4: -1.275 percentage pointsProjector + SIGReg minus SIGReg: mean -0.95; 95% interval [-2.14, 0.23] percentage pointsLower score for AHigher score for A
Difference in percentage points. Dots: five paired seeds. Diamond: mean difference. Horizontal line: paired Student-t 95% interval. Zero: equal scores. Axis limits stay fixed across attributes for this metric.

What the interval covers

For the five differences, let d̄ be their mean and sd their sample standard deviation. The interval is d̄ ± 2.776 × sd / √5, using Student’s t distribution with four degrees of freedom. One percentage point means, for example, a change from 90% to 91%.

These intervals describe training-seed variation on one fixed evaluation split. They do not cover a different dataset, image split, or architecture. The five comparisons were declared before these runs; no correction for multiple comparisons is applied. The classifier-fitting settings were selected after training. EMA remains a reference, rather than a paired mechanism contrast, because it changes multiple factors. [5]

Download the seed differences and intervals ↗
How are dimensionality measurements calculated?
Technical note / Rank diagnostics

How many directions
carry the representation?

An embedding has 192 coordinates, but these can move together. If every image varies along one line, 192 coordinates describe only one independent direction. These measurements give a smooth count of how broadly the representation uses its directions, rather than counting every tiny nonzero direction equally.

Start with the matrix of image embeddings.

Stack n images as the rows of a matrix Z ∈ ℝn×d, with d = 192 coordinates per image. A spectrum lists the strength of independent, perpendicular directions in that matrix. Both measurements turn a spectrum into relative shares, then ask how evenly those shares are distributed.

01 / Participation ratio

Count directions of variation around the mean.

Subtract the mean embedding μ from each row to obtain X. The covariance matrix is C = XᵀX / (n − 1). Its eigenvalues λᵢ ≥ 0 are the variances along its independent directions. Their sum is total variance.

qi=λi∑jλj PR=(∑iλi)2∑iλi2=1∑iqi2

Why this makes sense: squaring the shares makes a dominant direction weigh heavily. If one direction holds everything, the sum of squared shares is 1 and PR is 1. If k directions each hold 1/k, that sum is 1/k and PR is k. Unequal shares give an intermediate count.

02 / RankMe

Count directions in the raw matrix.

Use Z without subtracting its mean. Singular value decomposition gives strengths σᵢ ≥ 0 for its perpendicular directions. Normalize these strengths into shares pᵢ, then calculate their entropy. [4]

pi=σi∑jσj RankMe=exp(−∑ipilnpi)

Why exponentiate entropy? One direction has entropy 0, so its count is exp(0) = 1. With k equal shares, entropy is ln(k); exponentiating returns k. This puts the measurement in units of directions. Here ln is the natural logarithm and zero shares contribute zero to entropy.

A worked example: two directions, one much stronger.

Illustration only, not experiment data. Take a zero-mean matrix with singular values (3, 1, 0, 0). Because it is already centered, we can isolate the difference in weighting:

  1. Covariance eigenvalues are proportional to squared singular values: (9, 1, 0, 0). Variance shares are (0.9, 0.1, 0, 0), giving PR = 1 / (0.9² + 0.1²) ≈ 1.22.
  2. RankMe uses singular-value shares (0.75, 0.25, 0, 0), giving exp[−0.75 ln(0.75) − 0.25 ln(0.25)] ≈ 1.75.

Both counts lie between one and two, because two directions are present but contribute unequally. They differ because variance squares the singular values, while RankMe uses their unsquared strengths.

What these counts cannot tell you

Centering matters. Adding the same offset vector to every embedding leaves centered covariance and PR unchanged. Raw RankMe can change because that shared mean contributes to the singular-value spectrum.

Magnitude matters separately. Multiplying every embedding by a small positive number leaves both normalized spectra unchanged, while total variance falls by the square of that number. A representation can therefore shrink without either count falling. Neither count identifies which directions encode useful labels; read them alongside variance, classification, and retrieval.

The equations show the ideal nonzero-spectrum case. The implementation adds 10−7 to RankMe’s normalized singular values for numerical stability and returns 1 for an exactly zero spectrum in either measurement. That convention is not evidence of useful information. [5]