Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Online Graph Dictionary Learning

About

Dictionary learning is a key tool for representation learning, that explains the data as linear combination of few basic elements. Yet, this analysis is not amenable in the context of graph learning, as graphs usually belong to different metric spaces. We fill this gap by proposing a new online Graph Dictionary Learning approach, which uses the Gromov Wasserstein divergence for the data fitting term. In our work, graphs are encoded through their nodes' pairwise relations and modeled as convex combination of graph atoms, i.e. dictionary elements, estimated thanks to an online stochastic algorithm, which operates on a dataset of unregistered graphs with potentially different number of nodes. Our approach naturally extends to labeled graphs, and is completed by a novel upper bound that can be used as a fast approximation of Gromov Wasserstein in the embedding space. We provide numerical evidences showing the interest of our approach for unsupervised embedding of graph datasets and for online graph subspace estimation and tracking.

C\'edric Vincent-Cuaz, Titouan Vayer, R\'emi Flamary, Marco Corneli, Nicolas Courty• 2021

Related benchmarks

TaskDatasetResultRank
Graph ClassificationENZYMES
Accuracy71.47
305
Graph ClassificationIMDB-B (10-fold cross-validation)
Accuracy72.06
148
Graph ClassificationIMDB-M (10-fold cross-validation)
Accuracy50.64
84
Graph ClassificationPROTEIN
Accuracy74.86
48
Graph ClassificationCOX2
Accuracy78.11
40
Graph ClassificationBZR
Accuracy87.81
29
Graph ClassificationMUTAG discrete attributes (10-fold nested-cross val)
Accuracy87.09
13
Graph ClassificationPTC-MR discrete attributes (10-fold nested-cross val)
Accuracy58.45
13
Graph ClusteringMUTAG--
8
Graph ClusteringIMDB-B
Rand Index51.64
6
Showing 10 of 16 rows

Other info

Code

Follow for update