Sequence-Augmented SE(3)-Flow Matching For Conditional Protein Backbone Generation
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909424735485952 |
|---|---|
| author | Huguet, Guillaume Vuckovic, James Fatras, Kilian Thibodeau-Laufer, Eric Lemos, Pablo Islam, Riashat Liu, Cheng-Hao Rector-Brooks, Jarrid Akhound-Sadegh, Tara Bronstein, Michael Tong, Alexander Bose, Avishek Joey |
| author_facet | Huguet, Guillaume Vuckovic, James Fatras, Kilian Thibodeau-Laufer, Eric Lemos, Pablo Islam, Riashat Liu, Cheng-Hao Rector-Brooks, Jarrid Akhound-Sadegh, Tara Bronstein, Michael Tong, Alexander Bose, Avishek Joey |
| contents | Proteins are essential for almost all biological processes and derive their diverse functions from complex 3D structures, which are in turn determined by their amino acid sequences. In this paper, we exploit the rich biological inductive bias of amino acid sequences and introduce FoldFlow-2, a novel sequence-conditioned SE(3)-equivariant flow matching model for protein structure generation. FoldFlow-2 presents substantial new architectural features over the previous FoldFlow family of models including a protein large language model to encode sequence, a new multi-modal fusion trunk that combines structure and sequence representations, and a geometric transformer based decoder. To increase diversity and novelty of generated samples -- crucial for de-novo drug design -- we train FoldFlow-2 at scale on a new dataset that is an order of magnitude larger than PDB datasets of prior works, containing both known proteins in PDB and high-quality synthetic structures achieved through filtering. We further demonstrate the ability to align FoldFlow-2 to arbitrary rewards, e.g. increasing secondary structures diversity, by introducing a Reinforced Finetuning (ReFT) objective. We empirically observe that FoldFlow-2 outperforms previous state-of-the-art protein structure-based generative models, improving over RFDiffusion in terms of unconditional generation across all metrics including designability, diversity, and novelty across all protein lengths, as well as exhibiting generalization on the task of equilibrium conformation sampling. Finally, we demonstrate that a fine-tuned FoldFlow-2 makes progress on challenging conditional design tasks such as designing scaffolds for the VHH nanobody. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_20313 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Sequence-Augmented SE(3)-Flow Matching For Conditional Protein Backbone Generation Huguet, Guillaume Vuckovic, James Fatras, Kilian Thibodeau-Laufer, Eric Lemos, Pablo Islam, Riashat Liu, Cheng-Hao Rector-Brooks, Jarrid Akhound-Sadegh, Tara Bronstein, Michael Tong, Alexander Bose, Avishek Joey Machine Learning Biomolecules Proteins are essential for almost all biological processes and derive their diverse functions from complex 3D structures, which are in turn determined by their amino acid sequences. In this paper, we exploit the rich biological inductive bias of amino acid sequences and introduce FoldFlow-2, a novel sequence-conditioned SE(3)-equivariant flow matching model for protein structure generation. FoldFlow-2 presents substantial new architectural features over the previous FoldFlow family of models including a protein large language model to encode sequence, a new multi-modal fusion trunk that combines structure and sequence representations, and a geometric transformer based decoder. To increase diversity and novelty of generated samples -- crucial for de-novo drug design -- we train FoldFlow-2 at scale on a new dataset that is an order of magnitude larger than PDB datasets of prior works, containing both known proteins in PDB and high-quality synthetic structures achieved through filtering. We further demonstrate the ability to align FoldFlow-2 to arbitrary rewards, e.g. increasing secondary structures diversity, by introducing a Reinforced Finetuning (ReFT) objective. We empirically observe that FoldFlow-2 outperforms previous state-of-the-art protein structure-based generative models, improving over RFDiffusion in terms of unconditional generation across all metrics including designability, diversity, and novelty across all protein lengths, as well as exhibiting generalization on the task of equilibrium conformation sampling. Finally, we demonstrate that a fine-tuned FoldFlow-2 makes progress on challenging conditional design tasks such as designing scaffolds for the VHH nanobody. |
| title | Sequence-Augmented SE(3)-Flow Matching For Conditional Protein Backbone Generation |
| topic | Machine Learning Biomolecules |
| url | https://arxiv.org/abs/2405.20313 |