arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.01765cs.LGq-bio.BMstat.ML
收藏

一台Transformer可以同时理解2D和3D分子数据

One Transformer Can Understand Both 2D & 3D Molecular Data

Peking University(北京大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);Microsoft Research(微软研究院)

Shengjie Luo, Tianlang Chen, Yixian Xu, Shuxin Zheng, Tie-Yan Liu, Liwei Wang, Di He

摘要:

与通常具有独特格式的视觉和语言数据不同,分子可以自然地使用不同的化学配方来表征。人们可以将分子视为2D图形或将其定义为位于3D空间中的原子集合。对于分子表示学习,大多数以前的作品只为特定的数据格式设计了神经网络,使得学习的模型很可能对其他数据格式失败。我们相信,一个通用的化学神经网络模型应该能够处理跨数据模式的分子任务。为了实现这一目标,在这项工作中,我们开发了一种新的基于Transformer的分子模型,称为Transformer-M,它可以将2D或3D格式的分子数据作为输入,并生成有意义的语义表示。使用标准的Transformer作为主干架构,Transformer-M开发了两个独立的通道来编码2D和3D结构信息,并将它们与网络模块中的原子特征结合起来。当输入数据为特定格式时,相应的通道将被激活,而另一个通道将被禁用。通过使用适当设计的监督信号对2D和3D分子数据进行训练,Transformer-M自动学习利用来自不同数据模式的知识并正确捕获表示。我们对Transformer-M进行了广泛的实验。所有实证结果表明,Transformer-M可以同时在2D和3D任务上实现强大的性能,表明其广泛的适用性。代码和模型将在https://github.com/lsj2408/Transformer-M上公开提供。

英文摘要:

Unlike vision and language data which usually has a unique format, molecules can naturally be characterized using different chemical formulations. One can view a molecule as a 2D graph or define it as a collection of atoms located in a 3D space. For molecular representation learning, most previous works designed neural networks only for a particular data format, making the learned models likely to fail for other data formats. We believe a general-purpose neural network model for chemistry should be able to handle molecular tasks across data modalities. To achieve this goal, in this work, we develop a novel Transformer-based Molecular model called Transformer-M, which can take molecular data of 2D or 3D formats as input and generate meaningful semantic representations. Using the standard Transformer as the backbone architecture, Transformer-M develops two separated channels to encode 2D and 3D structural information and incorporate them with the atom features in the network modules. When the input data is in a particular format, the corresponding channel will be activated, and the other will be disabled. By training on 2D and 3D molecular data with properly designed supervised signals, Transformer-M automatically learns to leverage knowledge from different data modalities and correctly capture the representations. We conducted extensive experiments for Transformer-M. All empirical results show that Transformer-M can simultaneously achieve strong performance on 2D and 3D tasks, suggesting its broad applicability. The code and models will be made publicly available at https://github.com/lsj2408/Transformer-M.

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

↑