arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.03475cs.LGcs.AIq-bio.BM

分子联合自编码:利用2D和3D扩散进行轨迹预训练

Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D Diffusion

Huawei Technologies Co., Ltd.(华为技术有限公司);Chinese Academy of Sciences(中国科学院);Shenzhen Institute for Advanced Study, University of Electronic Science and Technology of China(电子科技大学深圳高等研究院);Université de Montréal(蒙特利尔大学)

Weitao Du, Jiujiu Chen, Xuecang Zhang, Zhiming Ma, Shengchao Liu

AI总结:

本文提出MoleculeJAE预训练方法,通过扩散模型联合建模分子2D拓扑与3D几何轨迹,以自监督学习化学结构表示,并在20项任务中15项达到SOTA。

摘要:

近年来,用于药物发现的人工智能在机器学习和化学领域都引起了越来越多的关注。药物发现的基本组成部分是分子几何结构,因此,分子的几何表示是更好地利用机器学习技术进行药物发现的主要瓶颈。在本工作中,我们提出了一种用于分子联合自编码的预训练方法(MoleculeJAE)。MoleculeJAE能够同时学习2D化学键(拓扑)和3D构象(几何)信息,并应用扩散过程模型来模拟这两种模态的增强轨迹;在此基础上,MoleculeJAE将以自监督方式学习内在化学结构。因此,MoleculeJAE中预训练得到的几何表示有望使下游几何相关任务受益。在实验方面,通过与12个具有竞争力的基线方法进行比较,MoleculeJAE在20项任务中的15项上达到了最先进性能,证明了其有效性。

英文摘要:

Recently, artificial intelligence for drug discovery has raised increasing interest in both machine learning and chemistry domains. The fundamental building block for drug discovery is molecule geometry and thus, the molecule's geometrical representation is the main bottleneck to better utilize machine learning techniques for drug discovery. In this work, we propose a pretraining method for molecule joint auto-encoding (MoleculeJAE). MoleculeJAE can learn both the 2D bond (topology) and 3D conformation (geometry) information, and a diffusion process model is applied to mimic the augmented trajectories of such two modalities, based on which, MoleculeJAE will learn the inherent chemical structure in a self-supervised manner. Thus, the pretrained geometrical representation in MoleculeJAE is expected to benefit downstream geometry-related tasks. Empirically, MoleculeJAE proves its effectiveness by reaching state-of-the-art performance on 15 out of 20 tasks by comparing it with 12 competitive baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑