Enhancing Molecular Design through Graph-based Topological Reinforcement Learning

Xiangyu Zhang

arXiv:2411.14726·cs.LG·Published 2024-11-22

The generation of drug-like molecules is crucial for drug design. Existing reinforcement learning (RL) methods often overlook structural information. However, feature engineering-based methods usually merely focus on binding affinity prediction without substantial molecular modification. To address this, we present Graph-based Topological Reinforcement Learning (GraphTRL), which integrates both chemical and structural data for improved molecular generation. GraphTRL leverages multiscale weighted colored graphs (MWCG) and persistent homology, combined with molecular fingerprints, as the state space for RL. Evaluations show that GraphTRL outperforms existing methods in binding affinity prediction, offering a promising approach to accelerate drug discovery.

TopicsGenerative Design & Molecule Optimization, Molecular Representation & Learning, Property Prediction & ADMET, Protein & Biomolecules

Tagsdrug-discovery molecular-generation molecular-representation protein-ligand reinforcement-learning

arXiv categoriescs.LG, q-bio.BM

arXiv abstract pagePDF