HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Optimal Transport for structured data with application on graphs

Titouan Vayer; Laetitia Chapel; Rémi Flamary; Romain Tavenard; Nicolas Courty

Optimal Transport for structured data with application on graphs

Abstract

This work considers the problem of computing distances between structured objects such as undirected graphs, seen as probability distributions in a specific metric space. We consider a new transportation distance (i.e. that minimizes a total cost of transporting probability masses) that unveils the geometric nature of the structured objects space. Unlike Wasserstein or Gromov-Wasserstein metrics that focus solely and respectively on features (by considering a metric in the feature space) or structure (by seeing structure as a metric space), our new distance exploits jointly both information, and is consequently called Fused Gromov-Wasserstein (FGW). After discussing its properties and computational aspects, we show results on a graph classification task, where our method outperforms both graph kernels and deep graph convolutional networks. Exploiting further on the metric properties of FGW, interesting geometric objects such as Fréchet means or barycenters of graphs are illustrated and discussed in a clustering context.

Code Repositories

tvayer/FGW
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
graph-classification-on-enzymesFGW sp
Accuracy: 71.00%
graph-classification-on-mutagFGW wl h=2 sp
Accuracy: 86.42%
graph-classification-on-mutagFGW wl h=4 sp
Accuracy: 88.42%
graph-classification-on-mutagFGW raw sp
Accuracy: 83.26%
graph-classification-on-nci1FGW wl h=4 sp
Accuracy: 86.42%
graph-classification-on-nci1FGW raw sp
Accuracy: 72.82%
graph-classification-on-nci1FGW wl h=2 sp
Accuracy: 85.82%
graph-classification-on-proteinsFGW sp
Accuracy: 74.55%

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
Optimal Transport for structured data with application on graphs | Papers | HyperAI