PLAID is a multimodal generative mannequin that concurrently generates protein 1D sequence and 3D construction, by studying the latent area of protein folding fashions.
The awarding of the 2024 Nobel Prize to AlphaFold2 marks an vital second of recognition for the of AI position in biology. What comes subsequent after protein folding?
In PLAID, we develop a technique that learns to pattern from the latent area of protein folding fashions to generate new proteins. It may well settle for compositional perform and organism prompts, and could be skilled on sequence databases, that are 2-4 orders of magnitude bigger than construction databases. In contrast to many earlier protein construction generative fashions, PLAID addresses the multimodal co-generation downside setting: concurrently producing each discrete sequence and steady all-atom structural coordinates.
From construction prediction to real-world drug design
Although current works exhibit promise for the power of diffusion fashions to generate proteins, there nonetheless exist limitations of earlier fashions that make them impractical for real-world functions, reminiscent of:
- All-atom era: Many current generative fashions solely produce the spine atoms. To supply the all-atom construction and place the sidechain atoms, we have to know the sequence. This creates a multimodal era downside that requires simultaneous era of discrete and steady modalities.
- Organism specificity: Proteins biologics meant for human use have to be humanized, to keep away from being destroyed by the human immune system.
- Management specification: Drug discovery and placing it into the arms of sufferers is a posh course of. How can we specify these advanced constraints? For instance, even after the biology is tackled, you may resolve that tablets are simpler to move than vials, including a brand new constraint on soluability.
Producing “helpful” proteins
Merely producing proteins isn’t as helpful as controlling the era to get helpful proteins. What may an interface for this appear like?
For inspiration, let’s think about how we might management picture era by way of compositional textual prompts (instance from Liu et al., 2022).
In PLAID, we mirror this interface for management specification. The last word aim is to regulate era totally by way of a textual interface, however right here we think about compositional constraints for 2 axes as a proof-of-concept: perform and organism:
Studying the function-structure-sequence connection. PLAID learns the tetrahedral cysteine-Fe2+/Fe3+ coordination sample usually present in metalloproteins, whereas sustaining excessive sequence-level variety.
Coaching utilizing sequence-only coaching information
One other vital facet of the PLAID mannequin is that we solely require sequences to coach the generative mannequin! Generative fashions be taught the information distribution outlined by its coaching information, and sequence databases are significantly bigger than structural ones, since sequences are less expensive to acquire than experimental construction.
Studying from a bigger and broader database. The price of acquiring protein sequences is far decrease than experimentally characterizing construction, and sequence databases are 2-4 orders of magnitude bigger than structural ones.
How does it work?
The rationale that we’re capable of practice the generative mannequin to generate construction by solely utilizing sequence information is by studying a diffusion mannequin over the latent area of a protein folding mannequin. Then, throughout inference, after sampling from this latent area of legitimate proteins, we are able to take frozen weights from the protein folding mannequin to decode construction. Right here, we use ESMFold, a successor to the AlphaFold2 mannequin which replaces a retrieval step with a protein language mannequin.
Our methodology. Throughout coaching, solely sequences are wanted to acquire the embedding; throughout inference, we are able to decode sequence and construction from the sampled embedding. ❄️ denotes frozen weights.
On this manner, we are able to use structural understanding data within the weights of pretrained protein folding fashions for the protein design activity. That is analogous to how vision-language-action (VLA) fashions in robotics make use of priors contained in vision-language fashions (VLMs) skilled on internet-scale information to provide notion and reasoning and understanding data.
Compressing the latent area of protein folding fashions
A small wrinkle with straight making use of this methodology is that the latent area of ESMFold – certainly, the latent area of many transformer-based fashions – requires a number of regularization. This area can also be very massive, so studying this embedding finally ends up mapping to high-resolution picture synthesis.
To deal with this, we additionally suggest CHEAP (Compressed Hourglass Embedding Diversifications of Proteins), the place we be taught a compression mannequin for the joint embedding of protein sequence and construction.
Investigating the latent area. (A) Once we visualize the imply worth for every channel, some channels exhibit “huge activations”. (B) If we begin analyzing the top-3 activations in comparison with the median worth (grey), we discover that this occurs over many layers. (C) Huge activations have additionally been noticed for different transformer-based fashions.
We discover that this latent area is definitely extremely compressible. By doing a little bit of mechanistic interpretability to higher perceive the bottom mannequin that we’re working with, we had been capable of create an all-atom protein generative mannequin.
What’s subsequent?
Although we look at the case of protein sequence and construction era on this work, we are able to adapt this methodology to carry out multi-modal era for any modalities the place there’s a predictor from a extra considerable modality to a much less considerable one. As sequence-to-structure predictors for proteins are starting to deal with more and more advanced techniques (e.g. AlphaFold3 can also be capable of predict proteins in advanced with nucleic acids and molecular ligands), it’s simple to think about performing multimodal era over extra advanced techniques utilizing the identical methodology.
In case you are fascinated with collaborating to increase our methodology, or to check our methodology within the wet-lab, please attain out!
Additional hyperlinks
For those who’ve discovered our papers helpful in your analysis, please think about using the next BibTeX for PLAID and CHEAP:
@article{lu2024generating,
title={Producing All-Atom Protein Construction from Sequence-Solely Coaching Knowledge},
writer={Lu, Amy X and Yan, Wilson and Robinson, Sarah A and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Bonneau, Richard and Abbeel, Pieter and Frey, Nathan},
journal={bioRxiv},
pages={2024--12},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
@article{lu2024tokenized,
title={Tokenized and Steady Embedding Compressions of Protein Sequence and Construction},
writer={Lu, Amy X and Yan, Wilson and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Abbeel, Pieter and Bonneau, Richard and Frey, Nathan},
journal={bioRxiv},
pages={2024--08},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
You can even checkout our preprints (PLAID, CHEAP) and codebases (PLAID, CHEAP).
Some bonus protein era enjoyable!
Extra function-prompted generations with PLAID.
Unconditional era with PLAID.
Transmembrane proteins have hydrophobic residues on the core, the place it’s embedded inside the fatty acid layer. These are constantly noticed when prompting PLAID with transmembrane protein key phrases.
Extra examples of energetic website recapitulation based mostly on perform key phrase prompting.
Evaluating samples between PLAID and all-atom baselines. PLAID samples have higher variety and captures the beta-strand sample that has been harder for protein generative fashions to be taught.
Acknowledgements
Due to Nathan Frey for detailed suggestions on this text, and to co-authors throughout BAIR, Genentech, Microsoft Analysis, and New York College: Wilson Yan, Sarah A. Robinson, Simon Kelow, Kevin Ok. Yang, Vladimir Gligorijevic, Kyunghyun Cho, Richard Bonneau, Pieter Abbeel, and Nathan C. Frey.
PLAID is a multimodal generative mannequin that concurrently generates protein 1D sequence and 3D construction, by studying the latent area of protein folding fashions.
The awarding of the 2024 Nobel Prize to AlphaFold2 marks an vital second of recognition for the of AI position in biology. What comes subsequent after protein folding?
In PLAID, we develop a technique that learns to pattern from the latent area of protein folding fashions to generate new proteins. It may well settle for compositional perform and organism prompts, and could be skilled on sequence databases, that are 2-4 orders of magnitude bigger than construction databases. In contrast to many earlier protein construction generative fashions, PLAID addresses the multimodal co-generation downside setting: concurrently producing each discrete sequence and steady all-atom structural coordinates.
From construction prediction to real-world drug design
Although current works exhibit promise for the power of diffusion fashions to generate proteins, there nonetheless exist limitations of earlier fashions that make them impractical for real-world functions, reminiscent of:
- All-atom era: Many current generative fashions solely produce the spine atoms. To supply the all-atom construction and place the sidechain atoms, we have to know the sequence. This creates a multimodal era downside that requires simultaneous era of discrete and steady modalities.
- Organism specificity: Proteins biologics meant for human use have to be humanized, to keep away from being destroyed by the human immune system.
- Management specification: Drug discovery and placing it into the arms of sufferers is a posh course of. How can we specify these advanced constraints? For instance, even after the biology is tackled, you may resolve that tablets are simpler to move than vials, including a brand new constraint on soluability.
Producing “helpful” proteins
Merely producing proteins isn’t as helpful as controlling the era to get helpful proteins. What may an interface for this appear like?
For inspiration, let’s think about how we might management picture era by way of compositional textual prompts (instance from Liu et al., 2022).
In PLAID, we mirror this interface for management specification. The last word aim is to regulate era totally by way of a textual interface, however right here we think about compositional constraints for 2 axes as a proof-of-concept: perform and organism:
Studying the function-structure-sequence connection. PLAID learns the tetrahedral cysteine-Fe2+/Fe3+ coordination sample usually present in metalloproteins, whereas sustaining excessive sequence-level variety.
Coaching utilizing sequence-only coaching information
One other vital facet of the PLAID mannequin is that we solely require sequences to coach the generative mannequin! Generative fashions be taught the information distribution outlined by its coaching information, and sequence databases are significantly bigger than structural ones, since sequences are less expensive to acquire than experimental construction.
Studying from a bigger and broader database. The price of acquiring protein sequences is far decrease than experimentally characterizing construction, and sequence databases are 2-4 orders of magnitude bigger than structural ones.
How does it work?
The rationale that we’re capable of practice the generative mannequin to generate construction by solely utilizing sequence information is by studying a diffusion mannequin over the latent area of a protein folding mannequin. Then, throughout inference, after sampling from this latent area of legitimate proteins, we are able to take frozen weights from the protein folding mannequin to decode construction. Right here, we use ESMFold, a successor to the AlphaFold2 mannequin which replaces a retrieval step with a protein language mannequin.
Our methodology. Throughout coaching, solely sequences are wanted to acquire the embedding; throughout inference, we are able to decode sequence and construction from the sampled embedding. ❄️ denotes frozen weights.
On this manner, we are able to use structural understanding data within the weights of pretrained protein folding fashions for the protein design activity. That is analogous to how vision-language-action (VLA) fashions in robotics make use of priors contained in vision-language fashions (VLMs) skilled on internet-scale information to provide notion and reasoning and understanding data.
Compressing the latent area of protein folding fashions
A small wrinkle with straight making use of this methodology is that the latent area of ESMFold – certainly, the latent area of many transformer-based fashions – requires a number of regularization. This area can also be very massive, so studying this embedding finally ends up mapping to high-resolution picture synthesis.
To deal with this, we additionally suggest CHEAP (Compressed Hourglass Embedding Diversifications of Proteins), the place we be taught a compression mannequin for the joint embedding of protein sequence and construction.
Investigating the latent area. (A) Once we visualize the imply worth for every channel, some channels exhibit “huge activations”. (B) If we begin analyzing the top-3 activations in comparison with the median worth (grey), we discover that this occurs over many layers. (C) Huge activations have additionally been noticed for different transformer-based fashions.
We discover that this latent area is definitely extremely compressible. By doing a little bit of mechanistic interpretability to higher perceive the bottom mannequin that we’re working with, we had been capable of create an all-atom protein generative mannequin.
What’s subsequent?
Although we look at the case of protein sequence and construction era on this work, we are able to adapt this methodology to carry out multi-modal era for any modalities the place there’s a predictor from a extra considerable modality to a much less considerable one. As sequence-to-structure predictors for proteins are starting to deal with more and more advanced techniques (e.g. AlphaFold3 can also be capable of predict proteins in advanced with nucleic acids and molecular ligands), it’s simple to think about performing multimodal era over extra advanced techniques utilizing the identical methodology.
In case you are fascinated with collaborating to increase our methodology, or to check our methodology within the wet-lab, please attain out!
Additional hyperlinks
For those who’ve discovered our papers helpful in your analysis, please think about using the next BibTeX for PLAID and CHEAP:
@article{lu2024generating,
title={Producing All-Atom Protein Construction from Sequence-Solely Coaching Knowledge},
writer={Lu, Amy X and Yan, Wilson and Robinson, Sarah A and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Bonneau, Richard and Abbeel, Pieter and Frey, Nathan},
journal={bioRxiv},
pages={2024--12},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
@article{lu2024tokenized,
title={Tokenized and Steady Embedding Compressions of Protein Sequence and Construction},
writer={Lu, Amy X and Yan, Wilson and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Abbeel, Pieter and Bonneau, Richard and Frey, Nathan},
journal={bioRxiv},
pages={2024--08},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
You can even checkout our preprints (PLAID, CHEAP) and codebases (PLAID, CHEAP).
Some bonus protein era enjoyable!
Extra function-prompted generations with PLAID.
Unconditional era with PLAID.
Transmembrane proteins have hydrophobic residues on the core, the place it’s embedded inside the fatty acid layer. These are constantly noticed when prompting PLAID with transmembrane protein key phrases.
Extra examples of energetic website recapitulation based mostly on perform key phrase prompting.
Evaluating samples between PLAID and all-atom baselines. PLAID samples have higher variety and captures the beta-strand sample that has been harder for protein generative fashions to be taught.
Acknowledgements
Due to Nathan Frey for detailed suggestions on this text, and to co-authors throughout BAIR, Genentech, Microsoft Analysis, and New York College: Wilson Yan, Sarah A. Robinson, Simon Kelow, Kevin Ok. Yang, Vladimir Gligorijevic, Kyunghyun Cho, Richard Bonneau, Pieter Abbeel, and Nathan C. Frey.
PLAID is a multimodal generative mannequin that concurrently generates protein 1D sequence and 3D construction, by studying the latent area of protein folding fashions.
The awarding of the 2024 Nobel Prize to AlphaFold2 marks an vital second of recognition for the of AI position in biology. What comes subsequent after protein folding?
In PLAID, we develop a technique that learns to pattern from the latent area of protein folding fashions to generate new proteins. It may well settle for compositional perform and organism prompts, and could be skilled on sequence databases, that are 2-4 orders of magnitude bigger than construction databases. In contrast to many earlier protein construction generative fashions, PLAID addresses the multimodal co-generation downside setting: concurrently producing each discrete sequence and steady all-atom structural coordinates.
From construction prediction to real-world drug design
Although current works exhibit promise for the power of diffusion fashions to generate proteins, there nonetheless exist limitations of earlier fashions that make them impractical for real-world functions, reminiscent of:
- All-atom era: Many current generative fashions solely produce the spine atoms. To supply the all-atom construction and place the sidechain atoms, we have to know the sequence. This creates a multimodal era downside that requires simultaneous era of discrete and steady modalities.
- Organism specificity: Proteins biologics meant for human use have to be humanized, to keep away from being destroyed by the human immune system.
- Management specification: Drug discovery and placing it into the arms of sufferers is a posh course of. How can we specify these advanced constraints? For instance, even after the biology is tackled, you may resolve that tablets are simpler to move than vials, including a brand new constraint on soluability.
Producing “helpful” proteins
Merely producing proteins isn’t as helpful as controlling the era to get helpful proteins. What may an interface for this appear like?
For inspiration, let’s think about how we might management picture era by way of compositional textual prompts (instance from Liu et al., 2022).
In PLAID, we mirror this interface for management specification. The last word aim is to regulate era totally by way of a textual interface, however right here we think about compositional constraints for 2 axes as a proof-of-concept: perform and organism:
Studying the function-structure-sequence connection. PLAID learns the tetrahedral cysteine-Fe2+/Fe3+ coordination sample usually present in metalloproteins, whereas sustaining excessive sequence-level variety.
Coaching utilizing sequence-only coaching information
One other vital facet of the PLAID mannequin is that we solely require sequences to coach the generative mannequin! Generative fashions be taught the information distribution outlined by its coaching information, and sequence databases are significantly bigger than structural ones, since sequences are less expensive to acquire than experimental construction.
Studying from a bigger and broader database. The price of acquiring protein sequences is far decrease than experimentally characterizing construction, and sequence databases are 2-4 orders of magnitude bigger than structural ones.
How does it work?
The rationale that we’re capable of practice the generative mannequin to generate construction by solely utilizing sequence information is by studying a diffusion mannequin over the latent area of a protein folding mannequin. Then, throughout inference, after sampling from this latent area of legitimate proteins, we are able to take frozen weights from the protein folding mannequin to decode construction. Right here, we use ESMFold, a successor to the AlphaFold2 mannequin which replaces a retrieval step with a protein language mannequin.
Our methodology. Throughout coaching, solely sequences are wanted to acquire the embedding; throughout inference, we are able to decode sequence and construction from the sampled embedding. ❄️ denotes frozen weights.
On this manner, we are able to use structural understanding data within the weights of pretrained protein folding fashions for the protein design activity. That is analogous to how vision-language-action (VLA) fashions in robotics make use of priors contained in vision-language fashions (VLMs) skilled on internet-scale information to provide notion and reasoning and understanding data.
Compressing the latent area of protein folding fashions
A small wrinkle with straight making use of this methodology is that the latent area of ESMFold – certainly, the latent area of many transformer-based fashions – requires a number of regularization. This area can also be very massive, so studying this embedding finally ends up mapping to high-resolution picture synthesis.
To deal with this, we additionally suggest CHEAP (Compressed Hourglass Embedding Diversifications of Proteins), the place we be taught a compression mannequin for the joint embedding of protein sequence and construction.
Investigating the latent area. (A) Once we visualize the imply worth for every channel, some channels exhibit “huge activations”. (B) If we begin analyzing the top-3 activations in comparison with the median worth (grey), we discover that this occurs over many layers. (C) Huge activations have additionally been noticed for different transformer-based fashions.
We discover that this latent area is definitely extremely compressible. By doing a little bit of mechanistic interpretability to higher perceive the bottom mannequin that we’re working with, we had been capable of create an all-atom protein generative mannequin.
What’s subsequent?
Although we look at the case of protein sequence and construction era on this work, we are able to adapt this methodology to carry out multi-modal era for any modalities the place there’s a predictor from a extra considerable modality to a much less considerable one. As sequence-to-structure predictors for proteins are starting to deal with more and more advanced techniques (e.g. AlphaFold3 can also be capable of predict proteins in advanced with nucleic acids and molecular ligands), it’s simple to think about performing multimodal era over extra advanced techniques utilizing the identical methodology.
In case you are fascinated with collaborating to increase our methodology, or to check our methodology within the wet-lab, please attain out!
Additional hyperlinks
For those who’ve discovered our papers helpful in your analysis, please think about using the next BibTeX for PLAID and CHEAP:
@article{lu2024generating,
title={Producing All-Atom Protein Construction from Sequence-Solely Coaching Knowledge},
writer={Lu, Amy X and Yan, Wilson and Robinson, Sarah A and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Bonneau, Richard and Abbeel, Pieter and Frey, Nathan},
journal={bioRxiv},
pages={2024--12},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
@article{lu2024tokenized,
title={Tokenized and Steady Embedding Compressions of Protein Sequence and Construction},
writer={Lu, Amy X and Yan, Wilson and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Abbeel, Pieter and Bonneau, Richard and Frey, Nathan},
journal={bioRxiv},
pages={2024--08},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
You can even checkout our preprints (PLAID, CHEAP) and codebases (PLAID, CHEAP).
Some bonus protein era enjoyable!
Extra function-prompted generations with PLAID.
Unconditional era with PLAID.
Transmembrane proteins have hydrophobic residues on the core, the place it’s embedded inside the fatty acid layer. These are constantly noticed when prompting PLAID with transmembrane protein key phrases.
Extra examples of energetic website recapitulation based mostly on perform key phrase prompting.
Evaluating samples between PLAID and all-atom baselines. PLAID samples have higher variety and captures the beta-strand sample that has been harder for protein generative fashions to be taught.
Acknowledgements
Due to Nathan Frey for detailed suggestions on this text, and to co-authors throughout BAIR, Genentech, Microsoft Analysis, and New York College: Wilson Yan, Sarah A. Robinson, Simon Kelow, Kevin Ok. Yang, Vladimir Gligorijevic, Kyunghyun Cho, Richard Bonneau, Pieter Abbeel, and Nathan C. Frey.
PLAID is a multimodal generative mannequin that concurrently generates protein 1D sequence and 3D construction, by studying the latent area of protein folding fashions.
The awarding of the 2024 Nobel Prize to AlphaFold2 marks an vital second of recognition for the of AI position in biology. What comes subsequent after protein folding?
In PLAID, we develop a technique that learns to pattern from the latent area of protein folding fashions to generate new proteins. It may well settle for compositional perform and organism prompts, and could be skilled on sequence databases, that are 2-4 orders of magnitude bigger than construction databases. In contrast to many earlier protein construction generative fashions, PLAID addresses the multimodal co-generation downside setting: concurrently producing each discrete sequence and steady all-atom structural coordinates.
From construction prediction to real-world drug design
Although current works exhibit promise for the power of diffusion fashions to generate proteins, there nonetheless exist limitations of earlier fashions that make them impractical for real-world functions, reminiscent of:
- All-atom era: Many current generative fashions solely produce the spine atoms. To supply the all-atom construction and place the sidechain atoms, we have to know the sequence. This creates a multimodal era downside that requires simultaneous era of discrete and steady modalities.
- Organism specificity: Proteins biologics meant for human use have to be humanized, to keep away from being destroyed by the human immune system.
- Management specification: Drug discovery and placing it into the arms of sufferers is a posh course of. How can we specify these advanced constraints? For instance, even after the biology is tackled, you may resolve that tablets are simpler to move than vials, including a brand new constraint on soluability.
Producing “helpful” proteins
Merely producing proteins isn’t as helpful as controlling the era to get helpful proteins. What may an interface for this appear like?
For inspiration, let’s think about how we might management picture era by way of compositional textual prompts (instance from Liu et al., 2022).
In PLAID, we mirror this interface for management specification. The last word aim is to regulate era totally by way of a textual interface, however right here we think about compositional constraints for 2 axes as a proof-of-concept: perform and organism:
Studying the function-structure-sequence connection. PLAID learns the tetrahedral cysteine-Fe2+/Fe3+ coordination sample usually present in metalloproteins, whereas sustaining excessive sequence-level variety.
Coaching utilizing sequence-only coaching information
One other vital facet of the PLAID mannequin is that we solely require sequences to coach the generative mannequin! Generative fashions be taught the information distribution outlined by its coaching information, and sequence databases are significantly bigger than structural ones, since sequences are less expensive to acquire than experimental construction.
Studying from a bigger and broader database. The price of acquiring protein sequences is far decrease than experimentally characterizing construction, and sequence databases are 2-4 orders of magnitude bigger than structural ones.
How does it work?
The rationale that we’re capable of practice the generative mannequin to generate construction by solely utilizing sequence information is by studying a diffusion mannequin over the latent area of a protein folding mannequin. Then, throughout inference, after sampling from this latent area of legitimate proteins, we are able to take frozen weights from the protein folding mannequin to decode construction. Right here, we use ESMFold, a successor to the AlphaFold2 mannequin which replaces a retrieval step with a protein language mannequin.
Our methodology. Throughout coaching, solely sequences are wanted to acquire the embedding; throughout inference, we are able to decode sequence and construction from the sampled embedding. ❄️ denotes frozen weights.
On this manner, we are able to use structural understanding data within the weights of pretrained protein folding fashions for the protein design activity. That is analogous to how vision-language-action (VLA) fashions in robotics make use of priors contained in vision-language fashions (VLMs) skilled on internet-scale information to provide notion and reasoning and understanding data.
Compressing the latent area of protein folding fashions
A small wrinkle with straight making use of this methodology is that the latent area of ESMFold – certainly, the latent area of many transformer-based fashions – requires a number of regularization. This area can also be very massive, so studying this embedding finally ends up mapping to high-resolution picture synthesis.
To deal with this, we additionally suggest CHEAP (Compressed Hourglass Embedding Diversifications of Proteins), the place we be taught a compression mannequin for the joint embedding of protein sequence and construction.
Investigating the latent area. (A) Once we visualize the imply worth for every channel, some channels exhibit “huge activations”. (B) If we begin analyzing the top-3 activations in comparison with the median worth (grey), we discover that this occurs over many layers. (C) Huge activations have additionally been noticed for different transformer-based fashions.
We discover that this latent area is definitely extremely compressible. By doing a little bit of mechanistic interpretability to higher perceive the bottom mannequin that we’re working with, we had been capable of create an all-atom protein generative mannequin.
What’s subsequent?
Although we look at the case of protein sequence and construction era on this work, we are able to adapt this methodology to carry out multi-modal era for any modalities the place there’s a predictor from a extra considerable modality to a much less considerable one. As sequence-to-structure predictors for proteins are starting to deal with more and more advanced techniques (e.g. AlphaFold3 can also be capable of predict proteins in advanced with nucleic acids and molecular ligands), it’s simple to think about performing multimodal era over extra advanced techniques utilizing the identical methodology.
In case you are fascinated with collaborating to increase our methodology, or to check our methodology within the wet-lab, please attain out!
Additional hyperlinks
For those who’ve discovered our papers helpful in your analysis, please think about using the next BibTeX for PLAID and CHEAP:
@article{lu2024generating,
title={Producing All-Atom Protein Construction from Sequence-Solely Coaching Knowledge},
writer={Lu, Amy X and Yan, Wilson and Robinson, Sarah A and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Bonneau, Richard and Abbeel, Pieter and Frey, Nathan},
journal={bioRxiv},
pages={2024--12},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
@article{lu2024tokenized,
title={Tokenized and Steady Embedding Compressions of Protein Sequence and Construction},
writer={Lu, Amy X and Yan, Wilson and Yang, Kevin Ok and Gligorijevic, Vladimir and Cho, Kyunghyun and Abbeel, Pieter and Bonneau, Richard and Frey, Nathan},
journal={bioRxiv},
pages={2024--08},
yr={2024},
writer={Chilly Spring Harbor Laboratory}
}
You can even checkout our preprints (PLAID, CHEAP) and codebases (PLAID, CHEAP).
Some bonus protein era enjoyable!
Extra function-prompted generations with PLAID.
Unconditional era with PLAID.
Transmembrane proteins have hydrophobic residues on the core, the place it’s embedded inside the fatty acid layer. These are constantly noticed when prompting PLAID with transmembrane protein key phrases.
Extra examples of energetic website recapitulation based mostly on perform key phrase prompting.
Evaluating samples between PLAID and all-atom baselines. PLAID samples have higher variety and captures the beta-strand sample that has been harder for protein generative fashions to be taught.
Acknowledgements
Due to Nathan Frey for detailed suggestions on this text, and to co-authors throughout BAIR, Genentech, Microsoft Analysis, and New York College: Wilson Yan, Sarah A. Robinson, Simon Kelow, Kevin Ok. Yang, Vladimir Gligorijevic, Kyunghyun Cho, Richard Bonneau, Pieter Abbeel, and Nathan C. Frey.
