Iterative Refinement Graph Neural Network for Antibody Sequence-Structure Co-design

Abstract

Antibodies are versatile proteins that bind to pathogens like viruses and stimulate the adaptive immune system. The specificity of antibody binding is determined by complementarity-determining regions (CDRs) at the tips of these Y-shaped proteins. In this paper, we propose a generative model to automatically design the CDRs of antibodies with enhanced binding specificity or neutralization capabilities. Previous generative approaches formulate protein design as a structure-conditioned sequence generation task, assuming the desired 3D structure is given a priori. In contrast, we propose to co-design the sequence and 3D structure of CDRs as graphs. Our model unravels a sequence autoregressively while iteratively refining its predicted global structure. The inferred structure in turn guides subsequent residue choices. For efficiency, we model the conditional dependence between residues inside and outside of a CDR in a coarse-grained manner. Our method achieves superior log-likelihood on the test set and outperforms previous baselines in designing antibodies capable of neutralizing the SARS-CoV-2 virus.

arXiv commentAccepted to ICLR 2022

References in corpus (7)

• Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
• Learning Deep Generative Models of Graphs
• Constrained Graph Variational Autoencoders for Molecule Design
• Efficient Graph Generation with Graph Recurrent Attention Networks
• Multi-Objective Molecule Generation using Interpretable Substructures
• Symmetry-adapted generation of 3d point sets for the targeted discovery of molecules
• Fold2Seq: A Joint Sequence(1D)-Fold(3D) Embedding-based Generative Model for Protein Design

Cited by in corpus (2)

• Computational protein design with evolutionary-based and physics-inspired modeling: current and future synergies
• Independent SE(3)-Equivariant Models for End-to-End Rigid Protein Docking