Fine Tuned Progen2 Small

🚀 Antibody Generator (Based on ProGen2)

The Antibody Generator is a specialized protein generation model designed to create therapeutic antibodies. It builds upon the ProGen2 architecture, offering high - performance in generating novel and viable protein sequences for immunology, vaccine development, and medical treatments.

🚀 Quick Start

Instructions on how to use the model, including example prompts and API documentation, are available in the Code Repository.

💻 Usage Examples

Basic Usage

from models.progen.modeling_progen import ProGenForCausalLM
import torch
from tokenizers import Tokenizer
import json

# Define the model identifier from Hugging Face's model hub
model_path = 'AntibodyGeneration/fine-tuned-progen2-small'

# Load the model and tokenizer
model = ProGenForCausalLM.from_pretrained(model_path)
tokenizer = Tokenizer.from_file('tokenizer.json')

# Define your sequence and other parameters
target_sequence = 'MQIPQAPWPVVWAVLQLGWRPGWFLDSPDRPWNPPTFSPALLVVTEGDNATFTCSFSNTSESFVLNWYRMSPSNQTDKLAAFPEDRSQPGQDCRFRVTQLPNGRDFHMSVVRARRNDSGTYLCGAISLAPKAQIKESLRAELRVTERRAEVPTAHPSPSPRPAGQFQTLVVGVVGGLLGSLVLLVWVLAVICSRAARGTIGARRTGQPLKEDPSAVPVFSVDYGELDFQWREKTPEPPVPCVPEQTEYATIVFPSGMGTSSPARRGSADGPRSAQPLRPEDGHCSWPL'
number_of_sequences = 2

# Tokenize the sequence
tokenized_sequence = tokenizer(target_sequence, return_tensors="pt")

# Move model and tensors to CUDA if available
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
model = model.to(device)
tokenized_sequence = tokenized_sequence.to(device)

# Generate sequences
with torch.no_grad():
    output = model.generate(**tokenized_sequence, max_length=1024, pad_token_id=tokenizer.pad_token_id, do_sample=True, top_p=0.9, temperature=0.8, num_return_sequences=number_of_sequences)

# Decoding the output to get generated sequences
generated_sequences = [tokenizer.decode(output_seq, skip_special_tokens=True) for output_seq in output]

✨ Features

• Specialized for Antibodies: Developed specifically for generating therapeutic antibody sequences, suitable for immunology, vaccine development, and medical treatments.
• Based on ProGen2: Leverages the advanced ProGen2 architecture, pre - trained on over 280 million protein sequences, with up to 6.4B parameters for state - of - the - art performance.

📦 Installation

No installation steps are provided in the original README, so this section is skipped.

📚 Documentation

Model Details

Model Overview

The Antibody Generator is a specialized protein generation model developed for creating therapeutic antibodies. It is based on the ProGen2 model, an advanced language model developed by Salesforce. ProGen2, an enhancement of the original ProGen model launched in 2020, is pre - trained on a vast dataset of over 280 million protein sequences. With up to 6.4B parameters, ProGen2 demonstrates state - of - the - art performance in generating novel, viable protein sequences and predicting protein fitness.

Intended Use

• Primary Use Case: Generation of therapeutic antibody sequences for use in immunology, vaccine development, and medical treatments.
• Target Users: Researchers and practitioners in bioinformatics, molecular biology, and related fields.

Training Data

• Baseline Model Data: ProGen2 was trained on a large collection of protein sequences from genomic, metagenomic, and immune repertoire databases, totaling over 280 million samples.
• Fine - tuning Data: For fine - tuning, the Structural Antibody Database was used, comprising approximately 5,000 experimentally - resolved crystal structures of antibodies and their antigens.

Model Variants

Models are labeled as progen2_<finetuning_type><prompting_type>

1. Size: This refers to the size of the progen2 base model that was used. There are 4 variants available:
   1. Small: 151M params
   2. Medium: 764M params
   3. Large: 2.7B params
   4. xLarge: 6.4B params
2. Finetuning_type: This refers to how the base model was finetuned. 2 types are supported:
   1. No finetuning
   2. Simple finetuning: The base model is finetuned with 5,000 experimentally - resolved crystal structures of antibodies and their antigens using hyperparameters below.
   3. Frozen layer finetuning: The base model is finetuned with 5,000 experimentally - resolved crystal structures of antibodies and their antigens using hyperparameters below. Additionally, all layers except last 3 are frozen to avoid overfitting.
3. Prompting_type: This refers to whether the model was provided with any prompting during inference.
   1. Prompted: Use prompt engineering for generating therapeutic antibody sequences.
   2. Zeroshot: No prompting is provided.

Model Hyperparameters

• Batch size: 40
• Epochs: 10
• Learning rate: 0.00001

Evaluation and Performance

Evaluation Tools

1. ANARCI: The model is evaluated using ANARCI, a tool for antibody numbering and receptor classification. ANARCI is employed to analyze the generated antibody sequences for their conformity to known antibody sequence patterns and structures. It helps in assessing the accuracy of the model's outputs in terms of their structural viability and alignment with known antibody frameworks. This evaluation is crucial to ensure that the generated sequences are not only novel but also biologically relevant and potentially functional. ANARCI
2. Diversity Score: We can measure how diverse each model’s outputs are by computing the sequence similarity between candidates for each possible pairing from the outputs. The average of this distribution indicates how widely the model’s outputs vary, which is useful to know for downstream evaluation of the generated candidates. We compute the average sequence similarity for both the entire variable sequence, as well as just the HCDR3 region.

Performance and analytics

Ethical Considerations

⚠️ Important Note

Generated antibodies should be validated experimentally before clinical or research applications. Users should be aware of the potential misuse of generated sequences in harmful applications.

Links

• Huggingface Model Repository
• Web Demo
• OpenSource RunPod Severless Rest API
• The Code for this App

Additional Resources and Links

• Progen Foundation Models
• ANARCI Github
• ANARCI Webserver
• TAP: Therapeutic Antibody Profiler
• ESM Fold

Limitations and Future Work

• Predictions require experimental validation for practical use.
• Future improvements will focus on incorporating diverse training data and enhancing prediction accuracy for the efficacy of generated antibodies.

Contact Information

For questions or feedback regarding this model, please contact [XYZ].

🔧 Technical Details

No specific technical details beyond what is already covered in other sections are provided, so this section is skipped.

📄 License

The model is released under the Apache 2.0 license.