arXiv:2609.04793cs.AI2026-09

用稀疏专家路由和局部记忆提升蛋白质语言模型效率

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

  • 引入条件记忆与专家路由机制,动态激活参数
  • 150M规模模型在突变预测上表现优异,计算成本低
  • 适合需要高效蛋白质功能预测的研究者使用

蛋白质执行多种细胞功能,单个氨基酸替换可能改变其稳定性、活性或分子互作。蛋白质语言模型(PLMs)可从无标签序列中建模序列-功能关系,但增加密集Transformer主干的规模常带来巨大计算开销,且未必提升突变敏感性预测性能。我们提出ProtLingo,一种高效PLM框架,通过在预训练单序列主干上添加条件局部记忆与稀疏专家路由来增强性能。ProtLingo将上下文残基表示映射为特定路由的离散代码,将中心局部窗口组合成潜在的N-gram地址,并检索与重复局部序列上下文相关的可复用残差信号。同时,选定的前馈模块被升级为共享与路由专家共存的稀疏专家混合层,实现残基依赖的计算,仅激活部分参数。在蛋白质适应度预测、FLIP基准和监督接触预测上的实验表明,ProtLingo以150M规模主干达到具有竞争力的性能,尤其在突变效应预测中展现强参数效率,并保持长程结构表示能力。

原文摘要 · Abstract (English)

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 $N$-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.

蛋白质语言模型专家路由高效建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。