In plain words: A capsule network, which tracks how small parts fit together into bigger shapes, reads 2D and 3D encodings of protein structures to tell HRAS and KRAS apart. It classified them more accurately than standard convolutional networks, which scan for local patterns.
Abstract
Capsule Networks have great potential to tackle problems in structural biology because of their attention to hierarchical relationships. This paper describes the implementation and application of a Capsule Network architecture to the classification of RAS protein family structures on GPU-based computational resources. The proposed Capsule Network trained on 2D and 3D structural encodings can successfully classify HRAS and KRAS structures. The Capsule Network can also classify a protein-based dataset derived from a PSI-BLAST search on sequences of KRAS and HRAS mutations. Our results show an accuracy improvement compared to traditional convolutional networks, while improving interpretability through visualization of activation vectors.
Dan Rosa de Jesus, Julian Cuevas, Wilson Rivera, Silvia Crivelli
arXiv:1808.07475 · cs.LG, q-bio.QM, stat.ML · submitted Aug 22, 2018
abstract · pdf · html