Technology

Researchers are using Generative AI to create New Proteins

Researchers are using Generative AI to create New Proteins

Generative AI, in particular, has been used to design new proteins by generating large numbers of protein sequences and selecting the ones that have desired properties. These methods rely on deep learning algorithms that can learn the rules governing protein folding and function from large datasets of existing proteins.

Researchers have developed an artificial intelligence system that can create proteins not found in nature using generative diffusion, the same technology behind popular image-creation platforms such as DALL-E and Midjourney.

Researchers at the University of Toronto have created an artificial intelligence system that can create proteins that do not exist in nature using generative diffusion, the same technology that powers popular image-creation platforms like DALL-E and Midjourney. The system will contribute to the advancement of the field of generative biology, which has the potential to accelerate drug development by making the design and testing of entirely new therapeutic proteins more efficient and flexible.

“Our model learns from image representations to generate fully new proteins at a very high rate,” says Philip M. Kim, a professor in U of T’s Temerty Faculty of Medicine’s Donnelly Centre for Cellular and Biomolecular Research. “All our proteins appear to be biophysically real, meaning they fold into configurations that enable them to carry out specific functions within cells.”

Our model learns from image representations to generate fully new proteins at a very high rate. All our proteins appear to be biophysically real, meaning they fold into configurations that enable them to carry out specific functions within cells.

Philip M. Kim

Today, the journal Nature Computational Science published the findings, the first of their kind in a peer-reviewed journal. Kim’s lab also published a pre-print on the model last summer through the open-access server bioRxiv, ahead of two similar pre-prints from last December, RF Diffusion by the University of Washington and Chroma by Generate Biomedicines.

Proteins are made from chains of amino acids that fold into three-dimensional shapes, which in turn dictate protein function. Those shapes evolved over billions of years and are varied and complex, but also limited in number. With a better understanding of how existing proteins fold, researchers have begun to design folding patterns not produced in nature.

But a major challenge, says Kim, has been to imagine folds that are both possible and functional. “It’s been very hard to predict which folds will be real and work in a protein structure,” says Kim, who is also a professor in the departments of molecular genetics and computer science at U of T. “By combining biophysics-based representations of protein structure with diffusion methods from the image generation space, we can begin to address this problem.”

Researchers use generative AI to design novel proteins

The new system, which the researchers call ProteinSGM, draws from a large set of image-like representations of existing proteins that encode their structure accurately. The researchers feed these images into a generative diffusion model, which gradually adds noise until each image becomes all noise. The model tracks how the images become noisier and then runs the process in reverse, learning how to transform random pixels into clear images that correspond to fully novel proteins.

Jin Sub (Michael) Lee, a doctoral student in the Kim lab and first author on the paper, says that optimizing the early stage of this image generation process was one of the biggest challenges in creating ProteinSGM. “A key idea was the proper image-like representation of protein structure, such that the diffusion model can learn how to generate novel proteins accurately,” says Lee, who is from Vancouver but did his undergraduate degree in South Korea and master’s in Switzerland before choosing U of T for his doctorate.

Also difficult was validation of the proteins produced by ProteinSGM. The system generates many structures, often unlike anything found in nature. Almost all of them look real according to standard metrics, says Lee, but the researchers needed further proof.

To test their new proteins, Lee and his colleagues first turned to OmegaFold, an improved version of DeepMind’s software AlphaFold 2. Both platforms use AI to predict the structure of proteins based on amino acid sequences.

With OmegaFold, the team confirmed that almost all their novel sequences fold into the desired and also novel protein structures. They then chose a smaller number to create physically in test tubes, to confirm the structures were proteins and not just stray strings of chemical compounds.

“With matches in OmegaFold and experimental testing in the lab, we could be confident these were properly folded proteins. It was amazing to see validation of these fully new protein folds that don’t exist anywhere in nature,” Lee says.

Next steps based on this work include further development of ProteinSGM for antibodies and other proteins with the most therapeutic potential, Kim says. “This will be a very exciting area for research and entrepreneurship,” he adds.