Geometric Structure Predicts Per-Example Neighborhood Loss Under PCA Truncation
Organizations: Department of Computer Science, San Jose State University, San Jose, CA, USA
Abstract
Dimensionality reduction can preserve the average performance of text embeddings even when a substantial fraction of their dimensions is removed. Aggregate metrics, however, do not reveal whether degradation is distributed uniformly across examples. We study per-example neighborhood stability under PCA truncation across static word embeddings and modern sentence encoders. Per-item neighbor retention varies widely and is predictable from geometric properties of the representation and its alignment with the retained PCA subspace. The fraction of centered-vector energy retained by the truncated subspace is strongest for CBOW and RI, with neighborhood contrast leading on SkipGram and the modern/Matryoshka settings (GTE approximately tied), while local density is associated with retention in marginal analysis but adds little unique information once energy and contrast are controlled (R-squared increments 0.001 to 0.010). Local intrinsic dimensionality accounts for more variation than density on modern sentence encoders in pairwise comparison. On SciFact, query density predicts retrieval degradation under aggressive compression; on NFCorpus the primary 32d association does not replicate, and a direct variance-restriction test leaves that null unresolved. Average compression performance conceals systematic per-example variation that geometry makes visible.
Figures & tables
| Family | ret100 | ret50 | ret32 |
|---|---|---|---|
| CBOW/fairytales | 0.616 | 0.378 | 0.264 |
| RI/text8sub | 0.819 | 0.620 | 0.478 |
| SkipGram/fairytales | 0.591 | 0.398 | 0.283 |
| Family | density | hub. | hub. den. | norm | margin |
|---|---|---|---|---|---|
| CBOW | 0.498 | 0.398 | 0.027 | 0.128 | 0.091 |
| RI | 0.666 | 0.473 | 0.040 | -0.065 | -0.171 |
| SkipGram | 0.541 | 0.377 | 0.020 | -0.081 | 0.171 |
| Dim | mean | partial | 95% CI |
|---|---|---|---|
| nDCG@10 | |||
| 128 | |||
| 64 | |||
| 32 | |||
| Recall@10 | |||
| 128 | |||
| Setting | density | energy | contrast |
|---|---|---|---|
| CBOW | -0.06 | 0.57 | 0.17 |
| SG | 0.04 | 0.12 | 0.17 |
| RI | 0.12 | 0.20 | 0.13 |
| MiniLM | 0.07 | 0.22 | 0.26 |
| GTE | 0.09 | 0.25 | 0.25 |
| MPNet | 0.13 | 0.11 | 0.24 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Analysis | Result |
|---|---|
| Median allocation | 95% CI ; no improvement |
| Quartile allocation | 95% CI ; no improvement |
| High-high pairs | vs (70 of 231 pairs); no statistically significant advantage |
| Inverse allocation | 95% CI ; no improvement |
| Random allocation | Mean , range ; comparable |
| Retrieval hybrid | Hybrid-minus-uniform , 95% CI , ; significantly worse |
| Setting | gap | ID | margin | |
|---|---|---|---|---|
| CBOW | 0.02 | -0.09 | -0.04 | 0.002 |
| SG | 0.02 | -0.10 | -0.07 | 0.001 |
| RI | 0.05 | -0.12 | -0.03 | 0.007 |
| MiniLM | 0.06 | -0.09 | -0.10 | 0.003 |
| GTE | 0.04 | -0.07 | -0.06 | 0.005 |
| MPNet | 0.08 | -0.09 | -0.11 | 0.010 |
| Setting | density F+ID | ID F+D |
|---|---|---|
| CBOW | 0.278 [0.229, 0.322] | -0.048 [-0.096, +0.000] |
| SkipGram | 0.296 [0.244, 0.346] | -0.124 [-0.170, -0.074] |
| RI | 0.466 [0.418, 0.510] | -0.224 [-0.276, -0.172] |
| MiniLM | 0.025 [-0.044, 0.094] | -0.390 [-0.447, -0.329] |
| GTE | 0.046 [-0.019, 0.108] | -0.386 [-0.444, -0.332] |
| MPNet | 0.091 [0.022, 0.157] | -0.315 [-0.377, -0.251] |
| Family | D10 | annulus 11–30 | annulus 11–50 |
|---|---|---|---|
| CBOW | 0.498 | 0.457 | 0.449 |
| SG | 0.541 | 0.472 | 0.464 |
| RI | 0.666 | 0.597 | 0.582 |
| MiniLM | 0.416 | 0.327 | 0.313 |
| GTE | 0.364 | 0.225 | 0.201 |
| MPNet | 0.485 | 0.416 | 0.405 |
| Band | n | SD | nDCG partial | recall partial |
|---|---|---|---|---|
| A full | 300 | 0.086 | 0.249 | 0.266 |
| B | 150 | 0.028 | 0.105 | 0.098 |
| C | 128 | 0.023 | 0.085 | 0.022 |
| D | 82 | 0.014 | 0.059 | 0.009 |
Text Embedding Evaluation
Text Embedding Evaluation measures how well vector representations capture meaning across retrieval, similarity, classification, and other tasks.
See all 78 papers