Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Optimal Transport for structured data with application on graphs

About

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\'echet means or barycenters of graphs are illustrated and discussed in a clustering context.

Titouan Vayer, Laetitia Chapel, R\'emi Flamary, Romain Tavenard, Nicolas Courty• 2018

Related benchmarks

TaskDatasetResultRank
Graph ClassificationIMDB-M
Accuracy48
425
Graph ClassificationIMDB-B
Accuracy63.8
425
Graph ClassificationENZYMES
Accuracy71
328
Graph ClassificationMUTAG (10-fold cross-validation)
Accuracy84.4
227
Graph ClassificationPROTEINS (10-fold cross-validation)
Accuracy74.3
223
Graph ClassificationBZR
Accuracy85.12
165
Graph ClassificationCOX2
Accuracy77.23
161
Graph ClassificationIMDB-B (10-fold cross-validation)
Accuracy70.8
155
Graph ClassificationPTC (10-fold cross-validation)
Accuracy55.5
120
Graph ClassificationNCI1 (10-fold cross-validation)
Accuracy74.4
110
Showing 10 of 44 rows

Other info

Follow for update