A Pretrained ELECTRA Model for Kinase-Specific Phosphorylation Site Prediction.

Journal: Methods in molecular biology (Clifton, N.J.)
PMID:

Abstract

Phosphorylation plays a vital role in signal transduction and cell cycle. Identifying and understanding phosphorylation through machine-learning methods has a long history. However, existing methods only learn representations of a protein sequence segment from a labeled dataset itself, which could result in biased or incomplete features, especially for kinase-specific phosphorylation site prediction in which training data are typically sparse. To learn a comprehensive contextual representation of a protein sequence segment for kinase-specific phosphorylation site prediction, we pretrained our model from over 24 million unlabeled sequence fragments using ELECTRA (Efficiently Learning an Encoder that Classifies Token Replacements Accurately). The pretrained model was applied to kinase-specific site prediction of kinases CDK, PKA, CK2, MAPK, and PKC. The pretrained ELECTRA model achieves 9.02% improvement over BERT and 11.10% improvement over MusiteDeep in the area under the precision-recall curve on the benchmark data.

Authors

  • Lei Jiang
    Department of Thoracic Surgery, Shanghai Pulmonary Hospital, Tongji University, Shanghai 200433, China.
  • Duolin Wang
    Department of Electrical Engineering and Computer Science, University of Missouri, Columbia, MO, USA.
  • Dong Xu
    Department of Electrical Engineering and Computer Science, University of Missouri, Columbia, MO, USA.