Graph Diffusion Wasserstein Distances

Abstract

Optimal Transport (OT) for structured data has received much attention in the machine learning community, especially for addressing graph classification or graph transfer learning tasks. In this paper, we present the Diffusion Wasserstein ( $\mathtt {DW}$ DW ) distance, as a generalization of the standard Wasserstein distance to undirected and connected graphs where nodes are described by feature vectors. $\mathtt {DW}$ DW is based on the Laplacian exponential kernel and benefits from the heat diffusion to catch both structural and feature information from the graphs. We further derive lower/upper bounds on $\mathtt {DW}$ DW and show that it can be directly plugged into the Fused Gromov Wasserstein ( $\mathtt {FGW}$ FGW ) distance that has been recently proposed, leading - for free - to a DifFused Gromov Wasserstein distance ( $\mathtt {DFGW}$ DFGW ) that allows a significant performance boost when solving graph domain adaptation tasks.

Cite

Text

Barbe et al. "Graph Diffusion Wasserstein Distances." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2020. doi:10.1007/978-3-030-67661-2_34

Markdown

[Barbe et al. "Graph Diffusion Wasserstein Distances." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2020.](https://mlanthology.org/ecmlpkdd/2020/barbe2020ecmlpkdd-graph/) doi:10.1007/978-3-030-67661-2_34

BibTeX

@inproceedings{barbe2020ecmlpkdd-graph,
  title     = {{Graph Diffusion Wasserstein Distances}},
  author    = {Barbe, Amélie and Sebban, Marc and Gonçalves, Paulo and Borgnat, Pierre and Gribonval, Rémi},
  booktitle = {European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
  year      = {2020},
  pages     = {577-592},
  doi       = {10.1007/978-3-030-67661-2_34},
  url       = {https://mlanthology.org/ecmlpkdd/2020/barbe2020ecmlpkdd-graph/}
}