arXiv:2608.24610cs.LGcs.DC2026-08

让生成分子更精准优化药物指标,还能保护数据隐私。

Conditional GraphGANFed: Optimizing Graph-Structured Molecule Generation in Federated Generative Adversarial Networks

  • 用判别器评估分子并引导生成,实现自定义指标优化。
  • 在7个指标上优于原模型,优化药效评分(QED)提升超10%。
  • 适合需定向设计药物分子的研究者,尤其关注隐私保护场景。

生成对抗网络(GAN)在新分子发现中备受关注,因其能生成高质量新分子。为在保护数据隐私的同时高效训练,已有研究提出GraphGANFed,结合联邦学习与图卷积网络。但该方法无法仅针对用户指定的指标生成优化分子。为此,本文提出条件图生成对抗网络联邦学习框架(cGraphGANFed),通过引入评鉴网络评估生成分子的用户自定义指标,将评鉴结果与判别器输出共同融入生成器损失函数,指导生成既保持真实分子化学特性又优化目标指标的新分子。在两种场景下进行大量模拟实验:其一,同时优化7个常用指标,结果表明cGraphGANFed在有效性(Validity)和LogP上显著优于GraphGANFed,QED略有提升;其二,仅优化药效评分(QED),生成分子的QED值较原模型提升超过10%。此外,该方法对非独立同分布数据具有更强鲁棒性,有效缓解模式崩溃与性能下降问题。

原文摘要 · Abstract (English)

Generative adversarial networks (GANs) have garnered considerable attention in molecular discovery for their ability to generate novel and high-quality molecules. To efficiently train a GAN model while preserving data privacy, GraphGANFed has been proposed to incorporate federated learning and graph convolutional networks into GAN. Yet, GraphGANFed cannot produce synthetic molecules that only optimize a user-defined metric(s) to facilitate the new drug discovery process. To address this issue, we introduce a novel extension to GraphGANFed, namely conditional GraphGANFed (cGraphGANFed), by incorporating the critic network to assess generated molecules using user-defined metric(s). The evaluation results from both the critic network and discriminator are integrated into the loss function of the generator, guiding it to generate novel molecules that maintain similar chemical properties to real ones while optimizing user-defined metrics. Extensive simulations are conducted in two scenarios. First, cGraphGANFed endeavors to optimize all seven commonly used metrics, and the results show that cGraphGANFed significantly outperforms GraphGANFed in Validity and LogP, with a slight advantage in QED, across different settings. Second, cGraphGANFed focuses solely on optimizing QED, and the results show that the synthetic molecules produced by cGraphGANFed can achieve more than 10% improvement in QED than GraphGANFed. Also, the results demonstrate cGraphGANFed has enhanced resilience against mode collapses and performance reduction caused by non-IID data.

分子生成联邦学习GAN药物设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。