提出JOENA框架,联合优化嵌入与最优传输,提升网络对齐精度与效率。
Joint Optimal Transport and Embedding for Network Alignment
- 用最优传输映射自适应采样,指导鲁棒嵌入学习。
- 端到端训练嵌入驱动的代价函数,使对齐准确率提升16%。
- 适合需要高精度跨网络匹配的研究者使用。
网络对齐旨在发现不同网络间的节点对应关系,是多网络和网络挖掘任务的核心。现有基于嵌入的方法通过手工策略采样正负节点对来间接建模跨网络关系,易受图噪声影响,导致对齐错误。另一类基于最优传输(OT)的方法直接建模跨网络关系,生成降噪对齐结果,但依赖固定预设代价函数,难以端到端训练且泛化能力差。本文提出联合最优传输与嵌入的框架JOENA,实现二者互益:一方面,利用降噪的OT映射作为自适应采样策略,直接建模所有跨网络节点对,增强嵌入学习鲁棒性;另一方面,在学习到的嵌入基础上,代价函数可端到端迭代优化,进一步提升对齐质量。通过统一目标函数与交替优化策略,算法保证收敛。在真实网络上的大量实验表明,JOENA在平均倒数排名(MRR)上最高提升16%,速度比当前最优方法快20倍。
原文摘要 · Abstract (English)
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 lead to potential misalignment of nodes. Another line of work based on the optimal transport (OT) theory directly models cross-network node relationships and generates 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 embeddings, the OT cost can be gradually trained 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 20x speedup compared with the state-of-the-art alignment methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。