# Pre-training of Equivariant Graph Matching Networks with Conformation Flexibility for Drug Binding

arXiv: https://arxiv.org/abs/2204.08663 · HTML: https://mlsanity.com/paper/2204.08663/

## Key facts

- Cited by 15 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 19 Apr 2022.

## Built on by (most cited first, 15 of 15)

- [Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasets](https://mlsanity.com/paper/2310.04292/) (2023)
- [A Hierarchical Training Paradigm for Antibody Structure-sequence Co-design](https://mlsanity.com/paper/2311.16126/) (2023)
- [Instructor-inspired Machine Learning for Robust Molecular Property Prediction](https://mlsanity.com/paper/2304.03906/) (2023)
- [Proteo-R1: Reasoning Foundation Models for De Novo Protein Design](https://mlsanity.com/paper/2605.02937/) (2026)
- [Discovering the Representation Bottleneck of Graph Neural Networks](https://mlsanity.com/paper/2205.07266/) (2022)
- [Holographic-(V)AE: an end-to-end SO(3)-Equivariant (Variational) Autoencoder in Fourier Space](https://mlsanity.com/paper/2209.15567/) (2022)
- [Surface-based Molecular Design with Multi-modal Flow Matching](https://mlsanity.com/paper/2601.04506/) (2026)
- [Dynamics-inspired Structure Hallucination for Protein-protein Interaction Modeling](https://mlsanity.com/paper/2601.06214/) (2026)
- [From Static to Dynamic Structures: Improving Binding Affinity Prediction with Graph-Based Deep Learning](https://mlsanity.com/paper/2208.10230/) (2022)
- [Graph Representation Learning for Interactive Biomolecule Systems](https://mlsanity.com/paper/2304.02656/) (2023)
- [DecoyDB: A Dataset for Graph Contrastive Learning in Protein-Ligand Binding Affinity Prediction](https://mlsanity.com/paper/2507.06366/) (2025)
- [ModuLM: Enabling Modular and Multimodal Molecular Relational Learning with Large Language Models](https://mlsanity.com/paper/2506.00880/) (2025)
- [Joint Design of Protein Surface and Structure Using a Diffusion Bridge Model](https://mlsanity.com/paper/2511.16675/) (2025)
- [SurfDesign: Effective Protein Design on Molecular Surfaces](https://mlsanity.com/paper/2606.07567/) (2026)
- [Deciphering Fingerprints of 3D Molecular Surfaces for Accurate Epitope Prediction](https://mlsanity.com/paper/2606.23830/) (2026)

## Abstract (arXiv)

The latest biological findings observe that the traditional motionless 'lock-and-key' theory is not generally applicable because the receptor and ligand are constantly moving. Nonetheless, remarkable changes in associated atomic sites and binding pose can provide vital information in understanding the process of drug binding. Based on this mechanism, molecular dynamics (MD) simulations were invented as a useful tool for investigating the dynamic properties of a molecular system. However, the computational expenditure limits the growth and application of protein trajectory-related studies, thus hindering the possibility of supervised learning. To tackle this obstacle, we present a novel spatial-temporal pre-training method based on the modified Equivariant Graph Matching Networks (EGMN), dubbed ProtMD, which has two specially designed self-supervised learning tasks: an atom-level prompt-based denoising generative task and a conformation-level snapshot ordering task to seize the flexibility information inside MD trajectories with very fine temporal resolutions. The ProtMD can grant the encoder network the capacity to capture the time-dependent geometric mobility of conformations along MD trajectories. Two downstream tasks are chosen, i.e., the binding affinity prediction and the ligand efficacy prediction, to verify the effectiveness of ProtMD through linear detection and task-specific fine-tuning. We observe a huge improvement from current state-of-the-art methods, with a decrease of 4.3% in RMSE for the binding affinity problem and an average increase of 13.8% in AUROC and AUPRC for the ligand efficacy problem. The results demonstrate valuable insight into a strong correlation between the magnitude of conformation's motion in the 3D space (i.e., flexibility) and the strength with which the ligand binds with its receptor.