Learning graph-structured data using Poincaré embeddings and Riemannian K-means algorithms

  • 2019-07-02 21:45:22
  • Hatem Hajri, Hadi Zaatiti, Georges Hebrail
  • 18

Abstract

Recent literature has shown several benefits of hyperbolic embedding ofgraph-structured data (GSD) in representing their structures and latentrelations. While several studies have explored the ability of hyperbolicembedding to represent data (for example, by quantifying their mean averageprecision) and their ability to produce better visualisations of clusters, onlyfew works exploited the effectiveness of hyperbolic embedding to performlearning on the initial GSD. Motivated by innovative ideas from the fields of Brain computer interfacesand Radar processing, this paper presents a new scheme for learning GSD basedon hyperbolic embedding, Riemannian barycentre (i.e. Fr\'echet or geometricmean) and $K$-means algorithms as a significant tool that derives from it. Themain idea is as follows. Relying on the Riemannian barycentre, we define anotion of minimal variance which allows us to choose an embedding betweendifferent ones. This embedding is used thereafter together with $K$-meansalgorithms to perform unsupervised clustering and in combination with thenearest neighbour rule to perform supervised learning. We demonstrate theperformance of the proposed framework through several experiments on real-worldsocial networks and hierarchical GSD. The obtained results outperform theircounterparts in high-dimensional Euclidean spaces and recent proposed geometricapproaches.

 

Quick Read (beta)

loading the full paper ...