English

Word2VecGD: Neural Graph Drawing with Cosine-Stress Optimization

Computational Geometry 2025-09-23 v1 Machine Learning

Abstract

We propose a novel graph visualization method leveraging random walk-based embeddings to replace costly graph-theoretical distance computations. Using word2vec-inspired embeddings, our approach captures both structural and semantic relationships efficiently. Instead of relying on exact shortest-path distances, we optimize layouts using cosine dissimilarities, significantly reducing computational overhead. Our framework integrates differentiable stress optimization with stochastic gradient descent (SGD), supporting multi-criteria layout objectives. Experimental results demonstrate that our method produces high-quality, semantically meaningful layouts while efficiently scaling to large graphs. Code available at: https://github.com/mlyann/graphv_nn

Keywords

Cite

@article{arxiv.2509.17333,
  title  = {Word2VecGD: Neural Graph Drawing with Cosine-Stress Optimization},
  author = {Minglai Yang and Reyan Ahmed},
  journal= {arXiv preprint arXiv:2509.17333},
  year   = {2025}
}
R2 v1 2026-07-01T05:48:46.926Z