Skip to Main content Skip to Navigation
Preprints, Working Papers, ...

Learning graph-structured data using Poincar\'e embeddings and Riemannian K-means algorithms

Abstract : Recent literature has shown several benefits of hyperbolic embedding of graph-structured data (GSD) in representing their structures and latent relations. While several studies have explored the ability of hyperbolic embedding to represent data (for example, by quantifying their mean average precision) and their ability to produce better visualisations of clusters, only few works exploited the effectiveness of hyperbolic embedding to perform learning on the initial GSD. Motivated by innovative ideas from the fields of Brain computer interfaces and Radar processing, this paper presents a new scheme for learning GSD based on hyperbolic embedding, Riemannian barycentre (i.e. Fr\'echet or geometric mean) and $K$-means algorithms as a significant tool that derives from it. The main idea is as follows. Relying on the Riemannian barycentre, we define a notion of minimal variance which allows us to choose an embedding between different ones. This embedding is used thereafter together with $K$-means algorithms to perform unsupervised clustering and in combination with the nearest neighbour rule to perform supervised learning. We demonstrate the performance of the proposed framework through several experiments on real-world social networks and hierarchical GSD. The obtained results outperform their counterparts in high-dimensional Euclidean spaces and recent proposed geometric approaches.
Document type :
Preprints, Working Papers, ...
Complete list of metadata
Contributor : Hadi Zaatiti Connect in order to contact the contributor
Submitted on : Wednesday, October 30, 2019 - 12:22:10 PM
Last modification on : Friday, January 21, 2022 - 12:56:02 PM

Links full text


  • HAL Id : hal-02339208, version 1
  • ARXIV : 1907.01662



Hatem Hajri, Hadi Zaatiti, Georges Hébrail. Learning graph-structured data using Poincar\'e embeddings and Riemannian K-means algorithms. 2019. ⟨hal-02339208⟩



Record views