Synergistic Benefits of Joint Molecule Generation and Property Prediction
本文介绍了 Hyformer,这是一种基于 Transformer 的模型,它通过交替注意力机制和联合预训练统一了分子生成与性质预测,展示了在条件采样、分布外预测以及新型抗菌肽发现方面的协同改进。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是一篇未经同行评审的预印本的AI生成解释。这不是医疗建议。请勿根据此内容做出健康决定。 阅读完整免责声明
想象一下,你正试图同时教一个机器人两项截然不同的技能:创作富有创意的故事,以及批判这些故事是否具备特定品质(例如“这个故事吓人吗?”或“它有趣吗?”)。
通常,科学家会构建两个独立的机器人:一个是擅长讲故事但拙于批判的大师,另一个是擅长批判却无法从零开始创作故事的大师。问题在于,当你想要一个既富有创意又具体“吓人”的故事时,你必须同时操控这两个独立的机器人,这往往会导致笨拙的结果。
本文介绍了Hyformer,这是一种新型人工智能,它利用单一“大脑”同时学习成为作者和评论家。
问题:“分裂人格”的挣扎
在药物发现领域(特别是设计新分子时),科学家通常分两步进行:
- 生成:一个模型创建新的化学结构(就像写一个新句子)。
- 预测:另一个模型观察该结构并推测其属性(就像检查这个句子是否通顺或具有某种特定情绪)。
作者认为,将这些任务分开是低效的。这就像雇佣一位厨师来烹饪菜肴,然后雇佣一个完全不同的人来品尝并告诉你是否太咸。如果他们彼此不交流,厨师可能会不断重复品尝者试图指出的错误。
此外,试图训练一个机器人同时完成这两项工作往往会失败,因为这两项任务会“互相冲突”。写句子所需的数学逻辑(仅关注 preceding 内容)与批判整个句子所需的数学逻辑(一次性审视全局)是不同的。这会导致“梯度干扰”,用通俗的话说,就是机器人感到困惑,无法很好地掌握任何一项技能。
解决方案:Hyformer 的“切换眼镜”
作者构建了Hyformer,这是一个基于Transformer技术(许多现代 AI 聊天机器人背后的同一种“大脑”)的模型。
他们使用的巧妙技巧是:交替注意力机制。
想象 Hyformer 戴着两副不同的眼镜:
- “作者”眼镜(因果性):当需要生成新分子时,它戴上只允许看到过去的眼镜。它观察已经放置的原子,并逐一决定下一个是什么。这就是它写作的方式。
- “评论家”眼镜(双向性):当需要预测属性时,它换上一副能同时看到整个分子的眼镜。它审视整个结构以理解其特征。
在训练过程中,模型在这两种模式之间不断切换。它不会试图在完全相同的瞬间同时做两件事;相反,它练习写作,然后立即练习批判,接着再次写作。这防止了两项任务互相干扰。
结果:为何这很重要
论文声称,通过这种“切换”方法,Hyformer 获得了协同效益。这意味着整体大于部分之和。
- 更好的控制:如果你要求 Hyformer“写一个可能是抗生素的分子”,它的表现远优于那些分别训练的模型。因为“作者”和“评论家”共享同一个大脑,作者在写作时就能确切地学习到评论家在寻找什么。
- 更聪明的推测:当模型遇到一种它从未见过的全新类型分子(分布外数据)时,它预测其属性的能力优于其他模型。它已经学会了关于分子本质的更深刻理解,而不仅仅是记住了模式。
- 现实世界测试:作者在现实世界的挑战中测试了这一点:设计抗菌肽(杀死细菌的微小蛋白质)。Hyformer 成功生成了极有可能是有效抗生素的新肽序列,表现优于以往的专业模型。
核心结论
将 Hyformer 想象成一位多任务艺术家,它学会在同一场会话中创作并批判自己的作品。通过在“创造”和“分析”模式之间切换,该模型学习了一种统一的化学语言。这使得它不仅能生成随机的化学物质,还能有目的地设计具有特定、所需超能力的新物质,从而使药物发现的过程更加高效和精准。
论文总结道,这种方法成功统一了生成与预测,证明你不需要两个独立的机器人来构建和测试新药;你只需要一个聪明且适应性强的机器人。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。