Guided protein LMs collapse off-manifold: solubility oracle scores 0.99 while output can't fold

rishabh16_ · x · 2026-10-12

A new arXiv paper, Off-Manifold Collapse in Guided Protein Language Models, shows that inference-time steering of protein language models like ESM-2 exhibits a reward-hacking-style failure mode.

Key findings

Proposed fix

Rather than modifying the generator, the authors detect failures at the output with a cheap density prior over natural protein activations, keeping only "typical" candidates — a training-free post-hoc step called Mahalanobis filtering that improves both property score and structural plausibility at matched guidance settings at negligible cost.

Original post →

More from Research

Research channel →