PROTEIN LATENT STRUCTURE TRAVERSAL
A latent space associated with a protein structure autoencoder is traversed. The traversal is performed for determining structure latent values of a target protein. A decoder is applied to the determined structure latent values. For example, the determined structure latent values are used to determine a structure of the target protein.
1 . A method, comprising:
traversing a latent space associated with a protein structure autoencoder to determine structure latent values of a target protein; and
applying a decoder to the determined structure latent values to determine a structure of the target protein.
2 . The method of claim 1 , further comprising:
using the protein structure autoencoder to generate initial latent values for the target protein, wherein the initial latent values are used for traversing the latent space.
3 . The method of claim 1 , wherein traversing the latent space associated with the protein structure autoencoder to determine the structure latent values of the target protein is performed using a machine learning model.
4 . The method of claim 3 , wherein the machine learning model is a diffusion model, a flow matching model, or a classifier model.
5 . The method of claim 3 , wherein the machine learning model is a continuous normalizing flows (CNF) model trained using flow matching.
6 . The method of claim 3 , wherein the machine learning model is used to determine a search criterion used for traversing the latent space.
7 . The method of claim 1 , further comprising:
for a specific amino acid in the target protein, determining physically neighboring amino acids of the specific amino acid in a local physical protein structure;
including representations of the determined physically neighboring amino acids in a structure encoder input for the specific amino acid; and
providing the structure encoder input to the protein structure autoencoder trained to determine an initial latent value representing the local physical protein structure for the specific amino acid, wherein the initial latent value is used for traversing the latent space.
8 . The method of claim 1 , wherein the protein structure autoencoder is configured with one or more geometric reasoning blocks, and wherein at least one of the one or more geometric reasoning blocks utilizes a geometric attention mechanism.
9 . The method of claim 1 , wherein the protein structure autoencoder is trained using geometric loss.
10 . The method of claim 9 , wherein the geometric loss is modeled using a function that determines an error loss value based on relative orientations of bond vectors in a predicted structure and a ground truth structure.
11 . A system, comprising:
one or more processors configured to:
traverse a latent space associated with a protein structure autoencoder to determine structure latent values of a target protein; and
apply a decoder to the determined structure latent values to determine a structure of the target protein; and
a memory coupled to at least one of the one or more processors and configured to provide instructions.
12 . The system of claim 11 , wherein the one or more processors are configured to:
use the protein structure autoencoder to generate initial latent values for the target protein, wherein the initial latent values are used for traversing the latent space.
13 . The system of claim 11 , wherein traversing the latent space associated with the protein structure autoencoder to determine the structure latent values of the target protein is performed using a machine learning model.
14 . The system of claim 13 , wherein the machine learning model is a diffusion model, a flow matching model, or a classifier model.
15 . The system of claim 13 , wherein the machine learning model is a continuous normalizing flows (CNF) model trained using flow matching.
16 . The system of claim 13 , wherein the machine learning model is used to determine a search criterion used for traversing the latent space.
17 . The system of claim 11 , wherein the one or more processors are configured to:
for a specific amino acid in the target protein, determine physically neighboring amino acids of the specific amino acid in a local physical protein structure;
include representations of the determined physically neighboring amino acids in a structure encoder input for the specific amino acid; and
provide the structure encoder input to the protein structure autoencoder trained to determine an initial latent value representing the local physical protein structure for the specific amino acid, wherein the initial latent value is used for traversing the latent space.
18 . The system of claim 11 , wherein the protein structure autoencoder is configured with one or more geometric reasoning blocks, and wherein at least one of the one or more geometric reasoning blocks utilizes a geometric attention mechanism.
19 . The system of claim 11 , wherein the protein structure autoencoder is trained using geometric loss.
20 . A computer program product, the computer program product being embodied in a non-transitory computer readable storage medium and comprising computer instructions for:
traversing a latent space associated with a protein structure autoencoder to determine structure latent values of a target protein; and
is applying a decoder to the determined structure latent values to determine a structure of the target protein.