Learning Hierarchical Protein Representations via Complete 3D Graph Networks
Not on the map. Why?
Abstract (arXiv)
We consider representation learning for proteins with 3D structures. We build 3D graphs based on protein structures and develop graph networks to learn their representations. Depending on the levels of details that we wish to capture, protein representations can be computed at different levels, e.g., the amino acid, backbone, or all-atom levels. Importantly, there exist hierarchical relations among different levels. In this work, we propose to develop a novel hierarchical graph network, known as ProNet, to capture the relations. Our ProNet is very flexible and can be used to compute protein representations at different levels of granularity. By treating each amino acid as a node in graph modeling as well as harnessing the inherent hierarchies, our ProNet is more effective and efficient than existing methods. We also show that, given a base 3D graph network that is complete, our ProNet representations are also complete at all levels. Experimental results show that ProNet outperforms recent methods on most datasets. In addition, results indicate that different downstream tasks may require representations at different levels. Our code is publicly available as part of the DIG library (urlhttps://github.com/divelab/DIG).
Key facts. Cited by 54 papers in the map (data through 7 Oct 2026). Builds on 39 of its 79 references that are in the map. First posted on arXiv 26 Jul 2022.
Builds on
39 of 79 references.
References · 39
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2018Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networkscited by 878
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2019Graph U-Netscited by 489
- 2020Learning from Protein Structure with Geometric Vector Perceptronscited by 308
- 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 258
- 2019Evaluating Protein Transfer Learning with TAPEcited by 229
27 more
- 2021Self-Supervised Learning of Graph Neural Networks: A Unified Reviewcited by 184
- 2018DeepDTA: Deep Drug-Target Binding Affinity Predictioncited by 171
- 2021Spherical Message Passing for 3D Graph Networkscited by 166
- 2020ProtTrans: Towards Cracking the Language of Life's Code Through Self-Supervised Deep Learning and High Performance Computingcited by 162
- 2022Protein Representation Learning by Geometric Structure Pretrainingcited by 161
- 2022EquiBind: Geometric Deep Learning for Drug Binding Structure Predictioncited by 143
- 2022GOOD: A Graph Out-of-Distribution Benchmarkcited by 140
- 2021Independent SE(3)-Equivariant Models for End-to-End Rigid Protein Dockingcited by 81
- 2021DIG: A Turnkey Library for Diving into Graph Deep Learning Researchcited by 76
- 2020ATOM3D: Tasks On Molecules in Three Dimensionscited by 73
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2021Equivariant Graph Neural Networks for 3D Macromolecular Structurecited by 53
- 2022Multi-Scale Representation Learning on Proteinscited by 52
- 2019Learning protein sequence embeddings using information from structurecited by 50
- 2017DeepSF: deep convolutional neural network for mapping protein sequences to foldscited by 46
- 2020Advanced Graph and Sequence Neural Networks for Molecular Property Prediction and Drug Discoverycited by 46
- 2020Topology-Aware Graph Pooling Networkscited by 44
- 2021Protein sequence design with deep generative modelscited by 35
- 2018End-to-End Learning on 3D Protein Structure for Interface Predictioncited by 32
- 2022GraphFM: Improving Large-Scale GNN Training via Feature Momentumcited by 31
- 2022Contrastive Representation Learning for 3D Protein Structurescited by 30
- 2021Fold2Seq: A Joint Sequence(1D)-Fold(3D) Embedding-based Generative Model for Protein Designcited by 23
- 2020Deep Learning of High-Order Interactions for Protein Interface Predictioncited by 22
- 2022Task-Agnostic Graph Explanationscited by 20
- 2021Geometric Transformers for Protein Interface Contact Predictioncited by 18
- 2022Orientation-Aware Graph Neural Networks for Protein Structure Representation Learningcited by 8
- 2022EGR: Equivariant Graph Refinement and Assessment of 3D Protein Complex Structurescited by 3
Built on by
Citing papers per year: 2022: 1; 2023: 10; 2024: 20; 2025: 12; 2026: 11.
54 papers.
Sort:
By area:
- 2023On the Expressive Power of Geometric Graph Neural Networkscited by 89
- 2023A Hitchhiker's Guide to Geometric GNNs for 3D Atomic Systemscited by 65
- 2023A new perspective on building efficient and expressive 3D equivariant graph neural networkscited by 46
- 2023Efficient Approximations of Complete Interatomic Potentials for Crystal Property Predictioncited by 37
- 2024Complete and Efficient Graph Transformers for Crystal Material Property Predictioncited by 34
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2023Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materialscited by 22
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2024Evaluating representation learning on the protein structure universecited by 16
- 2022Geometry-Complete Perceptron Networks for 3D Molecular Graphscited by 15
- 2023De novo protein design using geometric vector field networkscited by 14
- 2024Pre-training with Fractional Denoising to Enhance Molecular Property Predictioncited by 14
38 more
- 2023ProFSA: Self-supervised Pocket Pretraining via Protein Fragment-Surroundings Alignmentcited by 12
- 2023Protein-ligand binding representation learning from fine-grained interactionscited by 12
- 2023Multi-body SE(3) Equivariance for Unsupervised Rigid Segmentation and Motion Estimationcited by 7
- 2024Pre-Training Protein Bi-level Representation Through Span Mask Strategy On 3D Protein Chainscited by 6
- 2024Clustering for Protein Representation Learningcited by 6
- 2023AtomSurf : Surface Representation for Learning on Protein Structurescited by 5
- 2024An Equivariant Pretrained Transformer for Unified 3D Molecular Representation Learningcited by 5
- 2024Using pretrained graph neural networks with token mixers as geometric featurizers for conformational dynamicscited by 5
- 2024Binding Affinity Prediction: From Conventional to Machine Learning-Based Approachescited by 3
- 2024Pushing the Limits of All-Atom Geometric Graph Neural Networks: Pre-Training, Scaling and Zero-Shot Transfercited by 3
- 2024MeToken: Uniform Micro-environment Token Boosts Post-Translational Modification Predictioncited by 3
- 2025Advances in Protein Representation Learning: Methods, Applications, and Future Directionscited by 3
- 2025RISE: Radius of Influence based Subgraph Extraction for 3D Molecular Graph Explanationcited by 3
- 2026Self-Conditioned Denoising for Atomistic Representation Learningcited by 3
- 2024GOProteinGNN: Leveraging Protein Knowledge Graphs for Protein Representation Learningcited by 2
- 2024OneProt: Towards Multi-Modal Protein Foundation Modelscited by 2
- 2025Multi-Modality Representation Learning for Antibody-Antigen Interactions Predictioncited by 2
- 2024PepGB: Facilitating peptide drug discovery via graph neural networkscited by 1
- 2024Deep Manifold Transformation for Protein Representation Learningcited by 1
- 2025Molecule Generation for Target Protein Binding with Hierarchical Consistency Diffusion Modelcited by 1
- 2025A Comprehensive Benchmark for RNA 3D Structure-Function Modelingcited by 1
- 2025Bidirectional Hierarchical Protein Multi-Modal Representation Learningcited by 1
- 2025GAGA: Gaussianity-Aware Gaussian Approximation for Efficient 3D Molecular Generationcited by 1
- 2025Contrastive Multi-Task Learning with Solvent-Aware Augmentation for Drug Discoverycited by 1
- 2024NaNa and MiGu: Semantic Data Augmentation Techniques to Enhance Protein Classification in Graph Neural Networkscited by 0
- 2024HC-GAE: The Hierarchical Cluster-based Graph Auto-Encoder for Graph Representation Learningcited by 0
- 2024Geometric Self-Supervised Pretraining on 3D Protein Structures using Subgraphscited by 0
- 2024DeltaDock: A Unified Framework for Accurate, Efficient, and Physically Reliable Molecular Dockingcited by 0
- 2025Modeling All-Atom Glycan Structures via Hierarchical Message Passing and Multi-Scale Pre-trainingcited by 0
- 2025AlphaFold Database Debiasing for Robust Inverse Foldingcited by 0
- 2025Demystify Protein Generation with Hierarchical Conditional Diffusion Modelscited by 0
- 2025Tokenizing Loops of Antibodiescited by 0
- 2026MMPG: MoE-based Adaptive Multi-Perspective Graph Fusion for Protein Representation Learningcited by 0
- 2026Towards Multiscale Graph-based Protein Learning with Geometric Secondary Structural Motifscited by 0
- 2026Rigidity-Aware Geometric Pretraining for Protein Design and Conformational Ensemblescited by 0
- 2026h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Networkcited by 0
- 2026Structural Interpretations of Protein Language Model Representations via Differentiable Graph Partitioningcited by 0
- 2026Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generationcited by 0
50 most cited of 54 All 54 (JSON)
- 2026Hyper-Fold: Exploring the Expressive Limit of Sequence-Geometry Learning for Proteins via Hypergraph Modeling29 Aug 2026
- 2026Enhancing Protein Representation Learning via Manifold Restore Mixing21 Jun 2026
- 2026Protein Representation Learning with Secondary-Structure and Energy-Filtered Hydrogen-Bond Graphs12 Jun 2026
- 2026Curvature-Informed Potential Energy Surface for Protein-Ligand Binding Affinity Prediction12 Jun 2026
- 2026Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generation1 Jun 2026
- 2026Structural Interpretations of Protein Language Model Representations via Differentiable Graph Partitioning9 May 2026
- 2026h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Network25 Apr 2026
- 2026Self-Conditioned Denoising for Atomistic Representation Learning17 Mar 2026
- 2026Rigidity-Aware Geometric Pretraining for Protein Design and Conformational Ensembles2 Mar 2026
- 2026Towards Multiscale Graph-based Protein Learning with Geometric Secondary Structural Motifs31 Jan 2026
- 2026MMPG: MoE-based Adaptive Multi-Perspective Graph Fusion for Protein Representation Learning15 Jan 2026
- 2025Tokenizing Loops of Antibodies10 Sep 2025
38 more
- 2025Contrastive Multi-Task Learning with Solvent-Aware Augmentation for Drug Discovery3 Aug 2025
- 2025Demystify Protein Generation with Hierarchical Conditional Diffusion Models24 Jul 2025
- 2025GAGA: Gaussianity-Aware Gaussian Approximation for Efficient 3D Molecular Generation11 Jul 2025
- 2025AlphaFold Database Debiasing for Robust Inverse Folding10 Jun 2025
- 2025Modeling All-Atom Glycan Structures via Hierarchical Message Passing and Multi-Scale Pre-training2 Jun 2025
- 2025RISE: Radius of Influence based Subgraph Extraction for 3D Molecular Graph Explanation4 May 2025
- 2025Bidirectional Hierarchical Protein Multi-Modal Representation Learning7 Apr 2025
- 2025A Comprehensive Benchmark for RNA 3D Structure-Function Modeling27 Mar 2025
- 2025Multi-Modality Representation Learning for Antibody-Antigen Interactions Prediction22 Mar 2025
- 2025Advances in Protein Representation Learning: Methods, Applications, and Future Directions20 Mar 2025
- 2025Molecule Generation for Target Protein Binding with Hierarchical Consistency Diffusion Model2 Mar 2025
- 2024OneProt: Towards Multi-Modal Protein Foundation Models7 Nov 2024
- 2024MeToken: Uniform Micro-environment Token Boosts Post-Translational Modification Prediction4 Nov 2024
- 2024Pushing the Limits of All-Atom Geometric Graph Neural Networks: Pre-Training, Scaling and Zero-Shot Transfer29 Oct 2024
- 2024DeltaDock: A Unified Framework for Accurate, Efficient, and Physically Reliable Molecular Docking15 Oct 2024
- 2024Binding Affinity Prediction: From Conventional to Machine Learning-Based Approaches30 Sep 2024
- 2024Using pretrained graph neural networks with token mixers as geometric featurizers for conformational dynamics30 Sep 2024
- 2024GOProteinGNN: Leveraging Protein Knowledge Graphs for Protein Representation Learning31 Jul 2024
- 2024Pre-training with Fractional Denoising to Enhance Molecular Property Prediction14 Jul 2024
- 2024Geometric Self-Supervised Pretraining on 3D Protein Structures using Subgraphs20 Jun 2024
- 2024Evaluating representation learning on the protein structure universe19 Jun 2024
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency11 Jun 2024
- 2024HC-GAE: The Hierarchical Cluster-based Graph Auto-Encoder for Graph Representation Learning23 May 2024
- 2024Clustering for Protein Representation Learning30 Mar 2024
- 2024NaNa and MiGu: Semantic Data Augmentation Techniques to Enhance Protein Classification in Graph Neural Networks21 Mar 2024
- 2024Complete and Efficient Graph Transformers for Crystal Material Property Prediction18 Mar 2024
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applications1 Mar 2024
- 2024An Equivariant Pretrained Transformer for Unified 3D Molecular Representation Learning20 Feb 2024
- 2024Pre-Training Protein Bi-level Representation Through Span Mask Strategy On 3D Protein Chains2 Feb 2024
- 2024PepGB: Facilitating peptide drug discovery via graph neural networks26 Jan 2024
- 2024Deep Manifold Transformation for Protein Representation Learning12 Jan 2024
- 2023A Hitchhiker's Guide to Geometric GNNs for 3D Atomic Systems12 Dec 2023
- 2023Protein-ligand binding representation learning from fine-grained interactions9 Nov 2023
- 2023De novo protein design using geometric vector field networks18 Oct 2023
- 2023ProFSA: Self-supervised Pocket Pretraining via Protein Fragment-Surroundings Alignment11 Oct 2023
- 2023AtomSurf : Surface Representation for Learning on Protein Structures28 Sep 2023
- 2023Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materials15 Jun 2023
- 2023Efficient Approximations of Complete Interatomic Potentials for Crystal Property Prediction12 Jun 2023
50 newest of 54 All 54 (JSON)
- 2023On the Expressive Power of Geometric Graph Neural Networkscited by 89
- 2023A Hitchhiker's Guide to Geometric GNNs for 3D Atomic Systemscited by 65
- 2023A new perspective on building efficient and expressive 3D equivariant graph neural networkscited by 46
- 2023Efficient Approximations of Complete Interatomic Potentials for Crystal Property Predictioncited by 37
- 2024Complete and Efficient Graph Transformers for Crystal Material Property Predictioncited by 34
- 2024A Survey of Geometric Graph Neural Networks: Data Structures, Models and Applicationscited by 28
- 2023Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materialscited by 22
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2024Evaluating representation learning on the protein structure universecited by 16
- 2022Geometry-Complete Perceptron Networks for 3D Molecular Graphscited by 15
- 2023De novo protein design using geometric vector field networkscited by 14
- 2024Pre-training with Fractional Denoising to Enhance Molecular Property Predictioncited by 14
12 most cited of 52 All 52 (JSON)
- 2026Hyper-Fold: Exploring the Expressive Limit of Sequence-Geometry Learning for Proteins via Hypergraph Modeling29 Aug 2026
- 2026Enhancing Protein Representation Learning via Manifold Restore Mixing21 Jun 2026
- 2026Protein Representation Learning with Secondary-Structure and Energy-Filtered Hydrogen-Bond Graphs12 Jun 2026
- 2026Curvature-Informed Potential Energy Surface for Protein-Ligand Binding Affinity Prediction12 Jun 2026
- 2026Uncertainty-Calibrated Diffusion for Reliable 3D Molecular Graph Generation1 Jun 2026
- 2026Structural Interpretations of Protein Language Model Representations via Differentiable Graph Partitioning9 May 2026
- 2026h-MINT: Modeling Pocket-Ligand Binding with Hierarchical Molecular Interaction Network25 Apr 2026
- 2026Self-Conditioned Denoising for Atomistic Representation Learning17 Mar 2026
- 2026Rigidity-Aware Geometric Pretraining for Protein Design and Conformational Ensembles2 Mar 2026
- 2026Towards Multiscale Graph-based Protein Learning with Geometric Secondary Structural Motifs31 Jan 2026
- 2025Tokenizing Loops of Antibodies10 Sep 2025
- 2025Contrastive Multi-Task Learning with Solvent-Aware Augmentation for Drug Discovery3 Aug 2025
12 newest of 52 All 52 (JSON)