Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces GraphDETR, a deep learning framework that reformulates subgraph detection as a set prediction problem, leveraging a graph neural network for encoding and a transformer decoder for decoding. By employing bipartite matching during training, GraphDETR not only achieves exact structural matching but also extends to approximate matching, allowing it to detect a wide variety of patterns in larger graphs. The model demonstrates strong empirical performance, achieving an average precision of 91.2% in functional group detection on the ChEMBL dataset, showcasing its effectiveness in practical applications.
GraphDETR can detect diverse subgraph patterns in graphs with up to 1000 nodes, achieving an impressive average precision of 91.2% for molecular functional group detection.
Subgraph detection seeks to identify whether and where instances of query patterns occur within a larger graph. This problem is fundamental across scientific domains and is closely related to subgraph isomorphism, which is NP-complete, limiting combinatorial approaches to small patterns or moderately sized graphs. We introduce GraphDETR, a deep learning framework that formulates subgraph detection as a set prediction problem, analogous to DETR in object detection. GraphDETR encodes the target graph with a graph neural network, and employs a fixed set of learnable query vectors, decoded via a transformer decoder, to predict all pattern occurrences jointly in a single forward pass. This is enabled by training the model end-to-end with bipartite matching. Unlike traditional combinatorial methods that only solve exact structural matching, GraphDETR naturally extends to approximate matching, enabling detection beyond exact pattern correspondence. Empirically, we show that GraphDETR can detect diverse patterns, such as molecular structures, cycles, cliques, and fuzzy patterns of up to 50 nodes, in target graphs with up to 1000 nodes. We further evaluate on molecular functional group detection over the ChEMBL dataset, where GraphDETR predicts the complete set of functional groups per molecule, achieving a strong performance of $\text{AP}_{100} = 91.2$.
Max Planck