ML/DL
Antimicrobial Peptide Generator
- Reinforcement Learning
- Generative Adverserial Network
- Deep Learning
- PyTorch
Overview
This project leverages deep learning to address the critical issue of antibiotic resistance. By integrating Reinforcement Learning (RL) into a Generative Adversarial Network (GAN) framework, we developed a model capable of de novo design of Antimicrobial Peptides (AMPs). The core innovation lies in embedding a Monte Carlo search within the generator to optimize sequence generation through policy gradients.
Key highlights
Technical Architecture
- Implemented a Generative Adversarial Network (GAN) framework to generate novel peptide sequences.
- Enhanced the generator by embedding a Monte Carlo search to estimate intermediate action values, effectively treating sequence generation as a Reinforcement Learning problem.
- Utilized Policy Gradient methods where the discriminator provides the final reward signal to update the generator.
- Trained on experimentally verified peptide sequences curated from major databases including dbAMP, DADP, DBAASP, and APD.
Validation Pipeline
- Candidates were screened by comparing key properties (e.g., hydrophobicity, net charge, alpha-helical propensity, and Boman index) against known active peptides.
- High-confidence sequences were analyzed using AlphaFold to predict 3D structures, followed by Molecular Dynamic (MD) simulations to assess stability.