English

Improving Targeted Molecule Generation through Language Model Fine-Tuning Via Reinforcement Learning

Biomolecules 2025-05-20 v2 Machine Learning

Abstract

Developing new drugs is laborious and costly, demanding extensive time investment. In this paper, we introduce a de-novo drug design strategy, which harnesses the capabilities of language models to devise targeted drugs for specific proteins. Employing a Reinforcement Learning (RL) framework utilizing Proximal Policy Optimization (PPO), we refine the model to acquire a policy for generating drugs tailored to protein targets. The proposed method integrates a composite reward function, combining considerations of drug-target interaction and molecular validity. Following RL fine-tuning, the proposed method demonstrates promising outcomes, yielding notable improvements in molecular validity, interaction efficacy, and critical chemical properties, achieving 65.37 for Quantitative Estimation of Drug-likeness (QED), 321.55 for Molecular Weight (MW), and 4.47 for Octanol-Water Partition Coefficient (logP), respectively. Furthermore, out of the generated drugs, only 0.041% do not exhibit novelty.

Keywords

Cite

@article{arxiv.2405.06836,
  title  = {Improving Targeted Molecule Generation through Language Model Fine-Tuning Via Reinforcement Learning},
  author = {Salma J. Ahmed and Emad A. Mohammed},
  journal= {arXiv preprint arXiv:2405.06836},
  year   = {2025}
}
R2 v1 2026-06-28T16:23:51.625Z