G EOMETRIC T RANSFORMERS FOR P ROTEIN I NTERFACE C ONTACT P REDICTION
Anonymous authors
Paper under double-blind review
A BSTRACT
Computational methods for predicting the interface contacts between proteins come highly sought after for drug discovery as they can signiﬁcantly advance the accuracy of alternative approaches, such as protein-protein docking, protein function analysis tools, and other computational methods for protein bioinformatics.
In this work, we present the Geometric Transformer, a novel geometryevolving graph transformer for rotation and translation-invariant protein interface contact prediction, packaged within DeepInteract, an end-to-end prediction pipeline.
DeepInteract predicts partner-speciﬁc protein interface contacts (i e., inter-protein residue-residue contacts) given the 3D tertiary structures of two proteins as input.
In rigorous benchmarks, DeepInteract, on challenging protein complex targets from the 13th and 14th CASP-CAPRI experiments as well as Docking
Benchmark 5, achieves 14% and 1.1% top L/5 precision (L: length of a protein unit in a complex), respectively.
In doing so, DeepInteract, with the Geometric Transformer as its graph-based backbone, outperforms existing methods for interface contact prediction in addition to other graph-based neural network backbones compatible with DeepInteract, thereby validating the effectiveness of the
Geometric Transformer for learning rich relational-geometric features for downstream tasks on 3D protein structures.
I NTRODUCTION
Interactions of proteins often reﬂect and directly inﬂuence their functions in molecular processes, so understanding the relationship between protein interaction and protein function is of utmost importance to biologists and other life scientists.
Here, we study the residue-residue interaction between two protein structures that bind together to form a binary protein complex (i.e., dimer), to better understand how these coupled proteins will function in vivo .
Predicting where two proteins will interface in silico has become an appealing method for measuring the interactions between proteins since a computational approach saves time, energy, and resources compared to traditional methods for experimentally measuring such interfaces (Wells & McClendon (2007)).
A key motivation for determining these interface contacts is to decrease the time required to discover new drugs and to advance the study of newly designed proteins (Murakami et al.
Existing approaches to interface contact prediction include classical machine learning and deep learning-based methods.
These methods traditionally use hand-crafted features to predict which inter-chain pairs of amino acid residues will interact with one another upon the binding of the two protein chains, treating each of their residue pairs as being independent of one another.
Recent work on interface prediction (Liu et al.
(2020)), however, considers the biological insight that the interaction between two inter-chain residue pairs depends not only on the pairs’ features themselves but also on other residue pairs ordinally nearby in terms of the protein complex’s sequence.
As such, the problem of interface contact prediction became framed as one akin to image segmentation or object detection, opening the door to innovations in interface contact prediction by incorporating the latest techniques from computer vision.
Nonetheless, up to now, no works on partner-speciﬁc protein interface contact prediction have leveraged two recent innovations to better capture geometric shapes of protein structures and long-range interactions between amino acids important for accurate prediction of protein-protein interface contacts: (1) geometric deep learning for evolving proteins’ geometric representations and (2) graphbased self-attention similar to that of Vaswani et al.
Towards this end, we introduce DeepInteract, an end-to-end deep learning pipeline for protein interface prediction.
DeepInteract houses the Geometric Transformer, a new graph transformer designed to exploit protein structure-speciﬁc geometric properties, as well as a dilated convolution-based interaction module adapted from Chen et al.
(2021) to predict which inter-chain residue pairs comprise the interface between the two protein chains.
In response to the exponential rate of progress being made in predicting protein structures in silico , we trained DeepInteract end-to-end using DIPS-Plus (Morehead et al.
(2021)), to date the largest feature-rich dataset of protein complex structures for machine learning of protein interfaces, to close the gap on a proper solution to this fundamental problem in structural biology.
R ELATED W ORK
Over the past several years, geometric deep learning has become an effective means of automatically learning useful feature representations from structured data (Bronstein et al.
Previously, geometric learning algorithms like convolutional neural networks (CNNs) and graph neural networks (GNNs) have been used to model molecules and to predict protein interface contacts.
Sch ¨ utt et al.
(2017) introduced a deep tensor neural network designed for molecular tasks in quantum chemistry.
Fout et al.
(2017) designed a siamese GNN architecture to learn weight-tied feature representations of residue pairs.
This approach, in essence, processes subgraphs for each residue in each complex and aggregates node-level features locally using a nearest-neighbors approach.
Since this partner-speciﬁc method derives its training dataset from Docking Benchmark 5 (DB5) (Vreven et al.
, it is ultimately data-limited.
Townshend et al.
(2019) represent interacting protein complexes by voxelizing each residue into a 3D grid and encoding in each grid entry the presence and type of the residue’s underlying atoms.
This partner-speciﬁc encoding scheme captures static geometric features of interacting complexes, but it is not able to scale well due to its requiring a computationally-expensive spatial resolution of the residue voxels to achieve good results.
Continuing the trend of applying geometric learning to protein structures, Gainza et al.
(2020) developed MaSIF to perform partner-independent interface region prediction.
Likewise, Dai &
Bailey-Kellogg (2021) do so with an attention-based GNN.
These methods learn to perform binary classiﬁcation of the residues in both complex structures to identify regions where residues from both complexes are likely to interact with one another.
However, because these approaches predict partner-independent interface regions, they are less likely to be useful in helping solve related tasks such as drug-protein interaction prediction and protein-protein docking (Ahmad & Mizuguchi (2011)).
Liu et al.
(2021a) created a graph neural network for predicting the effects of mutations on protein-protein binding afﬁnities, and, more recently, Costa et al.
(2021) introduced a Euclidean equivariant transformer for protein docking.
Both of these methods may beneﬁt from the availability of precise interface predictors by using them to generate contact maps as input features.
To date, one of the best result sets obtained by any model for protein interface contact prediction comes from Liu et al.
(2020) where high-order (i.e. sequential and coevolution-based) interactions between residues are learned and preserved throughout the network in addition to static geometric features initially embedded in the protein complexes.
However, this work, like many of those preceding it, undesirably maintains the trend of reporting model performance in terms of the median area under the receiver operating characteristic which is not robust to extreme class imbalances as often occur in interface contact prediction.
In addition, this approach is data-limited as it uses the DB5 dataset and its predecessors to derive both its training data and makes use of only each residue’s carbon-alpha ( Cα ) atom in deriving its geometric features, ignoring important geometric details provided by an all-atom view of protein structures.
Our work builds on top of prior works by making the following contributions:
• We provide the ﬁrst example of graph self-attention applied to protein interface contact prediction, showcasing its effective use in learning representations of protein geometries to be exploited in downstream tasks.
• We propose the new Geometric Transformer which can be used for tasks on 3D protein structures and similar biomolecules.
For the problem of interface contact prediction, we train the Geometric Transformer to evolve a geometric representation of protein structures simultaneously with protein sequence and coevolutionary features for the prediction of inter-chain residue-residue contacts.
In doing so, we also demonstrate the merit of the recently-released Enhanced Database of Interacting Protein Structures (DIPS-Plus) for interface prediction (Morehead et al.
• Our experiments on challenging protein complex targets demonstrate that our proposed method, DeepInteract, achieves state-of-the-art results for interface contact prediction.
Through these experiments, we also highlight the need for more transparent and problemappropriate metrics as well as more stringent sequence-based ﬁltering on dataset crossvalidation partitions to assess the real-world performance of machine learning models for proteins.
D ATASETS
The current opinion in the bioinformatics community is that protein sequence features still carry important higher-order information concerning residue-residue interactions (Liu et al.
In particular, the residue-residue coevolution and residue conservation information obtained through multiple sequence alignments (MSAs) has been shown to contain powerful information concerning intra-chain and even inter-chain residue-residue interactions as they yield a compact representation of residues’ coevolutionary relationships (Jumper et al.
Keeping this in mind, for our training and validation datasets, we chose to use DIPS-Plus (Morehead et al.
(2021)), to our knowledge the largest feature-rich dataset of protein complexes for protein interface contact prediction to date.
In total, DIPS-Plus contains 42,112 binary protein complexes with positive labels (i.e., 1) for each inter-chain residue pair that are found within 6 ˚A of each other in the complex’s bound (i.e., structurally-conformed) state.
The dataset contains a variety of rich residuelevel features: (1) an 8-state one-hot encoding of the secondary structure in which the residue is found; (2) a scalar solvent accessibility; (3) a scalar residue depth; (4) a 1 × 6 vector detailing each residue’s protrusion concerning its side chain; (5) a 1 × 42 vector describing the composition of amino acids towards and away from each residue’s side chain; (6) each residue’s coordinate number conveying how many residues to which the residue meets a signiﬁcance threshold, (7) a 1 × 27 vector giving residues’ emission and transition probabilities derived from HH-suite3 (Steinegger et al.
proﬁle hidden Markov models constructed using MSAs; and (8) amide plane normal vectors for downstream calculation of the angle between each intra-chain residue pair’s amide planes.
To compare the performance of DeepInteract with that of state-of-the-art methods, we selecthomodimers and heterodimers from the test partition of DIPS-Plus to assess each method’s competency in predicting interface contacts.
We also evaluate each method on 14 homodimers and 5 heterodimers with PDB structures publicly available from the 13th and 14th sessions of CASP-CAPRI (Lensink et al.
(2019), Lensink et al.
as these targets are considered by the bioinformatics community to be challenging for existing interface predictors.
For any CASP-CAPRI test complexes derived from multimers (i.e., protein complexes that can contain more than two chains), to represent the complex we chose the pair of chains with the largest number of interface contacts.
Finally, we use the traditional 55 test complexes from the DB5 dataset (Fout et al.
; Townshend et al.
Liu et al.
to benchmark each heteromer-compatible method.
To expedite training and validation and to constrain memory usage, beginning with all remaining complexes not chosen for testing, we ﬁltered out all complexes where either chain contains fewer than 20 residues and where the number of possible interface contacts is more than 256 2 , leaving us with an intermediate total of 26,504 complexes for training and validation.
In DIPS-Plus, binary protein complexes are grouped into shared directories according to whether they are derived from the same parent complex.
As such, using a per-directory strategy, we randomly designate 80% of these complexes for training and 20% for validation to restrict overlap between our cross-validation datasets.
After choosing these targets for testing, we then ﬁlter out complexes from our training and validation partitions of DIPS-Plus that contain any chain with over 30% sequence identity to any chain in any complex in our test datasets.
This threshold of 30% sequence identity is commonly used in the bioinformatics literature (Jordan et al.
(2012), Yang et al.
to prevent large evolutionary overlap between a dataset’s cross-validation partitions.
However, most existing works for interface contact prediction do not employ such ﬁltering criteria, so the results reported in these works may be over-optimistic by nature.
In performing such sequence-based ﬁltering, we are left with 15, and 3,548 binary complexes for training and validation, respectively.
M ETHODS 4. P ROBLEM F ORMULATION
Summarized in Figure 2, we designed DeepInteract, our proposed pipeline for interface contact prediction, to frame the problem of predicting interface contacts in silico as a two-part task:
The ﬁrst part is to use attentive graph representation learning to inductively learn new node-level representations
h
A ∈ R
A × k and h B ∈ R B × k for a pair of graphs representing two protein chains.
The second part is to channel-wise interleave h A and h B into an interaction tensor
I ∈ R
A × B × 2 k , where A ∈ R and B ∈ R are the numbers of amino acid residues in the ﬁrst and second input protein chains, respectively, and k ∈ R is the number of hidden channels in both h A and h B .
We use interaction tensors such as I as input to our interaction module, a convolution-based dense predictor of inter-graph node-node interactions.
We denote each protein chain in an input complex as a graph G with edges E between the k -nearest neighbors of its nodes N , with nodes corresponding to the chain’s amino acid residues represented by their Cα atoms.
In this setting, we let k = 20 as we observed favorable cross entropy loss on our validation dataset with this level of connectivity.
We note that this level of graph connectivity has also proven to be advantageous for prior works developing deep learning approaches for graph-based protein representations (Fout et al.
(2017); Ingraham et al.
4. G EOMETRIC T RANSFORMER A RCHITECTURE
Hypothesizing that a self-attention mechanism that evolves proteins’ physical geometries is a key component missing from existing interface contact predictors, we propose the Geometric Transformer, a graph neural network explicitly designed for capturing and iteratively evolving protein geometric features.
As shown in Figure 3, the Geometric Transformer expands upon the existing Graph Transformer architecture (Dwivedi & Bresson (2021)) by introducing ( 1 ) an edge initialization module, ( 2 ) positional encoding for edges, and ( 3 ) a geometry-evolving conformation module.
Moreover, the Geometric Transformer includes subtle architectural enhancements to the original Transformer architecture (Vaswani et al.
such as moving the network’s ﬁrst normalization layer to precede any afﬁnity score computations for improved training stability (Hussain et al.
To our knowledge, the Geometric Transformer is the ﬁrst deep learning model that applies multi-head attention to the task of partner-speciﬁc protein interface contact prediction.
The following sections serve to distinguish our new Geometric Transformer from other Transformer-like architectures by describing its new neural network modules for geometric self-attention.
4.2. E DGE I NITIALIZATION M ODULE
The Geometric Transformer ﬁrst embeds each edge e ∈ E with the initial edge representation
[Equation]
[Equation]
(2) where f refers to a unique multi-layer perceptron; || denotes channel-wise concatenation; f 1 and f 2 , respectively, are trainable one-hot vectors indexed by P i and P j , the positions of nodes i and nodes j in the chain’s underlying amino acid sequence; f 3 are any user-predeﬁned features for e (in our case the normalized Euclidean distances between nodes i and nodes j ); λ e are edge-wise sinusoidal positional encodings sin ( P i − P j ) for e ; f 4 , f 5 , f 6 , and f 7 , in order, are the four proteinspeciﬁc geometric features deﬁned in Section A.3; and add and gate are feature summation and channel-wise gating functions, respectively.
4.2. C ONFORMATION M ODULE
The role of the Geometric Transformer’s subsequent conformation module, as illustrated in Figure 4, is for it to learn how to iteratively evolve geometric representations of protein graphs by applying repeated gating to our initial edge geometric features f 4 , f 5 , f 6 , and f 7 .
To do so, the conformation module updates e ij by introducing the notion of a geometric neighborhood of edge e , treating e as a pseudo-node.
Precisely, e k , the edge geometric neighborhood of e , is deﬁned as the 2 n edges
[Equation]
where N k ⊂ N are the source nodes for incoming edges on edge e (cid:48)
s source and destination nodes.
The intuition behind updating each edge according to its 2 n nearest neighboring edges is that the geometric relationship between a residue pair, described by their mutual edge’s features, can be inﬂuenced by the physical constraints imposed by proximal residue-residue geometries.
As such, we use these nearby edges during geometric feature updates.
In the conformation module, the processing of all geometric neighborhood features for edge e can be represented as
[Equation]
where (cid:12) refers to element-wise multiplication; e ij is e (cid:48)
s current representation; f 1 , f 2 , f 3 , and fare equivalent to f 4 , f 5 , f 6 , and f 7 as described in Section 4.2.1; and RB2 represents the application of two successive residual blocks, each deﬁned as RB( x )
= f RB 2 ( f RB 1 ( x ))
+ x .
As described in
Section A.3, by way of their construction, each of our selected edge geometric features is invariant to rotations and translations of the network’s input space.
Further, since each node and edge-wise operation we apply to such features is also geometrically invariant, we observe that the Geometric
Transformer operates invariantly with respect to translations and rotations in R 3 , a desirable property to have when employing deep learning for static interface prediction on 3D objects such as proteins.
4.2. R EMAINING T RANSFORMER I NITIALIZATIONS AND O PERATIONS
For the initial node features used within the Geometric Transformer, we include each of DIPS-Plus’ residue-level features described succinctly in Section 3.
Additionally, we append initial min-max normalizations of each residue’s index in P i to each node as node-wise positional encodings.
For the remainder of the Geometric Transformer’s operations, the network’s order of operations closely follows the deﬁnitions given by Dwivedi & Bresson (2021) for the Graph Transformer, with an exception being that the ﬁrst normalization layer now precedes any afﬁnity score calculations.
4. I NTERACTION M ODULE
Upon applying multiple layers of the Geometric Transformer to each pair of input protein chains, we then channel-wise interleave the Geometric Transformer’s learned node representations h A and h B into I to serve as input to our interaction module, consisting of a dilated ResNet module adapted from Chen et al.
The core residual network component in this interaction module consists of four residual blocks differing in the number of internal layers.
Each residual block is comprised of several consecutive instance normalization layers and convolutional layers with 64 kernels of size 3 × 3.
The number of layers in each block represents the number of 2D convolution layers in the corresponding component.
The ﬁnal values of the last convolutional layer are added to the output of a shortcut block, which is a convolutional layer with 64 kernels of size 1 × 1.
A squeeze-andexcitation (SE) block (Hu et al.
(2018)) is added at the end of each residual block to adaptively recalibrate its channel-wise feature responses.
Ultimately, the output of the interaction module is a probability-valued A x B matrix that can be viewed as a heatmap describing which inter-chain residue pairs are likely to be in contact with one another upon binding of the two chains.
E XPERIMENTS 5. S ETUP
For all experiments conducted with DeepInteract, we used 2 layers of the graph neural network chosen for the experiment and 128 intermediate GNN and CNN channels to restrict the time required to train each model.
For the Geometric Transformer, we used an edge geometric neighborhood of size n = 2 for each edge such that each edge’s geometric features are updated by their 4 -nearest incoming edges.
In addition, we used the Adam optimizer (Kingma & Ba (2014)), a learning rate of 1 e − 3 , a weight decay rate of 1 e − 2 , a dropout (i.e., forget) rate of 0.2, and a batch size of 1.
We also employed 0.5-threshold gradient value clipping and stochastic weight averaging (Izmailov et al.
With an early-stopping patience period of 5 epochs, we observed most models converging after approximately 30 training epochs on DIPS-Plus.
For our loss function, we used weighted cross entropy with a positive class weight of 5 to help the network overcome the large class imbalance present in interface prediction.
All DeepInteract models employed 14 layers of our dilated ResNet architecture described in Section 4.3 and had their top- k metrics averaged over three separate runs, each with a different random seed (standard deviation of top- k metrics in parentheses).
Prior to our experiments on the DB5 dataset’s 55 test complexes, we ﬁne-tuned each DeepInteract model using the held-out 140 and 35 complexes remaining in DB5 for training and validation, respectively.
Employing a similar training conﬁguration as described above, in this context we used a lower learning rate of 1 e − 5 to facilitate smoother transfer learning between DIPS-Plus and DB5.
5. H YPERPARAMETER S EARCH
To identify our optimal set of model hyperparameters, we performed a manual hyperparameter search over the ranges of [1 e − 1 , 1 e − 2 , 1 e − 3 , 1 e − 4 , 1 e − 5 , 1 e − 6 ] and [1 e − 1 , 1 e − 2 , 1 e − 3 , 1 e − 4 ] for the learning rate and weight decay rate, respectively.
In doing so, we found a learning rate of 1 e − 3 and a weight decay rate of 1 e − 2 to provide the lowest loss and the highest metric values on our DIPS-Plus validation dataset.
We restricted our hyperparameter search to the learning rate and weight decay rate of our models due to the large computational and environmental costs associated with training each model, however, this suggests that further improvements to our models could be found with a more extensive hyperparameter search such as a grid search over, for example, the models’ dropout rate and number of intermediate GNN and CNN channels.
5. S ELECTION OF B ASELINES
In choosing the baseline state-of-the-art methods against which we would compare DeepInteract with various graph-based backbones, we considered the reproducibility and accessibility of the method as the most important factor for its inclusion in our benchmarks to encourage adoption of accessible and transparent benchmarks for future works.
That is, we only selected baseline methods that are easy to reproduce or instead simple for the general public to use to make predictions.
[Table] [Table]
Each method predicts interfacing residue pairs subject to the (on average) 1 : 1000 positive-negative class imbalance imposed by the biological sparsity of true interface contacts.
Since two of the three baseline methods we chose to include, DeepHomo and ComplexContact, do not make both their feature generation and model training source code publicly available, we were restricted to using the version of their public web servers available on 11/15/2021 to make inter-chain interface contact predictions.
As such, our selection criterion for each baseline method consequently determined the number of complexes against which we could feasibly test each method, thereby restricting the size of our test datasets to 106 complexes in total.
In addition, not all baselines chosen were originally trained for both types of protein complexes (i.e., homodimers and heterodimers), so for these baselines we do not include their results for the type of complex for which they are not respectively designed.
For brevity, in all experiments, we refer to BIPSPI (Sanchez-Garcia et al.
, DeepHomo (Yan & Huang (2021)), ComplexContact (Zeng et al.
(2018)), and DeepInteract as BI, DH, CC, and
DI, respectively.
In addition, we refer to the Graph Convolutional Network of Kipf & Welling (2016), the Graph Transformer of Dwivedi & Bresson (2021), and the Geometric Transformer as
GCN, GraphTran, and GeoTran, respectively.
To assess models’ ability to accurately select residue pairs likely to be in interaction upon binding of two given chains, all methods are scored using the top- k precision and recall metrics (deﬁned in Section A.2) commonly used for intra-chain contact prediction (Chen et al.
(2021)) as well as recommender systems
(Jiang et al.
(2020)), where k ∈ { 10 , L/ 10 , L/ 5 , L/ 2 } with L being the length of the shortest chain in a given complex.
5. D ISCUSSION
Table 1 demonstrates that DeepInteract outperforms or achieves competitive results compared to existing state-of-the-art methods for interface contact prediction on DIPS-Plus with both types of protein complexes, homodimers (homo) where the two chains are of the same protein and heterodimers (hetero) where the two chains are of different proteins.
Table 2 shows that, when taking both types of complexes into account, DeepInteract outperforms all other methods’ predictions on DIPS-Plus.
Since future users of DeepInteract may want to predict interface contacts for either type of complex, we consider a method’s type-averaged top- k metrics as important metrics for which to optimize.
Likewise, Tables 3 and 4 present the average top- k metrics of DeepInteract on 19 challenging protein complexes (14 homodimers and 5 heterodimers) from the 13th and 14th rounds of the joint CASPCAPRI meeting.
In them, we once again see DeepInteract exceed the precision of state-of-the-art interface contact predictors for both complex types.
In particular, we see that combining DeepInteract with the Geometric Transformer offers improvements to the majority of our top- k metrics for both homodimers and heterodimers compared to using either a GCN or a Graph Transformer-based GNN backbone, notably for heteromeric complexes with largely asymmetric inter-chain geometries.
Such a result supports our hypothesis that the Geometric Transformer’s geometric self-attention mecha-
[Table]
[Table] [Table]
nism can enable enhanced prediction performance for downstream tasks on geometrically-intricate 3D objects such as protein structures, using interface contact prediction as a case study.
Finally
[Table]
in Table
[Table]
we observe that, in predicting the interface contacts between unbound protein chains in the DB5 test dataset
[Table]
the Geometric Transformer enables enhanced top- k precision and recall (deﬁnition in A.2) compared to all other baseline methods, including GCNs and Graph Transformers paired with DeepInteract.
Such as result conﬁrms, to a degree, the Geometric Transformer’s ability to predict how the structural conformations occurring upon the binding of two protein chains inﬂuence which inter-chain residue pairs will interact with one another in the complex’s bound state.
C ONCLUSION
In this work, we presented DeepInteract which introduces the geometry-evolving Geometric Transformer for protein structures and demonstrates its effectiveness in predicting residue-residue interactions in protein complexes.
We foresee several other uses of the Geometric Transformer in geometric deep learning of protein structures such as protein quaternary structure quality assessment and reﬁnement as well as residue disorder prediction, simply to name a few.
One limitation of the existing design for the Geometric Transformer is the O ( n 2 ) computational complexity associated with its dot product self-attention mechanism.
With the advent of efﬁcient alternatives to self-attention like that of the Nystr¨omformer (Xiong et al.
, we hope to overcome this limitation.
In addition, for future work, we plan to investigate how to adapt the Geometric Transformer’s conformation module to evolve geometric representations within equivariant graph neural networks such as those of Satorras et al.
(2021), to determine the effect of such network updates when modeling physical systems.
A CKNOWLEDGMENTS
The project is partially supported by two NSF grants (DBI 1759934 and IIS 1763246), one NIH grant (GM093123), three DOE grants (DE-SC0020400, DE-AR0001213, and DE-SC0021303), and the computing allocation on the Summit compute cluster provided by Oak Ridge Leadership Computing
Facility (Contract No. DE-AC05-00OR22725).
E THICS S TATEMENT
DeepInteract is designed to be used for machine learning of protein molecular data.
It makes use of only publicly available information concerning biomolecular structures and their interactions.
Consequently, all data used to create DeepInteract models do not contain any personally identiﬁable information or offensive content.
As such, we do not foresee negative societal impacts as a consequence of DeepInteract or the Geometric Transformer being made publicly available.
Furthermore, future adaptions or enhancements to DeepInteract may beneﬁt the machine learning community and, more broadly, the scientiﬁc community by providing meaningful reﬁnements to a transparent and extensible pipeline for geometric deep learning of protein-protein interactions
R EPRODUCIBILITY S TATEMENT
To enable this work to be reproducible by others we have thoroughly documented two methods for running predictions with DeepInteract using a provided PyTorch Lightning model checkpoint (Falcon (2019)) and one method for training DeepInteract models in an associated GitHub repository.
The most convenient prediction method describes how to run DeepInteract as a platform-agnostic
Docker container.
The second method, for both training and prediction, details how users can piecewise install DeepInteract and its data and software dependencies on a Linux-based operating system
A A PPENDIX A. S AMPLE I NTERFACE C ONTACT P REDICTIONS
In the ﬁrst row of Figure 5, we see predictions made by DeepInteract for a homodimer complex from our test partition of DIPS-Plus (i.e., PDB ID: 4HEQ).
The leftmost image represents the softmax contact probability map.
The center image corresponds to the same contact map after having a 0.
probability threshold applied to it such that residue pairs with at least a 50% probability of being in interaction with each other have their interaction probabilities rounded up to 1.0.
The rightmost image is the ground-truth contact map.
Similarly, in the second row of Figure 5, we are shown the cropped predictions made by DeepInteract for a CASP-CAPRI test heterodimer (i.e., PDB ID: 6TRI).
A. T OP - K T EST P RECISION AND R ECALL OF B OTH C OMPLEX T YPES IN DIPS-P LUS AND CASP-CAPRI
Formally, our deﬁnitions of a model’s top-
k precision prec k and recall rec k , where T pos k represents the number of true positive residue pairs selected from a model’s top- k most probable pairs and T pos corresponds to the total number of true positive pairs in the complex, are
[Equation]
(6) and
[Equation]
After deﬁning top-
k recall as such, in Tables 6 and 7 we provide the results of each model’s top-
k recall in the same set of experiments as given in Section 5.
A. D EFINITION OF E DGE G EOMETRIC F EATURES
Similar to Ingraham et al.
(2019), we construct a local reference frame (i.e., an orientation O i ) for each protein chain graph’s residues.
Representing each residue by its Cartesian coordinates x i , we formally deﬁne
[Table] [Table]
[Equation]
with n i being the unit vector normal to the plane formed by the rays ( x i − 1 − x i ) and ( x i +1 − x i ) and b i being the negative bisector of this plane.
We then deﬁne O i as
[Equation]
Having deﬁned the orientation O i for each residue that describes the local reference frame ( x i , O i ).
To provide the Geometric Transformer with an alternative notion of residue-residue orientations, we deﬁne the unit vector normal to the amide plane for residue i as
[Equation]
where x Cα i , x Cβi , and x Ni are the Cartesian coordinates of the residue’s carbon-alpha ( Cα ), carbon-beta ( Cβ ), and nitrogen ( N ) atoms, respectively.
Finally, we relate the reference frames for residues i and j by describing their edge geometric features as
[Equation]
with the ﬁrst term r () being a distance encoding of 16 Gaussian RBFs spaced isotropically from 0 to 20 ˚A, the second term describing the relative direction of x j with respect to reference frame ( x i , O i ), the third term detailing an orientation encoding q () of the quaternion representation of the rotation matrix
O Ti O j , representing each quaternion with respect to its vector of real coefﬁcients, and the fourth term a () representing the angle between the amide plane normal vectors
U i and U j .
Our deﬁnition of these edge geometric features makes use of the backbone atoms for each residue.
As such, the graph representation of protein chains we use with the Geometric Transformer encodes not only residue-level geometric features but also those derived from an atomic view of protein structures.
We hypothesized this hybrid approach to modeling protein structure geometries would have a noticeable downstream effect on interface contact prediction precision via the node and edge representations learned by the Geometric Transformer.
This hypothesis is conﬁrmed in Section 5.4.
[Table]
[Table] B Homo 6NQ A
B
Homo 6QEK
A B Homo 6UBL A
B
Homo 6UK
A B Homo 6YA A
B
Homo 7CWP C D Homo 6CP
A
C
Hetero 6D7Y
A
B Hetero 6TRI A B
Hetero 6XOD
A
B
Hetero 7M5F
A C Hetero A. P ROTEIN C OMPLEXES S ELECTED FOR T ESTING
To facilitate reproducibility of the results presented in Section 5.4, Table 8 displays the PDB and chain IDs of DIPS-Plus protein complexes chosen for testing.
Likewise, in Table 9, we provide the
PDB and chain IDs of CASP-CAPRI 13-14 targets chosen for testing.
These two tables describe precisely which targets were selected and ultimately used in our RCSB-derived benchmarks.
For full data provenance, the targets we selected from the Docking Benchmark 5 dataset (Vreven et al.
for benchmarking are the same 55 protein heterodimers used for testing in works such as that of Fout et al.
(2017), Townshend et al.
(2019), and Liu et al.
A. I NVARIANCE OR E QUIVARIANCE ?
In our view, a natural question to ask concerning a deep learning architecture designed for a speciﬁc task is whether invariance to translations and rotations in R 3 should be preferred over full equivariance to transformations in such a geometric space.
Our answer to this question, in the context of the Geometric Transformer, is that for tasks such as static interface prediction between proteins, invariance should be preferred over equivariance as the beneﬁts of equivariance primarily include equivariant updates to type-1 tensors such the coordinates representing an object in R 3 .
Our formulation of the interface prediction problem solely makes use of the features derived for each residue in our protein graphs to calculate our models’ cross entropy loss.
Therefore, in this context, we instead decided to focus on preserving translation and rotation invariance in R 3 to improve the learning efﬁciency and generalization capabilities of the Geometric Transformer (Bronstein et al.
A. R ATIONALE BEHIND THE N ODE I NITIALIZATION S CHEME
DIPS-Plus residue-level features are initially embedded in our protein chain graphs to accelerate the network’s training.
However, we also initially append node-wise min-max positional encodings in our network’s operations.
We do this to initialize the Geometric Transformer with information concerning the residue ordering of the chain’s underlying sequence as such ordering is important to understanding downstream protein structural, interactional, and functional properties of each residue.
A. R ATIONALE BEHIND THE E DGE I NITIALIZATION M ODULE ’ S D ESIGN
For the edge initializer module’s four protein geometric features, we sought to include enough geometric information for the network to be able to uniquely determine the Euclidean positions of each node’s neighboring nodes.
For this reason, we adopt similar distance, direction, and orientation descriptors as Ingraham et al.
We concatenate the protein backbone-geometric features provided by inter-residue distances, directions, and orientations with the angles between each residue pair’s amide plane normal vectors.
This is done ultimately to apply gating to each edges’ messages, distances, directions, orientations, and amide angles separately to encourage the network to learn the importance of speciﬁc channels in each of these input features.
Gating is a technique that has previously been shown to encourage neural networks to not become over-reliant on any particular input feature (Gu et al.
and, as such, in the Geometric Transformer can be seen as a form of channel-wise dropout for single feature sets.
By also employing residual connections from original edge representations to gating-learned edge representations, the network module can operate more stably in the presence of multiple neural network layers (He et al.
Furthermore, in the edge initialization module, we introduce edge-wise sinusoidal position encodings to provide the network with a directional notion of residue-to-residue distances in protein chains’ underlying sequences.
A. R ATIONALE BEHIND THE C ONFORMATION M ODULE ’ S D ESIGN
The conformation module’s design was inspired by SphereNet (Liu et al.
(2021b))
and similar graph neural network architectures designed for learning on 3D graphs.
What distinguishes our conformation module from the works of others is its introduction of the notion of 2 n edge geometric neighborhoods when updating edge representations as well as its incorporation of geometric insights speciﬁc to large biomolecules such as proteins.
Namely, by including the residue-residue distances, residue-residue local reference frame directions and (quaternion) orientations, and amide plane-amide plane angles, the network is provided with enough information to ascertain the relative coordinates of each neighboring residue from a given residue’s local reference frame (Liu et al.
(2021b)), thereby ensuring the network’s capability of adequately learning from 3D structures.
A. A LTERNATIVE N ETWORKS WITHIN THE I NTERACTION M ODULE
We, like Liu et al.
(2020), note that the task of interface prediction bears striking similarities to dense prediction tasks in computer vision (e.g., semantic segmentation).
In this train of thought, we experimented with several semantic segmentation models as replacements for our interaction module’s dilated ResNet, one namely being DeepLabV3Plus (Chen et al.
We observed a strong propensity of such semantic segmentation models to identify interaction regions well but to do so with low pixel-wise precision.
We hypothesize this is due to the downsampling and upsampling methods often employed within such architectures that invariably degrade the original input tensor’s representation resolution.
We also experimented with several state-of-the-art Vision
Transformer and MLP-based models for computer vision but ultimately found their algorithmic complexity, memory usage, or input shape requirements to be prohibitive for this task, since our test datasets’ input protein complexes can vary greatly in size to contain between 20 residues and over 2,000 residues in length.
As such, for the design of DeepInteract’s interaction module, we experimented primarily with convolution-based architectures that do not employ such sampling techniques or pose limited input size constraints.
A. H ARDWARE U SED
The Oak Ridge Leadership Facility (OLCF) at the Oak Ridge National Laboratory (ORNL) is an open science computing facility that supports HPC research.
The OLCF houses the Summit compute cluster.
Summit, launched in 2018, delivers 8 times the computational performance of Titan’s 18, nodes, using only 4,608 nodes.
Like Titan, Summit has a hybrid architecture, and each node contains multiple IBM POWER9 CPUs and NVIDIA Volta GPUs all connected with NVIDIA’s high-speed
NVLink.
Each node has over half a terabyte of coherent memory (high bandwidth memory + DDR4) addressable by all CPUs and GPUs plus 800GB of non-volatile RAM that can be used as a burst buffer or as extended memory.
To provide a high rate of I/O throughput, the nodes are connected in a non-blocking fat-tree using a dual-rail Mellanox EDR InﬁniBand interconnect.
We used the
Summit compute cluster to train all our models.
A. S OFTWARE U SED
In addition, we used Python 3.8 (Van Rossum & Drake (2009)), PyTorch 1.7.1 (Paszke et al.
(2019)), and PyTorch Lightning 1.4.
(Falcon (2019)) to run our deep learning experiments.
PyTorch Lightning was used to facilitate model checkpointing, metrics reporting, and distributed data parallelism across 72 Tesla V100 GPUs.
A more in-depth description of the software environment used to train and predict with DeepInteract models can be found at https://github.com/xxxxx/DeepInteract .