Back

Cerebra: a computationally efficient framework for accurate protein structure prediction

Hu, J.; Wang, W.; Gong, H.

2024-02-07 bioinformatics
10.1101/2024.02.02.578551 bioRxiv
Show abstract

AO_SCPLOWBSTRACTC_SCPLOWRemarkable progress has been made in the field of protein structure prediction in the past years. State-of-the-art methods like AlphaFold2 and RoseTTAFold2 achieve prediction accuracy close to experimental structural determination, but at the cost of heavy computational consumption for model training. In this work, we propose a new protein structure prediction framework, Cerebra, for improving the computational efficiency of protein structure prediction. In this innovative network architecture, multiple sets of atomic coordinates are predicted parallelly and their mutual complementary is leveraged to rapidly improve the quality of predicted structures through a novel attention mechanism. Consequently, Cerebra markedly reduces the model training consumption, achieving a training acceleration of at least 7 folds, in comparison to OpenFold, the academic version of AlphaFold2. When evaluated on the CAMEO and CASP15 sets, the Cerebra model insufficiently trained on a single GPU only shows slight performance inferiority to the published OpenFold model.

Matching journals

The top 5 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.