RNA Secondary Structure Prediction By Learning Unrolled Algorithms

Xinshi Chen, Yu Li, Ramzan Umarov, Xin Gao, Le Song


Tuesday: Biology and ML

Abstract: In this paper, we propose an end-to-end deep learning model, called E2Efold, for RNA secondary structure prediction which can effectively take into account the inherent constraints in the problem. The key idea of E2Efold is to directly predict the RNA base-pairing matrix, and use an unrolled algorithm for constrained programming as the template for deep architectures to enforce constraints. With comprehensive experiments on benchmark datasets, we demonstrate the superior performance of E2Efold: it predicts significantly better structures compared to previous SOTA (especially for pseudoknotted structures), while being as efficient as the fastest algorithms in terms of inference time.

Similar Papers

SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference
Lasse Espeholt, Raphaël Marinier, Piotr Stanczyk, Ke Wang, Marcin Michalski‎,