English

ProSpero: Active Learning for Robust Protein Design Beyond Wild-Type Neighborhoods

Machine Learning 2025-10-28 v2

Abstract

Designing protein sequences of both high fitness and novelty is a challenging task in data-efficient protein engineering. Exploration beyond wild-type neighborhoods often leads to biologically implausible sequences or relies on surrogate models that lose fidelity in novel regions. Here, we propose ProSpero, an active learning framework in which a frozen pre-trained generative model is guided by a surrogate updated from oracle feedback. By integrating fitness-relevant residue selection with biologically-constrained Sequential Monte Carlo sampling, our approach enables exploration beyond wild-type neighborhoods while preserving biological plausibility. We show that our framework remains effective even when the surrogate is misspecified. ProSpero consistently outperforms or matches existing methods across diverse protein engineering tasks, retrieving sequences of both high fitness and novelty.

Keywords

Cite

@article{arxiv.2505.22494,
  title  = {ProSpero: Active Learning for Robust Protein Design Beyond Wild-Type Neighborhoods},
  author = {Michal Kmicikiewicz and Vincent Fortuin and Ewa Szczurek},
  journal= {arXiv preprint arXiv:2505.22494},
  year   = {2025}
}

Comments

NeurIPS 2025

R2 v1 2026-07-01T02:46:41.550Z