Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces AgentPLM, a novel protein language model that enhances sequence design by integrating Reasoning-Augmented Decoding (RAD) and Contrastive Agent Policy Optimisation (CAPO). By allowing the model to interleave generation with real-time biophysical feedback and to learn from informative oracle feedback, AgentPLM significantly improves performance on various protein engineering tasks. The model achieves state-of-the-art results, particularly in antibody optimization, demonstrating effective online error correction and surpassing traditional passive baselines.
AgentPLM redefines protein sequence design by enabling real-time feedback integration, leading to a dramatic increase in antibody hit rates.
Protein language models (PLMs) are passive oracles: they generate sequences in a single forward pass with no mechanism to consult external biophysical feedback or redirect generation when a candidate violates thermodynamic or structural constraints. We introduce AgentPLM, which addresses this by equipping a pre-trained PLM with i) Reasoning-Augmented Decoding (RAD), which interleaves autoregressive generation with tool calls (ESMFold, FoldX, AutoDock Vina), and ii) Contrastive Agent Policy Optimisation (CAPO), a trajectory-level extension of direct preference optimisation that trains the policy end-to-end to learn when oracle feedback is informative rather than merely imitating high-fitness sequences. We evaluate AgentPLM on benchmark tasks spanning de novo enzyme design, antibody optimisation, thermostability, PPI interface design, and zero-shot fitness prediction with standardised oracle APIs and controlled sequence-identity splits. AgentPLM achieves state-of-the-art results with a gain in antibody top-10% hit rate over the strongest passive baseline, providing mechanistic evidence of online error correction without explicit backtracking.