Structure-aware Reinforcement Learning for Protein Directed Evolution

structure-aware-reinforcement-learning-for-protein-directed-evolution

Structure-aware Reinforcement Learning for Protein Directed Evolution

[Submitted on 30 Sep 2026 (v1), last revised 5 Nov 2026 (this version, v2)]

View PDF HTML (experimental)

Abstract:Protein optimization remains a longstanding goal in life sciences. Existing machine learning-assisted directed evolution (MLDE) methods primarily rely on sequence-only features, overlooking the critical spatial constraints and co-evolutionary interactions encoded in protein structures. However, directly integrating structural information remains challenging due to the scarcity of reliable mutant structures. To address these issues, we propose StructEvo, a novel structure-aware reinforcement learning framework for protein directed evolution. StructEvo employs a delta-structure fusion encoder to approximate mutant structure features via feature differences, enabling dynamic incorporation of spatial knowledge. The vast mutation space is then decomposed into manageable subspaces through a structure-aligned hierarchical action network, while a geometric constraint further stabilizes delta feature learning. Our approach outperforms prior state-of-the-art methods by 9.2% and 16.3% on two challenging optimization benchmarks, and further identifies an experimentally validated epistasis pattern in GFP, highlighting the importance of structural guidance for effective protein directed evolution.

Submission history

From: Zikun Nie [view email]
[v1] Wed, 30 Sep 2026 05:56:18 UTC (5,045 KB)
[v2] Thu, 5 Nov 2026 03:02:15 UTC (5,051 KB)