arXiv Machine Learning

Interpreting Protein Language Model Embeddings via Orthogonal Projection for Protein Fitness Prediction

The paper introduces a method that uses orthogonal projection to remove the influence of known biochemical features from protein language model (PLM) embeddings, allowing the authors to assess how much these features contribute to protein fitness predictions. By applying this technique to high‑order and interaction effects, they demonstrate that eliminating these interpretable features reduces downstream classifier performance, indicating that PLM embeddings encode patterns correlated with biochemical properties. The authors also show that these biochemical features explain a substantial portion of the variance in the classifier’s predictions, suggesting that PLM embeddings capture biologically relevant information.

arXiv Machine Learning
Aug 28

Interpreting Latent Protein Language Model Features with Geometric Annotations

The paper introduces a scalable method to interpret sparse autoencoder (SAE) features in the ESM-2 protein language model by leveraging geometrically inspired features of the protein α‑carbon backbone. Across 8M layers of ESM-2, a false discovery rate–controlled analysis shows that local geometry is significantly associated with many SAE features, revealing substructure within known biological labels and enabling annotation of unannotated metagenomic proteins. Ablation experiments demonstrate that removing these geometric features shifts ESM-2’s predicted contact maps toward the descriptor, linking mechanistic interpretability with structural biology.

By Siddharth Setlur, Djordje Mihajlovic, Darrick Lee
arXiv Machine Learning
Oct 15

ProtModernBERT: Compute-Efficient Protein Language Modeling

ProtModernBERT is a modernized BERT-style protein language model that reduces training cost by incorporating architectural refinements, achieving performance comparable to larger models while cutting pre‑training FLOPs by 80‑90%. The authors benchmark ProtModernBERT against ProtBert, Amplify, and ESM2 on supervised structural tasks and a zero‑shot functional benchmark, showing it matches or closely tracks Amplify‑120M and ProtBert. Their cluster‑based validation and random‑init experiments reveal that modern architectural components provide stronger inductive biases than parameter count in low‑compute regimes.

By Alberto Mart\'inez, Natalia Flechas Manrique, Lloren\c{c} Espinosa-Portal\'es, Luc Andrea
arXiv AI
Oct 7

CausalBind: Causal Modeling and Learning for Protein-Molecule Virtual Screening

CausalBind introduces a causal modeling framework for protein‑molecule virtual screening that focuses on sparse cross‑modality interactions rather than dense holistic alignment. By formalizing a V‑structure causal model with Heckman‑style selection, the authors prove that latent interaction concepts are identifiable only under sparsity constraints and propose three implementation variants. Experiments on DUD‑E and LIT‑PCBA benchmarks demonstrate that all variants outperform strong retrieval baselines, especially in early enrichment and out‑of‑distribution generalization.

By Loka Li, Jin Tian, Kun Zhang