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

通过基于片段的自监督嵌入预测进行层次分子表示学习

Hierarchical Molecular Representation Learning via Fragment-Based Self-Supervised Embedding Prediction

School of Computing, National University of Singapore, Singapore(新加坡国立大学计算机学院);Engineering, The Chinese University of Hong Kong, China(香港中文大学工程学院)

Jiele Wu, Haozhe Ma, Zhihan Guo, Thanh Vinh Vo, Tze Yun Leong

AI总结:

本文提出GraSPNet,通过层次自监督框架学习分子的原子和片段级语义,以提升分子性质预测的性能。

摘要:

图自监督学习(GSSL)已展示了在不需要人工标注的情况下生成具有表达力的图嵌入的强潜力,这使其在标注成本高的领域如分子图分析中尤为重要。然而,现有GSSL方法大多集中在节点或边级信息上,常常忽略化学相关的重要子结构,这些子结构对分子性质有重要影响。在本工作中,我们提出了图语义预测网络(GraSPNet),一种层次自监督框架,该框架明确建模了原子级和片段级的语义。GraSPNet将分子图分解为具有化学意义的片段,而无需预定义词汇,并通过多级消息传递学习节点和片段级的表示,同时在两个级别上进行掩码语义预测。这种层次语义监督使GraSPNet能够学习具有表达力和可转移性的多分辨率结构信息。在多个分子性质预测基准上的广泛实验表明,GraSPNet学习了具有化学意义的表示,并在迁移学习设置中一致优于最先进的GSSL方法。

英文摘要:

Graph self-supervised learning (GSSL) has demonstrated strong potential for generating expressive graph embeddings without the need for human annotations, making it particularly valuable in domains with high labeling costs such as molecular graph analysis. However, existing GSSL methods mostly focus on node- or edge-level information, often ignoring chemically relevant substructures which strongly influence molecular properties. In this work, we propose Graph Semantic Predictive Network (GraSPNet), a hierarchical self-supervised framework that explicitly models both atomic-level and fragment-level semantics. GraSPNet decomposes molecular graphs into chemically meaningful fragments without predefined vocabularies and learns node- and fragment-level representations through multi-level message passing with masked semantic prediction at both levels. This hierarchical semantic supervision enables GraSPNet to learn multi-resolution structural information that is both expressive and transferable. Extensive experiments on multiple molecular property prediction benchmarks demonstrate that GraSPNet learns chemically meaningful representations and consistently outperforms state-of-the-art GSSL methods in transfer learning settings.

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

↑