mlsanityData to 14 Oct 2026

The Impact of Dimensionality on the Stability of Node Embeddings

arXiv 2604.08492 ↗First posted 9 Apr 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Previous work has shown that node embedding methods can produce different representations and downstream predictions across repeated training runs, even when trained on the same data with identical hyperparameters. However, the role of embedding dimensionality in this instability remains poorly understood. In this work, we systematically analyze how embedding dimensionality affects the stability of embeddings from five widely used node embedding methods: ASNE, DGI, GraphSAGE, node2vec, and VERSE. We evaluate stability from both representational and functional perspectives across a broad range of dimensions, datasets, and repeated training runs, and relate the resulting stability patterns to predictive performance. Our results show that dimensionality can substantially affect embedding stability, although the observed effects depend strongly on the embedding method and stability notion considered. While node2vec and ASNE generally became more stable at higher dimensions, GraphSAGE and VERSE often exhibited non-monotonic behavior or decreasing stability. We further find that dimensions associated with high stability do not necessarily coincide with those yielding the strongest downstream performance. Overall, our findings demonstrate that embedding dimensionality can have a substantial impact on the stability of node embeddings and downstream predictions.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 42 of its 73 references that are in the map. First posted on arXiv 9 Apr 2026.

Builds on

42 of 73 references.

References · 42

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  2. 2017Inductive Representation Learning on Large Graphscited by 6,320
  3. 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,375
  4. 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,474
  5. 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,434
  6. 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
  7. 2016node2vec: Scalable Feature Learning for Networkscited by 1,918
  8. 2014DeepWalk: Online Learning of Social Representationscited by 1,547
  9. 2018Learning deep representations by mutual information estimation and maximizationcited by 1,273
  10. 2018Deep Graph Infomaxcited by 1,106
  11. 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 976
  12. 2014Distributed Representations of Sentences and Documentscited by 678
30 more
  1. 2015LINE: Large-scale Information Network Embeddingcited by 582
  2. 2019Multi-scale Attributed Node Embeddingcited by 548
  3. 2016Diachronic Word Embeddings Reveal Statistical Laws of Semantic Changecited by 221
  4. 2017struc2vec: Learning Node Representations from Structural Identitycited by 215
  5. 2021Accounting for Variance in Machine Learning Benchmarkscited by 201
  6. 2017Graph Embedding Techniques, Applications, and Performance: A Surveycited by 188
  7. 2023Similarity of Neural Network Models: A Survey of Functional and Representational Measurescited by 131
  8. 2019Graph Representation Learning: A Surveycited by 75
  9. 2016Cultural Shift or Linguistic Drift? Comparing Two Computational Measures of Semantic Changecited by 63
  10. 2018VERSE: Versatile Graph Embeddings from Similarity Measurescited by 54
  11. 2025PyG 2.0: Scalable Learning on Real World Graphscited by 52
  12. 2018On the Dimensionality of Word Embeddingcited by 46
  13. 2017Attributed Social Network Embeddingcited by 34
  14. 2018Factors Influencing the Surprising Instability of Word Embeddingscited by 33
  15. 2020PANE: scalable and effective attributed network embeddingcited by 23
  16. 2020Principled approach to the selection of the embedding dimension of networkscited by 19
  17. 2024ReSi: A Comprehensive Benchmark for Representational Similarity Measurescited by 19
  18. 2020The Effects of Randomness on the Stability of Node Embeddingscited by 15
  19. 2021Graph Entropy Guided Node Embedding Dimension Selection for Graph Neural Networkscited by 13
  20. 2015How to Generate a Good Word Embedding?cited by 9
  21. 2019Single Training Dimension Selection for Word Embedding with PCAcited by 9
  22. 2021GRAPE for Fast and Scalable Graph Processing and random walk-based Embeddingcited by 7
  23. 2022On the Prediction Instability of Graph Neural Networkscited by 6
  24. 2020Understanding the Downstream Instability of Word Embeddingscited by 5
  25. 2021Detecting the ultra low dimensionality of real networkscited by 5
  26. 2023On the Dimensionality of Sentence Embeddingscited by 5
  27. 2021Are Word Embedding Methods Stable and Should We Care About It?cited by 4
  28. 2025How Low Can You Go? Searching for the Intrinsic Dimensionality of Complex Networks using Metric Node Embeddingscited by 3
  29. 2023Geometric instability of graph neural networks on large graphscited by 2
  30. 2014Dimensionality of social networks using motifs and eigenvaluescited by 1

Built on by

Not cited yet.