English

Fast Graph Learning with Unique Optimal Solutions

Machine Learning 2021-04-23 v4 Mathematical Software Social and Information Networks

Abstract

We consider two popular Graph Representation Learning (GRL) methods: message passing for node classification and network embedding for link prediction. For each, we pick a popular model that we: (i) linearize and (ii) and switch its training objective to Frobenius norm error minimization. These simplifications can cast the training into finding the optimal parameters in closed-form. We program in TensorFlow a functional form of Truncated Singular Value Decomposition (SVD), such that, we could decompose a dense matrix M\mathbf{M}, without explicitly computing M\mathbf{M}. We achieve competitive performance on popular GRL tasks while providing orders of magnitude speedup. We open-source our code at http://github.com/samihaija/tf-fsvd

Keywords

Cite

@article{arxiv.2102.08530,
  title  = {Fast Graph Learning with Unique Optimal Solutions},
  author = {Sami Abu-El-Haija and Valentino Crespi and Greg Ver Steeg and Aram Galstyan},
  journal= {arXiv preprint arXiv:2102.08530},
  year   = {2021}
}
R2 v1 2026-06-23T23:14:00.992Z