EPO: Diverse and Realistic Protein Ensemble Generation via Energy Preference Optimization

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Sun, Yuancheng, Ren, Yuxuan, Chen, Zhaoming, Han, Xu, Liu, Kang, Ye, Qiwei
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908650452287488
author Sun, Yuancheng
Ren, Yuxuan
Chen, Zhaoming
Han, Xu
Liu, Kang
Ye, Qiwei
author_facet Sun, Yuancheng
Ren, Yuxuan
Chen, Zhaoming
Han, Xu
Liu, Kang
Ye, Qiwei
contents Accurate exploration of protein conformational ensembles is essential for uncovering function but remains hard because molecular-dynamics (MD) simulations suffer from high computational costs and energy-barrier trapping. This paper presents Energy Preference Optimization (EPO), an online refinement algorithm that turns a pretrained protein ensemble generator into an energy-aware sampler without extra MD trajectories. Specifically, EPO leverages stochastic differential equation sampling to explore the conformational landscape and incorporates a novel energy-ranking mechanism based on list-wise preference optimization. Crucially, EPO introduces a practical upper bound to efficiently approximate the intractable probability of long sampling trajectories in continuous-time generative models, making it easily adaptable to existing pretrained generators. On Tetrapeptides, ATLAS, and Fast-Folding benchmarks, EPO successfully generates diverse and physically realistic ensembles, establishing a new state-of-the-art in nine evaluation metrics. These results demonstrate that energy-only preference signals can efficiently steer generative models toward thermodynamically consistent conformational ensembles, providing an alternative to long MD simulations and widening the applicability of learned potentials in structural biology and drug discovery.
format Preprint
id arxiv_https___arxiv_org_abs_2511_10165
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle EPO: Diverse and Realistic Protein Ensemble Generation via Energy Preference Optimization
Sun, Yuancheng
Ren, Yuxuan
Chen, Zhaoming
Han, Xu
Liu, Kang
Ye, Qiwei
Machine Learning
Quantitative Methods
Accurate exploration of protein conformational ensembles is essential for uncovering function but remains hard because molecular-dynamics (MD) simulations suffer from high computational costs and energy-barrier trapping. This paper presents Energy Preference Optimization (EPO), an online refinement algorithm that turns a pretrained protein ensemble generator into an energy-aware sampler without extra MD trajectories. Specifically, EPO leverages stochastic differential equation sampling to explore the conformational landscape and incorporates a novel energy-ranking mechanism based on list-wise preference optimization. Crucially, EPO introduces a practical upper bound to efficiently approximate the intractable probability of long sampling trajectories in continuous-time generative models, making it easily adaptable to existing pretrained generators. On Tetrapeptides, ATLAS, and Fast-Folding benchmarks, EPO successfully generates diverse and physically realistic ensembles, establishing a new state-of-the-art in nine evaluation metrics. These results demonstrate that energy-only preference signals can efficiently steer generative models toward thermodynamically consistent conformational ensembles, providing an alternative to long MD simulations and widening the applicability of learned potentials in structural biology and drug discovery.
title EPO: Diverse and Realistic Protein Ensemble Generation via Energy Preference Optimization
topic Machine Learning
Quantitative Methods
url https://arxiv.org/abs/2511.10165