ChemBERTa-2: Fine-Tuning for Molecule’s HIV Replication Inhibition Prediction
ChemBERTa-2: Fine-Tuning for Molecule’s HIV Replication Inhibition Prediction
{Sylwia Nowakowska}

Abstract
Two versions of Large Language ChemBERTa-2 models, pre-trained with two different methods, were fine-tuned in this work for HIV replication inhibition prediction. The best model achieved AUROC of 0.793. The changes in distributions of molecular embeddings prior to and following fine-tuning reveal models’ enhanced ability to differentiate between active and inactive HIV molecules.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| molecular-property-prediction-on-hiv-dataset | ChemBERTa-2 Fine-tuned | AUC: 0.793 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.