HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Online Graph Dictionary Learning

Cédric Vincent-Cuaz Titouan Vayer Rémi Flamary Marco Corneli Nicolas Courty

Online Graph Dictionary Learning

Abstract

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.

Code Repositories

cedricvincentcuaz/GDL
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-bzrGDL-g (ADJ)
Accuracy: 87.81
graph-classification-on-cox2GDL-g (ADJ)
Accuracy(10-fold): 78.11
graph-classification-on-enzymesGDL-g (SP)
Accuracy: 71.47
graph-classification-on-imdb-bGDL
Accuracy: 72.06%
Rand index: 51.64
graph-classification-on-imdb-mGDL
Accuracy: 50.64%
graph-classification-on-mutagGDL-g (SP)
Accuracy: 87.09%
graph-classification-on-mutagGDL-g (ADJ)
Accuracy: 58.45%
graph-classification-on-proteinsGDL-g (SP)
Accuracy: 74.86

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Online Graph Dictionary Learning | Papers | HyperAI