DIG: A Turnkey Library for Diving into Graph Deep Learning Research
Not on the map. Why?
Abstract (arXiv)
Although there exist several libraries for deep learning on graphs, they are aiming at implementing basic operations for graph deep learning. In the research community, implementing and benchmarking various advanced tasks are still painful and time-consuming with existing libraries. To facilitate graph deep learning research, we introduce DIG: Dive into Graphs, a turnkey library that provides a unified testbed for higher level, research-oriented graph deep learning tasks. Currently, we consider graph generation, self-supervised learning on graphs, explainability of graph neural networks, and deep learning on 3D graphs. For each direction, we provide unified implementations of data interfaces, common algorithms, and evaluation metrics. Altogether, DIG is an extensible, open-source, and turnkey library for researchers to develop new methods and effortlessly compare with common baselines using widely used datasets and evaluation metrics. Source code is available at https://github.com/divelab/DIG.
Key facts. Cited by 76 papers in the map (data through 7 Oct 2026). Builds on 43 of its 57 references that are in the map. First posted on arXiv 23 Mar 2021.
Builds on
43 of 57 references.
References · 43
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2017Inductive Representation Learning on Large Graphscited by 6,284
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,906
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,148
- 2018Graph Neural Networks: A Review of Methods and Applicationscited by 1,571
- 2017Learning Important Features Through Propagating Activation Differencescited by 1,454
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,341
- 2020Graph Contrastive Learning with Augmentationscited by 1,203
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,042
31 more
- 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,037
- 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 970
- 2020Learning to Simulate Complex Physics with Graph Networkscited by 867
- 2019GNNExplainer: Generating Explanations for Graph Neural Networkscited by 735
- 2020TUDataset: A collection of benchmark datasets for learning with graphscited by 718
- 2020Contrastive Multi-View Representation Learning on Graphscited by 651
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 647
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2020Deep Graph Contrastive Representation Learningcited by 535
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 503
- 2017Representation Learning on Graphs: Methods and Applicationscited by 456
- 2019InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximizationcited by 441
- 2020Parameterized Explainer for Graph Neural Networkcited by 423
- 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 383
- 2020GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generationcited by 301
- 2021On Explainability of Graph Neural Networks via Subgraph Explorationscited by 280
- 2018Molecular Sets (MOSES): A Benchmarking Platform for Molecular Generation Modelscited by 270
- 2020Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Moleculescited by 254
- 2020XGNN: Towards Model-Level Explanations of Graph Neural Networkscited by 211
- 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
- 2021GraphDF: A Discrete Flow Model for Molecular Graph Generationcited by 177
- 2020Higher-Order Explanations of Graph Neural Networks via Relevant Walkscited by 138
- 2020Self-supervised Learning on Graphs: Deep Insights and New Directioncited by 108
- 2020A Systematic Survey on Deep Generative Models for Graph Generationcited by 94
- 2021GraphEBM: Molecular Graph Generation with Energy-Based Modelscited by 85
- 2020Graph Neural Networks in TensorFlow and Keras with Spektralcited by 59
- 2020Advanced Graph and Sequence Neural Networks for Molecular Property Prediction and Drug Discoverycited by 46
- 2020Deep Graph Generators: A Surveycited by 36
- 2021Efficient Graph Deep Learning in TensorFlow with tf_geometriccited by 17
- 2021GraphGallery: A Platform for Fast Benchmarking and Easy Development of Graph Neural Networks Based Intelligent Softwarecited by 8
Built on by
Citing papers per year: 2021: 7; 2022: 24; 2023: 25; 2024: 9; 2025: 7; 2026: 4.
76 papers.
Sort:
By area:
- 2022Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equationscited by 273
- 2023Towards Better Dynamic Graph Learning: New Architecture and Unified Librarycited by 154
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
- 2021PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Modelscited by 118
- 2022Generating 3D Molecules for Target Protein Bindingcited by 104
- 2021Trustworthy AI: A Computational Perspectivecited by 93
- 2022Periodic Graph Transformers for Crystal Material Property Predictioncited by 87
- 2022Evaluating Explainability for Graph Neural Networkscited by 85
- 2021Robust Counterfactual Explanations on Graph Neural Networkscited by 72
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2022DiffBP: Generative Diffusion of 3D Molecules for Target Protein Bindingcited by 58
- 2022Learning Hierarchical Protein Representations via Complete 3D Graph Networkscited by 54
38 more
- 2025PyG 2.0: Scalable Learning on Real World Graphscited by 49
- 2023A new perspective on building efficient and expressive 3D equivariant graph neural networkscited by 46
- 2022TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug Discoverycited by 45
- 2022GStarX: Explaining Graph Neural Networks with Structure-Aware Cooperative Gamescited by 42
- 2023Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasetscited by 37
- 2024Complete and Efficient Graph Transformers for Crystal Material Property Predictioncited by 34
- 2022Global Concept-Based Interpretability for Graph Neural Networks via Neuron Analysiscited by 32
- 2022Distributed Graph Neural Network Training: A Surveycited by 32
- 2022GraphFM: Improving Large-Scale GNN Training via Feature Momentumcited by 31
- 2023Functional-Group-Based Diffusion for Pocket-Specific Molecule Generation and Elaborationcited by 30
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalizationcited by 29
- 2023A Latent Diffusion Model for Protein Structure Generationcited by 28
- 2023QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Moleculescited by 26
- 2024PPFlow: Target-aware Peptide Design with Torsional Flow Matchingcited by 26
- 2023Towards Symmetry-Aware Generation of Periodic Materialscited by 25
- 2022Self-Supervised Representation Learning via Latent Graph Predictioncited by 22
- 2023Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materialscited by 22
- 2022Task-Agnostic Graph Explanationscited by 20
- 2022FlowX: Towards Explainable Graph Neural Networks via Message Flowscited by 20
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structurescited by 20
- 2022Automated Data Augmentations for Graph Classificationcited by 19
- 2022PyTorch Geometric Signed Directed: A Software Package on Graph Neural Networks for Signed and Directed Graphscited by 18
- 2023DEGREE: Decomposition Based Explanation For Graph Neural Networkscited by 18
- 2023Towards Robust Fidelity for Evaluating Explainability of Graph Neural Networkscited by 16
- 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
- 2023Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalizationcited by 12
- 2022ChemicalX: A Deep Learning Library for Drug Pair Scoringcited by 11
- 2023Interpretable Sparsification of Brain Graphs: Better Practices and Effective Designs for Graph Neural Networkscited by 11
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalizationcited by 10
- 20223D Graph Contrastive Learning for Molecular Property Predictioncited by 9
- 2022L2XGNN: Learning to Explain Graph Neural Networkscited by 7
- 2022Graph Generative Model for Benchmarking Graph Neural Networkscited by 6
- 2021FedGraphNN: A Federated Learning System and Benchmark for Graph Neural Networkscited by 5
- 2022Graph Learning Indexer: A Contributor-Friendly and Metadata-Rich Platform for Graph Learning Benchmarkscited by 4
- 2023Structural Explanations for Graph Neural Networks using HSICcited by 4
- 2023PyTorch Geometric High Order: A Unified Library for High Order Graph Neural Networkcited by 4
- 2024A Differential Geometric View and Explainability of GNN on Evolving Graphscited by 4
- 2021Graph Feature Gating Networkscited by 3
50 most cited of 76 All 76 (JSON)
- 2026WOMBAT: Whitebox Oracle for Molecular Benchmarking and Attribution Testing30 Sep 2026
- 2026Controllable GNN Explanations via Multi-Metric Preference Selection26 Sep 2026
- 2026Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization9 Apr 2026
- 2026FSX: Message Flow Sensitivity Enhanced Structural Explainer for Graph Neural Networks21 Jan 2026
- 2025PyG 2.0: Scalable Learning on Real World Graphs22 Jul 2025
- 2025Explainable Graph Neural Networks via Structural Externalities19 Jul 2025
- 2025Structure-Aware Compound-Protein Affinity Prediction via Graph Neural Networks with Group Lasso Regularization4 Jul 2025
- 2025Towards Comprehensive and Prerequisite-Free Explainer for Graph Neural Networks20 May 2025
- 2025Framework GNN-AID: Graph Neural Network Analysis Interpretation and Defense6 May 2025
- 2025RISE: Radius of Influence based Subgraph Extraction for 3D Molecular Graph Explanation4 May 2025
- 2025PyGDA: A Python Library for Graph Domain Adaptation13 Mar 2025
- 2024OneProt: Towards Multi-Modal Protein Foundation Models7 Nov 2024
38 more
- 2024Explanation-Preserving Augmentation for Semi-Supervised Graph Representation Learning16 Oct 2024
- 2024Beyond Euclid: An Illustrated Guide to Modern Machine Learning with Geometric, Topological, and Algebraic Structures12 Jul 2024
- 2024Complete and Efficient Graph Transformers for Crystal Material Property Prediction18 Mar 2024
- 2024A Differential Geometric View and Explainability of GNN on Evolving Graphs11 Mar 2024
- 2024PPFlow: Target-aware Peptide Design with Torsional Flow Matching5 Mar 2024
- 2024Unsupervised Optimisation of GNNs for Node Clustering12 Feb 2024
- 2024ADA-GNN: Atom-Distance-Angle Graph Neural Network for Crystal Material Property Prediction22 Jan 2024
- 2024On Discprecncies between Perturbation Evaluations of Graph Neural Network Attributions1 Jan 2024
- 2023PerCNet: Periodic Complete Representation for Crystal Graphs3 Dec 2023
- 2023PyTorch Geometric High Order: A Unified Library for High Order Graph Neural Network28 Nov 2023
- 2023Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasets6 Oct 2023
- 2023Towards Robust Fidelity for Evaluating Explainability of Graph Neural Networks3 Oct 2023
- 2023On the Robustness of Post-hoc GNN Explainers to Label Noise4 Sep 2023
- 2023Will More Expressive Graph Neural Networks do Better on Generative Tasks?23 Aug 2023
- 2023XFlow: Benchmarking Flow Behaviors over Graphs7 Aug 2023
- 2023With Flying Colors: Predicting Community Success in Large-scale Collaborative Campaigns18 Jul 2023
- 2023Towards Symmetry-Aware Generation of Periodic Materials6 Jul 2023
- 2023Interpretable Sparsification of Brain Graphs: Better Practices and Effective Designs for Graph Neural Networks26 Jun 2023
- 2023QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Molecules15 Jun 2023
- 2023Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materials15 Jun 2023
- 2023Graph Structure and Feature Extrapolation for Out-of-Distribution Generalization13 Jun 2023
- 2023LibAUC: A Deep Learning Library for X-Risk Optimization5 Jun 2023
- 2023Message-passing selection: Towards interpretable GNNs for graph classification3 Jun 2023
- 2023Joint Learning of Label and Environment Causal Independence for Graph Out-of-Distribution Generalization1 Jun 2023
- 2023Improving Expressivity of GNNs with Subgraph-specific Factor Embedded Normalization31 May 2023
- 2023Functional-Group-Based Diffusion for Pocket-Specific Molecule Generation and Elaboration30 May 2023
- 2023DEGREE: Decomposition Based Explanation For Graph Neural Networks22 May 2023
- 2023A Latent Diffusion Model for Protein Structure Generation6 May 2023
- 2023A new perspective on building efficient and expressive 3D equivariant graph neural networks7 Apr 2023
- 2023Towards Better Dynamic Graph Learning: New Architecture and Unified Library23 Mar 2023
- 2023Instance-incremental Scene Graph Generation from Real-world Point Clouds via Normalizing Flows21 Feb 2023
- 2023Structural Explanations for Graph Neural Networks using HSIC4 Feb 2023
- 2023A Network Science perspective of Graph Convolutional Networks: A survey12 Jan 2023
- 2022Graph Learning Indexer: A Contributor-Friendly and Metadata-Rich Platform for Graph Learning Benchmarks8 Dec 2022
- 2022DiffBP: Generative Diffusion of 3D Molecules for Target Protein Binding21 Nov 2022
- 2022Distributed Graph Neural Network Training: A Survey1 Nov 2022
- 2022L2XGNN: Learning to Explain Graph Neural Networks28 Sep 2022
- 2022Periodic Graph Transformers for Crystal Material Property Prediction23 Sep 2022
50 newest of 76 All 76 (JSON)
- 2023Towards Better Dynamic Graph Learning: New Architecture and Unified Librarycited by 154
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
- 2021PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Modelscited by 118
- 2022Generating 3D Molecules for Target Protein Bindingcited by 104
- 2021Trustworthy AI: A Computational Perspectivecited by 93
- 2022Periodic Graph Transformers for Crystal Material Property Predictioncited by 87
- 2022Evaluating Explainability for Graph Neural Networkscited by 85
- 2021Robust Counterfactual Explanations on Graph Neural Networkscited by 72
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2022DiffBP: Generative Diffusion of 3D Molecules for Target Protein Bindingcited by 58
- 2022Learning Hierarchical Protein Representations via Complete 3D Graph Networkscited by 54
- 2025PyG 2.0: Scalable Learning on Real World Graphscited by 49
12 most cited of 74 All 74 (JSON)
- 2026WOMBAT: Whitebox Oracle for Molecular Benchmarking and Attribution Testing30 Sep 2026
- 2026Controllable GNN Explanations via Multi-Metric Preference Selection26 Sep 2026
- 2026Adversarial Label Invariant Graph Data Augmentations for Out-of-Distribution Generalization9 Apr 2026
- 2026FSX: Message Flow Sensitivity Enhanced Structural Explainer for Graph Neural Networks21 Jan 2026
- 2025PyG 2.0: Scalable Learning on Real World Graphs22 Jul 2025
- 2025Explainable Graph Neural Networks via Structural Externalities19 Jul 2025
- 2025Structure-Aware Compound-Protein Affinity Prediction via Graph Neural Networks with Group Lasso Regularization4 Jul 2025
- 2025Towards Comprehensive and Prerequisite-Free Explainer for Graph Neural Networks20 May 2025
- 2025Framework GNN-AID: Graph Neural Network Analysis Interpretation and Defense6 May 2025
- 2025RISE: Radius of Influence based Subgraph Extraction for 3D Molecular Graph Explanation4 May 2025
- 2025PyGDA: A Python Library for Graph Domain Adaptation13 Mar 2025
- 2024OneProt: Towards Multi-Modal Protein Foundation Models7 Nov 2024
12 newest of 74 All 74 (JSON)