Joint Optimal Transport and Embedding for Network Alignment
Track: Social networks and social media
Keywords: Network Alignment, Optimal Transport, Network Embedding
TL;DR: We reveal the close relationship between graph optimal transport and node embedding learning, and propose a unified framework for network alignment.
Abstract: Network alignment, which aims to find node correspondence across different networks, is the cornerstone of various downstream multi-network and Web mining tasks. Most of the embedding-based methods indirectly model cross-network node relationships by contrasting positive and negative node pairs sampled from hand-crafted strategies, which are vulnerable to graph noises and leads to potential misalignment of nodes. Another line of works based on the optimal transport (OT) theory directly model cross-network node relationships and generate noise-reduced alignments. However, OT methods heavily rely on fixed, pre-defined cost functions that prohibit end-to-end training and are hard to generalize. In this paper, we aim to unify the embedding and OT-based methods in a mutually beneficial manner and propose a joint optimal transport and embedding framework for network alignment named JOENA. For one thing (OT for embedding), through a simple yet effective transformation, the noise-reduced OT mapping serves as an adaptive sampling strategy directly modeling all cross-network node pairs for robust embedding learning. For another (embedding for OT), on top of the learned node embeddings, the OT cost can be gradually trained along the learning process in an end-to-end fashion, which further enhances the alignment quality. With a unified objective, the mutual benefits of both methods can be achieved by an alternating optimization schema with guaranteed convergence. Extensive experiments on real-world networks validate the effectiveness and scalability of JOENA, achieving up to 16% improvement in MRR and 20 times speedup compared with the state-of-the-art alignment methods.
Submission Number: 908
Loading