首个统一分子生成与性质预测的模型,提升药物研发效率。
UniGEM: A Unified Approach to Generation and Property Prediction for Molecules
- 分两阶段生成:先构架后预测,避免任务冲突
- 在真实数据集上同时提升生成质量与预测精度
- 适合药物设计、化学信息学研究者使用
分子生成与分子性质预测对药物发现至关重要,但常被独立研究。受扩散模型可学习有效表示以增强预测任务的启发,我们提出UniGEM,首个成功整合分子生成与性质预测的统一模型。由于任务间存在固有不一致,简单多任务学习无效。UniGEM采用创新的两阶段生成流程,使预测任务在分子骨架形成后激活,并通过新颖训练策略实现任务平衡。理论分析与全面实验验证了其在两项任务上的显著提升。该方法原理可推广至自然语言处理与计算机视觉等领域。
原文摘要 · Abstract (English)
Molecular generation and molecular property prediction are both crucial for drug discovery, but they are often developed independently. Inspired by recent studies, which demonstrate that diffusion model, a prominent generative approach, can learn meaningful data representations that enhance predictive tasks, we explore the potential for developing a unified generative model in the molecular domain that effectively addresses both molecular generation and property prediction tasks. However, the integration of these tasks is challenging due to inherent inconsistencies, making simple multi-task learning ineffective. To address this, we propose UniGEM, the first unified model to successfully integrate molecular generation and property prediction, delivering superior performance in both tasks. Our key innovation lies in a novel two-phase generative process, where predictive tasks are activated in the later stages, after the molecular scaffold is formed. We further enhance task balance through innovative training strategies. Rigorous theoretical analysis and comprehensive experiments demonstrate our significant improvements in both tasks. The principles behind UniGEM hold promise for broader applications, including natural language processing and computer vision.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。