GLSearch: Maximum Common Subgraph Detection Via Learning to Search

Yunsheng Bai,Derek Xu,Yizhou Sun,Wei Wang
DOI: https://doi.org/10.48550/arxiv.2002.03129
2020-01-01
Abstract:Detecting the Maximum Common Subgraph (MCS) between two input graphs isfundamental for applications in drug synthesis, malware detection, cloudcomputing, etc. However, MCS computation is NP-hard, and state-of-the-art MCSsolvers rely on heuristic search algorithms which in practice cannot find goodsolution for large graph pairs given a limited computation budget. We proposeGLSearch, a Graph Neural Network (GNN) based learning to search model. Ourmodel is built upon the branch and bound algorithm, which selects one pair ofnodes from the two input graphs to expand at a time. Instead of usingheuristics, we propose a novel GNN-based Deep Q-Network (DQN) to select thenode pair, allowing the search process faster and more adaptive. To furtherenhance the training of DQN, we leverage the search process to providesupervision in a pre-training stage and guide our agent during an imitationlearning stage. Experiments on synthetic and real-world large graph pairsdemonstrate that our model learns a search strategy that is able to detectsignificantly larger common subgraphs given the same computation budget. OurGLSearch can be potentially extended to solve many other combinatorial problemswith constraints on graphs.
What problem does this paper attempt to address?