herokoking/cu-ssp

Implementation of High Quality Protein Q8 Secondary Structure Prediction by Diverse Neural Network Architectures

cu-ssp

Implementation of High Quality Prediction of Protein Q8 Secondary Structure by Diverse Neural Network Architectures, Iddo Drori, Isht Dwivedi, Pranav Shrestha, Jeffrey Wan, Yueqi Wang, Yunchu He, Anthony Mazza, Hugh Krogh-Freeman, Dimitri Leggas, Kendal Sandridge, Linyong Nan, Kaveri Thakoor, Chinmay Joshi, Sonam Goenka, Chen Keasar, Itsik Pe’er NIPS Workshop on Machine Learning for Molecules and Materials, 2018.

We tackle the problem of protein secondary structure prediction using a common task framework. This lead to the introduction of multiple ideas for neural architectures based on state of the art building blocks, used in this task for the first time. We take a principled machine learning approach, which provides genuine, unbiased performance measures, correcting longstanding errors in the application domain. We focus on the Q8 resolution of secondary structure, an active area for continuously improving methods. We use an ensemble of strong predictors to achieve accuracy of 70.7% (on the CB513 test set using the CB6133filtered training set). These results are statistically indistinguishable from those of the top existing predictors. In the spirit of reproducible research we make our data, models and code available, aiming to set a gold standard for purity of training and testing sets. Such good practices lower entry barriers to this domain and facilitate reproducible, extendable research.

Q3 (left) and Q8 (right) secondary structure spheres of protein 1AKD in CB513 dataset:

Models

Model 1: Bidirectional GRU with convolution blocks

Model 2: U-Net with convolution blocks

Model 3: Temoporal convolutional network

Model 4: Bidirectional GRUs

Model 5: Bidirectional LSTM with attention

Model 6: Convolutions and bidirectional LSTM

Acknowledgments

We would like to thank the 100 CS/DSI/Stats graduate students at Columbia University of the Fall 2018 Deep Learning course for their participation in an in class protein secondary structure prediction competition. The models which achieved top performance in the competition were invited to participate in this follow-up work, which lead to the discovery of new architectures with state of the art performance. We would like to thank Tomer Sidi of BGU for thorough examination of the correct measures used for performance comparison. We would like to thank Jian Zhou and Olga Troyanskaya of Princeton for making their CB6133 dataset available and for updating their CB6133 dataset splits following our work. Chen Keasar is partially supported by grants 1122/14 from the Israel Science Foundation (ISF).

Q3 (left) and Q8 (right) secondary structure spheres of protein 1F52 in CB513 dataset: