A PEPTIDE BASED VACCINE GENERATION SYSTEM WITH DUAL PROJECTION GENERATIVE ADVERSARIAL NETWORKS
A computer-implemented method is provided for generating new binding peptides to Major Histocompatibility Complex (MHC) proteins. The method includes training, by a processor device, a Generative Adversarial Network GAN having a generator and a discriminator only on a set of binding peptide sequences given training data comprising the set of binding peptide sequences and a set of non-binding peptide sequences. A GAN training objective includes the discriminator being iteratively updated to distinguish generated peptide sequences from sampled binding peptide sequences as fake or real and the generator being iteratively updated to fool the discriminator. The training includes optimizing the GAN training objective while learning two projection vectors for a binding class with two cross-entropy losses. A first loss discriminating binding peptide sequences in the training data from non-binding peptide sequences in the training data. A second loss discriminating generated binding peptide sequences from non-binding peptide sequences in the training data.
1 . A computer-implemented method for generating new binding peptides to Major Histocompatibility Complex (MHC) proteins, comprising:
training, by a processor device, a Generative Adversarial Network GAN having a generator and a discriminator only on a set of binding peptide sequences given training data comprising the set of binding peptide sequences and a set of non-binding peptide sequences, wherein a GAN training objective comprises the discriminator being iteratively updated to distinguish generated peptide sequences from sampled binding peptide sequences as fake or real and the generator being iteratively updated to fool the discriminator;
wherein said training comprises optimizing the GAN training objective while learning two projection vectors for a binding class with two cross-entropy losses, a first of the two cross-entropy losses discriminating binding peptide sequences in the training data from non-binding peptide sequences in the training data, and a second of the two cross-entropy losses discriminating generated binding peptide sequences from non-binding peptide sequences in the training data,
in the training at a beginning, an initial penalty coefficient is set for entropy regularization, and
in the training at a timing later than the beginning, a penalty coefficient larger than the initial penalty coefficient is used for the entropy regularization.
2 . The computer-implemented method of claim 1 , wherein the GAN is a Wasserstein GAN.
3 . The computer-implemented method of claim 1 , wherein said training further comprises updating the generator with tempering Softmax units to minimize the two cross-entropy losses.
4 . The computer-implemented method of claim 3 , wherein the tempering Softmax units are employed with the entropy regularization for implicit temperature control of the tempering Softmax units.
5 . The computer-implemented method of claim 1 , wherein the generator is a deep neural network, comprising a convolutional layer for receiving the set of binding peptide sequences.
6 . The computer-implemented method of claim 1 , wherein the GAN operates on sampled latent code vectors from a multivariate Gaussian distribution obtained from the training data.
7 . The computer-implemented method of claim 1 , wherein the two cross-entropy losses are implemented by dual classifiers.
8 . The computer-implemented method of claim 1 , further comprising generating peptide-based vaccines with user-specified properties using the trained GAN.
9 . The computer-implemented method of claim 8 , wherein the peptide-based vaccines are output from the generator as Softmax output units, and wherein the generator comprises a fully-connected layer for receiving an input random noise vector and another fully-connected layer for outputting the Softmax output units.