Quantitative evaluation of molecular generation performance of graph-based GANs

Jinli Zhang,Zhenbo Wang,Zongli Jiang,Man Wu,Chen Li,Yoshihiro Yamanishi
DOI: https://doi.org/10.1007/s11219-024-09671-7
2024-04-26
Software Quality Journal
Abstract:Deep generative models have been widely used in molecular generation tasks because they can save time and cost in drug development compared with traditional methods. Previous studies based on generative adversarial network (GAN) models typically employ reinforcement learning (RL) to constrain chemical properties, resulting in efficient and novel molecules. However, such models have poor performance in generating molecules due to instability in training. Therefore, quantitative evaluation of existing molecular generation models, especially GAN models, is necessary. This study aims to evaluate the performance of discrete GAN models using RL in molecular generation tasks and explore the impact of different factors on model performance. Through evaluation experiments on QM9 and ZINC datasets, the results show that noise sampling distributions, training epochs, and training data volumes can affect the performance of molecular generation. Finally, we provide strategies for stable training and improved performance for GAN models.
computer science, software engineering
What problem does this paper attempt to address?