cs.AISep 7, 2026

ProtLingo: Efficient Protein Language Modeling via Conditional Memory and Expert Routing

Authors: Mingrui LiSixian ShenMinzhang LiRuiyi ZhangKexin ZhangJiakai ZhangJingyi Yu

Organizations: ShanghaiTech University

Abstract

Proteins perform diverse cellular functions, and even single amino-acid substitutions can alter stability, activity, or molecular interactions. Protein language models (PLMs) provide a scalable approach for modeling such sequence--function relationships from unlabeled sequences, but increasing the size of dense Transformer backbones often brings substantial computational cost without consistently improving mutation-sensitive prediction. We introduce ProtLingo, an efficient PLM framework that augments a pretrained single-sequence backbone with conditional local memory and sparse expert routing. ProtLingo maps contextual residue representations into route-specific discrete codes, composes centered local windows into latent NN-gram addresses, and retrieves reusable residual signals associated with recurring local sequence contexts. In parallel, selected feed-forward blocks are upcycled into sparse Mixture-of-Experts layers with shared and routed experts, enabling residue-dependent computation while activating only a subset of parameters. Experiments on protein fitness prediction, FLIP benchmarks, and supervised contact prediction show that ProtLingo achieves competitive performance with a 150M-scale backbone, including strong parameter efficiency on mutation-effect prediction and preserved long-range structural representations.

Explore similar work

CardsList
  1. ProtSent: Protein Sentence Transformers

    May 7, 2026Dan Ofer, Oriel Perets, Michal Linial +1Protein Data BankGlobal Embeddings

  2. Task- and dataset-specific information in protein language models

    Aug 12, 2026Roman Joeres, Ilya Senatorov, Anastasia Kolchina +2