Viral Proteins Reveal Geometry of Protein Language Models

Publication information:

Bigot, A., Bhasin, H., Park, C. F., Shakhnovich, E. & Wang, D. Viral Proteins Reveal Geometry of Protein Language Models. (2026) doi:10.48550/arXiv.2606.12609.

Abstract

Protein language models are trained on highly imbalanced datasets, raising the question of how they represent underrepresented biological sequences. Using viral proteins as a case study across ESM model families, we identify a dominant nativeness axis in embedding space, aligned with masked reconstruction perplexity, that orders sequences from well-modeled cellular proteins through viral proteins to shuffled and random sequences. Scaling contracts this axis unevenly across viral families. Despite this, protein language model embeddings retain viral-specific signal: viral proteins remain linearly separable beyond zero-shot perplexity and shallow sequence features. Together, these results suggest that pLM representations are structured by a general notion of nativeness while preserving information specific to distinct biological groups.


Notes

Accepted at ICML 2026 GenBio Workshop and FM4LS Workshop