AlphaFold Meets Flow Matching for Generating Protein Ensembles

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Jing, Bowen, Berger, Bonnie, Jaakkola, Tommi
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929483023384576
author Jing, Bowen
Berger, Bonnie
Jaakkola, Tommi
author_facet Jing, Bowen
Berger, Bonnie
Jaakkola, Tommi
contents The biological functions of proteins often depend on dynamic structural ensembles. In this work, we develop a flow-based generative modeling approach for learning and sampling the conformational landscapes of proteins. We repurpose highly accurate single-state predictors such as AlphaFold and ESMFold and fine-tune them under a custom flow matching framework to obtain sequence-conditoned generative models of protein structure called AlphaFlow and ESMFlow. When trained and evaluated on the PDB, our method provides a superior combination of precision and diversity compared to AlphaFold with MSA subsampling. When further trained on ensembles from all-atom MD, our method accurately captures conformational flexibility, positional distributions, and higher-order ensemble observables for unseen proteins. Moreover, our method can diversify a static PDB structure with faster wall-clock convergence to certain equilibrium properties than replicate MD trajectories, demonstrating its potential as a proxy for expensive physics-based simulations. Code is available at https://github.com/bjing2016/alphaflow.
format Preprint
id arxiv_https___arxiv_org_abs_2402_04845
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AlphaFold Meets Flow Matching for Generating Protein Ensembles
Jing, Bowen
Berger, Bonnie
Jaakkola, Tommi
Biomolecules
Machine Learning
The biological functions of proteins often depend on dynamic structural ensembles. In this work, we develop a flow-based generative modeling approach for learning and sampling the conformational landscapes of proteins. We repurpose highly accurate single-state predictors such as AlphaFold and ESMFold and fine-tune them under a custom flow matching framework to obtain sequence-conditoned generative models of protein structure called AlphaFlow and ESMFlow. When trained and evaluated on the PDB, our method provides a superior combination of precision and diversity compared to AlphaFold with MSA subsampling. When further trained on ensembles from all-atom MD, our method accurately captures conformational flexibility, positional distributions, and higher-order ensemble observables for unseen proteins. Moreover, our method can diversify a static PDB structure with faster wall-clock convergence to certain equilibrium properties than replicate MD trajectories, demonstrating its potential as a proxy for expensive physics-based simulations. Code is available at https://github.com/bjing2016/alphaflow.
title AlphaFold Meets Flow Matching for Generating Protein Ensembles
topic Biomolecules
Machine Learning
url https://arxiv.org/abs/2402.04845