arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2112.00544cs.LGcs.AIq-bio.QM
收藏

基于化学元素知识图谱的分子对比学习

Molecular Contrastive Learning with Chemical Element Knowledge Graph

Zhejiang University(浙江大学);Westlake Laboratory of Life Sciences and Biomedicine(西湖大学生命科学与生物医学实验室)

Yin Fang, Qiang Zhang, Haihong Yang, Xiang Zhuang, Shumin Deng, Wen Zhang, Ming Qin, Zhuo Chen, Xiaohui Fan, Huajun Chen

摘要:

分子表示学习有助于多个下游任务,如分子性质预测和药物设计。为了正确地表示分子,图对比学习是一种有前途的范例,因为它利用自我监督信号并且不需要人类注释。然而,以前的工作未能将基本的领域知识到图的语义,从而忽略了原子之间的相关性,具有共同的属性,但不直接连接的债券。为了解决这些问题,我们构建了一个化学元素知识图谱(KG)来总结元素之间的微观关联,并提出了一个新的知识增强的对比学习(KCL)框架的分子表示学习。KCL框架由三个模块组成。第一个模块,知识引导的图形增强,基于化学元素KG增强原始分子图。第二个模块,知识感知图表示,提取分子表示与一个共同的图形编码器的原始分子图和知识感知的消息传递神经网络(KMPNN)的复杂信息编码的扩增分子图。最后一个模块是一个对比的目标,我们最大限度地提高这两个分子图的观点之间的一致性。广泛的实验表明,KCL在8个分子数据集上获得了优于最先进基线的性能。可视化实验正确地解释了KCL从增强分子图中的原子和属性中学到的东西。我们的代码和数据可在https://github.com/ZJU-Fangyin/KCL上获得。

英文摘要:

Molecular representation learning contributes to multiple downstream tasks such as molecular property prediction and drug design. To properly represent molecules, graph contrastive learning is a promising paradigm as it utilizes self-supervision signals and has no requirements for human annotations. However, prior works fail to incorporate fundamental domain knowledge into graph semantics and thus ignore the correlations between atoms that have common attributes but are not directly connected by bonds. To address these issues, we construct a Chemical Element Knowledge Graph (KG) to summarize microscopic associations between elements and propose a novel Knowledge-enhanced Contrastive Learning (KCL) framework for molecular representation learning. KCL framework consists of three modules. The first module, knowledge-guided graph augmentation, augments the original molecular graph based on the Chemical Element KG. The second module, knowledge-aware graph representation, extracts molecular representations with a common graph encoder for the original molecular graph and a Knowledge-aware Message Passing Neural Network (KMPNN) to encode complex information in the augmented molecular graph. The final module is a contrastive objective, where we maximize agreement between these two views of molecular graphs. Extensive experiments demonstrated that KCL obtained superior performances against state-of-the-art baselines on eight molecular datasets. Visualization experiments properly interpret what KCL has learned from atoms and attributes in the augmented molecular graphs. Our codes and data are available at https://github.com/ZJU-Fangyin/KCL.

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

↑