Learning Graphons via Structured Gromov-Wasserstein Barycenters

Hongteng Xu,Dixin Luo,Hongyuan Zha,Lawrence Carin Duke
2021-01-01
Abstract:In this paper, we propose a novel and theoretically-supportive method to learn a nonparametric graph model called \textit{graphon}, which is defined in an infinite-dimensional space and represents arbitrary-size graphs with limits in the space. Based on the weak regularity lemma in the theory of graphon, our method leverages a step function to approximate a graphon. We show that the cut distance of graphons can be relaxed to the Gromov-Wasserstein distance of their step functions. Accordingly, given a set of graphs generated by the proposed graphon, we learn the corresponding step function as the Gromov-Wasserstein barycenter of the graphs.Furthermore, we develop some variants of the algorithm, $e.g.$, the smoothed Gromov-Wasserstein barycenter for guaranteeing the continuity of the learned graphons, and the mixed Gromov-Wasserstein barycenters for learning multiple structured graphons. Compared with state-of-the-art methods, our method overcomes their drawbacks and outperforms them on both synthetic and real-world data.
What problem does this paper attempt to address?