用强化学习和图变换器提升子图匹配的效率与准确率。
Approximate Subgraph Matching with Neural Graph Representations and Reinforcement Learning
- 用图变换器提取全局图特征,替代传统启发式搜索。
- 结合模仿学习与PPO算法,显著提升匹配成功率与速度。
- 适合需要高效处理大规模图数据的研究者与工程师。
近似子图匹配(ASM)旨在判断给定查询图是否近似存在于大型目标图中,是图分析中的核心问题,广泛应用于数据库、网络科学、生物化学和隐私保护等领域。由于该问题为NP难,现有方法多依赖启发式策略,难以充分利用图结构信息,导致解的质量不佳。本文提出基于强化学习的近似子图匹配方法(RL-ASM),利用图变换器提取完整图表示,并构建基于策略的匹配机制。模型以分支定界算法为基础,每次选择一对节点进行匹配评估。通过监督信号进行模仿学习预训练,再使用近端策略优化(PPO)对策略进行微调,以最大化长期累积奖励。在合成与真实数据集上的大量实验表明,RL-ASM在效果与效率上均优于现有方法。代码已开源:https://github.com/KaiyangLi1992/RL-ASM。
原文摘要 · Abstract (English)
Approximate subgraph matching (ASM) is a task that determines the approximate presence of a given query graph in a large target graph. Being an NP-hard problem, ASM is critical in graph analysis with a myriad of applications ranging from database systems and network science to biochemistry and privacy. Existing techniques often employ heuristic search strategies, which cannot fully utilize the graph information, leading to sub-optimal solutions. This paper proposes a Reinforcement Learning based Approximate Subgraph Matching (RL-ASM) algorithm that exploits graph transformers to effectively extract graph representations and RL-based policies for ASM. Our model is built upon the branch-and-bound algorithm that selects one pair of nodes from the two input graphs at a time for potential matches. Instead of using heuristics, we exploit a Graph Transformer architecture to extract feature representations that encode the full graph information. To enhance the training of the RL policy, we use supervised signals to guide our agent in an imitation learning stage. Subsequently, the policy is fine-tuned with the Proximal Policy Optimization (PPO) that optimizes the accumulative long-term rewards over episodes. Extensive experiments on both synthetic and real-world datasets demonstrate that our RL-ASM outperforms existing methods in terms of effectiveness and efficiency. Our source code is available at https://github.com/KaiyangLi1992/RL-ASM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。