English

Layer Probing Improves Kinase Functional Prediction with Protein Language Models

Quantitative Methods 2025-12-02 v1 Artificial Intelligence Machine Learning

Abstract

Protein language models (PLMs) have transformed sequence-based protein analysis, yet most applications rely only on final-layer embeddings, which may overlook biologically meaningful information encoded in earlier layers. We systematically evaluate all 33 layers of ESM-2 for kinase functional prediction using both unsupervised clustering and supervised classification. We show that mid-to-late transformer layers (layers 20-33) outperform the final layer by 32 percent in unsupervised Adjusted Rand Index and improve homology-aware supervised accuracy to 75.7 percent. Domain-level extraction, calibrated probability estimates, and a reproducible benchmarking pipeline further strengthen reliability. Our results demonstrate that transformer depth contains functionally distinct biological signals and that principled layer selection significantly improves kinase function prediction.

Keywords

Cite

@article{arxiv.2512.00376,
  title  = {Layer Probing Improves Kinase Functional Prediction with Protein Language Models},
  author = {Ajit Kumar and IndraPrakash Jha},
  journal= {arXiv preprint arXiv:2512.00376},
  year   = {2025}
}

Comments

14 pages, 7 figures, 3 tables; includes code and dataset links

R2 v1 2026-07-01T08:00:37.870Z